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         log "github.com/sirupsen/logrus"
  18 )
  19
  20 var (
  21         version = "dev"
  22 )
  23
  24 // Config specifies server configuration.
  25 type Config struct {
  26         Client  arvados.Client
  27         Cache   cache
  28         cluster *arvados.Cluster
  29 }
  30
  31 func newConfig(arvCfg *arvados.Config) *Config {
  32         cfg := Config{}
  33         var cls *arvados.Cluster
  34         var err error
  35         if cls, err = arvCfg.GetCluster(""); err != nil {
  36                 log.Fatal(err)
  37         }
  38         cfg.cluster = cls
  39         cfg.Cache.config = &cfg.cluster.Collections.WebDAVCache
  40         return &cfg
  41 }
  42
  43 func init() {
  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")
  51         }
  52
  53         log.SetFormatter(&log.JSONFormatter{
  54                 TimestampFormat: "2006-01-02T15:04:05.000000000Z07:00",
  55         })
  56 }
  57
  58 func configure(logger log.FieldLogger, args []string) *Config {
  59         flags := flag.NewFlagSet(args[0], flag.ExitOnError)
  60
  61         loader := config.NewLoader(os.Stdin, logger)
  62         loader.SetupFlags(flags)
  63
  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.")
  68
  69         args = loader.MungeLegacyConfigArgs(logger, args[1:], "-legacy-keepweb-config")
  70         flags.Parse(args)
  71
  72         // Print version information if requested
  73         if *getVersion {
  74                 fmt.Printf("keep-web %s\n", version)
  75                 return nil
  76         }
  77
  78         arvCfg, err := loader.Load()
  79         if err != nil {
  80                 log.Fatal(err)
  81         }
  82         cfg := newConfig(arvCfg)
  83
  84         if *dumpConfig {
  85                 out, err := yaml.Marshal(cfg)
  86                 if err != nil {
  87                         log.Fatal(err)
  88                 }
  89                 _, err = os.Stdout.Write(out)
  90                 if err != nil {
  91                         log.Fatal(err)
  92                 }
  93                 return nil
  94         }
  95         return cfg
  96 }
  97
  98 func main() {
  99         logger := log.New()
 100
 101         cfg := configure(logger, os.Args)
 102         if cfg == nil {
 103                 return
 104         }
 105
 106         log.Printf("keep-web %s started", version)
 107
 108         if ext := ".txt"; mime.TypeByExtension(ext) == "" {
 109                 log.Warnf("cannot look up MIME type for %q -- this probably means /etc/mime.types is missing -- clients will see incorrect content types", ext)
 110         }
 111
 112         os.Setenv("ARVADOS_API_HOST", cfg.cluster.Services.Controller.ExternalURL.Host)
 113         srv := &server{Config: cfg}
 114         if err := srv.Start(); err != nil {
 115                 log.Fatal(err)
 116         }
 117         if _, err := daemon.SdNotify(false, "READY=1"); err != nil {
 118                 log.Printf("Error notifying init daemon: %v", err)
 119         }
 120         log.Println("Listening at", srv.Addr)
 121         if err := srv.Wait(); err != nil {
 122                 log.Fatal(err)
 123         }
 124 }