does apt_linux.go vs apt_windows.go work automagically?

This commit is contained in:
Jeff Carr 2024-11-15 09:50:13 -06:00
parent 7b06923395
commit ea96d4880e
3 changed files with 89 additions and 0 deletions

64
apt_linux.go Normal file
View File

@ -0,0 +1,64 @@
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
}

25
main.go
View File

@ -55,6 +55,31 @@ func main() {
for pkg, version := range installedPackages {
fmt.Printf("%s: %s\n", pkg, version)
}
// Detect the Linux distribution
distro := detectDistro()
if distro == "" {
fmt.Println("Unable to detect Linux distribution.")
return
}
fmt.Printf("Detected distribution: %s\n", distro)
// Get the list of installed packages for the detected distro
packages, err := getPackageList(distro)
if err != nil {
fmt.Println("Error:", err)
return
}
// Print the installed packages
fmt.Println("Installed Packages:")
for _, pkg := range packages {
if pkg != "" {
fmt.Println(pkg)
}
}
go NewWatchdog()
startHTTP()