authelia/main.go

package main

import (
	"errors"
	"flag"
	"log"
	"os"

	"github.com/clems4ever/authelia/regulation"

	"github.com/clems4ever/authelia/session"

	"github.com/clems4ever/authelia/authentication"
	"github.com/clems4ever/authelia/authorization"
	"github.com/clems4ever/authelia/configuration"
	"github.com/clems4ever/authelia/logging"
	"github.com/clems4ever/authelia/middlewares"
	"github.com/clems4ever/authelia/notification"
	"github.com/clems4ever/authelia/server"
	"github.com/clems4ever/authelia/storage"
	"github.com/sirupsen/logrus"
)

func tryExtractConfigPath() (string, error) {
	configPtr := flag.String("config", "", "The path to a configuration file.")
	flag.Parse()

	if *configPtr == "" {
		return "", errors.New("No config file path provided")
	}

	return *configPtr, nil
}

func main() {
	if os.Getenv("ENVIRONMENT") == "dev" {
		logging.Logger().Info("===> Authelia is running in development mode. <===")
	}

	configPath, err := tryExtractConfigPath()
	if err != nil {
		logging.Logger().Error(err)
	}

	config, errs := configuration.Read(configPath)

	if len(errs) > 0 {
		for _, err = range errs {
			logging.Logger().Error(err)
		}
		panic(errors.New("Some errors have been reported"))
	}

	switch config.LogsLevel {
	case "info":
		logging.SetLevel(logrus.InfoLevel)
		break
	case "debug":
		logging.SetLevel(logrus.TraceLevel)
	}

	var userProvider authentication.UserProvider

	if config.AuthenticationBackend.File != nil {
		userProvider = authentication.NewFileUserProvider(config.AuthenticationBackend.File.Path)
	} else if config.AuthenticationBackend.Ldap != nil {
		userProvider = authentication.NewLDAPUserProvider(*config.AuthenticationBackend.Ldap)
	} else {
		log.Fatalf("Unrecognized authentication backend")
	}

	var storageProvider storage.Provider
	if config.Storage.Mongo != nil {
		storageProvider = storage.NewMongoProvider(*config.Storage.Mongo)
	} else if config.Storage.Local != nil {
		storageProvider = storage.NewSQLiteProvider(config.Storage.Local.Path)
	} else {
		log.Fatalf("Unrecognized storage backend")
	}

	var notifier notification.Notifier
	if config.Notifier.SMTP != nil {
		notifier = notification.NewSMTPNotifier(*config.Notifier.SMTP)
	} else if config.Notifier.FileSystem != nil {
		notifier = notification.NewFileNotifier(*config.Notifier.FileSystem)
	} else {
		log.Fatalf("Unrecognized notifier")
	}

	authorizer := authorization.NewAuthorizer(*config.AccessControl)
	sessionProvider := session.NewProvider(config.Session)
	regulator := regulation.NewRegulator(config.Regulation, storageProvider)

	providers := middlewares.Providers{
		Authorizer:      authorizer,
		UserProvider:    userProvider,
		Regulator:       regulator,
		StorageProvider: storageProvider,
		Notifier:        notifier,
		SessionProvider: sessionProvider,
	}
	server.StartServer(*config, providers)
}