// Package system
// @Link  https://gitee.com/linngc/central-mirror
// @Copyright  Copyright (c) 2022 central CLI
// @Author  linngc
// @License
package system

import (
	"bytes"
	"os"
	"os/exec"
	"syscall"
)

// PackageUbuntuInstall 使用apt-get安装指定的Ubuntu软件包
func PackageUbuntuInstall(packageNames ...string) (mp map[string]bool, err error) {
	for _, packageName := range packageNames {
		// 构建apt-get install命令
		cmd := exec.Command("apt-get", "install", "-y", packageName)

		// 设置命令的标准输出和标准错误输出到当前进程的输出
		cmd.Stdout = os.Stdout
		cmd.Stderr = os.Stderr

		// 执行命令
		err := cmd.Run()
		if err != nil {
			mp[packageName] = false
			return mp, err
		}
		mp[packageName] = true
	}
	return mp, err
}

// PackageUbuntuInstalledCheck 检查Ubuntu系统中是否已安装指定的软件包
func PackageUbuntuInstalledCheck(packageNames ...string) (mp map[string]bool, err error) {
	for _, packageName := range packageNames {
		// 构建dpkg命令来查询软件包是否安装
		cmd := exec.Command("dpkg", "-l", packageName)

		// 执行命令并捕获输出
		var out bytes.Buffer
		cmd.Stdout = &out
		err := cmd.Run()

		// 检查命令执行是否有错误
		if err != nil {
			// 如果dpkg返回非零状态码，通常表示软件包未安装
			if exitErr, ok := err.(*exec.ExitError); ok {
				if exitErr.Sys().(syscall.WaitStatus).ExitStatus() != 0 {
					mp[packageName] = false
					// 如果命令返回状态码不为0，那么假设软件包未安装
					return mp, err
				}
			}
			mp[packageName] = false
			// 其他错误情况，比如无法执行dpkg命令等
			return mp, err
		}
		mp[packageName] = out.Len() > 0
	}

	// 如果命令成功执行，并且输出不为空，则软件包已安装
	return mp, err
}