2021-03-20 19:39:18 +01:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2021-03-20 20:47:05 +01:00
|
|
|
"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 {
|
2021-03-21 10:38:12 +01:00
|
|
|
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
|
|
|
|
}
|
|
|
|
|
2021-03-20 20:47:05 +01:00
|
|
|
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)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-03-20 20:47:05 +01:00
|
|
|
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
|
|
|
|
2021-03-20 20:47:05 +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
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-03-20 20:47:05 +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)
|
2021-03-20 20:22:38 +01:00
|
|
|
|
|
|
|
if argslen < 1 {
|
2021-03-21 10:29:41 +01:00
|
|
|
fmt.Fprintf(os.Stderr, "Usage: ./hs9001 <add/search/init/import>\n")
|
2021-03-20 20:22:38 +01:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2021-03-20 19:39:18 +01:00
|
|
|
cmd := args[0]
|
|
|
|
|
2021-03-20 20:47:05 +01:00
|
|
|
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)
|
2021-03-21 10:43:06 +01:00
|
|
|
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]
|
2021-03-20 20:47:05 +01:00
|
|
|
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)
|
|
|
|
}
|
2021-03-20 20:47:05 +01:00
|
|
|
initDatabase(conn)
|
|
|
|
} else if cmd == "import" {
|
|
|
|
importFromStdin(conn)
|
2021-03-20 19:39:18 +01:00
|
|
|
}
|
|
|
|
}
|