Merge branch '16314-testuserdb'
[arvados.git] / services / keep-web / main.go
1 // Copyright (C) The Arvados Authors. All rights reserved.
2 //
3 // SPDX-License-Identifier: AGPL-3.0
4
5 package main
6
7 import (
8         "flag"
9         "fmt"
10         "mime"
11         "os"
12
13         "git.arvados.org/arvados.git/lib/config"
14         "git.arvados.org/arvados.git/sdk/go/arvados"
15         "github.com/coreos/go-systemd/daemon"
16         "github.com/ghodss/yaml"
17         "github.com/sirupsen/logrus"
18         log "github.com/sirupsen/logrus"
19 )
20
21 var (
22         version = "dev"
23 )
24
25 // Config specifies server configuration.
26 type Config struct {
27         Client  arvados.Client
28         Cache   cache
29         cluster *arvados.Cluster
30 }
31
32 func newConfig(arvCfg *arvados.Config) *Config {
33         cfg := Config{}
34         var cls *arvados.Cluster
35         var err error
36         if cls, err = arvCfg.GetCluster(""); err != nil {
37                 log.Fatal(err)
38         }
39         cfg.cluster = cls
40         cfg.Cache.config = &cfg.cluster.Collections.WebDAVCache
41         return &cfg
42 }
43
44 func init() {
45         // MakeArvadosClient returns an error if this env var isn't
46         // available as a default token (even if we explicitly set a
47         // different token before doing anything with the client). We
48         // set this dummy value during init so it doesn't clobber the
49         // one used by "run test servers".
50         if os.Getenv("ARVADOS_API_TOKEN") == "" {
51                 os.Setenv("ARVADOS_API_TOKEN", "xxx")
52         }
53
54         log.SetFormatter(&log.JSONFormatter{
55                 TimestampFormat: "2006-01-02T15:04:05.000000000Z07:00",
56         })
57 }
58
59 func configure(logger log.FieldLogger, args []string) *Config {
60         flags := flag.NewFlagSet(args[0], flag.ExitOnError)
61
62         loader := config.NewLoader(os.Stdin, logger)
63         loader.SetupFlags(flags)
64
65         dumpConfig := flags.Bool("dump-config", false,
66                 "write current configuration to stdout and exit")
67         getVersion := flags.Bool("version", false,
68                 "print version information and exit.")
69
70         args = loader.MungeLegacyConfigArgs(logger, args[1:], "-legacy-keepweb-config")
71         flags.Parse(args)
72
73         // Print version information if requested
74         if *getVersion {
75                 fmt.Printf("keep-web %s\n", version)
76                 return nil
77         }
78
79         arvCfg, err := loader.Load()
80         if err != nil {
81                 log.Fatal(err)
82         }
83         cfg := newConfig(arvCfg)
84
85         if *dumpConfig {
86                 out, err := yaml.Marshal(cfg)
87                 if err != nil {
88                         log.Fatal(err)
89                 }
90                 _, err = os.Stdout.Write(out)
91                 if err != nil {
92                         log.Fatal(err)
93                 }
94                 return nil
95         }
96         return cfg
97 }
98
99 func main() {
100         logger := log.New()
101
102         cfg := configure(logger, os.Args)
103         if cfg == nil {
104                 return
105         }
106
107         log.Printf("keep-web %s started", version)
108
109         if ext := ".txt"; mime.TypeByExtension(ext) == "" {
110                 log.Warnf("cannot look up MIME type for %q -- this probably means /etc/mime.types is missing -- clients will see incorrect content types", ext)
111         }
112
113         os.Setenv("ARVADOS_API_HOST", cfg.cluster.Services.Controller.ExternalURL.Host)
114         srv := &server{Config: cfg}
115         if err := srv.Start(logrus.StandardLogger()); err != nil {
116                 log.Fatal(err)
117         }
118         if _, err := daemon.SdNotify(false, "READY=1"); err != nil {
119                 log.Printf("Error notifying init daemon: %v", err)
120         }
121         log.Println("Listening at", srv.Addr)
122         if err := srv.Wait(); err != nil {
123                 log.Fatal(err)
124         }
125 }