package cmd import ( "context" "os" "os/signal" "git.1in9.net/raider/wroofauth/internal/database" "git.1in9.net/raider/wroofauth/internal/entities" "git.1in9.net/raider/wroofauth/internal/logger" "git.1in9.net/raider/wroofauth/internal/server" "github.com/spf13/cobra" ) // serveCmd represents the serve command var serveCmd = &cobra.Command{ Use: "serve", Short: "Starts WroofAuth in Server mode", Long: `A longer description that spans multiple lines and likely contains examples and usage of using your command. For example: Cobra is a CLI library for Go that empowers applications. This application is a tool to generate the needed files to quickly create a Cobra application.`, Run: func(cmd *cobra.Command, args []string) { ctx := context.Background() database.MongoConnect(ctx) database.RedisConnect(ctx) entities.SetupEntityDatabases() c := make(chan os.Signal, 1) signal.Notify(c, os.Interrupt) go func() { for sig := range c { if sig == os.Interrupt { logger.Logger.Info("interrupt received, attempting graceful shutdown") server.Shutdown(ctx) } } }() server.Serve() logger.Logger.Info("serve loop exited, attempting to clean up connections") database.MongoDisconnect(ctx) database.RedisDisconnect() logger.Logger.Info("WroofAuth has gracefully shut down, good bye!") }, } func init() { rootCmd.AddCommand(serveCmd) // Here you will define your flags and configuration settings. // Cobra supports Persistent Flags which will work for this command // and all subcommands, e.g.: // serveCmd.PersistentFlags().String("foo", "", "A help for foo") // Cobra supports local flags which will only run when this command // is called directly, e.g.: // serveCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle") }