158 lines
3.3 KiB
Go
158 lines
3.3 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"path"
|
|
"strings"
|
|
"text/template"
|
|
"time"
|
|
|
|
cp "github.com/otiai10/copy"
|
|
)
|
|
|
|
type User struct {
|
|
IsLogin bool
|
|
Email string `form:"email" binding:"required"`
|
|
Password string `form:"password" binding:"required"`
|
|
}
|
|
|
|
func (u *User) Dir() string {
|
|
dir := path.Join(DATA_DIR, u.Email)
|
|
return dir
|
|
}
|
|
|
|
func (u *User) Mkdir() error {
|
|
return cp.Copy(ARCHETYPES_DIR, u.Dir())
|
|
}
|
|
|
|
func (u *User) FilePath(name string) string {
|
|
return path.Join(u.Dir(), name)
|
|
}
|
|
|
|
func (u *User) File(name string, mode int) (*os.File, error) {
|
|
return os.OpenFile(u.FilePath(name), mode, 0644)
|
|
}
|
|
|
|
func (u *User) AppendFile(name string) (*os.File, error) {
|
|
return u.File(name, os.O_WRONLY|os.O_CREATE|os.O_APPEND)
|
|
}
|
|
|
|
func (u *User) ReadFile(name string) (*os.File, error) {
|
|
return u.File(name, os.O_RDONLY|os.O_CREATE)
|
|
}
|
|
|
|
func (u *User) WriteFile(name string) (*os.File, error) {
|
|
return u.File(name, os.O_WRONLY|os.O_CREATE|os.O_TRUNC)
|
|
}
|
|
|
|
func (u *User) List() ([]string, error) {
|
|
files, err := os.ReadDir(u.Dir())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
result := make([]string, len(files))
|
|
for i, v := range files {
|
|
result[i] = v.Name()
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func (u *User) appendToFile(tx string) (err error) {
|
|
f, err := u.AppendFile(DEFAULT_JOURNAL)
|
|
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
|
|
}
|
|
|
|
func (u *User) overwriteFile(filename string, tx string) (err error) {
|
|
f, err := u.WriteFile(filename)
|
|
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
|
|
}
|
|
|
|
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.Stdin = strings.NewReader(query)
|
|
cmd.Stdout = &buf
|
|
cmd.Stderr = &buf
|
|
err = cmd.Run()
|
|
return buf.String(), err
|
|
}
|
|
|
|
func (u *User) scripts() (scripts map[string]string, err error) {
|
|
f, err := u.ReadFile(SCRIPTS_FILE)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer f.Close()
|
|
|
|
fileScanner := bufio.NewScanner(f)
|
|
fileScanner.Split(bufio.ScanLines)
|
|
|
|
scripts = make(map[string]string)
|
|
for fileScanner.Scan() {
|
|
arr := strings.SplitN(fileScanner.Text(), ":", 2)
|
|
if len(arr) < 2 {
|
|
err = fmt.Errorf("invalid data %s", arr)
|
|
return
|
|
}
|
|
scripts[arr[0]] = arr[1]
|
|
}
|
|
return
|
|
}
|
|
|
|
func (u *User) templates() (templates []string, err error) {
|
|
files, err := u.List()
|
|
if err != nil {
|
|
return
|
|
}
|
|
for _, v := range files {
|
|
if strings.HasSuffix(v, ".tpl") {
|
|
templates = append(templates, v)
|
|
}
|
|
}
|
|
log.Println(templates)
|
|
return
|
|
}
|
|
|
|
type TxData struct {
|
|
Action string `form:"action" binding:"required"`
|
|
Name string `form:"name"`
|
|
Date string
|
|
Amount string `form:"amount" binding:"required"`
|
|
Destination string `form:"dest"`
|
|
Source string `form:"src"`
|
|
}
|
|
|
|
func (u *User) newTx(data TxData) (result string, err error) {
|
|
data.Date = time.Now().Format("2006/01/02")
|
|
var buf bytes.Buffer
|
|
tpl, err := template.ParseFiles(u.FilePath(data.Action))
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = tpl.ExecuteTemplate(&buf, data.Action, data)
|
|
result = buf.String()
|
|
return
|
|
}
|