From c6c94bba40ceb98b23627f9c05c6c3d6d58517c7 Mon Sep 17 00:00:00 2001
From: HikariKnight <2557889+HikariKnight@users.noreply.github.com>
Date: Fri, 7 Apr 2023 22:50:53 +0200
Subject: [PATCH] setup a config handler

---
 internal/configs/configs.go | 83 +++++++++++++++++++++++++++++++++++++
 1 file changed, 83 insertions(+)
 create mode 100644 internal/configs/configs.go

diff --git a/internal/configs/configs.go b/internal/configs/configs.go
new file mode 100644
index 0000000..287fe2c
--- /dev/null
+++ b/internal/configs/configs.go
@@ -0,0 +1,83 @@
+package configs
+
+import (
+	"errors"
+	"fmt"
+	"os"
+	"regexp"
+
+	"github.com/HikariKnight/ls-iommu/pkg/errorcheck"
+)
+
+type Path struct {
+	MODPROBE   string
+	INITRAMFS  string
+	ETCMODULES string
+	DEFAULT    string
+	QUICKEMU   string
+	DRACUT     string
+	MKINITCPIO string
+}
+
+func GetConfigPaths() *Path {
+	Paths := &Path{
+		MODPROBE:   "config/etc/modprobe.d",
+		INITRAMFS:  "config/etc/initramfs-tools",
+		ETCMODULES: "config/etc/modules",
+		DEFAULT:    "config/etc/default",
+		QUICKEMU:   "config/quickemu",
+		DRACUT:     "config/etc/dracut.conf.d",
+		MKINITCPIO: "config/etc/mkinitcpio.conf",
+	}
+
+	return Paths
+}
+
+func InitConfigs() {
+	config := GetConfigPaths()
+
+	dirs := []string{
+		config.MODPROBE,
+		config.INITRAMFS,
+		config.DEFAULT,
+		config.DRACUT,
+	}
+
+	// Remove old config
+	os.RemoveAll("config")
+
+	// Make a regex to get the system path instead of the config path
+	syspath_re := regexp.MustCompile(`^config`)
+
+	// For each directory
+	for _, confpath := range dirs {
+		// Get the system path
+		syspath := syspath_re.ReplaceAllString(confpath, "")
+
+		// If the path exists
+		if _, err := os.Stat(syspath); !errors.Is(err, os.ErrNotExist) {
+			// Create the directories for our configs
+			err := os.MkdirAll(confpath, os.ModePerm)
+			errorcheck.ErrorCheck(err)
+		}
+	}
+
+	files := []string{
+		config.ETCMODULES,
+		config.MKINITCPIO,
+		fmt.Sprintf("%s/modules", config.INITRAMFS),
+	}
+
+	for _, conffile := range files {
+		// Get the system file path
+		sysfile := syspath_re.ReplaceAllString(conffile, "")
+
+		// If the file exists
+		if _, err := os.Stat(sysfile); !errors.Is(err, os.ErrNotExist) {
+			// Create the directories for our configs
+			file, err := os.Create(conffile)
+			errorcheck.ErrorCheck(err)
+			defer file.Close()
+		}
+	}
+}