123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188 |
- package cmd
- import (
- "context"
- "fmt"
- "net/http"
- "os"
- "os/signal"
- "syscall"
- "github.com/spf13/cobra"
- "github.com/spf13/viper"
- "go.uber.org/zap"
- "github.com/usememos/memos/internal/log"
- "github.com/usememos/memos/server"
- _profile "github.com/usememos/memos/server/profile"
- "github.com/usememos/memos/server/service/metric"
- "github.com/usememos/memos/store"
- "github.com/usememos/memos/store/db"
- )
- const (
- greetingBanner = `
- ███╗ ███╗███████╗███╗ ███╗ ██████╗ ███████╗
- ████╗ ████║██╔════╝████╗ ████║██╔═══██╗██╔════╝
- ██╔████╔██║█████╗ ██╔████╔██║██║ ██║███████╗
- ██║╚██╔╝██║██╔══╝ ██║╚██╔╝██║██║ ██║╚════██║
- ██║ ╚═╝ ██║███████╗██║ ╚═╝ ██║╚██████╔╝███████║
- ╚═╝ ╚═╝╚══════╝╚═╝ ╚═╝ ╚═════╝ ╚══════╝
- `
- )
- var (
- profile *_profile.Profile
- mode string
- addr string
- port int
- data string
- driver string
- dsn string
- enableMetric bool
- rootCmd = &cobra.Command{
- Use: "memos",
- Short: `An open-source, self-hosted memo hub with knowledge management and social networking.`,
- Run: func(_cmd *cobra.Command, _args []string) {
- ctx, cancel := context.WithCancel(context.Background())
- dbDriver, err := db.NewDBDriver(profile)
- if err != nil {
- cancel()
- log.Error("failed to create db driver", zap.Error(err))
- return
- }
- if err := dbDriver.Migrate(ctx); err != nil {
- cancel()
- log.Error("failed to migrate db", zap.Error(err))
- return
- }
- store := store.New(dbDriver, profile)
- s, err := server.NewServer(ctx, profile, store)
- if err != nil {
- cancel()
- log.Error("failed to create server", zap.Error(err))
- return
- }
- if profile.Metric {
- // nolint
- metric.NewMetricClient(s.ID, *profile)
- }
- c := make(chan os.Signal, 1)
- // Trigger graceful shutdown on SIGINT or SIGTERM.
- // The default signal sent by the `kill` command is SIGTERM,
- // which is taken as the graceful shutdown signal for many systems, eg., Kubernetes, Gunicorn.
- signal.Notify(c, os.Interrupt, syscall.SIGTERM)
- go func() {
- sig := <-c
- log.Info(fmt.Sprintf("%s received.\n", sig.String()))
- s.Shutdown(ctx)
- cancel()
- }()
- printGreetings()
- if err := s.Start(ctx); err != nil {
- if err != http.ErrServerClosed {
- log.Error("failed to start server", zap.Error(err))
- cancel()
- }
- }
- // Wait for CTRL-C.
- <-ctx.Done()
- },
- }
- )
- func Execute() error {
- defer log.Sync()
- return rootCmd.Execute()
- }
- func init() {
- cobra.OnInitialize(initConfig)
- rootCmd.PersistentFlags().StringVarP(&mode, "mode", "m", "demo", `mode of server, can be "prod" or "dev" or "demo"`)
- rootCmd.PersistentFlags().StringVarP(&addr, "addr", "a", "", "address of server")
- rootCmd.PersistentFlags().IntVarP(&port, "port", "p", 8081, "port of server")
- rootCmd.PersistentFlags().StringVarP(&data, "data", "d", "", "data directory")
- rootCmd.PersistentFlags().StringVarP(&driver, "driver", "", "", "database driver")
- rootCmd.PersistentFlags().StringVarP(&dsn, "dsn", "", "", "database source name(aka. DSN)")
- rootCmd.PersistentFlags().BoolVarP(&enableMetric, "metric", "", true, "allow metric collection")
- err := viper.BindPFlag("mode", rootCmd.PersistentFlags().Lookup("mode"))
- if err != nil {
- panic(err)
- }
- err = viper.BindPFlag("addr", rootCmd.PersistentFlags().Lookup("addr"))
- if err != nil {
- panic(err)
- }
- err = viper.BindPFlag("port", rootCmd.PersistentFlags().Lookup("port"))
- if err != nil {
- panic(err)
- }
- err = viper.BindPFlag("data", rootCmd.PersistentFlags().Lookup("data"))
- if err != nil {
- panic(err)
- }
- err = viper.BindPFlag("driver", rootCmd.PersistentFlags().Lookup("driver"))
- if err != nil {
- panic(err)
- }
- err = viper.BindPFlag("dsn", rootCmd.PersistentFlags().Lookup("dsn"))
- if err != nil {
- panic(err)
- }
- err = viper.BindPFlag("metric", rootCmd.PersistentFlags().Lookup("metric"))
- if err != nil {
- panic(err)
- }
- viper.SetDefault("mode", "demo")
- viper.SetDefault("driver", "sqlite")
- viper.SetDefault("addr", "")
- viper.SetDefault("port", 8081)
- viper.SetDefault("metric", true)
- viper.SetEnvPrefix("memos")
- }
- func initConfig() {
- viper.AutomaticEnv()
- var err error
- profile, err = _profile.GetProfile()
- if err != nil {
- fmt.Printf("failed to get profile, error: %+v\n", err)
- return
- }
- println("---")
- println("Server profile")
- println("data:", profile.Data)
- println("dsn:", profile.DSN)
- println("addr:", profile.Addr)
- println("port:", profile.Port)
- println("mode:", profile.Mode)
- println("driver:", profile.Driver)
- println("version:", profile.Version)
- println("metric:", profile.Metric)
- println("---")
- }
- func printGreetings() {
- print(greetingBanner)
- if len(profile.Addr) == 0 {
- fmt.Printf("Version %s has been started on port %d\n", profile.Version, profile.Port)
- } else {
- fmt.Printf("Version %s has been started on address '%s' and port %d\n", profile.Version, profile.Addr, profile.Port)
- }
- println("---")
- println("See more in:")
- fmt.Printf("👉Website: %s\n", "https://usememos.com")
- fmt.Printf("👉GitHub: %s\n", "https://github.com/usememos/memos")
- println("---")
- }
|