112 lines
2.6 KiB
Go
112 lines
2.6 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"log"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
yaml "gopkg.in/yaml.v2"
|
||
|
|
|
||
|
|
"github.com/coreos/poke/server"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
)
|
||
|
|
|
||
|
|
func commandServe() *cobra.Command {
|
||
|
|
return &cobra.Command{
|
||
|
|
Use: "serve [ config file ]",
|
||
|
|
Short: "Connect to the storage and begin serving requests.",
|
||
|
|
Long: ``,
|
||
|
|
Example: "poke serve c.yaml",
|
||
|
|
RunE: serve,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func serve(cmd *cobra.Command, args []string) error {
|
||
|
|
switch len(args) {
|
||
|
|
default:
|
||
|
|
return errors.New("surplus arguments")
|
||
|
|
case 0:
|
||
|
|
// TODO(ericchiang): Consider having a default config file location.
|
||
|
|
return errors.New("no config file specified")
|
||
|
|
case 1:
|
||
|
|
}
|
||
|
|
|
||
|
|
configFile := args[0]
|
||
|
|
configData, err := ioutil.ReadFile(configFile)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("read config file %s: %v", configFile, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
var c Config
|
||
|
|
if err := yaml.Unmarshal(configData, &c); err != nil {
|
||
|
|
return fmt.Errorf("parse config file %s: %v", configFile, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Fast checks. Perform these first for a more responsive CLI.
|
||
|
|
checks := []struct {
|
||
|
|
bad bool
|
||
|
|
errMsg string
|
||
|
|
}{
|
||
|
|
{c.Issuer == "", "no issuer specified in config file"},
|
||
|
|
{len(c.Connectors) == 0, "no connectors supplied in config file"},
|
||
|
|
{c.Storage.Config == nil, "no storage suppied in config file"},
|
||
|
|
{c.Web.HTTP == "" && c.Web.HTTPS == "", "must supply a HTTP/HTTPS address to listen on"},
|
||
|
|
{c.Web.HTTPS != "" && c.Web.TLSCert == "", "no cert specified for HTTPS"},
|
||
|
|
{c.Web.HTTPS != "" && c.Web.TLSKey == "", "no private key specified for HTTPS"},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, check := range checks {
|
||
|
|
if check.bad {
|
||
|
|
return errors.New(check.errMsg)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
connectors := make([]server.Connector, len(c.Connectors))
|
||
|
|
for i, conn := range c.Connectors {
|
||
|
|
if conn.Config == nil {
|
||
|
|
return fmt.Errorf("no config field for connector %q", conn.ID)
|
||
|
|
}
|
||
|
|
c, err := conn.Config.Open()
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("open %s: %v", conn.ID, err)
|
||
|
|
}
|
||
|
|
connectors[i] = server.Connector{
|
||
|
|
ID: conn.ID,
|
||
|
|
DisplayName: conn.Name,
|
||
|
|
Connector: c,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
s, err := c.Storage.Config.Open()
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("initializing storage: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
serverConfig := server.Config{
|
||
|
|
Issuer: c.Issuer,
|
||
|
|
Connectors: connectors,
|
||
|
|
Storage: s,
|
||
|
|
}
|
||
|
|
|
||
|
|
serv, err := server.New(serverConfig)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("initializing server: %v", err)
|
||
|
|
}
|
||
|
|
errc := make(chan error, 2)
|
||
|
|
if c.Web.HTTP != "" {
|
||
|
|
go func() {
|
||
|
|
log.Printf("listening on %s", c.Web.HTTP)
|
||
|
|
errc <- http.ListenAndServe(c.Web.HTTP, serv)
|
||
|
|
}()
|
||
|
|
}
|
||
|
|
if c.Web.HTTPS != "" {
|
||
|
|
go func() {
|
||
|
|
log.Printf("listening on %s", c.Web.HTTPS)
|
||
|
|
errc <- http.ListenAndServeTLS(c.Web.HTTPS, c.Web.TLSCert, c.Web.TLSKey, serv)
|
||
|
|
}()
|
||
|
|
}
|
||
|
|
return <-errc
|
||
|
|
}
|