main.go

package main

import (
	"context"
	"os"

	awsLambda "github.com/aws/aws-lambda-go/lambda"
	cfg "github.com/kvendingoldo/aws-letsencrypt-lambda/internal/config"
	"github.com/kvendingoldo/aws-letsencrypt-lambda/internal/lambda"
	"github.com/kvendingoldo/aws-letsencrypt-lambda/internal/types"
	log "github.com/sirupsen/logrus"
)

func init() {
	log.SetReportCaller(false)

	var formatter log.Formatter

	if formatterType, ok := os.LookupEnv("FORMATTER_TYPE"); ok {
		if formatterType == "JSON" || formatterType == "json" {
			formatter = &log.JSONFormatter{PrettyPrint: false}
		}

		if formatterType == "TEXT" || formatterType == "text" {
			formatter = &log.TextFormatter{DisableColors: false}
		}
	}

	if formatter == nil {
		formatter = &log.TextFormatter{DisableColors: false}
	}

	log.SetFormatter(formatter)

	var logLevel log.Level
	var err error

	if ll, ok := os.LookupEnv("LOG_LEVEL"); ok {
		logLevel, err = log.ParseLevel(ll)
		if err != nil {
			logLevel = log.DebugLevel
		}
	} else {
		logLevel = log.DebugLevel
	}

	log.SetLevel(logLevel)
}

func Handler(ctx context.Context, event types.Event) (types.Response, error) {
	log.Infof("Handling lambda for event: %+v", event)
	config, err := cfg.New(event)
	if err != nil {
		return types.Response{Message: "Lambda has been failed"}, err
	}

	var msg string
	err = lambda.Execute(ctx, *config)
	if err != nil {
		msg = "Lambda has been failed"
	} else {
		msg = "Lambda has been completed successfully"
	}

	return types.Response{Message: msg}, err
}

func main() {
	log.Info("Starting lambda execution ...")
	if mode, ok := os.LookupEnv("MODE"); ok {
		if !(mode == "local" || mode == "cloud") {
			log.Errorf("Environment variable 'MODE' has unknown value '%v'. Value should be 'local' or 'cloud'", mode)
			os.Exit(1)
		}

		if mode == "local" {
			config, err := cfg.New(nil)
			if err != nil {
				log.Errorf("Lambda has been failed. Error: %s", err)
				os.Exit(1)
			}

			err = lambda.Execute(context.TODO(), *config)
			if err != nil {
				log.Errorf("Lambda has been failed. Error: %s", err)
				os.Exit(1)
			}
			log.Info("Lambda has been completed")
		} else if mode == "cloud" {
			awsLambda.Start(Handler)
		}
	} else {
		log.Errorf("Environment variable 'MODE' is unspecified. Please, specify it. Value should be 'local' or 'cloud'")
		os.Exit(1)
	}
}