2dph/internal/brain/db.go

//go:build cgo && system_ladybug

package brain

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"

	lbug "github.com/LadybugDB/go-ladybug"
)

var (
	db   *lbug.Database
	conn *lbug.Connection
)

func repoRoot() string {
	// Try KB_ROOT env, then walk up from binary
	if v := os.Getenv("KB_ROOT"); v != "" {
		return v
	}
	self, err := os.Executable()
	if err == nil {
		dir := filepath.Dir(self)
		for i := 0; i < 5; i++ {
			if _, err := os.Stat(filepath.Join(dir, "var")); err == nil {
				return dir
			}
			if _, err := os.Stat(filepath.Join(dir, ".git")); err == nil {
				return dir
			}
			parent := filepath.Dir(dir)
			if parent == dir {
				break
			}
			dir = parent
		}
	}
	return "."
}

func dbPath() string {
	return filepath.Join(repoRoot(), "var", "kb.lbug")
}

func openBrain() error {
	return openWithSandbox(eps())
}

func openWithSandbox(epsv string) error {
	cfg := lbug.DefaultSystemConfig()
	cfg.MaxNumThreads = 8
	cfg.BufferPoolSize = 1 << 30 // 1GB

	var err error
	db, err = lbug.OpenDatabase(dbPath(), cfg)
	if err != nil {
		return fmt.Errorf("OpenDatabase: %w", err)
	}

	conn, err = lbug.OpenConnection(db)
	if err != nil {
		closeBrain()
		return fmt.Errorf("OpenConnection: %w", err)
	}
	// Session settings need a live connection; running this before
	// OpenConnection dereferenced a nil *Connection.
	if epsv != "" {
		if strings.ContainsAny(epsv, "'\\") {
			closeBrain()
			return fmt.Errorf("SET STREAM_SANDBOX: invalid value")
		}
		if _, err := conn.Query("SET STREAM_SANDBOX = '" + epsv + "'"); err != nil {
			closeBrain()
			return fmt.Errorf("SET STREAM_SANDBOX: %w", err)
		}
	}
	if _, err := conn.Query("LOAD EXTENSION FTS"); err != nil {
		closeBrain()
		return fmt.Errorf("LOAD EXTENSION FTS: %w", err)
	}
	if _, err := conn.Query("LOAD EXTENSION VECTOR"); err != nil {
		closeBrain()
		return fmt.Errorf("LOAD EXTENSION VECTOR: %w", err)
	}
	migrateIntervalColumns()
	return nil
}

// migrateIntervalColumns adds D24 valid_from/valid_to on existing Leaf tables.
// Fresh CREATE already has them; ALTER is a no-op when the column exists.
func migrateIntervalColumns() {
	for _, col := range []string{"valid_from", "valid_to"} {
		_, _ = conn.Query("ALTER TABLE Leaf ADD " + col + " STRING")
	}
}

func closeBrain() {
	if conn != nil {
		conn.Close()
		conn = nil
	}
	if db != nil {
		db.Close()
		db = nil
	}
}