approve_list_svc.go

package main

import (
	"errors"
	"fmt"
	"os"

	"github.com/ecadlabs/signatory/cmd/approve-list-svc/server"
	log "github.com/sirupsen/logrus"
	"github.com/spf13/cobra"
)

var rootCmd = &cobra.Command{
	Use:   "approve-list-svc",
	Short: "Example IP approve list external policy service",
}

var confFile string

var pubCmd = &cobra.Command{
	Use:   "pub",
	Short: "Print the authentication public key",
	RunE: func(cmd *cobra.Command, args []string) error {
		conf, err := ReadConfig(confFile)
		if err != nil {
			return err
		}
		priv, err := conf.GetPrivateKey()
		if err != nil {
			return err
		}
		if priv == nil {
			return errors.New("private key is not specified")
		}
		fmt.Printf("Public key: %v\n", priv.Public())
		return nil
	},
}

var serveCmd = &cobra.Command{
	Use:   "serve",
	Short: "Start the server",
	RunE: func(cmd *cobra.Command, args []string) error {
		conf, err := ReadConfig(confFile)
		if err != nil {
			return err
		}

		pk, err := conf.GetPrivateKey()
		if err != nil {
			return err
		}

		ips, nets, err := conf.Addresses()
		if err != nil {
			return err
		}

		srv := server.Server{
			Address:    conf.Address,
			PrivateKey: pk,
			Addresses:  ips,
			Nets:       nets,
		}

		s := srv.New()
		log.Printf("HTTP server is listening for connections on %s", srv.Address)
		log.Println(s.ListenAndServe())

		return nil
	},
}

func init() {
	rootCmd.PersistentFlags().StringVarP(&confFile, "config", "c", "", "Config file")
	rootCmd.AddCommand(pubCmd)
	rootCmd.AddCommand(serveCmd)
}

func main() {
	if err := rootCmd.Execute(); err != nil {
		os.Exit(1)
	}
}