db/js_export.go

package db

import (
	"context"
	"errors"

	"apigo.cc/go/jsmod"
)

func init() {
	jsmod.Register("db", map[string]any{
		"get": func(ctx context.Context, name string) (*jsDB, error) {
			d := GetDB(name, nil)
			if d.Error != nil {
				return nil, d.Error
			}
			return &jsDB{db: d, ctx: ctx}, nil
		},
	})
}

type jsDB struct {
	db  *DB
	ctx context.Context
}

var errSafeMode = errors.New("database write operation is restricted in safe mode")

func (jd *jsDB) checkSafe() error {
	if jsmod.IsSafeMode(jd.ctx) {
		return errSafeMode
	}
	return nil
}

// Read Operations
func (jd *jsDB) Query(query string, args ...any) *QueryResult {
	return jd.db.Query(query, args...)
}

// Write Operations
func (jd *jsDB) Exec(query string, args ...any) *ExecResult {
	if jd.checkSafe() != nil {
		return &ExecResult{Error: errSafeMode}
	}
	return jd.db.Exec(query, args...)
}

func (jd *jsDB) Insert(table string, data any) *ExecResult {
	if jd.checkSafe() != nil {
		return &ExecResult{Error: errSafeMode}
	}
	return jd.db.Insert(table, data)
}

func (jd *jsDB) Update(table string, data any, conditions string, args ...any) *ExecResult {
	if jd.checkSafe() != nil {
		return &ExecResult{Error: errSafeMode}
	}
	return jd.db.Update(table, data, conditions, args...)
}

func (jd *jsDB) Delete(table string, conditions string, args ...any) *ExecResult {
	if jd.checkSafe() != nil {
		return &ExecResult{Error: errSafeMode}
	}
	return jd.db.Delete(table, conditions, args...)
}

func (jd *jsDB) Replace(table string, data any) *ExecResult {
	if jd.checkSafe() != nil {
		return &ExecResult{Error: errSafeMode}
	}
	return jd.db.Replace(table, data)
}

// Transaction Support
func (jd *jsDB) Begin() (*jsTx, error) {
	// Note: We don't block Begin() itself, but destructive operations within Tx will be blocked
	tx := jd.db.Begin()
	if tx.Error != nil {
		return nil, tx.Error
	}
	return &jsTx{tx: tx, ctx: jd.ctx}, nil
}

// jsTx wraps *Tx for JS environment
type jsTx struct {
	tx  *Tx
	ctx context.Context
}

func (jt *jsTx) checkSafe() error {
	if jsmod.IsSafeMode(jt.ctx) {
		return errSafeMode
	}
	return nil
}

func (jt *jsTx) Query(query string, args ...any) *QueryResult {
	return jt.tx.Query(query, args...)
}

func (jt *jsTx) Exec(query string, args ...any) *ExecResult {
	if jt.checkSafe() != nil {
		return &ExecResult{Error: errSafeMode}
	}
	return jt.tx.Exec(query, args...)
}

func (jt *jsTx) Insert(table string, data any) *ExecResult {
	if jt.checkSafe() != nil {
		return &ExecResult{Error: errSafeMode}
	}
	return jt.tx.Insert(table, data)
}

func (jt *jsTx) Update(table string, data any, conditions string, args ...any) *ExecResult {
	if jt.checkSafe() != nil {
		return &ExecResult{Error: errSafeMode}
	}
	return jt.tx.Update(table, data, conditions, args...)
}

func (jt *jsTx) Delete(table string, conditions string, args ...any) *ExecResult {
	if jt.checkSafe() != nil {
		return &ExecResult{Error: errSafeMode}
	}
	return jt.tx.Delete(table, conditions, args...)
}

func (jt *jsTx) Replace(table string, data any) *ExecResult {
	if jt.checkSafe() != nil {
		return &ExecResult{Error: errSafeMode}
	}
	return jt.tx.Replace(table, data)
}

func (jt *jsTx) Commit() error   { return jt.tx.Commit() }
func (jt *jsTx) Rollback() error { return jt.tx.Rollback() }

// Metadata
func (jd *jsDB) InKeys(numArgs int) string { return jd.db.InKeys(numArgs) }
func (jd *jsDB) Quote(text string) string  { return jd.db.Quote(text) }