1
0
mirror of https://github.com/aclindsa/moneygo.git synced 2024-10-30 07:40:05 -04:00
moneygo/internal/handlers/handlers.go
Aaron Lindsay 04d85cd681 Fix gorp to internally lowercase fields for Postgres
Postgres folds all unquoted column names to lowercase, and doing this
allows that to work nicely without screwing up the queries for the other
database engines.
2017-11-17 21:01:06 -05:00

111 lines
2.3 KiB
Go

package handlers
import (
"github.com/aclindsa/gorp"
"log"
"net/http"
"path"
"strconv"
"strings"
)
// But who writes the ResponseWriterWriter?
type ResponseWriterWriter interface {
Write(http.ResponseWriter) error
}
type Context struct {
Tx *Tx
User *User
remainingURL string // portion of URL path not yet reached in the hierarchy
}
func (c *Context) SetURL(url string) {
c.remainingURL = path.Clean("/" + url)[1:]
}
func (c *Context) NextLevel() string {
split := strings.SplitN(c.remainingURL, "/", 2)
if len(split) == 2 {
c.remainingURL = split[1]
} else {
c.remainingURL = ""
}
return split[0]
}
func (c *Context) NextID() (int64, error) {
return strconv.ParseInt(c.NextLevel(), 0, 64)
}
func (c *Context) LastLevel() bool {
return len(c.remainingURL) == 0
}
type Handler func(*http.Request, *Context) ResponseWriterWriter
type APIHandler struct {
DB *gorp.DbMap
}
func (ah *APIHandler) txWrapper(h Handler, r *http.Request, context *Context) (writer ResponseWriterWriter) {
tx, err := GetTx(ah.DB)
if err != nil {
log.Print(err)
return NewError(999 /*Internal Error*/)
}
defer func() {
if r := recover(); r != nil {
tx.Rollback()
panic(r)
}
if _, ok := writer.(*Error); ok {
tx.Rollback()
} else {
err = tx.Commit()
if err != nil {
log.Print(err)
writer = NewError(999 /*Internal Error*/)
}
}
}()
context.Tx = tx
return h(r, context)
}
func (ah *APIHandler) route(r *http.Request) ResponseWriterWriter {
context := &Context{}
context.SetURL(r.URL.Path)
if context.NextLevel() != "v1" {
return NewError(3 /*Invalid Request*/)
}
route := context.NextLevel()
switch route {
case "sessions":
return ah.txWrapper(SessionHandler, r, context)
case "users":
return ah.txWrapper(UserHandler, r, context)
case "securities":
return ah.txWrapper(SecurityHandler, r, context)
case "securitytemplates":
return SecurityTemplateHandler(r, context)
case "accounts":
return ah.txWrapper(AccountHandler, r, context)
case "transactions":
return ah.txWrapper(TransactionHandler, r, context)
case "imports":
return ah.txWrapper(ImportHandler, r, context)
case "reports":
return ah.txWrapper(ReportHandler, r, context)
default:
return NewError(3 /*Invalid Request*/)
}
}
func (ah *APIHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
ah.route(r).Write(w)
}