driftctl/main.go

87 lines
2.0 KiB
Go
Raw Normal View History

package main
import (
"fmt"
"os"
2021-01-04 17:28:40 +00:00
"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"
2021-01-04 17:28:40 +00:00
"github.com/sirupsen/logrus"
)
func init() {
_ = godotenv.Load(".env.local")
_ = godotenv.Load() // The Original .env
}
func main() {
2021-01-04 17:28:40 +00:00
os.Exit(run())
}
func run() int {
config.Init()
2021-02-10 13:25:07 +00:00
logger.Init()
driftctlCmd := cmd.NewDriftctlCmd(build.Build{})
checkVersion := driftctlCmd.ShouldCheckVersion()
latestVersionChan := make(chan string)
if checkVersion {
go func() {
latestVersion := version.CheckLatest()
latestVersionChan <- latestVersion
}()
}
2021-01-04 17:28:40 +00:00
// 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)
2021-01-04 17:28:40 +00:00
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
}
2021-01-04 17:28:40 +00:00
if cmd.IsReportingEnabled(&driftctlCmd.Command) {
sentry.CaptureException(err)
}
_, _ = fmt.Fprintln(os.Stderr, color.RedString("%s", err))
2021-01-04 17:28:40 +00:00
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)
}
}
2021-01-04 17:28:40 +00:00
return 0
}
func flushSentry() {
fmt.Print("Sending error report ...")
gosentry.Flush(60 * time.Second)
2021-01-04 17:28:40 +00:00
fmt.Printf(" done, thank you %s\n", color.RedString("❤️"))
}