package sqlx

import (
	"context"
	"database/sql"
)

// TxBuilder represents a transaction builder.
type TxBuilder struct {
	db        *DB
	ctx       context.Context
	readonly  bool
	isolation IsolationLevel
}

// Context sets the context.
// Default is Background.
func (tb *TxBuilder) Context(ctx context.Context) *TxBuilder {
	tb.ctx = ctx
	return tb
}

// ReadOnly sets the transaction readonly or not.
// Default is false.
func (tb *TxBuilder) ReadOnly(readonly bool) *TxBuilder {
	tb.readonly = readonly
	return tb
}

// Isolation sets the isolation level.
// Default is LevelDefault.
func (tb *TxBuilder) Isolation(isolation IsolationLevel) *TxBuilder {
	tb.isolation = isolation
	return tb
}

// Build builds the transaction.
func (tb *TxBuilder) Build() (*Tx, error) {
	raw, err := tb.db.raw.BeginTx(tb.ctx, &sql.TxOptions{
		Isolation: tb.isolation,
		ReadOnly:  tb.readonly,
	})
	return &Tx{raw: raw, ctx: tb.ctx}, err
}