122 lines
2.7 KiB
Go
122 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"flag"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
"text/template"
|
|
"time"
|
|
)
|
|
|
|
var ledgerTpl *template.Template
|
|
var htmlTpl *template.Template
|
|
|
|
var LEDGER_FILE string
|
|
var LEDGER_INIT string
|
|
|
|
type TxData struct {
|
|
Name string
|
|
Date string
|
|
Amount string
|
|
Account string
|
|
}
|
|
|
|
func init() {
|
|
ledgerTpl = template.Must(template.ParseGlob("tx/*"))
|
|
htmlTpl = template.Must(template.ParseGlob("templates/*.html"))
|
|
flag.StringVar(&LEDGER_FILE, "f", "example.txt", "ledger journal file")
|
|
flag.StringVar(&LEDGER_INIT, "i", "~/.ledgerrc", "ledger initiation file")
|
|
flag.Parse()
|
|
}
|
|
|
|
func main() {
|
|
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
htmlTpl.ExecuteTemplate(w, "index.html", struct {
|
|
Templates []*template.Template
|
|
Scripts map[string][]string
|
|
}{
|
|
ledgerTpl.Templates(),
|
|
SCRIPTS,
|
|
})
|
|
})
|
|
|
|
http.HandleFunc("/new", func(w http.ResponseWriter, r *http.Request) {
|
|
if err := r.ParseForm(); err != nil {
|
|
http.Error(w, err.Error(), 400)
|
|
return
|
|
}
|
|
tx, err := newTx(r.Form)
|
|
if err != nil {
|
|
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)
|
|
}
|
|
})
|
|
|
|
http.HandleFunc("/exec", func(w http.ResponseWriter, r *http.Request) {
|
|
name := r.FormValue("name")
|
|
if err := executeScript(w, name); err != nil {
|
|
http.Error(w, err.Error(), 500)
|
|
log.Println(err)
|
|
return
|
|
}
|
|
})
|
|
|
|
log.Println("Listen on http://localhost:8000")
|
|
log.Fatal(http.ListenAndServe(":8000", nil))
|
|
}
|
|
|
|
func newTx(params url.Values) (result string, err error) {
|
|
action := 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, action, 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 {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
|
|
buf := strings.NewReader(strings.ReplaceAll(tx, "\r", "")) // Remove CR generated from browser
|
|
_, err = io.Copy(f, buf)
|
|
return err
|
|
}
|