hs9001/main.go

162 lines
3.1 KiB
Go
Raw Normal View History

2021-03-20 19:39:18 +01:00
package main
import (
"bufio"
2021-03-20 19:39:18 +01:00
"database/sql"
"flag"
"fmt"
"log"
"os"
2021-03-20 20:18:27 +01:00
"path/filepath"
2021-03-20 19:39:18 +01:00
"regexp"
2021-03-20 20:18:27 +01:00
_ "modernc.org/sqlite"
2021-03-20 19:39:18 +01:00
)
2021-03-20 20:18:27 +01:00
func databaseLocation() string {
envOverride := os.Getenv("HS9001_DB_PATH")
if envOverride != "" {
return envOverride
}
2021-03-20 20:18:27 +01:00
return filepath.Join(xdgOrFallback("XDG_DATA_HOME", filepath.Join(os.Getenv("HOME"), ".local/share")), "hs9001/db.sqlite")
}
func createConnection() *sql.DB {
2021-03-20 19:39:18 +01:00
2021-03-20 20:18:27 +01:00
db, err := sql.Open("sqlite", databaseLocation())
2021-03-20 19:39:18 +01:00
if err != nil {
log.Panic(err)
}
return db
}
func initDatabase(conn *sql.DB) {
2021-03-20 20:18:27 +01:00
queryStmt := "CREATE TABLE history(id INTEGER PRIMARY KEY, command varchar(512), timestamp datetime DEFAULT current_timestamp, user varchar(25), hostname varchar(32));\n" +
"CREATE VIEW count_by_date AS SELECT COUNT(id), STRFTIME('%Y-%m-%d', timestamp) FROM history GROUP BY strftime('%Y-%m-%d', timestamp)"
_, err := conn.Exec(queryStmt)
if err != nil {
log.Panic(err)
}
}
func importFromStdin(conn *sql.DB) {
scanner := bufio.NewScanner(os.Stdin)
_, err := conn.Exec("BEGIN;")
if err != nil {
log.Panic(err)
}
for scanner.Scan() {
add(conn, scanner.Text())
}
2021-03-20 19:39:18 +01:00
_, err = conn.Exec("END;")
if err != nil {
log.Panic(err)
}
}
func search(conn *sql.DB, q string) {
2021-03-20 19:39:18 +01:00
queryStmt := "SELECT command FROM history WHERE command LIKE ? ORDER BY timestamp ASC"
2021-03-20 20:18:27 +01:00
rows, err := conn.Query(queryStmt, "%"+q+"%")
2021-03-20 19:39:18 +01:00
if err != nil {
log.Panic(err)
}
for rows.Next() {
var command string
err = rows.Scan(&command)
if err != nil {
log.Panic(err)
}
2021-03-20 20:18:27 +01:00
fmt.Printf("%s\n", command)
2021-03-20 19:39:18 +01:00
}
}
func add(conn *sql.DB, cmd string) {
2021-03-20 19:39:18 +01:00
user := os.Getenv("USER")
hostname, err := os.Hostname()
if err != nil {
log.Panic(err)
}
stmt, err := conn.Prepare("INSERT INTO history (user, command, hostname) VALUES (?, ?, ?)")
if err != nil {
log.Panic(err)
}
_, err = stmt.Exec(user, cmd, hostname)
if err != nil {
log.Panic(err)
}
}
2021-03-20 20:18:27 +01:00
func xdgOrFallback(xdg string, fallback string) string {
dir := os.Getenv(xdg)
if dir != "" {
if ok, err := exists(dir); ok && err == nil {
return dir
}
}
return fallback
}
func exists(path string) (bool, error) {
_, err := os.Stat(path)
if err == nil {
return true, nil
}
if os.IsNotExist(err) {
return false, nil
}
return false, err
}
2021-03-20 19:39:18 +01:00
func main() {
flag.Parse()
args := flag.Args()
argslen := len(args)
if argslen < 1 {
2021-03-21 10:29:41 +01:00
fmt.Fprintf(os.Stderr, "Usage: ./hs9001 <add/search/init/import>\n")
return
}
2021-03-20 19:39:18 +01:00
cmd := args[0]
conn := createConnection()
2021-03-20 19:39:18 +01:00
if cmd == "add" {
if argslen < 2 {
fmt.Fprint(os.Stderr, "Error: You need to provide the command to be added")
}
historycmd := args[1]
var rgx = regexp.MustCompile("\\s+\\d+\\s+(.*)")
rs := rgx.FindStringSubmatch(historycmd)
if len(rs) == 2 {
add(conn, rs[1])
}
2021-03-20 19:39:18 +01:00
} else if cmd == "search" {
if argslen < 2 {
fmt.Fprint(os.Stderr, "Please provide the search query\n")
}
q := args[1]
search(conn, q)
2021-03-20 20:18:27 +01:00
} else if cmd == "init" {
err := os.MkdirAll(filepath.Dir(databaseLocation()), 0755)
if err != nil {
log.Panic(err)
}
initDatabase(conn)
} else if cmd == "import" {
importFromStdin(conn)
2021-03-20 19:39:18 +01:00
}
}