113 lines
2.2 KiB
Go
113 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
type Config struct {
|
|
Admins []string
|
|
Port int
|
|
Host string
|
|
InstanceName string `yaml:"instance_name"`
|
|
AllowAnon bool `yaml:"allow_anon"`
|
|
Debug bool
|
|
DBPath string `yaml:"db_path"`
|
|
}
|
|
|
|
type iostreams struct {
|
|
Err io.Writer
|
|
Out io.Writer
|
|
}
|
|
|
|
type Opts struct {
|
|
ConfigPath string
|
|
IO iostreams
|
|
Log func(string)
|
|
Logf func(string, ...interface{})
|
|
Config Config
|
|
DB *sql.DB
|
|
}
|
|
|
|
func main() {
|
|
var configFlag = flag.String("config", "config.yml", "A path to a config file.")
|
|
flag.Parse()
|
|
io := iostreams{
|
|
Err: os.Stderr,
|
|
Out: os.Stdout,
|
|
}
|
|
opts := &Opts{
|
|
ConfigPath: *configFlag,
|
|
IO: io,
|
|
// TODO use real logger
|
|
Log: func(s string) {
|
|
fmt.Fprintln(io.Out, s)
|
|
},
|
|
Logf: func(s string, args ...interface{}) {
|
|
fmt.Fprintf(io.Out, s, args...)
|
|
},
|
|
}
|
|
|
|
err := _main(opts)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "failed: %s", err)
|
|
}
|
|
}
|
|
|
|
type Teardown func()
|
|
|
|
func setupDB(opts *Opts) (Teardown, error) {
|
|
db, err := sql.Open("sqlite3", opts.Config.DBPath)
|
|
fmt.Printf("DBG %#v\n", db)
|
|
|
|
opts.DB = db
|
|
|
|
return func() { db.Close() }, err
|
|
}
|
|
|
|
func _main(opts *Opts) error {
|
|
cfg, err := parseConfig(opts.ConfigPath)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "could not read config file '%s'", opts.ConfigPath)
|
|
os.Exit(1)
|
|
}
|
|
|
|
opts.Config = *cfg
|
|
|
|
teardown, err := setupDB(opts)
|
|
if err != nil {
|
|
return fmt.Errorf("could not initialize DB: %w", err)
|
|
}
|
|
defer teardown()
|
|
|
|
setupAPI(*opts)
|
|
|
|
// TODO TLS or SSL or something
|
|
opts.Logf("starting server at %s:%d\n", cfg.Host, cfg.Port)
|
|
if err := http.ListenAndServe(fmt.Sprintf("%s:%d", cfg.Host, cfg.Port), nil); err != nil {
|
|
return fmt.Errorf("http server exited with error: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func handler(opts Opts, f http.HandlerFunc) http.HandlerFunc {
|
|
// TODO make this more real
|
|
return func(w http.ResponseWriter, req *http.Request) {
|
|
opts.Log(req.URL.Path)
|
|
f(w, req)
|
|
}
|
|
}
|
|
|
|
func setupAPI(opts Opts) {
|
|
http.HandleFunc("/instance", handler(opts, func(w http.ResponseWriter, req *http.Request) {
|
|
io.WriteString(w, opts.Config.InstanceName)
|
|
}))
|
|
}
|