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(logger logrus.FieldLogger, 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         cfg.Cache.cluster = cls
  42         cfg.Cache.logger = logger
  43         return &cfg
  44 }
  45
  46 func init() {
  47         // MakeArvadosClient returns an error if this env var isn't
  48         // available as a default token (even if we explicitly set a
  49         // different token before doing anything with the client). We
  50         // set this dummy value during init so it doesn't clobber the
  51         // one used by "run test servers".
  52         if os.Getenv("ARVADOS_API_TOKEN") == "" {
  53                 os.Setenv("ARVADOS_API_TOKEN", "xxx")
  54         }
  55
  56         log.SetFormatter(&log.JSONFormatter{
  57                 TimestampFormat: "2006-01-02T15:04:05.000000000Z07:00",
  58         })
  59 }
  60
  61 func configure(logger log.FieldLogger, args []string) (*Config, error) {
  62         flags := flag.NewFlagSet(args[0], flag.ContinueOnError)
  63
  64         loader := config.NewLoader(os.Stdin, logger)
  65         loader.SetupFlags(flags)
  66
  67         dumpConfig := flags.Bool("dump-config", false,
  68                 "write current configuration to stdout and exit")
  69         getVersion := flags.Bool("version", false,
  70                 "print version information and exit.")
  71
  72         args = loader.MungeLegacyConfigArgs(logger, args[1:], "-legacy-keepweb-config")
  73         err := flags.Parse(args)
  74         if err == flag.ErrHelp {
  75                 return nil, nil
  76         } else if err != nil {
  77                 return nil, err
  78         } else if flags.NArg() != 0 {
  79                 return nil, fmt.Errorf("unrecognized command line arguments: %v", flags.Args())
  80         }
  81
  82         // Print version information if requested
  83         if *getVersion {
  84                 fmt.Printf("keep-web %s\n", version)
  85                 return nil, nil
  86         }
  87
  88         arvCfg, err := loader.Load()
  89         if err != nil {
  90                 log.Fatal(err)
  91         }
  92         cfg := newConfig(logger, arvCfg)
  93
  94         if *dumpConfig {
  95                 out, err := yaml.Marshal(cfg)
  96                 if err != nil {
  97                         return nil, err
  98                 }
  99                 _, err = os.Stdout.Write(out)
 100                 return nil, err
 101         }
 102         return cfg, nil
 103 }
 104
 105 func main() {
 106         logger := log.New()
 107
 108         cfg, err := configure(logger, os.Args)
 109         if err != nil {
 110                 logger.Error(err)
 111                 os.Exit(1)
 112         } else if cfg == nil {
 113                 return
 114         }
 115
 116         log.Printf("keep-web %s started", version)
 117
 118         if ext := ".txt"; mime.TypeByExtension(ext) == "" {
 119                 log.Warnf("cannot look up MIME type for %q -- this probably means /etc/mime.types is missing -- clients will see incorrect content types", ext)
 120         }
 121
 122         os.Setenv("ARVADOS_API_HOST", cfg.cluster.Services.Controller.ExternalURL.Host)
 123         srv := &server{Config: cfg}
 124         if err := srv.Start(logrus.StandardLogger()); err != nil {
 125                 log.Fatal(err)
 126         }
 127         if _, err := daemon.SdNotify(false, "READY=1"); err != nil {
 128                 log.Printf("Error notifying init daemon: %v", err)
 129         }
 130         log.Println("Listening at", srv.Addr)
 131         if err := srv.Wait(); err != nil {
 132                 log.Fatal(err)
 133         }
 134 }