17344: -create-db=false to use $POSTGRES_HOST/USER/etc instead.
[arvados.git] / 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         "context"
9         "encoding/json"
10         "fmt"
11         "io"
12         "net/http"
13         "os"
14         "path"
15         "path/filepath"
16         "strings"
17
18         "git.arvados.org/arvados.git/lib/boot"
19         "git.arvados.org/arvados.git/lib/cloud/cloudtest"
20         "git.arvados.org/arvados.git/lib/cmd"
21         "git.arvados.org/arvados.git/lib/config"
22         "git.arvados.org/arvados.git/lib/controller"
23         "git.arvados.org/arvados.git/lib/crunchrun"
24         "git.arvados.org/arvados.git/lib/dispatchcloud"
25         "git.arvados.org/arvados.git/lib/install"
26         "git.arvados.org/arvados.git/lib/lsf"
27         "git.arvados.org/arvados.git/lib/recovercollection"
28         "git.arvados.org/arvados.git/lib/service"
29         "git.arvados.org/arvados.git/sdk/go/arvados"
30         "git.arvados.org/arvados.git/sdk/go/health"
31         "git.arvados.org/arvados.git/services/githttpd"
32         keepbalance "git.arvados.org/arvados.git/services/keep-balance"
33         keepweb "git.arvados.org/arvados.git/services/keep-web"
34         "git.arvados.org/arvados.git/services/keepproxy"
35         "git.arvados.org/arvados.git/services/keepstore"
36         "git.arvados.org/arvados.git/services/ws"
37         "github.com/prometheus/client_golang/prometheus"
38 )
39
40 var (
41         handler = cmd.Multi(map[string]cmd.Handler{
42                 "version":   cmd.Version,
43                 "-version":  cmd.Version,
44                 "--version": cmd.Version,
45
46                 "boot":               boot.Command,
47                 "check":              health.CheckCommand,
48                 "cloudtest":          cloudtest.Command,
49                 "config-check":       config.CheckCommand,
50                 "config-defaults":    config.DumpDefaultsCommand,
51                 "config-dump":        config.DumpCommand,
52                 "controller":         controller.Command,
53                 "crunch-run":         crunchrun.Command,
54                 "dispatch-cloud":     dispatchcloud.Command,
55                 "dispatch-lsf":       lsf.DispatchCommand,
56                 "git-httpd":          githttpd.Command,
57                 "health":             healthCommand,
58                 "install":            install.Command,
59                 "init":               install.InitCommand,
60                 "keep-balance":       keepbalance.Command,
61                 "keep-web":           keepweb.Command,
62                 "keepproxy":          keepproxy.Command,
63                 "keepstore":          keepstore.Command,
64                 "recover-collection": recovercollection.Command,
65                 "workbench2":         wb2command{},
66                 "ws":                 ws.Command,
67         })
68 )
69
70 func main() {
71         os.Exit(handler.RunCommand(os.Args[0], os.Args[1:], os.Stdin, os.Stdout, os.Stderr))
72 }
73
74 type wb2command struct{}
75
76 func (wb2command) RunCommand(prog string, args []string, stdin io.Reader, stdout, stderr io.Writer) int {
77         if len(args) != 3 {
78                 fmt.Fprintf(stderr, "usage: %s api-host listen-addr app-dir\n", prog)
79                 return 1
80         }
81         configJSON, err := json.Marshal(map[string]string{"API_HOST": args[0]})
82         if err != nil {
83                 fmt.Fprintf(stderr, "json.Marshal: %s\n", err)
84                 return 1
85         }
86         servefs := http.FileServer(http.Dir(args[2]))
87         mux := http.NewServeMux()
88         mux.Handle("/", http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
89                 for _, ent := range strings.Split(req.URL.Path, "/") {
90                         if ent == ".." {
91                                 http.Error(w, "invalid URL path", http.StatusBadRequest)
92                                 return
93                         }
94                 }
95                 fnm := filepath.Join(args[2], filepath.FromSlash(path.Clean("/"+req.URL.Path)))
96                 if _, err := os.Stat(fnm); os.IsNotExist(err) {
97                         req.URL.Path = "/"
98                 }
99                 servefs.ServeHTTP(w, req)
100         }))
101         mux.HandleFunc("/config.json", func(w http.ResponseWriter, _ *http.Request) {
102                 w.Write(configJSON)
103         })
104         mux.HandleFunc("/_health/ping", func(w http.ResponseWriter, _ *http.Request) {
105                 io.WriteString(w, `{"health":"OK"}`)
106         })
107         err = http.ListenAndServe(args[1], mux)
108         if err != nil {
109                 fmt.Fprintln(stderr, err.Error())
110                 return 1
111         }
112         return 0
113 }
114
115 var healthCommand cmd.Handler = service.Command(arvados.ServiceNameHealth, func(ctx context.Context, cluster *arvados.Cluster, _ string, reg *prometheus.Registry) service.Handler {
116         mClockSkew := prometheus.NewGauge(prometheus.GaugeOpts{
117                 Namespace: "arvados",
118                 Subsystem: "health",
119                 Name:      "clock_skew_seconds",
120                 Help:      "Clock skew observed in most recent health check",
121         })
122         reg.MustRegister(mClockSkew)
123         return &health.Aggregator{
124                 Cluster:         cluster,
125                 MetricClockSkew: mClockSkew,
126         }
127 })