hangar-tech
/
listmonk
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add --yes flag to skip prompts

This commit is contained in:
Kailash Nadh 2019-11-30 16:55:14 +05:30
parent 1dc26e2cb7
commit 427dd93d3b
2 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
install.go
View File

@ -17,21 +17,23 @@ import (
// install runs the first time setup of creating and // install runs the first time setup of creating and
// migrating the database and creating the super user. // migrating the database and creating the super user.
func install(app *App, qMap goyesql.Queries) { func install(app *App, qMap goyesql.Queries, prompt bool) {
fmt.Println("") fmt.Println("")
fmt.Println("** First time installation **") fmt.Println("** First time installation **")
fmt.Printf("** IMPORTANT: This will wipe existing listmonk tables and types in the DB '%s' **", fmt.Printf("** IMPORTANT: This will wipe existing listmonk tables and types in the DB '%s' **",
ko.String("db.database")) ko.String("db.database"))
fmt.Println("") fmt.Println("")
var ok string if prompt {
fmt.Print("Continue (y/n)? ") var ok string
if _, err := fmt.Scanf("%s", &ok); err != nil { fmt.Print("Continue (y/n)? ")
logger.Fatalf("Error reading value from terminal: %v", err) if _, err := fmt.Scanf("%s", &ok); err != nil {
} logger.Fatalf("Error reading value from terminal: %v", err)
if strings.ToLower(ok) != "y" { }
fmt.Println("Installation cancelled.") if strings.ToLower(ok) != "y" {
return fmt.Println("Installation cancelled.")
return
}
} }
// Migrate the tables. // Migrate the tables.

3
main.go
View File

@ -84,6 +84,7 @@ func init() {
f.Bool("install", false, "Run first time installation") f.Bool("install", false, "Run first time installation")
f.Bool("version", false, "Current version of the build") f.Bool("version", false, "Current version of the build")
f.Bool("new-config", false, "Generate sample config file") f.Bool("new-config", false, "Generate sample config file")
f.Bool("yes", false, "Assume 'yes' to prompts, eg: during --install")
// Process flags. // Process flags.
if err := f.Parse(os.Args[1:]); err != nil { if err := f.Parse(os.Args[1:]); err != nil {
@ -274,7 +275,7 @@ func main() {
// Run the first time installation. // Run the first time installation.
if ko.Bool("install") { if ko.Bool("install") {
install(app, qMap) install(app, qMap, !ko.Bool("yes"))
return return
} }