mirror of
https://github.com/nestriness/cli.git
synced 2025-12-13 04:35:37 +02:00
✨ feat: Write cli executable as a go program (#4)
## Description **What issue are you solving (or what feature are you adding) and how are you doing it?** I have opted to go with Golang as it has inbuilt support for writing cross-platform executables unlike something like JS. And also, it has support for running docker using the unix socket directly in code.
This commit is contained in:
76
pkg/specs/system.go
Normal file
76
pkg/specs/system.go
Normal file
@@ -0,0 +1,76 @@
|
||||
package specs
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os/exec"
|
||||
"runtime"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type SysSpecs struct {
|
||||
osx string
|
||||
}
|
||||
|
||||
func New() *SysSpecs {
|
||||
return &SysSpecs{osx: runtime.GOOS}
|
||||
}
|
||||
|
||||
func (s SysSpecs) GetGPUInfo() (string, error) {
|
||||
var output []byte
|
||||
var err error
|
||||
|
||||
switch s.osx {
|
||||
case "windows":
|
||||
output, err = exec.Command("wmic", "path", "win32_VideoController", "get", "name").Output()
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("error retrieving GPU information on Windows: %v", err)
|
||||
}
|
||||
case "darwin":
|
||||
output, err = exec.Command("system_profiler", "SPDisplaysDataType").Output()
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("error retrieving GPU information on macOS: %v", err)
|
||||
}
|
||||
case "linux":
|
||||
output, err = exec.Command("lspci", "-vnn").Output()
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("error retrieving GPU information on Linux: %v", err)
|
||||
}
|
||||
default:
|
||||
return "", fmt.Errorf("error: GPU information retrieval not implemented for %s", runtime.GOOS)
|
||||
}
|
||||
|
||||
outputStr := strings.TrimSpace(string(output))
|
||||
|
||||
if s.osx == "windows" {
|
||||
lines := strings.Split(outputStr, "\r\n")[1:]
|
||||
gpuName := strings.TrimSpace(strings.Join(lines, " "))
|
||||
return gpuName, nil
|
||||
}
|
||||
|
||||
if s.osx == "darwin" {
|
||||
lines := strings.Split(outputStr, "\n")
|
||||
for _, line := range lines {
|
||||
if strings.Contains(line, "Chipset Model:") {
|
||||
fields := strings.Split(line, ":")
|
||||
if len(fields) >= 2 {
|
||||
gpuName := strings.TrimSpace(fields[1])
|
||||
return gpuName, nil
|
||||
}
|
||||
}
|
||||
}
|
||||
return "", fmt.Errorf("error parsing GPU information on macOS")
|
||||
}
|
||||
|
||||
lines := strings.Split(outputStr, "\n")
|
||||
|
||||
for _, line := range lines {
|
||||
if strings.Contains(line, "VGA compatible controller") {
|
||||
fields := strings.Fields(line)
|
||||
if len(fields) > 2 {
|
||||
gpuName := strings.Join(fields[2:], " ")
|
||||
return gpuName, nil
|
||||
}
|
||||
}
|
||||
}
|
||||
return "", fmt.Errorf("error parsing GPU information on Linux")
|
||||
}
|
||||
Reference in New Issue
Block a user