69 lines
1.7 KiB
Go
69 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/spf13/viper"
|
|
"yves-biener.de/wish-serve/internal/serve"
|
|
)
|
|
|
|
func createDefaultConfig() {
|
|
viper.SetDefault("host", "127.0.0.1")
|
|
viper.SetDefault("port", "8022")
|
|
viper.SetDefault("users", map[string]string{})
|
|
viper.SetDefault("key.dir", ".ssh")
|
|
viper.SetDefault("key.name", "ssh_ed25519_key")
|
|
viper.SetDefault("app.name", "echo")
|
|
viper.SetDefault("app.args", []string{"Hello World"})
|
|
}
|
|
|
|
// Load the configuration file from (`$HOME/.config/serve/config.yml` or
|
|
// `./config.yml`) if it exists. Otherwise use the default values from
|
|
// `createDefaultConfig()`.
|
|
func loadConfig() {
|
|
viper.SetConfigName("config")
|
|
viper.SetConfigType("yml")
|
|
viper.AddConfigPath("$HOME/.config/serve")
|
|
viper.AddConfigPath(".")
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
if _, ok := err.(viper.ConfigFileNotFoundError); !ok {
|
|
panic(fmt.Errorf("fatal error config file: %w", err))
|
|
}
|
|
}
|
|
}
|
|
|
|
// Overwrite (default) configuration value for the application to serve with
|
|
// given command line arguments.
|
|
//
|
|
// NOTE: This will also overwrite the configuration for the `app` entry from the
|
|
// configuration file.
|
|
func overwriteServeCommand() {
|
|
if len(os.Args) >= 2 {
|
|
name := os.Args[1]
|
|
viper.Set("app.name", name)
|
|
if len(os.Args) >= 3 {
|
|
args := os.Args[2:]
|
|
viper.Set("app.args", args)
|
|
} else {
|
|
// no arguments provided -> clear app arguments
|
|
viper.Set("app.args", []string{})
|
|
}
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
createDefaultConfig()
|
|
loadConfig()
|
|
overwriteServeCommand()
|
|
|
|
serve.Serve(
|
|
viper.GetString("host"),
|
|
viper.GetString("port"),
|
|
fmt.Sprintf("%s/%s", viper.GetString("key.dir"), viper.GetString("key.name")),
|
|
viper.GetStringMapString("users"),
|
|
viper.GetString("app.name"),
|
|
viper.GetStringSlice("app.args"),
|
|
)
|
|
}
|