wancat/ledger-quicknote
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add comfirm page, add account and name parameter

Browse Source
This commit is contained in:
Wancat
2022-10-19 08:50:44 +08:00
parent 482c293dec
commit 268b488fea
7 changed files with 94 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
main.go
View File

@@ -1,71 +1,104 @@
package main package main
import ( import (
"bytes"
"fmt" "fmt"
"io" "io"
"log" "log"
"net/http" "net/http"
"net/url" "net/url"
"os" "os"
"strings"
"text/template" "text/template"
"time" "time"
) )
var tpl *template.Template var ledgerTpl *template.Template
var htmlTpl *template.Template
const LEDGER_FILE = "test.txt" const LEDGER_FILE = "test.txt"
type TxData struct { type TxData struct {
Name string
Date string Date string
Amount string Amount string
Account string
} }
func init() { func init() {
var err error ledgerTpl = template.Must(template.ParseGlob("tx/*.txt"))
tpl, err = template.ParseGlob("templates/*") log.Println(ledgerTpl.DefinedTemplates())
if err != nil { htmlTpl = template.Must(template.ParseGlob("templates/*.html"))
panic(err)
}
log.Println(tpl.DefinedTemplates())
} }
func main() { func main() {
fmt.Println("Hi")
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
tpl.ExecuteTemplate(w, "index.html", tpl) htmlTpl.ExecuteTemplate(w, "index.html", ledgerTpl)
}) })
http.HandleFunc("/action", func(w http.ResponseWriter, r *http.Request) { http.HandleFunc("/new", func(w http.ResponseWriter, r *http.Request) {
if err := r.ParseForm(); err != nil { if err := r.ParseForm(); err != nil {
panic(err) http.Error(w, err.Error(), 400)
return
} }
fmt.Println(r.Form) fmt.Println(r.Form)
err := renderTx(w, r.Form) tx, err := newTx(r.Form)
if err != nil { if err != nil {
panic(err) http.Error(w, err.Error(), 400)
log.Println(err, r.Form)
return
}
if err := htmlTpl.ExecuteTemplate(w, "new.html", struct {
Tx string
}{tx}); err != nil {
http.Error(w, err.Error(), 500)
} }
}) })
http.HandleFunc("/submit", func(w http.ResponseWriter, r *http.Request) {
if err := r.ParseForm(); err != nil {
http.Error(w, err.Error(), 400)
return
}
tx := r.FormValue("tx")
if err := appendToFile(tx); err != nil {
http.Error(w, err.Error(), 500)
return
}
if err := htmlTpl.ExecuteTemplate(w, "success.html", struct {
Tx string
}{tx}); err != nil {
http.Error(w, err.Error(), 500)
}
})
log.Println("Listen on http://localhost:8000")
log.Fatal(http.ListenAndServe(":8000", nil)) log.Fatal(http.ListenAndServe(":8000", nil))
} }
func renderTx(w io.Writer, params url.Values) (err error) { func newTx(params url.Values) (result string, err error) {
name := params.Get("action") name := params.Get("action")
data := TxData{ data := TxData{
Date: time.Now().Format("2006/01/02"), Date: time.Now().Format("2006/01/02"),
Amount: params.Get("amount"), Amount: params.Get("amount"),
Account: params.Get("account"),
Name: params.Get("name"),
} }
var buf bytes.Buffer
err = ledgerTpl.ExecuteTemplate(&buf, name, data)
return buf.String(), nil
}
func appendToFile(tx string) (err error) {
f, err := os.OpenFile(LEDGER_FILE, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0644) f, err := os.OpenFile(LEDGER_FILE, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0644)
if err != nil { if err != nil {
panic(err) return err
} }
defer f.Close() defer f.Close()
if err = tpl.ExecuteTemplate(f, name, data); err != nil {
buf := strings.NewReader(strings.ReplaceAll(tx, "\r", "")) // Remove CR generated from browser
_, err = io.Copy(f, buf)
return err return err
} }
if err = tpl.ExecuteTemplate(w, name, data); err != nil {
return err
}
return nil
}

10
templates/index.html
View File

@@ -5,15 +5,15 @@
</head> </head>
<body> <body>
<h1>Ledger Quick Note</h1> <h1>Ledger Quick Note</h1>
<form action="/action" method="POST"> <form action="/new" method="POST">
<label>Action: <select name="action"> <label>Action: <select name="action">
{{ range .Templates }} {{ range .Templates }}
{{ if ne .Name "index.html" }}
<option value="{{ .Name }}">{{ .Name }}</option> <option value="{{ .Name }}">{{ .Name }}</option>
{{ end }} {{ end }}
{{ end }} </select></label><br>
</select></label> <label>Amount: <input name="amount" type="number"></label><br>
<label>Amount: <input name="amount" type="number"></label> <label>Account: <input name="account" type="text"></label></br>
<label>Tx Name: <input name="name" type="text"></label></br>
<input type="submit"> <input type="submit">
</form> </form>
</body> </body>

13
templates/new.html Normal file
View File

@@ -0,0 +1,13 @@
<!DOCTYPE html>
<html>
<head>
<title>Ledger Quick Note</title>
</head>
<body>
<h1>Comfirm new Tx</h1>
<form action="/submit" method="POST">
<textarea name="tx" rows="15" cols="40">{{ .Tx }}</textarea><br>
<input type="submit">
</form>
</body>
</html>

12
templates/success.html Normal file
View File

@@ -0,0 +1,12 @@
<!DOCTYPE html>
<html>
<head>
<title>Ledger Quick Note</title>
</head>
<body>
<h1>Ledger Quick Note</h1>
<p><strong>Success</strong></p>
<pre><code>{{ .Tx }}</code></pre>
<p><a href="/">Back to home</a></p>
</body>
</html>

0
templates/cash.txt → tx/cash.txt
View File

4
tx/expense.txt Normal file
View File

@@ -0,0 +1,4 @@
{{ .Date }} {{ with .Name }}{{ . }}{{ else }}{{ .Account }}{{ end }}
{{ .Account }} ${{ .Amount }}
cash

0
templates/withdraw.txt → tx/withdraw.txt
View File