Add: Windows support

config/config.go

@@ -5,14 +5,15 @@ import (
 	"fmt"
 	"io/ioutil"
 	"os"
+	"runtime"
 
 	"github.com/rs/zerolog"
 	"github.com/spf13/viper"
 )
 
 const (
-	version = "0.2"
-	title   = "HellPot"
+	Version = "0.2"
+	Title   = "HellPot"
 )
 
 var (
@@ -34,6 +35,7 @@ var (
 	f   *os.File
 	err error
 
+	NoColor         bool
 	customconfig    = false
 	home            string
 	configLocations []string
@@ -58,7 +60,7 @@ func init() {
 	if home, err = os.UserHomeDir(); err != nil {
 		panic(err)
 	}
-	prefConfigLocation = home + "/.config/" + title
+	prefConfigLocation = home + "/.config/" + Title
 	Opt = make(map[string]map[string]interface{})
 	snek = viper.New()
 }
@@ -83,10 +85,10 @@ func Init() {
 		snek.AddConfigPath(loc)
 	}
 
-	if err = snek.MergeInConfig(); err != nil {
+	if err = snek.MergeInConfig(); err != nil && runtime.GOOS != "windows" {
 		if _, err := os.Stat(prefConfigLocation); os.IsNotExist(err) {
 			if err = os.Mkdir(prefConfigLocation, 0755); err != nil {
-				panic(err)
+				println("error writing new config: " + err.Error())
 			}
 		}
 
@@ -100,6 +102,19 @@ func Init() {
 		Filename = newconfig
 	}
 
+	if runtime.GOOS == "windows" {
+		newconfig := "hellpot-config"
+		snek.SetConfigName(newconfig)
+		if err = snek.MergeInConfig(); err != nil {
+			if err = snek.SafeWriteConfigAs(newconfig + ".toml"); err != nil {
+				fmt.Println(err.Error())
+				os.Exit(1)
+			}
+		}
+
+		Filename = newconfig
+	}
+
 	if len(Filename) < 1 {
 		Filename = snek.ConfigFileUsed()
 	}
@@ -108,11 +123,21 @@ func Init() {
 }
 
 func setDefaults() {
-	var configSections = []string{"logger", "http"}
+	var (
+		configSections = []string{"logger", "http"}
+		deflogdir      = home + "/.config/" + Title + "/logs/"
+		defNoColor     = false
+	)
+
+	if runtime.GOOS == "windows" {
+		deflogdir = "logs/"
+		defNoColor = true
+	}
 
 	Opt["logger"] = map[string]interface{}{
 		"debug":             true,
-		"directory":         home + "/.config/" + title + "/logs/",
+		"directory":         deflogdir,
+		"nocolor":           defNoColor,
 		"use_date_filename": true,
 	}
 	Opt["http"] = map[string]interface{}{
@@ -130,11 +155,14 @@ func setDefaults() {
 }
 
 func acquireClue() {
-	configLocations = append(configLocations, prefConfigLocation)
-	configLocations = append(configLocations, "/etc/"+title+"/")
 	configLocations = append(configLocations, "./")
-	configLocations = append(configLocations, "../")
-	configLocations = append(configLocations, "../../")
+
+	if runtime.GOOS != "windows" {
+		configLocations = append(configLocations, prefConfigLocation)
+		configLocations = append(configLocations, "/etc/"+Title+"/")
+		configLocations = append(configLocations, "../")
+		configLocations = append(configLocations, "../../")
+	}
 }
 
 func loadCustomConfig(path string) {
@@ -159,6 +187,11 @@ func loadCustomConfig(path string) {
 func argParse() {
 	for i, arg := range os.Args {
 		switch arg {
+		case "-h":
+			println("HellPot: use -c <file.toml> to specify config file.")
+			os.Exit(0)
+		case "--config":
+			fallthrough
 		case "-c":
 			if len(os.Args) <= i-1 {
 				panic("syntax error! expected file after -c")
@@ -184,6 +217,7 @@ func associate() {
 	Opt = newOpt
 	Debug = snek.GetBool("logger.debug")
 	logDir = snek.GetString("logger.directory")
+	NoColor = snek.GetBool("logger.nocolor")
 	BindAddr = snek.GetString("http.bind_addr")
 	BindPort = snek.GetString("http.bind_port")
 	Paths = snek.GetStringSlice("http.paths")

config/logger.go

@@ -1,10 +1,11 @@
 package config
 
 import (
-	"github.com/rs/zerolog"
 	"os"
 	"strings"
 	"time"
+
+	"github.com/rs/zerolog"
 )
 
 var (
@@ -34,6 +35,7 @@ func Logger() zerolog.Logger {
 		println("cannot create log file: " + err.Error())
 		os.Exit(1)
 	}
-	multi := zerolog.MultiLevelWriter(zerolog.ConsoleWriter{Out: os.Stderr}, logFile)
+
+	multi := zerolog.MultiLevelWriter(zerolog.ConsoleWriter{NoColor: NoColor, Out: os.Stdout}, logFile)
 	return zerolog.New(multi).With().Timestamp().Logger()
 }

extra/decorate.go

@@ -5,7 +5,10 @@ import (
 	"bufio"
 	"encoding/base64"
 	"fmt"
+	"runtime"
 	"strings"
+
+	"github.com/yunginnanet/HellPot/config"
 )
 
 const banner = "ChtbOTc7NDBtIBtbOTc7NDNt4paEG1szMzs0MG3ilojilpIbWzk3OzQwbSAbWzMzOzQwbeKWiOKWiBtbMzc7NDBtIBtbMG0bWzBtICAbWzMzOzQwbeKWkxtbOTc7NDNtIOKWhOKWhBtbMzM7NDBt4paI4paIG1szNzs0MG0gG1swbRtbMG0gIBtbOTc7NDBtIBtbMzM7NDBt4paI4paI4paTG1szNzs0MG0gICAgG1swbRtbMG0gIBtbOTc7NDBtIBtbMzM7NDBt4paI4paI4paTG1szNzs0MG0gICAgG1swbRtbMG0gIBtbOTc7NDBtIBtbMzM7NDBt4paI4paI4paS4paI4paI4paIG1szNzs0MG0gIBtbMG0bWzBtICAbWzk3OzQwbSAbWzMzOzQwbeKWk+KWiBtbOTc7NDNt4paE4paEG1szMzs0MG3ilojilogbWzM3OzQwbSAgG1swbRtbMG0gIBtbMzM7NDBt4paE4paE4paE4paIG1s5Nzs0M23iloQbWzMzOzQwbeKWiOKWiOKWiOKWkxtbMG0bWzBtICAbWzBtChtbMzM7NDBt4paTG1s5Nzs0M23ilogbWzMzOzQwbeKWiOKWkRtbOTc7NDBtIBtbMzM7NDBt4paIG1s5Nzs0M23iloQbWzMzOzQwbeKWkhtbMG0bWzBtICAbWzMzOzQwbeKWkxtbOTc7NDNt4paIIBtbOTc7NDBtICAbWzMzOzQwbeKWgBtbMzc7NDBtIBtbMG0bWzBtICAbWzMzOzQwbeKWk+KWiOKWiOKWkhtbMzc7NDBtICAgIBtbMG0bWzBtICAbWzMzOzQwbeKWk+KWiOKWiOKWkhtbMzc7NDBtICAgIBtbMG0bWzBtICAbWzMzOzQwbeKWkxtbOTc7NDNt4paIG1szMzs0MG3ilojilpEbWzk3OzQwbSAbWzMzOzQwbSDilojilojilpMbWzBtG1swbSAgG1szMzs0MG3ilpPilogbWzk3OzQzbeKWiBtbMzM7NDBt4paSG1s5Nzs0MG0gIBtbMzM7NDBt4paI4paI4paTG1swbRtbMG0gIBtbMzM7NDBt4paTG1s5Nzs0MG0gIBtbOTc7NDNt4paIG1szMzs0MG3ilojilpIbWzk3OzQwbSAbWzMzOzQwbeKWk+KWkhtbMG0bWzBtICAbWzBtChtbMzM7NDBt4paSG1szMzs0M23ilogbWzMzOzQwbeKWiOKWiOKWgOKWiBtbOTc7NDNt4paIG1szMzs0MG3ilpEbWzBtG1swbSAgG1szMzs0MG3ilpIbWzk3OzQzbeKWiCAbWzMzOzQwbeKWiBtbMzc7NDBtICAgG1swbRtbMG0gIBtbMzM7NDBt4paSG1s5Nzs0M23ilogbWzMzOzQwbeKWiOKWkRtbMzc7NDBtICAgIBtbMG0bWzBtICAbWzMzOzQwbeKWkhtbOTc7NDNt4paIG1szMzs0MG3ilojilpEbWzM3OzQwbSAgICAbWzBtG1swbSAgG1szMzs0MG3ilpMbWzk3OzQzbeKWiBtbMzM7NDBt4paI4paRIOKWiOKWiOKWk+KWkhtbMG0bWzBtICAbWzMzOzQwbeKWkuKWiBtbOTc7NDNt4paIG1szMzs0MG3ilpEbWzk3OzQwbSAgG1s5Nzs0M23ilogbWzMzOzQwbeKWiOKWkhtbMG0bWzBtICAbWzMzOzQwbeKWkhtbOTc7NDBtIBtbMzM7NDBt4paTG1s5Nzs0M23ilogbWzMzOzQwbeKWiOKWkRtbOTc7NDBtIBtbMzM7NDBt4paS4paRG1swbRtbMG0gIBtbMG0KG1szMzs0MG3ilpHilpPilogbWzk3OzQwbSAbWzMzOzQwbeKWk+KWiBtbOTc7NDNt4paIG1szNzs0MG0gG1swbRtbMG0gIBtbMzM7NDBt4paS4paT4paIG1s5Nzs0MG0gIBtbMzM7NDBt4paEG1szNzs0MG0gG1swbRtbMG0gIBtbMzM7NDBt4paSG1s5Nzs0M23iloggG1szMzs0MG3ilpEbWzM3OzQwbSAgICAbWzBtG1swbSAgG1szMzs0MG3ilpIbWzk3OzQzbeKWiCAbWzMzOzQwbeKWkRtbMzc7NDBtICAgIBtbMG0bWzBtICAbWzMzOzQwbeKWkhtbOTc7NDNt4paEIBtbMzM7NDBt4paE4paI4paS4paTG1s5Nzs0MG0gG1szMzs0MG3ilpIbWzBtG1swbSAgG1szMzs0MG3ilpLilojilogbWzk3OzQwbSAgG1s5Nzs0M23iloTiloAbWzMzOzQwbeKWkuKWkRtbMG0bWzBtICAbWzMzOzQwbeKWkRtbOTc7NDBtIBtbMzM7NDBt4paTG1s5Nzs0M23iloQbWzMzOzQwbeKWiOKWkxtbOTc7NDBtIBtbMzM7NDBt4paRG1szNzs0MG0gG1swbRtbMG0gIBtbMG0KG1szMzs0MG3ilpHilpPilojilpLilpMbWzk3OzQzbSDiloQbWzMzOzQwbeKWkxtbMG0bWzBtICAbWzMzOzQwbeKWkeKWkuKWiOKWiOKWiOKWiOKWkhtbMG0bWzBtICAbWzMzOzQwbeKWkRtbOTc7NDNt4paEG1szMzs0MG3ilojilojilojilojilojilpIbWzBtG1swbSAgG1szMzs0MG3ilpEbWzk3OzQzbeKWhBtbMzM7NDBt4paI4paI4paI4paI4paI4paSG1swbRtbMG0gIBtbMzM7NDBt4paS4paI4paI4paSG1s5Nzs0MG0gG1szMzs0MG3ilpEbWzk3OzQwbSAgG1szMzs0MG3ilpEbWzBtG1swbSAgG1szMzs0MG3ilpEbWzk3OzQwbSAbWzMzOzQwbeKWiBtbOTc7NDNt4paA4paAIBtbMzM7NDBt4paT4paS4paRG1swbRtbMG0gIBtbOTc7NDBtICAbWzMzOzQwbeKWkuKWiOKWiOKWkhtbOTc7NDBtIBtbMzM7NDBt4paRG1szNzs0MG0gG1swbRtbMG0gIBtbMG0KG1s5Nzs0MG0gG1szMzs0MG3ilpIbWzk3OzQwbSAbWzMzOzQwbeKWkeKWk+KWkuKWkeKWkhtbMG0bWzBtICAbWzMzOzQwbeKWkeKWkRtbOTc7NDBtIBtbMzM7NDBt4paS4paRG1s5Nzs0MG0gG1szMzs0MG3ilpEbWzBtG1swbSAgG1szMzs0MG3ilpEbWzk3OzQwbSAbWzMzOzQwbeKWkuKWkeKWkxtbOTc7NDBtICAbWzMzOzQwbeKWkRtbMG0bWzBtICAbWzMzOzQwbeKWkRtbOTc7NDBtIBtbMzM7NDBt4paS4paR4paTG1s5Nzs0MG0gIBtbMzM7NDBt4paRG1swbRtbMG0gIBtbMzM7NDBt4paS4paT4paS4paRG1s5Nzs0MG0gG1szMzs0MG3ilpEbWzk3OzQwbSAgG1szMzs0MG3ilpEbWzBtG1swbSAgG1szMzs0MG3ilpEbWzk3OzQwbSAbWzMzOzQwbeKWkuKWkeKWk+KWkeKWk+KWkRtbMzc7NDBtIBtbMG0bWzBtICAbWzk3OzQwbSAgG1szMzs0MG3ilpIbWzk3OzQwbSAbWzMzOzQwbeKWkeKWkRtbMzc7NDBtICAgG1swbRtbMG0gIBtbMG0KG1s5Nzs0MG0gG1szMzs0MG3ilpIbWzk3OzQwbSAbWzMzOzQwbeKWkeKWkuKWkRtbOTc7NDBtIBtbMzM7NDBt4paRG1swbRtbMG0gIBtbMzM7NDBtIOKWkRtbOTc7NDBtIBtbMzM7NDBt4paRG1s5Nzs0MG0gIBtbMzM7NDBt4paRG1swbRtbMG0gIBtbMzM7NDBt4paRG1s5Nzs0MG0gG1szMzs0MG3ilpEbWzk3OzQwbSAbWzMzOzQwbeKWkhtbOTc7NDBtICAbWzMzOzQwbeKWkRtbMG0bWzBtICAbWzMzOzQwbeKWkRtbOTc7NDBtIBtbMzM7NDBt4paRG1s5Nzs0MG0gG1szMzs0MG3ilpIbWzk3OzQwbSAgG1szMzs0MG3ilpEbWzBtG1swbSAgG1szMzs0MG3ilpHilpIbWzk3OzQwbSAbWzMzOzQwbeKWkRtbMzc7NDBtICAgICAbWzBtG1swbSAgG1s5Nzs0MG0gIBtbMzM7NDBt4paRG1s5Nzs0MG0gG1szMzs0MG3ilpIbWzk3OzQwbSAbWzMzOzQwbeKWkuKWkRtbMzc7NDBtIBtbMG0bWzBtICAbWzk3OzQwbSAgICAbWzMzOzQwbeKWkRtbMzc7NDBtICAgIBtbMG0bWzBtICAbWzBtChtbOTc7NDBtIBtbMzM7NDBt4paRG1s5Nzs0MG0gIBtbMzM7NDBt4paR4paRG1s5Nzs0MG0gG1szMzs0MG3ilpEbWzBtG1swbSAgG1szMzs0MG0gG1s5Nzs0MG0gIBtbMzM7NDBt4paRG1szNzs0MG0gICAbWzBtG1swbSAgG1s5Nzs0MG0gIBtbMzM7NDBt4paRG1s5Nzs0MG0gG1szMzs0MG3ilpEbWzM3OzQwbSAgIBtbMG0bWzBtICAbWzk3OzQwbSAgG1szMzs0MG3ilpEbWzk3OzQwbSAbWzMzOzQwbeKWkRtbMzc7NDBtICAgG1swbRtbMG0gIBtbMzM7NDBt4paR4paRG1szNzs0MG0gICAgICAgG1swbRtbMG0gIBtbMzM7NDBt4paRG1s5Nzs0MG0gG1szMzs0MG3ilpEbWzk3OzQwbSAbWzMzOzQwbeKWkRtbOTc7NDBtIBtbMzM7NDBt4paSG1szNzs0MG0gIBtbMG0bWzBtICAbWzk3OzQwbSAgG1szMzs0MG3ilpEbWzM3OzQwbSAgICAgIBtbMG0bWzBtICAbWzBtChtbOTc7NDBtIBtbMzM7NDBt4paRG1s5Nzs0MG0gIBtbMzM7NDBt4paRG1s5Nzs0MG0gIBtbMzM7NDBt4paRG1swbRtbMG0gIBtbOTc7NDBtICAgG1szMzs0MG3ilpEbWzk3OzQwbSAgG1szMzs0MG3ilpEbWzBtG1swbSAgG1s5Nzs0MG0gICAgG1szMzs0MG3ilpEbWzk3OzQwbSAgG1szMzs0MG3ilpEbWzBtG1swbSAgG1s5Nzs0MG0gICAgG1szMzs0MG3ilpEbWzk3OzQwbSAgG1szMzs0MG3ilpEbWzBtG1swbSAgG1szMDs0MG0gICAgICAgICAbWzBtG1swbSAgG1s5Nzs0MG0gICAgG1szMzs0MG3ilpEbWzk3OzQwbSAbWzMzOzQwbeKWkRtbMzc7NDBtICAbWzBtG1swbSAgG1szMDs0MG0gICAgICAgICAbWzBtG1swbSAgG1swbQoK"
@@ -21,6 +24,12 @@ const banner = "ChtbOTc7NDBtIBtbOTc7NDNt4paEG1szMzs0MG3ilojilpIbWzk3OzQwbSAbWzMz
   and just fmt.Println(dec) to print the banner without additional styling
 */
 func Banner() {
+	if runtime.GOOS == "windows" {
+		println(config.Title + " " + config.Version)
+		println(" ")
+		return
+	}
+
 	if len(banner) < 1 {
 		panic("invalid banner data")
 	}