From 26fe0de3542336af1608b20f21dac8ef9b481a51 Mon Sep 17 00:00:00 2001
From: Naveenraj M <22456988+naveenrajm7@users.noreply.github.com>
Date: Sun, 22 Dec 2024 23:31:12 -0700
Subject: [PATCH] scripting: expose icon in config * Allows to get/set built-in
 icons iconURL is URL in UTMConfig, but stored as string in config.plist, same
 is done for scripting interface.

---
 Scripting/UTM.sdef                     |  6 ++++++
 Scripting/UTMScriptingConfigImpl.swift | 18 ++++++++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/Scripting/UTM.sdef b/Scripting/UTM.sdef
index 234408ce4..b0692271b 100644
--- a/Scripting/UTM.sdef
+++ b/Scripting/UTM.sdef
@@ -405,6 +405,9 @@
         <record-type name="qemu configuration" code="QeCf" description="QEMU virtual machine configuration.">
           <property name="name" code="pnam" type="text"
             description="Virtual machine name."/>
+        
+          <property name="icon" code="IcOn" type="text"
+            description="Virtual machine icon."/>
             
           <property name="notes" code="NoTe" type="text"
             description="User-specified notes."/>
@@ -567,6 +570,9 @@
           <property name="name" code="pnam" type="text"
             description="Virtual machine name."/>
             
+          <property name="icon" code="IcOn" type="text"
+            description="Virtual machine icon."/>
+            
           <property name="notes" code="NoTe" type="text"
             description="User-specified notes."/>
             
diff --git a/Scripting/UTMScriptingConfigImpl.swift b/Scripting/UTMScriptingConfigImpl.swift
index b00edca15..6fc7bf0cc 100644
--- a/Scripting/UTMScriptingConfigImpl.swift
+++ b/Scripting/UTMScriptingConfigImpl.swift
@@ -96,6 +96,7 @@ extension UTMScriptingConfigImpl {
     private func serializeQemuConfiguration(_ config: UTMQemuConfiguration) -> [AnyHashable : Any] {
         [
             "name": config.information.name,
+            "icon": config.information.iconURL?.deletingPathExtension().lastPathComponent ?? "",
             "notes": config.information.notes ?? "",
             "architecture": config.system.architecture.rawValue,
             "machine": config.system.target.rawValue,
@@ -200,6 +201,7 @@ extension UTMScriptingConfigImpl {
     private func serializeAppleConfiguration(_ config: UTMAppleConfiguration) -> [AnyHashable : Any] {
         [
             "name": config.information.name,
+            "icon": config.information.iconURL?.deletingPathExtension().lastPathComponent ?? "",
             "notes": config.information.notes ?? "",
             "memory": config.system.memorySize,
             "cpuCores": config.system.cpuCount,
@@ -307,6 +309,13 @@ extension UTMScriptingConfigImpl {
         if let name = record["name"] as? String, !name.isEmpty {
             config.information.name = name
         }
+        if let icon = record["icon"] as? String, !icon.isEmpty {
+            if let url = UTMConfigurationInfo.builtinIcon(named: icon) {
+                config.information.iconURL = url
+            } else {
+                throw ConfigurationError.iconNotFound(icon: icon)
+            }
+        }
         if let notes = record["notes"] as? String, !notes.isEmpty {
             config.information.notes = notes
         }
@@ -530,6 +539,13 @@ extension UTMScriptingConfigImpl {
         if let name = record["name"] as? String, !name.isEmpty {
             config.information.name = name
         }
+        if let icon = record["icon"] as? String, !icon.isEmpty {
+            if let url = UTMConfigurationInfo.builtinIcon(named: icon) {
+                config.information.iconURL = url
+            } else {
+                throw ConfigurationError.iconNotFound(icon: icon)
+            }
+        }
         if let notes = record["notes"] as? String, !notes.isEmpty {
             config.information.notes = notes
         }
@@ -652,6 +668,7 @@ extension UTMScriptingConfigImpl {
         case invalidDriveDescription
         case indexNotFound(index: Int)
         case deviceNotSupported
+        case iconNotFound(icon: String)
         
         var errorDescription: String? {
             switch self {
@@ -659,6 +676,7 @@ extension UTMScriptingConfigImpl {
             case .invalidDriveDescription: return NSLocalizedString("Drive description is invalid.", comment: "UTMScriptingConfigImpl")
             case .indexNotFound(let index): return String.localizedStringWithFormat(NSLocalizedString("Index %lld cannot be found.", comment: "UTMScriptingConfigImpl"), index)
             case .deviceNotSupported: return NSLocalizedString("This device is not supported by the target.", comment: "UTMScriptingConfigImpl")
+            case .iconNotFound(let icon): return String.localizedStringWithFormat(NSLocalizedString("The icon named '%@' cannot be found in the built-in icons.", comment: "UTMScriptingConfigImpl"), icon)
             }
         }
     }