65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"os/exec"
|
||
|
|
"runtime"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// detectDistro returns the Linux distribution name (if possible)
|
||
|
|
func detectDistro() string {
|
||
|
|
// Check if we're on Linux
|
||
|
|
if runtime.GOOS != "linux" {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
|
||
|
|
// Try to read /etc/os-release to determine the distro
|
||
|
|
file, err := os.Open("/etc/os-release")
|
||
|
|
if err != nil {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
defer file.Close()
|
||
|
|
|
||
|
|
scanner := bufio.NewScanner(file)
|
||
|
|
for scanner.Scan() {
|
||
|
|
line := scanner.Text()
|
||
|
|
if strings.HasPrefix(line, "ID=") {
|
||
|
|
parts := strings.SplitN(line, "=", 2)
|
||
|
|
if len(parts) == 2 {
|
||
|
|
return strings.Trim(parts[1], `"`)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
|
||
|
|
// getPackageList returns the list of installed packages based on the distro
|
||
|
|
func getPackageList(distro string) ([]string, error) {
|
||
|
|
var cmd *exec.Cmd
|
||
|
|
|
||
|
|
// Run the appropriate command based on the detected distribution
|
||
|
|
switch distro {
|
||
|
|
case "ubuntu", "debian":
|
||
|
|
cmd = exec.Command("dpkg-query", "-W", "-f=${Package} ${Version}\n")
|
||
|
|
case "fedora", "centos", "rhel":
|
||
|
|
cmd = exec.Command("rpm", "-qa")
|
||
|
|
case "arch", "manjaro":
|
||
|
|
cmd = exec.Command("pacman", "-Q")
|
||
|
|
default:
|
||
|
|
return nil, fmt.Errorf("unsupported distribution: %s", distro)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Capture the command's output
|
||
|
|
output, err := cmd.CombinedOutput()
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("error running command: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Split the output into lines and return
|
||
|
|
lines := strings.Split(string(output), "\n")
|
||
|
|
return lines, nil
|
||
|
|
}
|