package sql

import (
	"database/sql"

	"go.nhat.io/otelsql"
	semconv "go.opentelemetry.io/otel/semconv/v1.20.0"
)

func openDB(driver, dsn string) (*sql.DB, error) {
	system := semconv.DBSystemPostgreSQL
	if driver == "sqlite" {
		system = semconv.DBSystemSqlite
	}
	
	// Register the otelsql wrapper for the provided postgres driver.
	driverName, err := otelsql.Register(driver,
		otelsql.AllowRoot(),
		otelsql.TraceQueryWithoutArgs(),
		otelsql.TraceRowsClose(),
		otelsql.TraceRowsAffected(),
		// otelsql.WithDatabaseName("my_database"),        // Optional.
		otelsql.WithSystem(system), // Optional.
	)
	if err != nil {
		return nil, err
	}
	
	// Connect to a Postgres database using the postgres driver wrapper.
	db, err := sql.Open(driverName, dsn)
	if err != nil {
		return nil, err
	}

	if err := otelsql.RecordStats(db); err != nil {
		return nil, err
	}

	return db, nil
}