From ae86204c847ea712fd975d01925de586803ab402 Mon Sep 17 00:00:00 2001
From: Wancat <wancat@wancat.cc>
Date: Sat, 5 Nov 2022 00:13:21 +0800
Subject: [PATCH] Add query function

---
 config.go                      |  8 ++++----
 route.go                       | 31 ++++++++++++++++++++-----------
 templates/dashboard.html       |  7 +------
 templates/layouts/base.html    |  3 ++-
 templates/layouts/scripts.html |  8 ++++++++
 templates/query.html           | 12 ++++++++++++
 tx.go                          | 19 +++++++++----------
 7 files changed, 56 insertions(+), 32 deletions(-)
 create mode 100644 templates/layouts/scripts.html
 create mode 100644 templates/query.html

diff --git a/config.go b/config.go
index 5fe6c7c..082405f 100644
--- a/config.go
+++ b/config.go
@@ -1,7 +1,7 @@
 package main
 
-var SCRIPTS = map[string][]string{
-	"balance assets":     {"b", "assets", "-X", "$"},
-	"register":           {"r", "--tail", "10"},
-	"balance this month": {"b", "-b", "this month"},
+var SCRIPTS = map[string]string{
+	"balance assets":     "b assets -X $",
+	"register":           "r --tail 10",
+	"balance this month": "b -b \"this month\"",
 }
diff --git a/route.go b/route.go
index 5ccd909..7bb0639 100644
--- a/route.go
+++ b/route.go
@@ -33,7 +33,7 @@ func router() *gin.Engine {
 	authZone.GET("/dashboard", func(c *gin.Context) {
 		HTML(c, 200, "dashboard.html", struct {
 			Templates []*template.Template
-			Scripts   map[string][]string
+			Scripts   map[string]string
 		}{
 			ledgerTpl.Templates(),
 			SCRIPTS,
@@ -69,16 +69,6 @@ func router() *gin.Engine {
 		}{tx})
 	})
 
-	authZone.GET("/exec", func(c *gin.Context) {
-		user := getUser(c)
-		name, _ := c.GetQuery("name")
-		if err := user.executeScript(c.Writer, name); err != nil {
-			c.AbortWithError(500, err)
-			log.Println(err)
-			return
-		}
-	})
-
 	authZone.GET("/edit", func(c *gin.Context) {
 		user := getUser(c)
 		f, err := user.ReadFile(DEFAULT_JOURNAL)
@@ -110,5 +100,24 @@ func router() *gin.Engine {
 		c.FileAttachment(user.FilePath(DEFAULT_JOURNAL), DEFAULT_JOURNAL)
 	})
 
+	authZone.GET("/query", func(c *gin.Context) {
+		response := struct {
+			Query   string
+			Result  string
+			Scripts map[string]string
+		}{Scripts: SCRIPTS}
+		user := getUser(c)
+		var ok bool
+		var err error
+		response.Query, ok = c.GetQuery("query")
+		if ok && response.Query != "" {
+			response.Result, err = user.query(response.Query)
+			if err != nil {
+				panic(err)
+			}
+		}
+		HTML(c, 200, "query.html", response)
+	})
+
 	return r
 }
diff --git a/templates/dashboard.html b/templates/dashboard.html
index 7db6bf7..975b57b 100644
--- a/templates/dashboard.html
+++ b/templates/dashboard.html
@@ -11,10 +11,5 @@
     <label>Tx Name: <input name="name" type="text"></label></br>
     <input type="submit">
 </form>
-<h2>Scripts</h2>
-<ul>
-    {{ range $k, $v := .Scripts }}
-    <li><a href="/exec?name={{ $k }}">{{ $k }}</a></li>
-    {{ end }}
-</ul>
+{{ template "scripts" .Scripts }}
 {{ end }}
diff --git a/templates/layouts/base.html b/templates/layouts/base.html
index fb88e5c..fe6492d 100644
--- a/templates/layouts/base.html
+++ b/templates/layouts/base.html
@@ -9,9 +9,10 @@
             <ul>
             {{ with .User }}
                 {{ .Email }}
+                <li><a href="dashboard">儀表板</a></li>
                 <li><a href="edit">編輯</a></li>
                 <li><a href="download">下載</a></li>
-                <li><a href="exec">查詢</a></li>
+                <li><a href="query">查詢</a></li>
             {{ end }}
         </nav>
         {{ block "main" .Data }}
diff --git a/templates/layouts/scripts.html b/templates/layouts/scripts.html
new file mode 100644
index 0000000..b049d4d
--- /dev/null
+++ b/templates/layouts/scripts.html
@@ -0,0 +1,8 @@
+{{ define "scripts" }}
+<h3>常用指令</h3>
+<ul>
+    {{ range $k, $v := . }}
+    <li><a href="/query?query={{ urlquery $v }}">{{ $k }}</a></li>
+    {{ end }}
+</ul>
+{{ end }}
diff --git a/templates/query.html b/templates/query.html
new file mode 100644
index 0000000..c5aa88c
--- /dev/null
+++ b/templates/query.html
@@ -0,0 +1,12 @@
+{{ define "title" }}查詢{{ end }}
+{{ define "main" }}
+<h1>查詢</h1>
+{{ with .Result }}
+<pre><code>{{ . }}</code></pre>
+{{ end }}
+<form action="/query" method="GET">
+    <input type="text" name="query" value="{{ .Query }}" autofocus>
+    <input type="submit" value="查詢">
+</form>
+{{ template "scripts" .Scripts }}
+{{ end }}
diff --git a/tx.go b/tx.go
index 2253ded..34deb33 100644
--- a/tx.go
+++ b/tx.go
@@ -2,7 +2,6 @@ package main
 
 import (
 	"bytes"
-	"fmt"
 	"io"
 	"os/exec"
 	"strings"
@@ -48,14 +47,14 @@ func (u *User) overwriteFile(tx string) (err error) {
 	return err
 }
 
-func (u *User) executeScript(w io.Writer, name string) (err error) {
-	script, ok := SCRIPTS[name]
-	if !ok {
-		return fmt.Errorf("%s script not found", name)
-	}
-	cmd := exec.Command("ledger", append([]string{"--file", DEFAULT_JOURNAL}, script...)...)
+func (u *User) query(query string) (result string, err error) {
+	var buf bytes.Buffer
+
+	cmd := exec.Command("ledger", "--file", DEFAULT_JOURNAL)
 	cmd.Dir = u.Dir()
-	cmd.Stdout = w
-	cmd.Stderr = w
-	return cmd.Run()
+	cmd.Stdin = strings.NewReader(query)
+	cmd.Stdout = &buf
+	cmd.Stderr = &buf
+	err = cmd.Run()
+	return buf.String(), err
 }