initial commit - from neko open source repository..

2025-06-01 10:22:35 +02:00 · 2020-10-22 16:54:50 +02:00 · 2020-10-22 16:54:50 +02:00 · 56de805f54
commit 56de805f54
66 changed files with 5498 additions and 0 deletions
--- a/cmd/root.go
+++ b/cmd/root.go
@ -0,0 +1,127 @@
+package cmd
+
+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/cobra"
+	"github.com/spf13/viper"
+
+	"n.eko.moe/neko"
+)
+
+func Execute() error {
+	return root.Execute()
+}
+
+var root = &cobra.Command{
+	Use:     "neko",
+	Short:   "neko streaming server",
+	Long:    `neko streaming server`,
+	Version: neko.Service.Version.String(),
+}
+
+func init() {
+	cobra.OnInitialize(func() {
+		//////
+		// logs
+		//////
+		zerolog.TimeFieldFormat = ""
+		zerolog.SetGlobalLevel(zerolog.InfoLevel)
+
+		if viper.GetBool("debug") {
+			zerolog.SetGlobalLevel(zerolog.DebugLevel)
+		}
+
+		console := zerolog.ConsoleWriter{Out: os.Stdout}
+
+		if !viper.GetBool("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, "neko-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))
+		}
+
+		//////
+		// configs
+		//////
+		config := viper.GetString("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("neko")
+		}
+
+		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("debug")).
+			Str("logging", viper.GetString("logs")).
+			Str("config", file).
+			Logger()
+
+		if file == "" {
+			logger.Warn().Msg("preflight complete without config file")
+		} else {
+			logger.Info().Msg("preflight complete")
+		}
+
+		neko.Service.Root.Set()
+	})
+
+	if err := neko.Service.Root.Init(root); err != nil {
+		log.Panic().Err(err).Msg("unable to run root command")
+	}
+
+	root.SetVersionTemplate(neko.Service.Version.Details())
+}