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.LogPath = cfg.LogPath
|
---|
65 | srv.Debug = debug
|
---|
66 |
|
---|
67 | log.Printf("server listening on %q", cfg.Addr)
|
---|
68 | go func() {
|
---|
69 | if err := srv.Run(); err != nil {
|
---|
70 | log.Fatal(err)
|
---|
71 | }
|
---|
72 | }()
|
---|
73 | log.Fatal(srv.Serve(ln))
|
---|
74 | }
|
---|