mirror of https://github.com/aclindsa/moneygo.git
103 lines
2.6 KiB
Go
103 lines
2.6 KiB
Go
package main
|
|
|
|
//go:generate make
|
|
|
|
import (
|
|
"flag"
|
|
"gopkg.in/gorp.v1"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"net/http/fcgi"
|
|
"os"
|
|
"path"
|
|
"strconv"
|
|
)
|
|
|
|
var configFile string
|
|
var config *Config
|
|
|
|
func init() {
|
|
var err error
|
|
flag.StringVar(&configFile, "config", "/etc/moneygo/config.ini", "Path to config file")
|
|
flag.Parse()
|
|
|
|
config, err = readConfig(configFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
static_path := path.Join(config.MoneyGo.Basedir, "static")
|
|
|
|
// Ensure base directory is valid
|
|
dir_err_str := "The base directory doesn't look like it contains the " +
|
|
"'static' directory. Check to make sure your config file contains the" +
|
|
"right path for 'base-directory'."
|
|
static_dir, err := os.Stat(static_path)
|
|
if err != nil {
|
|
log.Print(err)
|
|
log.Fatal(dir_err_str)
|
|
}
|
|
if !static_dir.IsDir() {
|
|
log.Fatal(dir_err_str)
|
|
}
|
|
|
|
// Setup the logging flags to be printed
|
|
log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
|
|
}
|
|
|
|
func rootHandler(w http.ResponseWriter, r *http.Request) {
|
|
http.ServeFile(w, r, path.Join(config.MoneyGo.Basedir, "static/index.html"))
|
|
}
|
|
|
|
func staticHandler(w http.ResponseWriter, r *http.Request) {
|
|
http.ServeFile(w, r, path.Join(config.MoneyGo.Basedir, r.URL.Path))
|
|
}
|
|
|
|
// Create a closure over db, allowing the handlers to look like a
|
|
// http.HandlerFunc
|
|
type DB = gorp.DbMap
|
|
type DBHandler func(http.ResponseWriter, *http.Request, *DB)
|
|
|
|
func DBHandlerFunc(h DBHandler, db *DB) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
h(w, r, db)
|
|
}
|
|
}
|
|
|
|
func GetHandler(db *DB) http.Handler {
|
|
servemux := http.NewServeMux()
|
|
servemux.HandleFunc("/", rootHandler)
|
|
servemux.HandleFunc("/static/", staticHandler)
|
|
servemux.HandleFunc("/session/", DBHandlerFunc(SessionHandler, db))
|
|
servemux.HandleFunc("/user/", DBHandlerFunc(UserHandler, db))
|
|
servemux.HandleFunc("/security/", DBHandlerFunc(SecurityHandler, db))
|
|
servemux.HandleFunc("/securitytemplate/", SecurityTemplateHandler)
|
|
servemux.HandleFunc("/account/", DBHandlerFunc(AccountHandler, db))
|
|
servemux.HandleFunc("/transaction/", DBHandlerFunc(TransactionHandler, db))
|
|
servemux.HandleFunc("/import/gnucash", DBHandlerFunc(GnucashImportHandler, db))
|
|
servemux.HandleFunc("/report/", DBHandlerFunc(ReportHandler, db))
|
|
|
|
return servemux
|
|
}
|
|
|
|
func main() {
|
|
database, err := initDB(config)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
handler := GetHandler(database)
|
|
|
|
listener, err := net.Listen("tcp", ":"+strconv.Itoa(config.MoneyGo.Port))
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
log.Printf("Serving on port %d out of directory: %s", config.MoneyGo.Port, config.MoneyGo.Basedir)
|
|
if config.MoneyGo.Fcgi {
|
|
fcgi.Serve(listener, handler)
|
|
} else {
|
|
http.Serve(listener, handler)
|
|
}
|
|
}
|