From 920db7d9254411f1ef70059c5066a116ff8d6a56 Mon Sep 17 00:00:00 2001
From: Albert S <mail@quitesimple.org>
Date: Sat, 20 Mar 2021 19:39:18 +0100
Subject: [PATCH] First commit

---
 main.go | 87 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 87 insertions(+)
 create mode 100644 main.go

diff --git a/main.go b/main.go
new file mode 100644
index 0000000..3668936
--- /dev/null
+++ b/main.go
@@ -0,0 +1,87 @@
+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)
+	}
+}