From 879bbd9523596bc94b8aaa60ecd2c12799b74b8a Mon Sep 17 00:00:00 2001
From: "Ward Nakchbandi (Cosmic Fusion)" <hotrod.master@hotmail.com>
Date: Thu, 17 Aug 2023 18:56:21 +0300
Subject: [PATCH] add auto init for apx v2

---
 alpine/commands.go |  4 +++
 arch/commands.go   |  4 +++
 command/command.go | 67 ++++++++++++++++++++++++++++++++++++++++++++++
 fedora/commands.go |  4 +++
 4 files changed, 79 insertions(+)

diff --git a/alpine/commands.go b/alpine/commands.go
index 88f8d64..229a077 100644
--- a/alpine/commands.go
+++ b/alpine/commands.go
@@ -2,6 +2,10 @@ package alpine
 
 var PackageManager = "apx"
 
+var ContainerSubsystem = "pikman-alpine"
+
+var ApxSubsystem = "alpine"
+
 var Commands = map[string]string{
 	"autoremove": "pikman-alpine autoremove",
 	"clean":      "pikman-alpine clean",
diff --git a/arch/commands.go b/arch/commands.go
index 8a0c16e..47ce4f6 100644
--- a/arch/commands.go
+++ b/arch/commands.go
@@ -2,6 +2,10 @@ package arch
 
 var PackageManager = "apx"
 
+var ContainerSubsystem = "pikman-arch-linux"
+
+var ApxSubsystem = "arch-linux"
+
 var Commands = map[string]string{
 	"autoremove": "pikman-arch-linux autoremove",
 	"clean":      "pikman-arch-linux clean",
diff --git a/command/command.go b/command/command.go
index e97173f..8b7f023 100644
--- a/command/command.go
+++ b/command/command.go
@@ -33,6 +33,26 @@ var packageManagerMap = map[types.OSType]string{
 	types.Flatpak: flatpak.PackageManager,
 }
 
+var  containerSubsystemMap = map[types.OSType]string{
+	types.Arch:    arch.ContainerSubsystem,
+	types.Fedora:    fedora.ContainerSubsystem,
+	types.Alpine:    alpine.ContainerSubsystem,
+}
+
+func (c *Command) ContainerSubsystem() string {
+    return containerSubsystemMap[c.OsType]
+}
+
+var  apxSubsystemMap = map[types.OSType]string{
+	types.Arch:    arch.ApxSubsystem,
+	types.Fedora:    fedora.ApxSubsystem,
+	types.Alpine:    alpine.ApxSubsystem,
+}
+
+func (c *Command) ApxSubsystem() string {
+    return apxSubsystemMap[c.OsType]
+}
+
 type Command struct {
 	Command       string
 	OsType        types.OSType
@@ -55,6 +75,53 @@ func (c *Command) processCommand() error {
 	}
 
 	var err error
+	
+	if c.OsType != types.Ubuntu && c.OsType != types.Flatpak && c.Command != "init" && c.ContainerName == "" {
+		cmd := exec.Command("/bin/bash", "-c", "apx subsystems list | grep ^" + c.ContainerSubsystem() + "$")
+    		cmd.Stdin = os.Stdin
+    		cmd.Stderr = os.Stderr
+    		if err != nil {
+       			fmt.Println("Error 1")
+    		}
+    		if err := cmd.Start(); err != nil {
+        		fmt.Println("Error 2")
+    		}
+    		if err != nil {
+        		fmt.Println("Error 3")
+    		}
+    		if err := cmd.Wait(); err != nil {
+    			fmt.Println("Warning: Subsystem hasn't been pre-initialized, initializing...")
+        		cmd_exec := exec.Command("/bin/bash", "-c", "apx subsystems new -n " + c.ContainerSubsystem() + " -s " + c.ApxSubsystem())
+        		cmd_exec.Stdout = os.Stdout
+			if err := cmd_exec.Run(); err != nil {
+  				fmt.Println("Apx Error: ", err)
+			}
+    		}
+	}
+	
+	if c.OsType != types.Ubuntu && c.OsType != types.Flatpak && c.Command != "init" && c.ContainerName != "" {
+		cmd := exec.Command("/bin/bash", "-c", "apx subsystems list | grep ^" + c.ContainerName + "$")
+    		cmd.Stdin = os.Stdin
+    		cmd.Stderr = os.Stderr
+    		if err != nil {
+       			fmt.Println("Error 1")
+    		}
+    		if err := cmd.Start(); err != nil {
+        		fmt.Println("Error 2")
+    		}
+    		if err != nil {
+        		fmt.Println("Error 3")
+    		}
+    		if err := cmd.Wait(); err != nil {
+    			fmt.Println("Warning: Subsystem hasn't been pre-initialized, initializing...")
+        		cmd_exec := exec.Command("/bin/bash", "-c", "apx subsystems new -n " + c.ContainerName + " -s " + c.ApxSubsystem())
+        		cmd_exec.Stdout = os.Stdout
+			if err := cmd_exec.Run(); err != nil {
+  				fmt.Println("Apx Error: ", err)
+			}
+    		}
+	}
+	
 	if c.OsType != types.Ubuntu && c.OsType != types.Flatpak && c.ContainerName != "" {
 		c.PackageName = append([]string{"--name " + c.ContainerName}, c.PackageName...)
 	}
diff --git a/fedora/commands.go b/fedora/commands.go
index d0c35a9..976d59c 100644
--- a/fedora/commands.go
+++ b/fedora/commands.go
@@ -2,6 +2,10 @@ package fedora
 
 var PackageManager = "apx"
 
+var ContainerSubsystem = "pikman-fedora-workstation"
+
+var ApxSubsystem = "fedora-workstation"
+
 var Commands = map[string]string{
 	"autoremove": "pikman-fedora-workstation autoremove",
 	"clean":      "pikman-fedora-workstation clean",