réplica de
https://github.com/quitesimpleorg/hs9001.git
sincronizado 2024-11-14 06:37:52 +01:00
Comparar comentimentos
Não há cometimentos em comum. "216e59747cd1a99b173cf7fbff5b0dc0d0acfc1a" e "3b01d7b898346050003159a902e112797128755b" têm históricos completamente diferentes.
216e59747c
...
3b01d7b898
6
Makefile
6
Makefile
@ -1,6 +0,0 @@
|
||||
GIT_COMMIT=$(shell git rev-list -1 HEAD)
|
||||
GIT_TAG=$(shell git tag --sort="-version:refname" | head -n 1)
|
||||
all:
|
||||
go build -ldflags "-X main.GitCommit=${GIT_COMMIT} -X main.GitTag=${GIT_TAG}"
|
||||
|
||||
|
52
main.go
52
main.go
@ -23,13 +23,9 @@ type HistoryEntry struct {
|
||||
cwd string
|
||||
hostname string
|
||||
user string
|
||||
retval int
|
||||
timestamp time.Time
|
||||
}
|
||||
|
||||
var GitTag string
|
||||
var GitCommit string
|
||||
|
||||
func databaseLocation() string {
|
||||
envOverride := os.Getenv("HS9001_DB_PATH")
|
||||
if envOverride != "" {
|
||||
@ -61,8 +57,7 @@ func initDatabase(conn *sql.DB) {
|
||||
func migrateDatabase(conn *sql.DB, currentVersion int) {
|
||||
|
||||
migrations := []string{
|
||||
"ALTER TABLE history ADD COLUMN workdir varchar(4096) DEFAULT ''",
|
||||
"ALTER TABLE history ADD COLUMN retval integer DEFAULT -9001",
|
||||
"ALTER TABLE history add column workdir varchar(4096) DEFAULT ''",
|
||||
}
|
||||
|
||||
if !(len(migrations) > currentVersion) {
|
||||
@ -108,7 +103,7 @@ func setDBVersion(conn *sql.DB, ver int) {
|
||||
}
|
||||
}
|
||||
|
||||
func NewHistoryEntry(cmd string, retval int) HistoryEntry {
|
||||
func NewHistoryEntry(cmd string) HistoryEntry {
|
||||
wd, err := os.Getwd()
|
||||
if err != nil {
|
||||
log.Panic(err)
|
||||
@ -123,7 +118,6 @@ func NewHistoryEntry(cmd string, retval int) HistoryEntry {
|
||||
cmd: cmd,
|
||||
cwd: wd,
|
||||
timestamp: time.Now(),
|
||||
retval: retval,
|
||||
}
|
||||
}
|
||||
|
||||
@ -136,7 +130,7 @@ func importFromStdin(conn *sql.DB) {
|
||||
}
|
||||
|
||||
for scanner.Scan() {
|
||||
entry := NewHistoryEntry(scanner.Text(), -9001)
|
||||
entry := NewHistoryEntry(scanner.Text())
|
||||
entry.cwd = ""
|
||||
entry.timestamp = time.Unix(0, 0)
|
||||
add(conn, entry)
|
||||
@ -148,29 +142,10 @@ func importFromStdin(conn *sql.DB) {
|
||||
}
|
||||
}
|
||||
|
||||
func search(conn *sql.DB, q string, workdir string, beginTime time.Time, endTime time.Time, retval int) list.List {
|
||||
var sb strings.Builder
|
||||
sb.WriteString("SELECT id, command, workdir, user, hostname, retval ")
|
||||
sb.WriteString("FROM history ")
|
||||
sb.WriteString("WHERE timestamp BETWEEN datetime(?, 'unixepoch') ")
|
||||
sb.WriteString("AND datetime(?, 'unixepoch') ")
|
||||
sb.WriteString("AND command LIKE ? ")
|
||||
sb.WriteString("AND workdir LIKE ? ")
|
||||
if retval != -9001 {
|
||||
sb.WriteString("AND retval = ? ")
|
||||
}
|
||||
sb.WriteString("ORDER BY timestamp ASC ")
|
||||
func search(conn *sql.DB, q string, workdir string, beginTime time.Time, endTime time.Time) list.List {
|
||||
queryStmt := "SELECT id, command, workdir, user, hostname FROM history WHERE timestamp BETWEEN datetime(?, 'unixepoch') AND datetime(?, 'unixepoch') AND command LIKE ? AND workdir LIKE ? ORDER BY timestamp ASC"
|
||||
|
||||
queryStmt := sb.String()
|
||||
args := make([]interface{}, 0)
|
||||
args = append(args, beginTime.Unix())
|
||||
args = append(args, endTime.Unix())
|
||||
args = append(args, q)
|
||||
args = append(args, workdir)
|
||||
if retval != -9001 {
|
||||
args = append(args, retval)
|
||||
}
|
||||
rows, err := conn.Query(queryStmt, args...)
|
||||
rows, err := conn.Query(queryStmt, beginTime.Unix(), endTime.Unix(), q, workdir)
|
||||
if err != nil {
|
||||
log.Panic(err)
|
||||
}
|
||||
@ -179,7 +154,7 @@ func search(conn *sql.DB, q string, workdir string, beginTime time.Time, endTime
|
||||
defer rows.Close()
|
||||
for rows.Next() {
|
||||
var entry HistoryEntry
|
||||
err = rows.Scan(&entry.id, &entry.cmd, &entry.cwd, &entry.user, &entry.hostname, &entry.retval)
|
||||
err = rows.Scan(&entry.id, &entry.cmd, &entry.cwd, &entry.user, &entry.hostname)
|
||||
if err != nil {
|
||||
log.Panic(err)
|
||||
}
|
||||
@ -198,12 +173,12 @@ func delete(conn *sql.DB, entryId uint32) {
|
||||
}
|
||||
|
||||
func add(conn *sql.DB, entry HistoryEntry) {
|
||||
stmt, err := conn.Prepare("INSERT INTO history (user, command, hostname, workdir, timestamp, retval) VALUES (?, ?, ?, ?, datetime(?, 'unixepoch'),?)")
|
||||
stmt, err := conn.Prepare("INSERT INTO history (user, command, hostname, workdir, timestamp) VALUES (?, ?, ?, ?, datetime(?, 'unixepoch'))")
|
||||
if err != nil {
|
||||
log.Panic(err)
|
||||
}
|
||||
|
||||
_, err = stmt.Exec(entry.user, entry.cmd, entry.hostname, entry.cwd, entry.timestamp.Unix(), entry.retval)
|
||||
_, err = stmt.Exec(entry.user, entry.cmd, entry.hostname, entry.cwd, entry.timestamp.Unix())
|
||||
if err != nil {
|
||||
log.Panic(err)
|
||||
}
|
||||
@ -283,7 +258,7 @@ func main() {
|
||||
var rgx = regexp.MustCompile("\\s+\\d+\\s+(.*)")
|
||||
rs := rgx.FindStringSubmatch(historycmd)
|
||||
if len(rs) == 2 {
|
||||
add(conn, NewHistoryEntry(rs[1], ret))
|
||||
add(conn, NewHistoryEntry(rs[1]))
|
||||
}
|
||||
case "search":
|
||||
fallthrough
|
||||
@ -292,12 +267,11 @@ func main() {
|
||||
var beginTime string
|
||||
var endTime string
|
||||
var distinct bool = true
|
||||
var retVal int
|
||||
|
||||
searchCmd.StringVar(&workDir, "workdir", "%", "Search only within this workdir")
|
||||
searchCmd.StringVar(&beginTime, "begin", "50 years ago", "Start searching from this timeframe")
|
||||
searchCmd.StringVar(&endTime, "end", "now", "End searching from this timeframe")
|
||||
searchCmd.BoolVar(&distinct, "distinct", true, "Remove consecutive duplicate commands from output")
|
||||
searchCmd.IntVar(&retVal, "ret", -9001, "Only query commands that returned with this exit code. -9001=all (default)")
|
||||
|
||||
searchCmd.Parse(globalargs)
|
||||
|
||||
@ -314,7 +288,7 @@ func main() {
|
||||
}
|
||||
|
||||
q := strings.Join(args, " ")
|
||||
results := search(conn, "%"+q+"%", workDir, beginTimestamp, endTimeStamp, retVal)
|
||||
results := search(conn, "%"+q+"%", workDir, beginTimestamp, endTimeStamp)
|
||||
|
||||
previousCmd := ""
|
||||
for e := results.Front(); e != nil; e = e.Next() {
|
||||
@ -357,8 +331,6 @@ func main() {
|
||||
os.Exit(23)
|
||||
case "import":
|
||||
importFromStdin(conn)
|
||||
case "version":
|
||||
fmt.Fprintf(os.Stdout, "Git Tag: %s\nGit Commit: %s\n", GitTag, GitCommit)
|
||||
default:
|
||||
fmt.Fprintf(os.Stderr, "Error: Unknown subcommand '%s' supplied\n\n", cmd)
|
||||
printUsage()
|
||||
|
Carregando…
Criar uma nova questão referindo esta
Bloquear um utilizador