package sql

import (
	"context"
	"database/sql/driver"
	"errors"
)

func ctxDriverPrepare(ctx context.Context, ci driver.Conn, query string) (driver.Stmt, error) {
	if ciCtx, is := ci.(driver.ConnPrepareContext); is {
		return ciCtx.PrepareContext(ctx, query)
	}
	si, err := ci.Prepare(query)
	if err == nil {
		select {
		default:
		case <-ctx.Done():
			si.Close()
			return nil, ctx.Err()
		}
	}

	// Begin starts and returns a new transaction.
	//
	// Deprecated: Drivers should implement ConnBeginTx instead (or additionally).
	Begin() (Tx, error)
}

// ConnPrepareContext enhances the [Conn] interface with context.
type ConnPrepareContext interface {
	// PrepareContext returns a prepared statement, bound to this connection.
	// context is for the preparation of the statement,

pkg database/sql/driver, type ConnBeginTx interface { BeginTx }
pkg database/sql/driver, type ConnBeginTx interface, BeginTx(context.Context, TxOptions) (Tx, error)
pkg database/sql/driver, type ConnPrepareContext interface { PrepareContext }
pkg database/sql/driver, type ConnPrepareContext interface, PrepareContext(context.Context, string) (Stmt, error)
pkg database/sql/driver, type ExecerContext interface { ExecContext }

		{"(RowsAffected).LastInsertId", Method, 0},
		{"(RowsAffected).RowsAffected", Method, 0},
		{"Bool", Var, 0},
		{"ColumnConverter", Type, 0},
		{"Conn", Type, 0},
		{"ConnBeginTx", Type, 8},
		{"ConnPrepareContext", Type, 8},
		{"Connector", Type, 10},
		{"DefaultParameterConverter", Var, 0},
		{"Driver", Type, 0},
		{"DriverContext", Type, 10},
		{"ErrBadConn", Var, 0},
		{"ErrRemoveArgument", Var, 9},