Add comfirm page, add account and name parameter
This commit is contained in:
Wancat
77
main.go
77
main.go
@@ -1,71 +1,104 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"os"
|
||||
"strings"
|
||||
"text/template"
|
||||
"time"
|
||||
)
|
||||
|
||||
var tpl *template.Template
|
||||
var ledgerTpl *template.Template
|
||||
var htmlTpl *template.Template
|
||||
|
||||
const LEDGER_FILE = "test.txt"
|
||||
|
||||
type TxData struct {
|
||||
Name string
|
||||
Date string
|
||||
Amount string
|
||||
Account string
|
||||
}
|
||||
|
||||
func init() {
|
||||
var err error
|
||||
tpl, err = template.ParseGlob("templates/*")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
log.Println(tpl.DefinedTemplates())
|
||||
ledgerTpl = template.Must(template.ParseGlob("tx/*.txt"))
|
||||
log.Println(ledgerTpl.DefinedTemplates())
|
||||
htmlTpl = template.Must(template.ParseGlob("templates/*.html"))
|
||||
}
|
||||
|
||||
func main() {
|
||||
fmt.Println("Hi")
|
||||
|
||||
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 {
|
||||
panic(err)
|
||||
http.Error(w, err.Error(), 400)
|
||||
return
|
||||
}
|
||||
fmt.Println(r.Form)
|
||||
err := renderTx(w, r.Form)
|
||||
tx, err := newTx(r.Form)
|
||||
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))
|
||||
}
|
||||
|
||||
func renderTx(w io.Writer, params url.Values) (err error) {
|
||||
func newTx(params url.Values) (result string, err error) {
|
||||
name := params.Get("action")
|
||||
data := TxData{
|
||||
Date: time.Now().Format("2006/01/02"),
|
||||
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)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
return err
|
||||
}
|
||||
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
|
||||
}
|
||||
if err = tpl.ExecuteTemplate(w, name, data); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -5,15 +5,15 @@
|
||||
</head>
|
||||
<body>
|
||||
<h1>Ledger Quick Note</h1>
|
||||
<form action="/action" method="POST">
|
||||
<form action="/new" method="POST">
|
||||
<label>Action: <select name="action">
|
||||
{{ range .Templates }}
|
||||
{{ if ne .Name "index.html" }}
|
||||
<option value="{{ .Name }}">{{ .Name }}</option>
|
||||
{{ end }}
|
||||
{{ end }}
|
||||
</select></label>
|
||||
<label>Amount: <input name="amount" type="number"></label>
|
||||
</select></label><br>
|
||||
<label>Amount: <input name="amount" type="number"></label><br>
|
||||
<label>Account: <input name="account" type="text"></label></br>
|
||||
<label>Tx Name: <input name="name" type="text"></label></br>
|
||||
<input type="submit">
|
||||
</form>
|
||||
</body>
|
||||
|
||||
13
templates/new.html
Normal file
13
templates/new.html
Normal 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
12
templates/success.html
Normal 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>
|
||||
4
tx/expense.txt
Normal file
4
tx/expense.txt
Normal file
@@ -0,0 +1,4 @@
|
||||
{{ .Date }} {{ with .Name }}{{ . }}{{ else }}{{ .Account }}{{ end }}
|
||||
{{ .Account }} ${{ .Amount }}
|
||||
cash
|
||||
|
||||
Reference in New Issue
Block a user