From c4c71fc49aae972a963601fa6e2be55bc722e587 Mon Sep 17 00:00:00 2001
From: CapCreeperGR <nickvavour@gmail.com>
Date: Sun, 9 Jun 2024 14:45:17 +0300
Subject: [PATCH] Added hidden_gpus option in the config

---
 config/config.yaml | 1 +
 main.go            | 2 ++
 utils.go           | 5 ++++-
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/config/config.yaml b/config/config.yaml
index 3ad64eb..b604356 100644
--- a/config/config.yaml
+++ b/config/config.yaml
@@ -3,3 +3,4 @@ fetch_script: auto
 ansii_colors: []
 force_config_ansii: false
 dependency_warning: true
+hidden_gpus: []
diff --git a/main.go b/main.go
index 6e30e71..ea6ac9e 100644
--- a/main.go
+++ b/main.go
@@ -22,6 +22,7 @@ var config = StormfetchConfig{
 	AnsiiColors:       make([]int, 0),
 	ForceConfigAnsii:  false,
 	DependencyWarning: true,
+	HiddenGPUS:        make([]int, 0),
 }
 
 type StormfetchConfig struct {
@@ -31,6 +32,7 @@ type StormfetchConfig struct {
 	AnsiiColors       []int  `yaml:"ansii_colors"`
 	ForceConfigAnsii  bool   `yaml:"force_config_ansii"`
 	DependencyWarning bool   `yaml:"dependency_warning"`
+	HiddenGPUS        []int  `yaml:"hidden_gpus"`
 }
 
 func main() {
diff --git a/utils.go b/utils.go
index 62f377a..5bdd168 100644
--- a/utils.go
+++ b/utils.go
@@ -130,7 +130,10 @@ func getGPUNames() []string {
 	if err != nil {
 		return nil
 	}
-	for _, graphics := range gpu.GraphicsCards {
+	for i, graphics := range gpu.GraphicsCards {
+		if slices.Contains(config.HiddenGPUS, i+1) {
+			continue
+		}
 		if graphics.DeviceInfo != nil {
 			ret = append(ret, graphics.DeviceInfo.Product.Name)
 		}