first commit

server/internal/preflight/config.go

@@ -0,0 +1,48 @@
+package preflight
+
+import (
+	"runtime"
+
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/viper"
+)
+
+func Config(name string) {
+	config := viper.GetString("neko.config")
+
+	if config != "" {
+		viper.SetConfigFile(config) // Use config file from the flag.
+	} else {
+		if runtime.GOOS == "linux" {
+			viper.AddConfigPath("/etc/neko/")
+		}
+
+		viper.AddConfigPath(".")
+		viper.SetConfigName(name)
+	}
+
+	viper.SetEnvPrefix("NEKO")
+	viper.AutomaticEnv() // read in environment variables that match
+
+	if err := viper.ReadInConfig(); err != nil {
+		if _, ok := err.(viper.ConfigFileNotFoundError); !ok {
+			log.Error().Err(err)
+		}
+		if config != "" {
+			log.Error().Err(err)
+		}
+	}
+
+	file := viper.ConfigFileUsed()
+	logger := log.With().
+		Bool("debug", viper.GetBool("neko.debug")).
+		Str("logging", viper.GetString("neko.logs")).
+		Str("config", file).
+		Logger()
+
+	if file == "" {
+		logger.Warn().Msg("Preflight complete without config file")
+	} else {
+		logger.Info().Msg("Preflight complete")
+	}
+}

server/internal/preflight/logs.go

@@ -0,0 +1,60 @@
+package preflight
+
+import (
+	"fmt"
+	"io"
+	"os"
+	"path/filepath"
+	"runtime"
+	"time"
+
+	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/diode"
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/viper"
+)
+
+func Logs(name string) {
+	zerolog.TimeFieldFormat = ""
+	zerolog.SetGlobalLevel(zerolog.InfoLevel)
+
+	if viper.GetBool("neko.debug") {
+		zerolog.SetGlobalLevel(zerolog.DebugLevel)
+	}
+
+	console := zerolog.ConsoleWriter{Out: os.Stdout}
+
+	if !viper.GetBool("neko.logs") {
+		log.Logger = log.Output(console)
+	} else {
+
+		logs := filepath.Join(".", "logs")
+		if runtime.GOOS == "linux" {
+			logs = "/var/log/neko"
+		}
+
+		if _, err := os.Stat(logs); os.IsNotExist(err) {
+			os.Mkdir(logs, os.ModePerm)
+		}
+
+		latest := filepath.Join(logs, name+"-latest.log")
+		_, err := os.Stat(latest)
+		if err == nil {
+			err = os.Rename(latest, filepath.Join(logs, "neko."+time.Now().Format("2006-01-02T15-04-05Z07-00")+".log"))
+			if err != nil {
+				log.Panic().Err(err).Msg("Failed to rotate log file")
+			}
+		}
+
+		logf, err := os.OpenFile(latest, os.O_RDWR|os.O_CREATE, 0666)
+		if err != nil {
+			log.Panic().Err(err).Msg("Failed to create log file")
+		}
+
+		logger := diode.NewWriter(logf, 1000, 10*time.Millisecond, func(missed int) {
+			fmt.Printf("Logger Dropped %d messages", missed)
+		})
+
+		log.Logger = log.Output(io.MultiWriter(console, logger))
+	}
+}