miroir de
https://github.com/quitesimpleorg/hs9001.git
synchronisé 2024-11-01 02:04:37 +01:00
88 lignes
1.5 KiB
Go
88 lignes
1.5 KiB
Go
|
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)
|
||
|
}
|
||
|
}
|