diff --git a/README.md b/README.md index da8e4cd..22d9080 100644 --- a/README.md +++ b/README.md @@ -31,10 +31,6 @@ apk add hs9001 ### Setup / Config -``` -hs9001 init -``` - Add this to .bashrc ``` diff --git a/go.mod b/go.mod index 4d3ea54..b261697 100644 --- a/go.mod +++ b/go.mod @@ -2,4 +2,4 @@ module hs9001 go 1.16 -require modernc.org/sqlite v1.10.0 // indirect +require modernc.org/sqlite v1.10.0 diff --git a/go.sum b/go.sum index e3caea6..2fca74a 100644 --- a/go.sum +++ b/go.sum @@ -1,9 +1,12 @@ +github.com/dustin/go-humanize v1.0.0 h1:VSnTsYCnlFHaM2/igO1h6X3HA71jcobQuxemgkq4zYo= github.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk= +github.com/google/go-cmp v0.5.3 h1:x95R7cp+rSeeqAMI2knLtQ0DKlaBhv2NrtrOvafPHRo= github.com/google/go-cmp v0.5.3/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= github.com/kballard/go-shellquote v0.0.0-20180428030007-95032a82bc51 h1:Z9n2FFNUXsshfwJMBgNA0RU6/i7WVaAegv3PtuIHPMs= github.com/kballard/go-shellquote v0.0.0-20180428030007-95032a82bc51/go.mod h1:CzGEWj7cYgsdH8dAjBGEr58BoE7ScuLd+fwFZ44+/x8= github.com/mattn/go-isatty v0.0.12 h1:wuysRhFDzyxgEmMf5xjvJ2M9dZoWAXNNr5LSBS7uHXY= github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU= +github.com/mattn/go-sqlite3 v1.14.6 h1:dNPt6NO46WmLVt2DLNpwczCmdV5boIZ6g/tlDrlRUbg= github.com/mattn/go-sqlite3 v1.14.6/go.mod h1:NyWgC/yNuGj7Q9rpYnZvas74GogHl5/Z4A/KQRfk6bU= github.com/remyoudompheng/bigfft v0.0.0-20200410134404-eec4a21b6bb0 h1:OdAsTTz6OkFY5QxjkYwrChwuRruF69c169dPK26NUlk= github.com/remyoudompheng/bigfft v0.0.0-20200410134404-eec4a21b6bb0/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo= @@ -40,6 +43,7 @@ modernc.org/cc/v3 v3.31.5-0.20210308123301-7a3e9dab9009 h1:u0oCo5b9wyLr++HF3AN9J modernc.org/cc/v3 v3.31.5-0.20210308123301-7a3e9dab9009/go.mod h1:0R6jl1aZlIl2avnYfbfHBS1QB6/f+16mihBObaBC878= modernc.org/ccgo/v3 v3.9.0 h1:JbcEIqjw4Agf+0g3Tc85YvfYqkkFOv6xBwS4zkfqSoA= modernc.org/ccgo/v3 v3.9.0/go.mod h1:nQbgkn8mwzPdp4mm6BT6+p85ugQ7FrGgIcYaE7nSrpY= +modernc.org/httpfs v1.0.6 h1:AAgIpFZRXuYnkjftxTAZwMIiwEqAfk8aVB2/oA6nAeM= modernc.org/httpfs v1.0.6/go.mod h1:7dosgurJGp0sPaRanU53W4xZYKh14wfzX420oZADeHM= modernc.org/libc v1.7.13-0.20210308123627-12f642a52bb8/go.mod h1:U1eq8YWr/Kc1RWCMFUWEdkTg8OTcfLw2kY8EDwl039w= modernc.org/libc v1.8.0 h1:Pp4uv9g0csgBMpGPABKtkieF6O5MGhfGo6ZiOdlYfR8= @@ -55,8 +59,10 @@ modernc.org/sqlite v1.10.0 h1:0QNqx4EzfZzNEG13sFbS/L+egh0X5WXSckHrxHkySX8= modernc.org/sqlite v1.10.0/go.mod h1:PGzq6qlhyYjL6uVbSgS6WoF7ZopTW/sI7+7p+mb4ZVU= modernc.org/strutil v1.1.0 h1:+1/yCzZxY2pZwwrsbH+4T7BQMoLQ9QiBshRC9eicYsc= modernc.org/strutil v1.1.0/go.mod h1:lstksw84oURvj9y3tn8lGvRxyRC1S2+g5uuIzNfIOBs= +modernc.org/tcl v1.5.0 h1:euZSUNfE0Fd4W8VqXI1Ly1v7fqDJoBuAV88Ea+SnaSs= modernc.org/tcl v1.5.0/go.mod h1:gb57hj4pO8fRrK54zveIfFXBaMHK3SKJNWcmRw1cRzc= modernc.org/token v1.0.0 h1:a0jaWiNMDhDUtqOj09wvjWWAqd3q7WpBulmL9H2egsk= modernc.org/token v1.0.0/go.mod h1:UGzOrNV1mAFSEB63lOFHIpNRUVMvYTc6yu1SMY/XTDM= modernc.org/z v1.0.1-0.20210308123920-1f282aa71362/go.mod h1:8/SRk5C/HgiQWCgXdfpb+1RvhORdkz5sw72d3jjtyqA= +modernc.org/z v1.0.1 h1:WyIDpEpAIx4Hel6q/Pcgj/VhaQV5XPJ2I6ryIYbjnpc= modernc.org/z v1.0.1/go.mod h1:8/SRk5C/HgiQWCgXdfpb+1RvhORdkz5sw72d3jjtyqA= diff --git a/main.go b/main.go index db644e0..d8eaebf 100644 --- a/main.go +++ b/main.go @@ -28,6 +28,7 @@ func createConnection() *sql.DB { if err != nil { log.Panic(err) } + return db } @@ -41,6 +42,53 @@ func initDatabase(conn *sql.DB) { } } +func migrateDatabase(conn *sql.DB, currentVersion int) { + + migrations := []string{"ALTER TABLE history add column workdir varchar(4096)"} + + if !(len(migrations) > currentVersion) { + return + } + + _, err := conn.Exec("BEGIN;") + if err != nil { + log.Panic(err) + } + for _, m := range migrations[currentVersion:] { + _, err := conn.Exec(m) + if err != nil { + log.Panic(err) + } + + } + + setDBVersion(conn, len(migrations)) + + _, err = conn.Exec("END;") + if err != nil { + log.Panic(err) + } +} + +func fetchDBVersion(conn *sql.DB) int { + rows, err := conn.Query("PRAGMA user_version;") + if err != nil { + log.Panic(err) + } + defer rows.Close() + rows.Next() + var res int + rows.Scan(&res) + return res +} + +func setDBVersion(conn *sql.DB, ver int) { + _, err := conn.Exec(fmt.Sprintf("PRAGMA user_version=%d", ver)) + if err != nil { + log.Panic(err) + } +} + func importFromStdin(conn *sql.DB) { scanner := bufio.NewScanner(os.Stdin) @@ -67,6 +115,7 @@ func search(conn *sql.DB, q string) { log.Panic(err) } + defer rows.Close() for rows.Next() { var command string err = rows.Scan(&command) @@ -135,7 +184,7 @@ func exists(path string) (bool, error) { } func printUsage() { - fmt.Fprintf(os.Stderr, "Usage: ./hs9001 \n") + fmt.Fprintf(os.Stderr, "Usage: ./hs9001 \n") } func main() { @@ -151,7 +200,21 @@ func main() { cmd := os.Args[1] globalargs := os.Args[2:] - conn := createConnection() + var conn *sql.DB + ok, _ := exists(databaseLocation()) + + if !ok { + err := os.MkdirAll(filepath.Dir(databaseLocation()), 0755) + if err != nil { + log.Panic(err) + } + conn = createConnection() + initDatabase(conn) + } else { + conn = createConnection() + } + + migrateDatabase(conn, fetchDBVersion(conn)) switch cmd { case "add": @@ -197,12 +260,7 @@ func main() { //we do not want to leak what we just deleted :^) os.Exit(23) - case "init": - err := os.MkdirAll(filepath.Dir(databaseLocation()), 0755) - if err != nil { - log.Panic(err) - } - initDatabase(conn) + case "import": importFromStdin(conn) default: