diff options
Diffstat (limited to 'pkg/data/db.go')
| -rw-r--r-- | pkg/data/db.go | 21 |
1 files changed, 18 insertions, 3 deletions
diff --git a/pkg/data/db.go b/pkg/data/db.go index 178c19d..cf58caf 100644 --- a/pkg/data/db.go +++ b/pkg/data/db.go @@ -5,12 +5,13 @@ import ( "database/sql" "fmt" "log/slog" + "regexp" "strings" "time" "github.com/jpappel/atlas/pkg/index" "github.com/jpappel/atlas/pkg/query" - _ "github.com/mattn/go-sqlite3" + "github.com/mattn/go-sqlite3" ) type Query struct { @@ -51,7 +52,7 @@ func NewQuery(filename string) *Query { func NewDB(filename string) *sql.DB { connStr := "file:" + filename + "?_fk=true&_journal=WAL" - db, err := sql.Open("sqlite3", connStr) + db, err := sql.Open("sqlite3_regex", connStr) if err != nil { panic(err) } @@ -64,7 +65,7 @@ func NewDB(filename string) *sql.DB { } func NewMemDB() *sql.DB { - db, err := sql.Open("sqlite3", ":memory:?_fk=true") + db, err := sql.Open("sqlite3_regex", ":memory:?_fk=true") if err != nil { panic(err) } @@ -420,3 +421,17 @@ func (q Query) Execute(artifact query.CompilationArtifact) (map[string]*index.Do return f.docs, nil } + +func regex(re, s string) (bool, error) { + return regexp.MatchString(re, s) +} + +func init() { + sql.Register("sqlite3_regex", + &sqlite3.SQLiteDriver{ + ConnectHook: func(sc *sqlite3.SQLiteConn) error { + return sc.RegisterFunc("regexp", regex, true) + }, + }, + ) +} |