moneygo/main.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)
	}
}