cmd/arvados-server/cmd.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         "encoding/json"
   9         "fmt"
  10         "io"
  11         "net/http"
  12         "os"
  13
  14         "git.arvados.org/arvados.git/lib/boot"
  15         "git.arvados.org/arvados.git/lib/cloud/cloudtest"
  16         "git.arvados.org/arvados.git/lib/cmd"
  17         "git.arvados.org/arvados.git/lib/config"
  18         "git.arvados.org/arvados.git/lib/controller"
  19         "git.arvados.org/arvados.git/lib/crunchrun"
  20         "git.arvados.org/arvados.git/lib/dispatchcloud"
  21         "git.arvados.org/arvados.git/lib/install"
  22         "git.arvados.org/arvados.git/lib/lsf"
  23         "git.arvados.org/arvados.git/lib/recovercollection"
  24         "git.arvados.org/arvados.git/services/keepstore"
  25         "git.arvados.org/arvados.git/services/ws"
  26 )
  27
  28 var (
  29         handler = cmd.Multi(map[string]cmd.Handler{
  30                 "version":   cmd.Version,
  31                 "-version":  cmd.Version,
  32                 "--version": cmd.Version,
  33
  34                 "boot":               boot.Command,
  35                 "cloudtest":          cloudtest.Command,
  36                 "config-check":       config.CheckCommand,
  37                 "config-defaults":    config.DumpDefaultsCommand,
  38                 "config-dump":        config.DumpCommand,
  39                 "controller":         controller.Command,
  40                 "crunch-run":         crunchrun.Command,
  41                 "dispatch-cloud":     dispatchcloud.Command,
  42                 "dispatch-lsf":       lsf.DispatchCommand,
  43                 "install":            install.Command,
  44                 "init":               install.InitCommand,
  45                 "keepstore":          keepstore.Command,
  46                 "recover-collection": recovercollection.Command,
  47                 "workbench2":         wb2command{},
  48                 "ws":                 ws.Command,
  49         })
  50 )
  51
  52 func main() {
  53         os.Exit(handler.RunCommand(os.Args[0], os.Args[1:], os.Stdin, os.Stdout, os.Stderr))
  54 }
  55
  56 type wb2command struct{}
  57
  58 func (wb2command) RunCommand(prog string, args []string, stdin io.Reader, stdout, stderr io.Writer) int {
  59         if len(args) != 3 {
  60                 fmt.Fprintf(stderr, "usage: %s api-host listen-addr app-dir\n", prog)
  61                 return 1
  62         }
  63         configJSON, err := json.Marshal(map[string]string{"API_HOST": args[0]})
  64         if err != nil {
  65                 fmt.Fprintf(stderr, "json.Marshal: %s\n", err)
  66                 return 1
  67         }
  68         mux := http.NewServeMux()
  69         mux.Handle("/", http.FileServer(http.Dir(args[2])))
  70         mux.HandleFunc("/config.json", func(w http.ResponseWriter, _ *http.Request) {
  71                 w.Write(configJSON)
  72         })
  73         mux.HandleFunc("/_health/ping", func(w http.ResponseWriter, _ *http.Request) {
  74                 io.WriteString(w, `{"health":"OK"}`)
  75         })
  76         err = http.ListenAndServe(args[1], mux)
  77         if err != nil {
  78                 fmt.Fprintln(stderr, err.Error())
  79                 return 1
  80         }
  81         return 0
  82 }