1
0
mirror of https://github.com/aclindsa/moneygo.git synced 2024-10-30 07:40:05 -04:00

Add initial implementation of accounts

This commit is contained in:
Aaron Lindsay 2015-06-27 17:46:06 -04:00
parent 5b71c181fc
commit 3367106e41
2 changed files with 247 additions and 5 deletions

View File

@ -1,5 +1,13 @@
package main package main
import (
"encoding/json"
"errors"
"log"
"net/http"
"strings"
)
type AccountType int64 type AccountType int64
const ( const (
@ -13,9 +21,242 @@ const (
) )
type Account struct { type Account struct {
AccountId int64 AccountId int64
UserId int64 UserId int64
SecurityId int64 SecurityId int64
Type AccountType ParentAccountId int64 // -1 if this account is at the root
Name string Type AccountType
Name string
}
type AccountList struct {
Accounts *[]Account `json:"accounts"`
}
func (a *Account) Write(w http.ResponseWriter) error {
enc := json.NewEncoder(w)
return enc.Encode(a)
}
func (a *Account) Read(json_str string) error {
dec := json.NewDecoder(strings.NewReader(json_str))
return dec.Decode(a)
}
func (al *AccountList) Write(w http.ResponseWriter) error {
enc := json.NewEncoder(w)
return enc.Encode(al)
}
func GetAccount(accountid int64, userid int64) (*Account, error) {
var a Account
err := DB.SelectOne(&a, "SELECT * from accounts where UserId=? AND AccountId=?", userid, accountid)
if err != nil {
return nil, err
}
return &a, nil
}
func GetAccounts(userid int64) (*[]Account, error) {
var accounts []Account
_, err := DB.Select(&accounts, "SELECT * from accounts where UserId=?", userid)
if err != nil {
return nil, err
}
return &accounts, nil
}
type ParentAccountMissingError struct{}
func (pame ParentAccountMissingError) Error() string {
return "Parent account missing"
}
func insertUpdateAccount(a *Account, insert bool) error {
transaction, err := DB.Begin()
if err != nil {
return err
}
if a.ParentAccountId != -1 {
existing, err := transaction.SelectInt("SELECT count(*) from accounts where AccountId=?", a.ParentAccountId)
if err != nil {
transaction.Rollback()
return err
}
if existing != 1 {
transaction.Rollback()
return ParentAccountMissingError{}
}
}
if insert {
err = transaction.Insert(a)
if err != nil {
transaction.Rollback()
return err
}
} else {
count, err := transaction.Update(a)
if err != nil {
transaction.Rollback()
return err
}
if count != 1 {
transaction.Rollback()
return errors.New("Updated more than one account")
}
}
err = transaction.Commit()
if err != nil {
transaction.Rollback()
return err
}
return nil
}
func InsertAccount(a *Account) error {
return insertUpdateAccount(a, true)
}
func UpdateAccount(a *Account) error {
return insertUpdateAccount(a, false)
}
func AccountHandler(w http.ResponseWriter, r *http.Request) {
user, err := GetUserFromSession(r)
if err != nil {
WriteError(w, 1 /*Not Signed In*/)
return
}
if r.Method == "POST" {
account_json := r.PostFormValue("account")
if account_json == "" {
WriteError(w, 3 /*Invalid Request*/)
return
}
var account Account
err := account.Read(account_json)
if err != nil {
WriteError(w, 3 /*Invalid Request*/)
return
}
account.AccountId = -1
account.UserId = user.UserId
if GetSecurity(account.SecurityId) == nil {
WriteError(w, 3 /*Invalid Request*/)
return
}
err = InsertAccount(&account)
if err != nil {
if _, ok := err.(ParentAccountMissingError); ok {
WriteError(w, 3 /*Invalid Request*/)
} else {
WriteError(w, 999 /*Internal Error*/)
log.Print(err)
}
return
WriteError(w, 999 /*Internal Error*/)
log.Print(err)
return
}
WriteSuccess(w)
} else if r.Method == "GET" {
accountid, err := GetURLID(r.URL.Path)
if err != nil {
//Return all Accounts
var al AccountList
accounts, err := GetAccounts(user.UserId)
if err != nil {
WriteError(w, 999 /*Internal Error*/)
log.Print(err)
return
}
al.Accounts = accounts
err = (&al).Write(w)
if err != nil {
WriteError(w, 999 /*Internal Error*/)
log.Print(err)
return
}
} else {
//Return Account with this Id
account, err := GetAccount(accountid, user.UserId)
if err != nil {
WriteError(w, 3 /*Invalid Request*/)
return
}
err = account.Write(w)
if err != nil {
WriteError(w, 999 /*Internal Error*/)
log.Print(err)
return
}
}
} else {
accountid, err := GetURLID(r.URL.Path)
if err != nil {
WriteError(w, 3 /*Invalid Request*/)
return
}
if r.Method == "PUT" {
account_json := r.PostFormValue("account")
if account_json == "" {
WriteError(w, 3 /*Invalid Request*/)
return
}
var account Account
err := account.Read(account_json)
if err != nil || account.AccountId != accountid {
WriteError(w, 3 /*Invalid Request*/)
return
}
account.UserId = user.UserId
if GetSecurity(account.SecurityId) == nil {
WriteError(w, 3 /*Invalid Request*/)
return
}
err = UpdateAccount(&account)
if err != nil {
WriteError(w, 999 /*Internal Error*/)
log.Print(err)
return
}
WriteSuccess(w)
} else if r.Method == "DELETE" {
accountid, err := GetURLID(r.URL.Path)
if err != nil {
WriteError(w, 3 /*Invalid Request*/)
return
}
account, err := GetAccount(accountid, user.UserId)
if err != nil {
WriteError(w, 3 /*Invalid Request*/)
return
}
count, err := DB.Delete(&account)
if count != 1 || err != nil {
WriteError(w, 999 /*Internal Error*/)
log.Print(err)
return
}
WriteSuccess(w)
}
}
} }

View File

@ -63,6 +63,7 @@ func main() {
servemux.HandleFunc("/session/", SessionHandler) servemux.HandleFunc("/session/", SessionHandler)
servemux.HandleFunc("/user/", UserHandler) servemux.HandleFunc("/user/", UserHandler)
servemux.HandleFunc("/security/", SecurityHandler) servemux.HandleFunc("/security/", SecurityHandler)
servemux.HandleFunc("/account/", AccountHandler)
listener, err := net.Listen("tcp", ":"+strconv.Itoa(port)) listener, err := net.Listen("tcp", ":"+strconv.Itoa(port))
if err != nil { if err != nil {