mirror of
https://github.com/aclindsa/moneygo.git
synced 2024-10-30 07:40:05 -04:00
Aaron Lindsay
147a00e429
Because MoneyGo requires sending passwords and session cookies, we should never serve over HTTP. While we're at it, make it more convenient for folks to test this out by adding a config option to auto-generate self-signed certificates.
114 lines
3.1 KiB
Go
114 lines
3.1 KiB
Go
package main
|
|
|
|
//go:generate make
|
|
|
|
import (
|
|
"database/sql"
|
|
"flag"
|
|
"github.com/aclindsa/moneygo/internal/config"
|
|
"github.com/aclindsa/moneygo/internal/db"
|
|
"github.com/aclindsa/moneygo/internal/handlers"
|
|
"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() {
|
|
dsn := db.GetDSN(cfg.MoneyGo.DBType, cfg.MoneyGo.DSN)
|
|
database, err := sql.Open(cfg.MoneyGo.DBType.String(), dsn)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer database.Close()
|
|
|
|
dbmap, err := db.GetDbMap(database, cfg.MoneyGo.DBType)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// Get ServeMux for API and add our own handlers for files
|
|
servemux := http.NewServeMux()
|
|
servemux.Handle("/v1/", &handlers.APIHandler{DB: dbmap})
|
|
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)
|
|
}
|
|
}
|