Scaffolding for configuration + preparation for setting passphrase (#192)

* Scaffolding for configuration + preparation for setting passphrase

* Setup config on dev & user installation

* dev installation/uninstallation fixes

Author: vaslabs

.goreleaser.yml

@@ -90,6 +90,10 @@ nfpms:
         file_info:
           mode: 0644
 
+      - src: etc/piwebagent2/config/config.env
+        dst: /etc/piwebagent2/config/config.env
+        type: "config|noreplace"
+        
       - src: usr/share/piwebagent2/assets
         dst: /usr/share/piwebagent2/assets
 

install.sh

@@ -29,6 +29,12 @@ function install_assets() {
     cp -r $SHARED_PATH /$SHARED_PATH
 }
 
+function install_config() {
+    mkdir -p $CONFIG_DIR
+    cp $CONFIG_FILE /$CONFIG_FILE
+    chown -R piwebagent2 /$CONFIG_FILE
+}
+
 function prepare_unpack() {
     target=$(mktemp -d)
     cp target/piwebagent2.zip $target/
@@ -45,6 +51,7 @@ install_binary
 install_assets
 create_user
 chown piwebagent2 -R /$SHARED_PATH
+install_config
 sudoer_user_priviledges
 register_service
 start_service

installation_paths

@@ -3,3 +3,5 @@ SERVICE_PATH=lib/systemd/system/piwebagent2.service
 BINARY_PATH=usr/bin/piwebagent2
 SHARED_PATH=usr/share/piwebagent2
 SUDOERS_PATH=etc/sudoers.d/piwebagent2
+CONFIG_DIR=etc/piwebagent2/config
+CONFIG_FILE=${CONFIG_DIR}/config.env

scripts/postinstall.sh

@@ -6,3 +6,6 @@ echo "Enabling piwebagent2.service"
 systemctl enable piwebagent2.service
 echo "Starting piwebagent2.service"
 systemctl start piwebagent2.service
+
+echo "Giving permissions to the config directory"
+chown -R piwebagent2 /etc/piwebagent2/config

service/system/etc/piwebagent2/config/config.env

@@ -0,0 +1,2 @@
+PASSWORD_HASH=
+PORT_NUMBER=8080

service/web/cmd/pi-web-agent.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"io"
 	"log"
 	"net/http"
@@ -27,5 +28,7 @@ func main() {
 	http.HandleFunc("/api/control/stream", action_dispatcher_handler)
 	http.Handle("/", http.FileServer(http.Dir("assets")))
 
-	log.Fatal(http.ListenAndServe(":8080", nil))
+	addr := fmt.Sprintf(":%d", single_user_session.PWA_Config.Port())
+
+	log.Fatal(http.ListenAndServe(addr, nil))
 }

service/web/configs/.actions

@@ -0,0 +1,61 @@
+package shell
+
+import (
+	"fmt"
+
+	"github.com/spf13/viper"
+)
+
+type RPI_Config interface {
+	Port() uint32
+	Password_Hash() string
+	Update_Passphrase(phrase string)
+}
+
+type Dynamic_RPI_Config struct {
+	config *viper.Viper
+}
+
+const (
+	PASSWORD_HASH_KEY = "PASSWORD_HASH"
+	PORT_NUMBER_KEY   = "PORT_NUMBER"
+)
+
+func (config *Dynamic_RPI_Config) Port() uint32 {
+	return config.config.GetUint32(PORT_NUMBER_KEY)
+}
+
+func (config *Dynamic_RPI_Config) Password_Hash() string {
+	return config.config.GetString(PASSWORD_HASH_KEY)
+}
+
+func (config *Dynamic_RPI_Config) Update_Passphrase(phrase string) {
+	config.config.Set(PASSWORD_HASH_KEY, phrase)
+	config.config.WriteConfig()
+}
+
+func Load_PWA_config(path string) RPI_Config {
+	viper_config, _ := load_PWA_config(path)
+	return &Dynamic_RPI_Config{viper_config}
+
+}
+
+func Load_RPI_PWA_Config() RPI_Config {
+	return Load_PWA_config("/etc/piwebagent2/config")
+}
+
+func load_PWA_config(path string) (*viper.Viper, error) {
+	config := viper.New()
+	config.SetDefault(PASSWORD_HASH_KEY, "")
+	config.SetDefault(PORT_NUMBER_KEY, uint32(8080))
+	config.AddConfigPath(path)
+	config.SetConfigType("env")
+	config.WatchConfig()
+	config.ReadInConfig()
+	err := config.SafeWriteConfig()
+	if err != nil {
+		fmt.Printf("Error writing config for the first time %s\n", err.Error())
+	}
+	return config, err
+
+}