mirror of https://github.com/aclindsa/moneygo.git
107 lines
3.0 KiB
Go
107 lines
3.0 KiB
Go
package main
|
|
|
|
//go:generate make
|
|
|
|
import (
|
|
"flag"
|
|
"github.com/aclindsa/moneygo/internal/config"
|
|
"github.com/aclindsa/moneygo/internal/handlers"
|
|
"github.com/aclindsa/moneygo/internal/store/db"
|
|
"github.com/kabukky/httpscerts"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"net/http/fcgi"
|
|
"os"
|
|
"path"
|
|
"strconv"
|
|
)
|
|
|
|
var configFile string
|
|
var cfg *config.Config
|
|
|
|
func init() {
|
|
var err error
|
|
flag.StringVar(&configFile, "config", "/etc/moneygo/config.ini", "Path to config file")
|
|
flag.Parse()
|
|
|
|
cfg, err = config.ReadConfig(configFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
static_path := path.Join(cfg.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)
|
|
}
|
|
|
|
type FileHandler func(http.ResponseWriter, *http.Request, string)
|
|
|
|
func FileHandlerFunc(h FileHandler, basedir string) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
h(w, r, basedir)
|
|
}
|
|
}
|
|
|
|
func rootHandler(w http.ResponseWriter, r *http.Request, basedir string) {
|
|
http.ServeFile(w, r, path.Join(basedir, "static/index.html"))
|
|
}
|
|
|
|
func staticHandler(w http.ResponseWriter, r *http.Request, basedir string) {
|
|
http.ServeFile(w, r, path.Join(basedir, r.URL.Path))
|
|
}
|
|
|
|
func main() {
|
|
db, err := db.GetStore(cfg.MoneyGo.DBType, cfg.MoneyGo.DSN)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer db.Close()
|
|
|
|
// Get ServeMux for API and add our own handlers for files
|
|
servemux := http.NewServeMux()
|
|
servemux.Handle("/v1/", &handlers.APIHandler{Store: db})
|
|
servemux.HandleFunc("/", FileHandlerFunc(rootHandler, cfg.MoneyGo.Basedir))
|
|
servemux.HandleFunc("/static/", FileHandlerFunc(staticHandler, cfg.MoneyGo.Basedir))
|
|
|
|
listener, err := net.Listen("tcp", ":"+strconv.Itoa(cfg.MoneyGo.Port))
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
if cfg.MoneyGo.Fcgi {
|
|
log.Printf("Serving via FCGI on port %d out of directory: %s", cfg.MoneyGo.Port, cfg.MoneyGo.Basedir)
|
|
fcgi.Serve(listener, servemux)
|
|
} else {
|
|
cert := cfg.Https.CertFile
|
|
key := cfg.Https.KeyFile
|
|
|
|
if err := httpscerts.Check(cert, key); err != nil {
|
|
if !cfg.Https.GenerateCerts {
|
|
log.Fatalf("HTTPS certficates not found at '%s' and '%s'. If you would like for them to be auto-generated for you, specify 'generate-certs-if-absent = true' in your config file at '%s'", cert, key, configFile)
|
|
}
|
|
|
|
err = httpscerts.Generate(cert, key, cfg.Https.GenerateCertsHosts)
|
|
if err != nil {
|
|
log.Fatalf("Error: Generating HTTPS cert/key at '%s' and '%s' failed: %s", cert, key, err)
|
|
}
|
|
}
|
|
log.Printf("Serving via HTTPS on port %d out of directory: %s", cfg.MoneyGo.Port, cfg.MoneyGo.Basedir)
|
|
http.ServeTLS(listener, servemux, cert, key)
|
|
}
|
|
}
|