writefreely/cmd/writefreely/main.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
}