1
0
peilaus alkaen https://github.com/quitesimpleorg/hs9001.git synced 2024-10-31 23:44:38 +01:00
hs9001/main.go

88 rivejä
1.5 KiB
Go
Raaka Normaali näkymä Historia

2021-03-20 19:39:18 +01:00
package main
import (
"database/sql"
"flag"
"fmt"
"log"
"os"
"regexp"
_ "github.com/mattn/go-sqlite3"
)
func create_connection() * sql.DB {
homedir := os.Getenv("HOME");
db, err := sql.Open("sqlite3", homedir + "./local/share/hs0001/db.sqlite")
if err != nil {
log.Panic(err)
}
return db
}
func search(q string) {
conn := create_connection()
queryStmt := "SELECT command FROM history WHERE command LIKE ? ORDER BY timestamp ASC"
rows, err := conn.Query(queryStmt,"%"+q+"%");
if err != nil {
log.Panic(err)
}
for rows.Next() {
var command string
err = rows.Scan(&command)
if err != nil {
log.Panic(err)
}
log.Printf("%s", command)
}
}
func add(cmd string) {
conn := create_connection()
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)
}
}
func main() {
flag.Parse()
args := flag.Args()
argslen := len(args)
cmd := args[0]
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)
add(rs[1])
} else if cmd == "search" {
if argslen < 2 {
fmt.Fprint(os.Stderr, "Please provide the search query\n")
}
q := args[1]
search(q)
}
}