package main

import (

"fmt"

"os"

"github.com/getsentry/sentry-go"

"github.com/rawnly/splash-cli/config"

"github.com/rawnly/splash-cli/lib/analytics"

"github.com/rs/zerolog"

"github.com/sirupsen/logrus"

"github.com/spf13/cobra"

"github.com/voxelite-ai/env"

)

// Setup the logs

func setupLogs() {

debug := env.Bool("SPLASH_CLI_DEBUG", false)

logLevel := env.String("LOG_LEVEL", "info")

lvl, err := zerolog.ParseLevel(logLevel)

if err != nil {

panic(err)

}

zerolog.SetGlobalLevel(lvl)

// You could set this to any `io.Writer` such as a file

file, err := os.CreateTemp("", "splash-cli.log")

if err == nil {

logrus.SetOutput(file)

if debug {

fmt.Println("Log file created at:", file.Name())

}

} else {

logrus.Warn("Failed to log to file, using default stderr")

}

if config.GetFormatterType() == config.LOG_FORMAT_JSON {

logrus.SetFormatter(&logrus.JSONFormatter{

TimestampFormat: "2006-01-02 15:04:05",

DisableTimestamp: true,

})

} else {

logrus.SetFormatter(&logrus.TextFormatter{

DisableColors: false,

FullTimestamp: true,

DisableTimestamp: true,

TimestampFormat: "2006-01-02 15:04:05",

})

}

if config.IsDebug() {

if logLevel == "trace" {

logrus.SetLevel(logrus.TraceLevel)

} else {

logrus.SetLevel(logrus.DebugLevel)

}

logrus.SetReportCaller(true)

} else {

logrus.SetLevel(logrus.WarnLevel)

}

}

// Setup sentry if not in DEBUG mode

func setupSentry() {

if !config.IsSentryEnabled() {

logrus.WithFields(logrus.Fields{

"dsn": config.SentryDSN,

"debug": config.IsDebug(),

}).Trace("Sentry disabled")

return

}

err := sentry.Init(sentry.ClientOptions{

Dsn: config.SentryDSN,

TracesSampleRate: 1.0,

Environment: config.GetEnvironment(),

Debug: config.SentryDebug == "true",

})

cobra.CheckErr(err)

}

// Setup posthog

func setupPosthog() *analytics.Analytics {

return analytics.New(config.PostHogKey)

}