238 lines
9.1 KiB
Go
238 lines
9.1 KiB
Go
package packagemanager
|
|
|
|
// This package is responsible for handling our actual install logic. We could have probably gotten away with
|
|
// implementing this into the packagemanager package, but this seems like a better way
|
|
// because this provides a single interface for all our install functions
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
"os"
|
|
"os/exec"
|
|
"os/user"
|
|
"strings"
|
|
"path/filepath"
|
|
"pixelridgesoftworks.com/AllPac/pkg/logger"
|
|
)
|
|
|
|
// installs a package using Pacman and logs the installation
|
|
func InstallPackagePacman(packageName string) error {
|
|
cmd := exec.Command("sudo", "pacman", "-Syu", "--noconfirm", packageName)
|
|
if output, err := cmd.CombinedOutput(); err != nil {
|
|
logger.Errorf("error installing package with Pacman: %s, %v", output, err)
|
|
return fmt.Errorf("error installing package with Pacman: %s, %v", output, err)
|
|
}
|
|
|
|
version, err := GetPacmanPackageVersion(packageName)
|
|
if err != nil {
|
|
logger.Errorf("An error has occured:", err)
|
|
return err
|
|
}
|
|
|
|
if err := LogInstallation(packageName, "pacman", version); err != nil {
|
|
logger.Errorf("error logging installation: %v", err)
|
|
return fmt.Errorf("error logging installation: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// installs a package using Snap and logs the installation
|
|
func InstallPackageSnap(packageName string) error {
|
|
cmd := exec.Command("sudo", "snap", "install", packageName)
|
|
output, err := cmd.CombinedOutput()
|
|
|
|
if err != nil {
|
|
outputStr := string(output)
|
|
logger.Errorf("error installing package with Snap: %s, %v", outputStr, err)
|
|
|
|
// Check if the error is due to the need for classic confinement
|
|
if strings.Contains(outputStr, "using classic") {
|
|
fmt.Println("This package requires installation in classic mode, which may perform arbitrary system changes outside of the security sandbox. Do you want to proceed? (yes/no)")
|
|
var response string
|
|
fmt.Scanln(&response)
|
|
if strings.ToLower(response) == "yes" {
|
|
// Retry installation with --classic flag
|
|
classicCmd := exec.Command("sudo", "snap", "install", "--classic", packageName)
|
|
if classicOutput, classicErr := classicCmd.CombinedOutput(); classicErr != nil {
|
|
logger.Errorf("error installing package with Snap in classic mode: %s, %v", classicOutput, classicErr)
|
|
return fmt.Errorf("error installing package with Snap in classic mode: %s, %v", classicOutput, classicErr)
|
|
}
|
|
} else {
|
|
return fmt.Errorf("installation aborted by user")
|
|
}
|
|
} else {
|
|
return fmt.Errorf("error installing package with Snap: %s, %v", outputStr, err)
|
|
}
|
|
}
|
|
|
|
version, err := GetVersionFromSnap(packageName)
|
|
if err != nil {
|
|
logger.Errorf("An error has occurred:", err)
|
|
return err
|
|
}
|
|
|
|
if err := LogInstallation(packageName, "snap", version); err != nil {
|
|
logger.Errorf("error logging installation: %v", err)
|
|
return fmt.Errorf("error logging installation: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// installs a package using Flatpak and logs the installation
|
|
func InstallPackageFlatpak(packageName string) error {
|
|
cmd := exec.Command("flatpak", "install", "-y", packageName)
|
|
if output, err := cmd.CombinedOutput(); err != nil {
|
|
logger.Errorf("error installing package with Flatpak: %s, %v", output, err)
|
|
return fmt.Errorf("error installing package with Flatpak: %s, %v", output, err)
|
|
}
|
|
|
|
version, err := GetVersionFromFlatpak(packageName)
|
|
if err != nil {
|
|
logger.Errorf("An error has occured:", err)
|
|
return err
|
|
}
|
|
|
|
if err := LogInstallation(packageName, "flatpak", version); err != nil {
|
|
logger.Errorf("error logging installation: %v", err)
|
|
return fmt.Errorf("error logging installation: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// clones the given AUR repository and installs it
|
|
func CloneAndInstallFromAUR(repoURL string, skipConfirmation bool) (string, error) {
|
|
// System update
|
|
if !skipConfirmation && !confirmAction("Do you want to update the system before proceeding? (skipping this step may result in partial updates, and break your system)") {
|
|
logger.Warnf("user aborted the system update")
|
|
return "", fmt.Errorf("user aborted the system update")
|
|
}
|
|
|
|
cmdUpdate := exec.Command("sudo", "pacman", "-Syu", "--noconfirm")
|
|
if output, err := cmdUpdate.CombinedOutput(); err != nil {
|
|
logger.Errorf("error updating system: %s, %v", output, err)
|
|
return "", fmt.Errorf("error updating system: %s, %v", output, err)
|
|
}
|
|
// Confirm before proceeding with each step
|
|
if !skipConfirmation && !confirmAction("Do you want to download and build package from " + repoURL + "?") {
|
|
logger.Warnf("user aborted the action")
|
|
return "", fmt.Errorf("user aborted the action")
|
|
}
|
|
|
|
// Get the current user's home directory
|
|
usr, err := user.Current()
|
|
if err != nil {
|
|
logger.Errorf("error getting current user: %v", err)
|
|
return "", fmt.Errorf("error getting current user: %v", err)
|
|
}
|
|
|
|
// Determine the name of the package from the repo URL
|
|
repoName := filepath.Base(repoURL)
|
|
|
|
// Remove .git suffix
|
|
repoName = strings.TrimSuffix(repoName, ".git")
|
|
|
|
// Get the current date in YYYYMMDD format
|
|
currentDate := time.Now().Format("20060102")
|
|
|
|
// Define the directory for this specific package clone
|
|
cloneDir := filepath.Join(usr.HomeDir, ".allpac", "cache", repoName+"-"+currentDate)
|
|
|
|
// Ensure the clone directory exists
|
|
if err := os.MkdirAll(cloneDir, 0755); err != nil {
|
|
logger.Errorf("error creating clone directory: %v", err)
|
|
return "", fmt.Errorf("error creating clone directory: %v", err)
|
|
}
|
|
|
|
// Define the base directory for AllPac cache
|
|
baseDir := filepath.Join(usr.HomeDir, ".allpac", "cache")
|
|
|
|
// Ensure the base directory exists
|
|
if err := os.MkdirAll(baseDir, 0755); err != nil {
|
|
logger.Errorf("error creating base directory: %v", err)
|
|
return "", fmt.Errorf("error creating base directory: %v", err)
|
|
}
|
|
|
|
// Clone the repository
|
|
cmdGitClone := exec.Command("git", "clone", repoURL, cloneDir)
|
|
if output, err := cmdGitClone.CombinedOutput(); err != nil {
|
|
logger.Errorf("error cloning AUR repo: %s, %v", output, err)
|
|
return "", fmt.Errorf("error cloning AUR repo: %s, %v", output, err)
|
|
}
|
|
|
|
// Change directory to the cloned repository
|
|
if err := os.Chdir(cloneDir); err != nil {
|
|
logger.Errorf("error changing directory: %v", err)
|
|
return "", fmt.Errorf("error changing directory: %v", err)
|
|
}
|
|
|
|
// Build the package using makepkg as the non-root user
|
|
cmdMakePkg := exec.Command("makepkg", "-si", "--noconfirm")
|
|
cmdMakePkg.Env = []string{"HOME=" + usr.HomeDir, "USER=" + usr.Username, "LOGNAME=" + usr.Username}
|
|
if output, err := cmdMakePkg.CombinedOutput(); err != nil {
|
|
logger.Errorf("error building package with makepkg: %s, %v", output, err)
|
|
return "", fmt.Errorf("error building package with makepkg: %s, %v", output, err)
|
|
}
|
|
|
|
// Extract the version from PKGBUILD
|
|
version, err := ExtractVersionFromPKGBUILD(cloneDir)
|
|
if err != nil {
|
|
logger.Errorf("error extracting version from PKGBUILD: %v", err)
|
|
return "", fmt.Errorf("error extracting version from PKGBUILD: %v", err)
|
|
}
|
|
|
|
// Confirm before installing
|
|
if !skipConfirmation && !confirmAction("Do you want to install the built package " + repoName + "?") {
|
|
logger.Warnf("user aborted the installation")
|
|
return "", fmt.Errorf("user aborted the installation")
|
|
}
|
|
|
|
if err := LogInstallation(repoName, "aur", version); err != nil {
|
|
logger.Errorf("error logging installation")
|
|
return "", fmt.Errorf("error logging installation: %v", err)
|
|
}
|
|
|
|
return version, nil
|
|
}
|
|
|
|
// installs Snap manually from the AUR
|
|
func InstallSnap() error {
|
|
version, err := CloneAndInstallFromAUR("https://aur.archlinux.org/snapd.git", true)
|
|
if err != nil {
|
|
logger.Errorf("error installing Snap: %v", err)
|
|
return fmt.Errorf("error installing Snap: %v", err)
|
|
}
|
|
|
|
if err := LogInstallation("snapd", "aur", version); err != nil {
|
|
logger.Errorf("error logging installation")
|
|
return fmt.Errorf("error logging installation: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// installs Git using Pacman
|
|
func InstallGit() error {
|
|
if err := InstallPackagePacman("git"); err != nil {
|
|
logger.Errorf("error installing Git: %v", err)
|
|
return fmt.Errorf("error installing Git: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// installs the base-devel group using Pacman
|
|
func InstallBaseDevel() error {
|
|
if err := InstallPackagePacman("base-devel"); err != nil {
|
|
logger.Errorf("error installing base-devel: %v", err)
|
|
return fmt.Errorf("error installing base-devel: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// installs the Flatpak package using Pacman
|
|
func InstallFlatpak() error {
|
|
if err := InstallPackagePacman("flatpak"); err != nil {
|
|
logger.Errorf("error installing flatpak: %v", err)
|
|
return fmt.Errorf("error installing flatpak: %v", err)
|
|
}
|
|
return nil
|
|
}
|