From aaa9136a2c90030e8d011e99ad85d01ac76ff398 Mon Sep 17 00:00:00 2001 From: netbenix Date: Sat, 20 Nov 2021 15:40:24 +0100 Subject: [PATCH] Updated Config Handler --- Lotus/ConfigHandler.go | 36 +++++++++++++++++++++++++++++++++--- 1 file changed, 33 insertions(+), 3 deletions(-) diff --git a/Lotus/ConfigHandler.go b/Lotus/ConfigHandler.go index d9319a6..7e69bf8 100644 --- a/Lotus/ConfigHandler.go +++ b/Lotus/ConfigHandler.go @@ -5,9 +5,6 @@ import ( "os" ) -type Config struct{ - Registries []Registry -} type Registry struct { Address string @@ -36,3 +33,36 @@ func GetRegistries() []Registry{ return registires } + +func checkIfConfigFolderExists(){ + logger := DetailedLogger("ConfigHandler", "checkIfConfigFolderExists") + + if _, err := os.Stat("config"); os.IsNotExist(err) { + dirErr := os.Mkdir("config", 600) + if dirErr != nil { + logger.Fatal(dirErr) + } + } +} + +func CheckIfRegistryListExists(){ + if _, err := os.Stat("config/registrylist.json"); os.IsNotExist(err){ + writeDefaultConfig() + } +} + +func writeDefaultConfig(){ + logger := DetailedLogger("ConfigHandler", "WriteDefaultConfig") + + checkIfConfigFolderExists() + + bytes, err := json.MarshalIndent([]Registry{ + {Address: "core.lotus-pm.net:9090", Register: "core"}, + {Address: "community.lotus-pm.net:9090", Register: "community"}, + }, "", " ") + if err != nil{ + logger.Fatal(err) + } + + _ = os.WriteFile("config/registrylist.json", bytes, 600) +} \ No newline at end of file