main.go

package main

import (
	"errors"
	"fmt"
	"log"
	"net/url"
	"os"
	"strings"

	"github.com/urfave/cli"

	"github.com/spaceuptech/space-cloud/config"
	"github.com/spaceuptech/space-cloud/utils"
	"github.com/spaceuptech/space-cloud/utils/server"
	"github.com/spaceuptech/space-cloud/utils/validate"
)

func main() {
	app := cli.NewApp()
	app.Version = utils.BuildVersion
	app.Name = "space-cloud-ee"
	app.Usage = "core binary to run space cloud"

	app.Commands = []cli.Command{
		{
			Name:   "run",
			Usage:  "runs the space cloud instance",
			Action: actionRun,
			Flags: []cli.Flag{
				cli.StringFlag{
					Name:  "port",
					Value: "8080",
					Usage: "Start HTTP server on port `PORT`",
				},
				cli.StringFlag{
					Name:  "grpc-port",
					Value: "8081",
					Usage: "Start grpc on port `GRPC_PORT`",
				},
				cli.IntFlag{
					Name:  "nats-port",
					Value: 4222,
					Usage: "Start nats on port `NATS_PORT`",
				},
				cli.IntFlag{
					Name:  "cluster-port",
					Value: 4248,
					Usage: "Start nats on port `NATS_PORT`",
				},
				cli.StringFlag{
					Name:   "secret",
					Value:  "none",
					Usage:  "The secret to use for authentication",
					EnvVar: "SECRET",
				},
				cli.StringFlag{
					Name:   "account",
					Value:  "none",
					Usage:  "Start space-cloud with `ACCOUNT`",
					EnvVar: "ACCOUNT",
				},
				cli.BoolFlag{
					Name:   "prod",
					Usage:  "Run space-cloud in production mode",
					EnvVar: "PROD",
				},
				cli.BoolFlag{
					Name:   "disable-metrics",
					Usage:  "Disable anonymous metric collection",
					EnvVar: "DISABLE_METRICS",
				},
				cli.BoolFlag{
					Name:   "disable-nats",
					Usage:  "Disable embedded nats server",
					EnvVar: "DISABLE_NATS",
				},
				cli.StringFlag{
					Name:   "seeds",
					Value:  "none",
					Usage:  "Seed nodes to cluster with",
					EnvVar: "SEEDS",
				},
			},
		},
		{
			Name:   "init",
			Usage:  "creates a config file with sensible defaults",
			Action: actionInit,
		},
	}

	err := app.Run(os.Args)
	if err != nil {
		log.Fatal(err)
	}
}

func actionRun(c *cli.Context) error {
	// Load cli flags
	port := c.String("port")
	grpcPort := c.String("grpc-port")
	secret := c.String("secret")
	account := c.String("account")
	natsPort := c.Int("nats-port")
	clusterPort := c.Int("cluster-port")
	isProd := c.Bool("prod")
	disableMetrics := c.Bool("disable-metrics")
	disableNats := c.Bool("disable-nats")
	seeds := c.String("seeds")

	if account == "none" || secret == "none" {
		return errors.New("Cannot start space-cloud with no account")
	}

	// Project and env cannot be changed once space cloud has started
	s := server.New(isProd)

	if !disableNats {
		// TODO read nats config from the yaml file if it exists
		if seeds != "" {
			array := strings.Split(seeds, ",")
			urls := []*url.URL{}
			for _, v := range array {
				if v != "" {
					u, err := url.Parse("nats://" + v)
					if err != nil {
						return err
					}
					urls = append(urls, u)
				}
			}
			server.DefaultNatsOptions.Routes = urls
		}
		server.DefaultNatsOptions.Port = natsPort
		server.DefaultNatsOptions.Cluster.Port = clusterPort
		s.RunNatsServer(server.DefaultNatsOptions)
		fmt.Println("Started NATS server on port ", server.DefaultNatsOptions.Port)
	}

	// Anonymously collect usage metrics if not explicitly disabled
	if !disableMetrics {
		go s.RoutineMetrics()
	}

	// Start the validation process
	validate.New(s.GetProjects()).Start(s.GetID(), account, secret)

	s.Routes()
	return s.Start(port, grpcPort)
}

func actionInit(*cli.Context) error {
	return config.GenerateConfig()
}