[98] | 1 | package main
|
---|
| 2 |
|
---|
| 3 | import (
|
---|
| 4 | "crypto/tls"
|
---|
| 5 | "flag"
|
---|
| 6 | "log"
|
---|
| 7 | "net"
|
---|
| 8 |
|
---|
| 9 | "git.sr.ht/~emersion/soju"
|
---|
| 10 | "git.sr.ht/~emersion/soju/config"
|
---|
| 11 | )
|
---|
| 12 |
|
---|
| 13 | func main() {
|
---|
| 14 | var addr, configPath string
|
---|
| 15 | var debug bool
|
---|
| 16 | flag.StringVar(&addr, "listen", "", "listening address")
|
---|
| 17 | flag.StringVar(&configPath, "config", "", "path to configuration file")
|
---|
| 18 | flag.BoolVar(&debug, "debug", false, "enable debug logging")
|
---|
| 19 | flag.Parse()
|
---|
| 20 |
|
---|
| 21 | var cfg *config.Server
|
---|
| 22 | if configPath != "" {
|
---|
| 23 | var err error
|
---|
| 24 | cfg, err = config.Load(configPath)
|
---|
| 25 | if err != nil {
|
---|
| 26 | log.Fatalf("failed to load config file: %v", err)
|
---|
| 27 | }
|
---|
| 28 | } else {
|
---|
| 29 | cfg = config.Defaults()
|
---|
| 30 | }
|
---|
| 31 |
|
---|
| 32 | if addr != "" {
|
---|
| 33 | cfg.Addr = addr
|
---|
| 34 | }
|
---|
| 35 |
|
---|
| 36 | db, err := soju.OpenSQLDB(cfg.SQLDriver, cfg.SQLSource)
|
---|
| 37 | if err != nil {
|
---|
| 38 | log.Fatalf("failed to open database: %v", err)
|
---|
| 39 | }
|
---|
| 40 |
|
---|
| 41 | var ln net.Listener
|
---|
| 42 | if cfg.TLS != nil {
|
---|
| 43 | cert, err := tls.LoadX509KeyPair(cfg.TLS.CertPath, cfg.TLS.KeyPath)
|
---|
| 44 | if err != nil {
|
---|
| 45 | log.Fatalf("failed to load TLS certificate and key: %v", err)
|
---|
| 46 | }
|
---|
| 47 |
|
---|
| 48 | tlsCfg := &tls.Config{Certificates: []tls.Certificate{cert}}
|
---|
| 49 | ln, err = tls.Listen("tcp", cfg.Addr, tlsCfg)
|
---|
| 50 | if err != nil {
|
---|
| 51 | log.Fatalf("failed to start TLS listener: %v", err)
|
---|
| 52 | }
|
---|
| 53 | } else {
|
---|
| 54 | var err error
|
---|
| 55 | ln, err = net.Listen("tcp", cfg.Addr)
|
---|
| 56 | if err != nil {
|
---|
| 57 | log.Fatalf("failed to start listener: %v", err)
|
---|
| 58 | }
|
---|
| 59 | }
|
---|
| 60 |
|
---|
| 61 | srv := soju.NewServer(db)
|
---|
| 62 | // TODO: load from config/DB
|
---|
| 63 | srv.Hostname = cfg.Hostname
|
---|
| 64 | srv.Debug = debug
|
---|
| 65 |
|
---|
| 66 | log.Printf("server listening on %q", cfg.Addr)
|
---|
| 67 | go func() {
|
---|
| 68 | if err := srv.Run(); err != nil {
|
---|
| 69 | log.Fatal(err)
|
---|
| 70 | }
|
---|
| 71 | }()
|
---|
| 72 | log.Fatal(srv.Serve(ln))
|
---|
| 73 | }
|
---|