driftctl/main.go

87 lines
2.0 KiB
Go

package main
import (
"fmt"
"os"
"time"
"github.com/cloudskiff/driftctl/build"
"github.com/cloudskiff/driftctl/logger"
"github.com/cloudskiff/driftctl/pkg/cmd"
cmderrors "github.com/cloudskiff/driftctl/pkg/cmd/errors"
"github.com/cloudskiff/driftctl/pkg/config"
"github.com/cloudskiff/driftctl/pkg/version"
"github.com/cloudskiff/driftctl/sentry"
"github.com/fatih/color"
gosentry "github.com/getsentry/sentry-go"
"github.com/joho/godotenv"
"github.com/sirupsen/logrus"
)
func init() {
_ = godotenv.Load(".env.local")
_ = godotenv.Load() // The Original .env
}
func main() {
os.Exit(run())
}
func run() int {
config.Init()
logger.Init()
driftctlCmd := cmd.NewDriftctlCmd(build.Build{})
checkVersion := driftctlCmd.ShouldCheckVersion()
latestVersionChan := make(chan string)
if checkVersion {
go func() {
latestVersion := version.CheckLatest()
latestVersionChan <- latestVersion
}()
}
// Handle panic and log them to sentry if error reporting is enabled
defer func() {
if cmd.IsReportingEnabled(&driftctlCmd.Command) {
err := recover()
if err != nil {
gosentry.CurrentHub().Recover(err)
flushSentry()
logrus.Fatalf("Captured panic: %s", err)
os.Exit(2)
}
flushSentry()
}
}()
if _, err := driftctlCmd.ExecuteC(); err != nil {
if _, isNotInSync := err.(cmderrors.InfrastructureNotInSync); isNotInSync {
return 1
}
if cmd.IsReportingEnabled(&driftctlCmd.Command) {
sentry.CaptureException(err)
}
_, _ = fmt.Fprintln(os.Stderr, color.RedString("%s", err))
return 1
}
if checkVersion {
newVersion := <-latestVersionChan
if newVersion != "" {
_, _ = fmt.Fprintln(os.Stderr, "\n\nYour version of driftctl is outdated, please upgrade !")
_, _ = fmt.Fprintf(os.Stderr, "Current: %s; Latest: %s\n", version.Current(), newVersion)
}
}
return 0
}
func flushSentry() {
fmt.Print("Sending error report ...")
gosentry.Flush(60 * time.Second)
fmt.Printf(" done, thank you %s\n", color.RedString("❤️"))
}