143 lines
3.5 KiB
Go
143 lines
3.5 KiB
Go
/*
|
|
* Copyright © 2018-2019 A Bunch Tell LLC.
|
|
*
|
|
* This file is part of WriteFreely.
|
|
*
|
|
* WriteFreely is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License, included
|
|
* in the LICENSE file in this source code package.
|
|
*/
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"github.com/gorilla/mux"
|
|
"github.com/writeas/web-core/log"
|
|
"github.com/writeas/writefreely"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
func main() {
|
|
// General options usable with other commands
|
|
debugPtr := flag.Bool("debug", false, "Enables debug logging.")
|
|
configFile := flag.String("c", "config.ini", "The configuration file to use")
|
|
|
|
// Setup actions
|
|
createConfig := flag.Bool("create-config", false, "Creates a basic configuration and exits")
|
|
doConfig := flag.Bool("config", false, "Run the configuration process")
|
|
genKeys := flag.Bool("gen-keys", false, "Generate encryption and authentication keys")
|
|
createSchema := flag.Bool("init-db", false, "Initialize app database")
|
|
migrate := flag.Bool("migrate", false, "Migrate the database")
|
|
|
|
// Admin actions
|
|
createAdmin := flag.String("create-admin", "", "Create an admin with the given username:password")
|
|
createUser := flag.String("create-user", "", "Create a regular user with the given username:password")
|
|
resetPassUser := flag.String("reset-pass", "", "Reset the given user's password")
|
|
outputVersion := flag.Bool("v", false, "Output the current version")
|
|
flag.Parse()
|
|
|
|
app := writefreely.NewApp(*configFile)
|
|
|
|
if *outputVersion {
|
|
writefreely.OutputVersion()
|
|
os.Exit(0)
|
|
} else if *createConfig {
|
|
err := writefreely.CreateConfig(app)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
os.Exit(1)
|
|
}
|
|
os.Exit(0)
|
|
} else if *doConfig {
|
|
writefreely.DoConfig(app)
|
|
os.Exit(0)
|
|
} else if *genKeys {
|
|
err := writefreely.GenerateKeyFiles(app)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
os.Exit(1)
|
|
}
|
|
os.Exit(0)
|
|
} else if *createSchema {
|
|
err := writefreely.CreateSchema(app)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
os.Exit(1)
|
|
}
|
|
os.Exit(0)
|
|
} else if *createAdmin != "" {
|
|
username, password, err := userPass(*createAdmin, true)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
os.Exit(1)
|
|
}
|
|
err = writefreely.CreateUser(app, username, password, true)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
os.Exit(1)
|
|
}
|
|
os.Exit(0)
|
|
} else if *createUser != "" {
|
|
username, password, err := userPass(*createUser, false)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
os.Exit(1)
|
|
}
|
|
err = writefreely.CreateUser(app, username, password, false)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
os.Exit(1)
|
|
}
|
|
os.Exit(0)
|
|
} else if *resetPassUser != "" {
|
|
err := writefreely.ResetPassword(app, *resetPassUser)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
os.Exit(1)
|
|
}
|
|
os.Exit(0)
|
|
} else if *migrate {
|
|
err := writefreely.Migrate(app)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
os.Exit(1)
|
|
}
|
|
os.Exit(0)
|
|
}
|
|
|
|
// Initialize the application
|
|
var err error
|
|
app, err = writefreely.Initialize(app, *debugPtr)
|
|
if err != nil {
|
|
log.Error("%s", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Set app routes
|
|
r := mux.NewRouter()
|
|
app.InitRoutes(r)
|
|
app.InitStaticRoutes(r)
|
|
|
|
// Serve the application
|
|
writefreely.Serve(app, r)
|
|
}
|
|
|
|
func userPass(credStr string, isAdmin bool) (user string, pass string, err error) {
|
|
creds := strings.Split(credStr, ":")
|
|
if len(creds) != 2 {
|
|
c := "user"
|
|
if isAdmin {
|
|
c = "admin"
|
|
}
|
|
err = fmt.Errorf("usage: writefreely --create-%s username:password", c)
|
|
return
|
|
}
|
|
|
|
user = creds[0]
|
|
pass = creds[1]
|
|
return
|
|
}
|