Something went wrong. Try again.
A local-first bookmark manager with ambition
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295/*Copyright © 2025 Lukas Werner <me@lukaswerner.com>*/package cmd
import ( "bufio" "database/sql" "fmt" "log" "os" "strings"
"github.com/charmbracelet/lipgloss" "github.com/charmbracelet/lipgloss/table" "github.com/charmbracelet/x/term" "github.com/lukasmwerner/mark/search" "github.com/lukasmwerner/mark/store" "github.com/spf13/cobra")
// Style definitionsvar ( promptStyle = lipgloss.NewStyle(). Foreground(highlight). Bold(true)
resultHeaderStyle = lipgloss.NewStyle(). Foreground(lipgloss.Color("15")). Background(highlight). Padding(0, 1)
resultRowStyle = lipgloss.NewStyle(). Padding(0, 1)
errorStyle = lipgloss.NewStyle(). Foreground(lipgloss.Color("9")). Bold(true))
// sqlCmd represents the sql commandvar sqlCmd = &cobra.Command{ Use: "sql", Short: "Lets you run sql queries on your database", Long: ``, Run: func(cmd *cobra.Command, args []string) { db, err := store.Open(store.Options{ Flags: []store.Flag{store.Embedding}, }) if err != nil { fmt.Println(err.Error()) return }
defer db.Close()
if len(args) > 0 { fmt.Println(args) query := strings.Join(args, " ") fmt.Println(query) rows, err := executeQuery(db, query) if err != nil { log.Fatalln(err.Error()) } renderResults(rows) return } reader := bufio.NewReader(os.Stdin) fmt.Println("mark SQL REPL - Enter SQL queries to execute (type '.exit' or '.quit' to exit)") fmt.Println("Bookmarks are in the 'Bookmarks' table") fmt.Println("Type '.help' for available commands")
var query string for { fmt.Print(promptStyle.Render("mark-sql> ")) input, _ := reader.ReadString('\n') input = strings.TrimSpace(input)
if input == "" { continue }
if input == ".exit" || input == ".quit" || input == ".q" { break }
if input == ".help" || input == "?" { printHelp() continue }
if input == ".tables" { query = "SELECT name FROM sqlite_master WHERE type='table' ORDER BY name;" } else if input == ".schema" { query = "SELECT sql FROM sqlite_master WHERE type IN ('table', 'trigger') ORDER BY name;" } else if s, ok := strings.CutPrefix(input, ".web-search"); ok { results, err := search.Search(db, s) if err != nil { fmt.Println("err: " + err.Error()) continue } renderSearchResults(results) query = "" continue
} else if s, ok := strings.CutPrefix(input, ".semantic "); ok { fmt.Printf("semantic: '%s'\n", s) results, err := search.Semantic(db, s) if err != nil { fmt.Println("err: " + err.Error()) continue }
renderSearchBookmarks(results) query = "" continue } else if s, ok := strings.CutPrefix(input, ".search "); ok { fmt.Printf("search: '%s'\n", s) results, err := search.FullText5(db, s) if err != nil { fmt.Println("err: " + err.Error()) continue }
renderSearchBookmarks(results) query = "" continue } else { // Append to current query if it doesn't end with semicolon if !strings.HasSuffix(query, ";") && query != "" { query += " " + input } else { query = input } }
// Execute only if query ends with semicolon if strings.HasSuffix(query, ";") { rows, _ := executeQuery(db, query) renderResults(rows) query = "" rows.Close() } } },}
func init() { rootCmd.AddCommand(sqlCmd)
// Here you will define your flags and configuration settings.
// Cobra supports Persistent Flags which will work for this command // and all subcommands, e.g.: // sqlCmd.PersistentFlags().String("foo", "", "A help for foo")
// Cobra supports local flags which will only run when this command // is called directly, e.g.: // sqlCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")}
func renderSearchResults(results []search.Result) { bookmarks := make([]store.Bookmark, len(results)) for i, r := range results { bookmarks[i] = r.Bookmark } renderBookmarks(bookmarks)
}
func renderBookmarks(bookmarks []store.Bookmark) { if bookmarks == nil { return }
width, _, err := term.GetSize(0) if err != nil { fmt.Println(errorStyle.Render("Error getting terminal size: " + err.Error())) return }
t := table.New().Headers("url", "title", "description", "tags").Width(width) for _, b := range bookmarks { t.Row(b.Url, b.Title, b.Description, strings.Join(b.Tags, ", ")) } fmt.Println(t.Render())
fmt.Printf("Query executed successfully. %d row(s) returned.\n", len(bookmarks))}
func renderResults(rows *sql.Rows) { if rows == nil { return } columns, err := rows.Columns() if err != nil { fmt.Println(errorStyle.Render("Error getting columns: " + err.Error())) return }
values := make([]any, len(columns)) valuePtrs := make([]any, len(columns)) for i := range columns { valuePtrs[i] = &values[i] }
width, _, err := term.GetSize(0) if err != nil { fmt.Println(errorStyle.Render("Error getting terminal size: " + err.Error())) return }
t := table.New().Headers(columns...).Width(width)
rowCount := 0 for rows.Next() { rowCount++ err = rows.Scan(valuePtrs...) if err != nil { fmt.Println(errorStyle.Render("Error scanning row: " + err.Error())) continue }
row := []string{} for _, val := range values { var displayVal string switch v := val.(type) { case []byte: displayVal = string(v) case nil: displayVal = "NULL" default: displayVal = fmt.Sprintf("%v", v) } row = append(row, resultRowStyle.Render(displayVal)) } t.Row(row...) }
fmt.Println(t.Render())
if err = rows.Err(); err != nil { fmt.Println(errorStyle.Render("Error during iteration: " + err.Error())) return }
fmt.Printf("Query executed successfully. %d row(s) returned.\n", rowCount)
}func renderSearchBookmarks(bookmarks []search.Bookmark) { if bookmarks == nil { return }
width, _, err := term.GetSize(0) if err != nil { fmt.Println(errorStyle.Render("Error getting terminal size: " + err.Error())) return }
t := table.New().Headers("url", "title", "description", "tags", "distance").Width(width) for _, bm := range bookmarks { b := bm.Get() t.Row(b.Url, b.Title, b.Description, strings.Join(b.Tags, ", "), fmt.Sprintf("%.4f", bm.Score())) } fmt.Println(t.Render())
fmt.Printf("Query executed successfully. %d row(s) returned.\n", len(bookmarks))}
func executeQuery(db *store.DB, query string) (*sql.Rows, error) { query = strings.TrimSpace(query)
rows, err := db.Query(query) if err != nil { fmt.Println(errorStyle.Render("Error: " + err.Error())) return nil, err } return rows, err}
func printHelp() { fmt.Println("Available commands:") fmt.Println(" .help, ? - Show this help") fmt.Println(" .exit, .quit - Exit the REPL") fmt.Println(" .tables - List all tables") fmt.Println(" .schema - Show schema for all tables") fmt.Println(" .search - Use FTS search to find bookmarks") fmt.Println(" .semantic - Use vector embeddings to find bookmarks") fmt.Println("") fmt.Println("Notes:") fmt.Println("- SQL queries can span multiple lines until a semicolon is entered") fmt.Println("- Press Enter on an empty line to submit the query")}