main.go

package main

import (
	"net/http"
	"time"

	"github.com/ARGOeu/argo-api-authn/version"

	"github.com/gorilla/handlers"

	"flag"

	"crypto/tls"

	"strconv"

	"github.com/ARGOeu/argo-api-authn/auth"
	"github.com/ARGOeu/argo-api-authn/config"
	"github.com/ARGOeu/argo-api-authn/routing"
	"github.com/ARGOeu/argo-api-authn/stores"
	log "github.com/sirupsen/logrus"
)

func init() {
	log.SetFormatter(&log.TextFormatter{FullTimestamp: true, DisableColors: true})

	// display binary version information during start up
	version.LogInfo()
}

func main() {

	// Retrieve configuration file location through cmd argument
	var cfgPath = flag.String("config", "/etc/argo-api-authn/conf.d/argo-api-authn-config.json",
		"Path for the required configuration file.")
	flag.Parse()

	// initialize the config
	var cfg = config.WithDefaults()
	if err := cfg.ConfigSetUp(*cfgPath); err != nil {
		log.Error(err.Error())
		panic(err.Error())
	}

	//configure datastore
	store := &stores.MongoStoreWithOfficialDriver{
		Server:   cfg.MongoHost,
		Database: cfg.MongoDB,
	}
	store.SetUp()

	defer store.Close()

	// configure the TLS config for the server
	tlsConfig := &tls.Config{
		MinVersion: tls.VersionTLS12,
		ClientAuth: cfg.ClientAuthPolicy(),
		ClientCAs:  auth.LoadCAs(cfg.CertificateAuthorities),
	}

	api := routing.NewRouting(routing.ApiRoutes, store, cfg)

	xReqWithConType := handlers.AllowedHeaders([]string{"X-Requested-With", "Content-AuthModel"})
	allowVerbs := handlers.AllowedMethods([]string{"OPTIONS", "POST", "GET", "PUT", "DELETE", "HEAD"})

	server := &http.Server{
		Addr:              ":" + strconv.Itoa(cfg.ServicePort),
		Handler:           handlers.CORS(xReqWithConType, allowVerbs)(api.Router),
		TLSConfig:         tlsConfig,
		ReadTimeout:       time.Duration(cfg.ServerReadTimeout) * time.Second,
		ReadHeaderTimeout: time.Duration(cfg.ServerHeaderReadTimeout) * time.Second,
		WriteTimeout:      time.Duration(cfg.ServerWriteTimeout) * time.Second,
		IdleTimeout:       time.Duration(cfg.ServerIdleTimeout) * time.Second,
	}

	//Start the server
	err := server.ListenAndServeTLS(cfg.Certificate, cfg.CertificateKey)
	if err != nil {
		log.WithFields(
			log.Fields{
				"type":    "service_log",
				"details": err.Error(),
			},
		).Fatal("Service failed to start")
	}
}