main.go

package main

import (
	"fmt"
	"os"
)

const (
	baseDir = "/usr/bin/arfpkg"
	tempDir = baseDir + "/temp"
)

func main() {
	// Make sure everything is running as root
	if os.Getuid() != 0 {
		fmt.Println("Please run as root (or with sudo).")
		os.Exit(1)
	} else {
		// Make sure temp exists
		if err := os.MkdirAll(tempDir, 0755); err != nil {
			fmt.Println("Failed to create temp directory:", err)
			os.Exit(1)
		}

		// Always clean up
		defer func() {
			_ = os.Remove(tempDir + "/packages.toml")
		}()

		// Get package list TOML [WILL BE REPLACED WITH SOURCE LISTS SOON]
		if err := download("packages.toml", "https://ixrepo.pages.dev/packages.toml"); err != nil {
			fmt.Println("Could not fetch package list:", err)
			os.Exit(1)
		}

		// Command arg mode
		if len(os.Args) >= 2 {
			cmd := os.Args[1]

			switch cmd {
			case "install", "fetch":
				if len(os.Args) < 3 {
					fmt.Println("Usage: arfpkg install <package>")
					os.Exit(1)
				}
				pkg := os.Args[2]
				if err := install(pkg); err != nil {
					fmt.Println("Install failed:", err)
					os.Exit(1)
				}

			case "remove":
				if len(os.Args) < 3 {
					fmt.Println("Usage: arfpkg remove <package>")
					os.Exit(1)
				}
				pkg := os.Args[2]
				if err := remove(pkg); err != nil {
					fmt.Println("Remove failed:", err)
					os.Exit(1)
				}

			default:
				fmt.Printf("Unknown command: %s\n", cmd)
				os.Exit(1)
			}

			fmt.Println("All done! 🦴")
			return
		}

		// Interactive mode(aka TUI/CLI)
		var oper string
		fmt.Println("-----| arfpkg v1.4 |-----")
		fmt.Print("Enter operation (install/remove): ")
		fmt.Scanln(&oper)

		switch oper {
		case "install", "fetch":
			var pkg string
			fmt.Print("Enter package name: ")
			fmt.Scanln(&pkg)
			if err := install(pkg); err != nil {
				fmt.Println("Install failed:", err)
				return
			}

		case "remove":
			var pkg string
			fmt.Print("Enter package name: ")
			fmt.Scanln(&pkg)
			if err := remove(pkg); err != nil {
				fmt.Println("Remove failed:", err)
				return
			}

		default:
			fmt.Printf("Invalid operation: %s\n", oper)
			return
		}

		fmt.Println("All done! 🦴")
	}
}