1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
12 "git.curoverse.com/arvados.git/lib/config"
13 "git.curoverse.com/arvados.git/sdk/go/arvados"
14 "github.com/coreos/go-systemd/daemon"
15 "github.com/ghodss/yaml"
16 log "github.com/sirupsen/logrus"
23 // Config specifies server configuration.
27 cluster *arvados.Cluster
30 // DefaultConfig returns the default configuration.
31 func DefaultConfig(arvCfg *arvados.Config) *Config {
33 var cls *arvados.Cluster
35 if cls, err = arvCfg.GetCluster(""); err != nil {
39 cfg.Cache.config = &cfg.cluster.Collections.WebDAVCache
44 // MakeArvadosClient returns an error if this env var isn't
45 // available as a default token (even if we explicitly set a
46 // different token before doing anything with the client). We
47 // set this dummy value during init so it doesn't clobber the
48 // one used by "run test servers".
49 if os.Getenv("ARVADOS_API_TOKEN") == "" {
50 os.Setenv("ARVADOS_API_TOKEN", "xxx")
53 log.SetFormatter(&log.JSONFormatter{
54 TimestampFormat: "2006-01-02T15:04:05.000000000Z07:00",
58 func configure(logger log.FieldLogger, args []string) *Config {
59 flags := flag.NewFlagSet(args[0], flag.ExitOnError)
61 loader := config.NewLoader(os.Stdin, logger)
62 loader.SetupFlags(flags)
64 dumpConfig := flags.Bool("dump-config", false,
65 "write current configuration to stdout and exit")
66 getVersion := flags.Bool("version", false,
67 "print version information and exit.")
69 args = loader.MungeLegacyConfigArgs(logger, args[1:], "-legacy-keepweb-config")
72 // Print version information if requested
74 fmt.Printf("keep-web %s\n", version)
78 arvCfg, err := loader.Load()
82 cfg := DefaultConfig(arvCfg)
85 out, err := yaml.Marshal(cfg)
89 _, err = os.Stdout.Write(out)
101 cfg := configure(logger, os.Args)
106 log.Printf("keep-web %s started", version)
108 os.Setenv("ARVADOS_API_HOST", cfg.cluster.Services.Controller.ExternalURL.Host)
109 srv := &server{Config: cfg}
110 if err := srv.Start(); err != nil {
113 if _, err := daemon.SdNotify(false, "READY=1"); err != nil {
114 log.Printf("Error notifying init daemon: %v", err)
116 log.Println("Listening at", srv.Addr)
117 if err := srv.Wait(); err != nil {