From 2c6a279b9bcf7957f9760b46767f835f794d9e01 Mon Sep 17 00:00:00 2001
From: alteredCoder <kevin@crowdsec.net>
Date: Wed, 6 Oct 2021 18:12:12 +0200
Subject: [PATCH] add load configuration file

---
 pkg/csconfig/api.go | 42 ++++++++++++++++++++++++++++++++----------
 1 file changed, 32 insertions(+), 10 deletions(-)

diff --git a/pkg/csconfig/api.go b/pkg/csconfig/api.go
index 7862cd7a4..7dde7be65 100644
--- a/pkg/csconfig/api.go
+++ b/pkg/csconfig/api.go
@@ -3,6 +3,7 @@ package csconfig
 import (
 	"fmt"
 	"io/ioutil"
+	"os"
 	"strings"
 
 	"github.com/crowdsecurity/crowdsec/pkg/apiclient"
@@ -78,16 +79,18 @@ func (l *LocalApiClientCfg) Load() error {
 
 /*local api service configuration*/
 type LocalApiServerCfg struct {
-	ListenURI              string              `yaml:"listen_uri,omitempty"` //127.0.0.1:8080
-	TLS                    *TLSCfg             `yaml:"tls"`
-	DbConfig               *DatabaseCfg        `yaml:"-"`
-	LogDir                 string              `yaml:"-"`
-	LogMedia               string              `yaml:"-"`
-	OnlineClient           *OnlineApiClientCfg `yaml:"online_client"`
-	ProfilesPath           string              `yaml:"profiles_path,omitempty"`
-	Profiles               []*ProfileCfg       `yaml:"-"`
-	LogLevel               *log.Level          `yaml:"log_level"`
-	UseForwardedForHeaders bool                `yaml:"use_forwarded_for_headers,omitempty"`
+	ListenURI              string                 `yaml:"listen_uri,omitempty"` //127.0.0.1:8080
+	TLS                    *TLSCfg                `yaml:"tls"`
+	DbConfig               *DatabaseCfg           `yaml:"-"`
+	LogDir                 string                 `yaml:"-"`
+	LogMedia               string                 `yaml:"-"`
+	OnlineClient           *OnlineApiClientCfg    `yaml:"online_client"`
+	ProfilesPath           string                 `yaml:"profiles_path,omitempty"`
+	ConsoleConfigPath      string                 `yaml:"console_path,omitempty"`
+	Profiles               []*ProfileCfg          `yaml:"-"`
+	LogLevel               *log.Level             `yaml:"log_level"`
+	UseForwardedForHeaders bool                   `yaml:"use_forwarded_for_headers,omitempty"`
+	ConsoleConfig          map[string]interface{} `yaml:"-"`
 }
 
 type TLSCfg struct {
@@ -105,6 +108,7 @@ func (c *Config) LoadAPIServer() error {
 		if err := c.API.Server.LoadProfiles(); err != nil {
 			return errors.Wrap(err, "while loading profiles for LAPI")
 		}
+
 		if c.API.Server.OnlineClient != nil && c.API.Server.OnlineClient.CredentialsFilePath != "" {
 			if err := c.API.Server.OnlineClient.Load(); err != nil {
 				return errors.Wrap(err, "loading online client credentials")
@@ -124,6 +128,24 @@ func (c *Config) LoadAPIServer() error {
 	return nil
 }
 
+func (c *LocalApiServerCfg) LoadConsoleConfig() error {
+	c.ConsoleConfig = make(map[string]interface{})
+	if _, err := os.Stat(c.ConsoleConfigPath); err != nil && os.IsNotExist(err) {
+		return nil
+	}
+
+	yamlFile, err := ioutil.ReadFile(c.ConsoleConfigPath)
+	if err != nil {
+		return fmt.Errorf("reading console config file '%s': %s", c.ConsoleConfigPath, err)
+	}
+	err = yaml.Unmarshal(yamlFile, c.ConsoleConfig)
+	if err != nil {
+		return fmt.Errorf("unmarshaling console config file '%s': %s", c.ConsoleConfigPath, err)
+	}
+
+	return nil
+}
+
 func (c *Config) LoadAPIClient() error {
 	if c.API != nil && c.API.Client != nil && c.API.Client.CredentialsFilePath != "" && !c.DisableAgent {
 		if err := c.API.Client.Load(); err != nil {