ok

0_setup/main.go

@@ -0,0 +1,74 @@
+package main
+
+import (
+	"database/sql"
+	"flag"
+	"fmt"
+	"log"
+	"strings"
+
+	_ "github.com/mattn/go-sqlite3"
+)
+
+func main() {
+	dbpath := flag.String("db", "./db.db", "path to db file")
+	q := flag.String("q", "show tables", "query to execute")
+	flag.Parse()
+
+	query := strings.Trim(strings.TrimSpace(*q), ";")
+	if len(query) == 0 {
+		panic("refusing empty query")
+	}
+	query += ";"
+	log.Printf("%s", query)
+
+	db, err := sql.Open("sqlite3", *dbpath)
+	if err != nil {
+		panic(err)
+	}
+
+	if strings.HasPrefix(strings.ToUpper(strings.TrimSpace(query)), "SELECT") {
+		rows, err := db.Query(query)
+		if err != nil {
+			panic(err)
+		}
+		defer rows.Close()
+
+		columns, err := rows.Columns()
+		if err != nil {
+			panic(err)
+		}
+
+		ptrs := make([]any, len(columns))
+		for i := range ptrs {
+			var v interface{}
+			ptrs[i] = &v
+		}
+
+		for rows.Next() {
+			if err := rows.Scan(ptrs...); err != nil {
+				panic(err)
+			}
+			result := make([]string, len(columns))
+			for i := range ptrs {
+				ptr := ptrs[i].(*interface{})
+				result[i] = fmt.Sprintf("%s:%+v", columns[i], *ptr)
+			}
+			log.Println(result)
+		}
+	} else {
+		stmt, err := db.Prepare(query)
+		if err != nil {
+			panic(err)
+		}
+
+		result, err := stmt.Exec()
+		if err != nil {
+			panic(err)
+		}
+
+		affected, err := result.RowsAffected()
+		log.Printf("%d: %v", affected, err)
+	}
+
+}