package main import ( _ "embed" "flag" "fmt" "io" "log" "os" "path/filepath" "syscall" "github.com/bouncepaw/mycorrhiza/cfg" "github.com/bouncepaw/mycorrhiza/files" "github.com/bouncepaw/mycorrhiza/user" "golang.org/x/term" ) // CLI options are read and parsed here. // printHelp prints the help message. func printHelp() { fmt.Fprintf( flag.CommandLine.Output(), "Usage: %s WIKI_PATH\n", os.Args[0], ) flag.PrintDefaults() } // CreateUserCommand is parameters for admin creation flag. Only a single parameter, though. type CreateUserCommand struct { name string } // parseCliArgs parses CLI options and sets several important global variables. Call it early. func parseCliArgs() { var createAdminName string flag.StringVar(&cfg.ListenAddr, "listen-addr", "", "Address to listen on. For example, 127.0.0.1:1737 or /run/mycorrhiza.sock.") flag.StringVar(&createAdminName, "create-admin", "", "Create a new admin. The password will be prompted in the terminal.") flag.Usage = printHelp flag.Parse() args := flag.Args() if len(args) == 0 { log.Fatal("error: pass a wiki directory") } wikiDir, err := filepath.Abs(args[0]) if err != nil { log.Fatal(err) } cfg.WikiDir = wikiDir if createAdminName != "" { createAdminCommand(createAdminName) os.Exit(0) } } func createAdminCommand(name string) { wr := log.Writer() log.SetFlags(0) if err := files.PrepareWikiRoot(); err != nil { log.Fatal("error: ", err) } cfg.UseAuth = true cfg.AllowRegistration = true log.SetOutput(io.Discard) user.InitUserDatabase() log.SetOutput(wr) handle := syscall.Stdin if !term.IsTerminal(handle) { log.Fatal("error: not a terminal") } fmt.Print("Password: ") passwordBytes, err := term.ReadPassword(handle) fmt.Print("\n") if err != nil { log.Fatal("error: ", err) } password := string(passwordBytes) log.SetOutput(io.Discard) err = user.Register(name, password, "admin", "local", true) log.SetOutput(wr) if err != nil { log.Fatal("error: ", err) } }