From f3d0c88c54838d68e667c97bc10a2a47f893e153 Mon Sep 17 00:00:00 2001
From: Naveenraj M <22456988+naveenrajm7@users.noreply.github.com>
Date: Sat, 25 Jan 2025 14:18:33 -0700
Subject: [PATCH] scripting: allow creation of macos vm By accepting 'operating
 system' field, we will configure the os in boot section If macos, we required
 'ipsw' to be present and configure the same to vm.

---
 Scripting/UTM.sdef                        |  6 +++++
 Scripting/UTMScriptingCreateCommand.swift | 32 ++++++++++++++++++++++-
 2 files changed, 37 insertions(+), 1 deletion(-)

diff --git a/Scripting/UTM.sdef b/Scripting/UTM.sdef
index 234408ce4..8d02efa4b 100644
--- a/Scripting/UTM.sdef
+++ b/Scripting/UTM.sdef
@@ -570,6 +570,12 @@
           <property name="notes" code="NoTe" type="text"
             description="User-specified notes."/>
             
+          <property name="operating system" code="OpSy" type="text" access="r"
+            description="Operating system (cannot be changed after creation)."/>
+            
+          <property name="ipsw" code="IpSw" type="file"
+            description="Macos recovery IPSW (cannot be changed after creation)."/>
+              
           <property name="memory" code="MeMy" type="integer"
             description="RAM size (in mebibytes)."/>
             
diff --git a/Scripting/UTMScriptingCreateCommand.swift b/Scripting/UTMScriptingCreateCommand.swift
index fb6fbdf8a..2c803dfa9 100644
--- a/Scripting/UTMScriptingCreateCommand.swift
+++ b/Scripting/UTMScriptingCreateCommand.swift
@@ -15,6 +15,7 @@
 //
 
 import Foundation
+import Virtualization
 
 @MainActor
 @objc(UTMScriptingCreateCommand)
@@ -102,8 +103,29 @@ class UTMScriptingCreateCommand: NSCreateCommand, UTMScriptable {
         guard record["name"] as? String != nil else {
             throw ScriptingError.nameNotSpecified
         }
+        guard let osString = record["operatingSystem"] as? String  else {
+            throw ScriptingError.osNotSpecified
+        }
         let config = UTMAppleConfiguration()
-        config.system.boot = try UTMAppleConfigurationBoot(for: .linux)
+        // validate os
+        switch osString {
+        case "macos":
+            // need ipsw for macos
+            guard let ipswURL = record["ipsw"] as? URL else {
+                throw ScriptingError.ipswNotSpecified
+            }
+            let image = try await VZMacOSRestoreImage.image(from: ipswURL)
+            guard let model = image.mostFeaturefulSupportedConfiguration?.hardwareModel else {
+                throw ScriptingError.ipswNotSupported
+            }
+            config.system.macPlatform = UTMAppleConfigurationMacPlatform(newHardware: model)
+            config.system.boot = try UTMAppleConfigurationBoot(for: .macOS)
+            config.system.boot.macRecoveryIpswURL = ipswURL
+        case "linux":
+            config.system.boot = try UTMAppleConfigurationBoot(for: .linux)
+        default:
+            throw ScriptingError.osNotSupported
+        }
         config.virtualization.hasBalloon = true
         config.virtualization.hasEntropy = true
         config.networks = [UTMAppleConfigurationNetwork()]
@@ -131,6 +153,10 @@ class UTMScriptingCreateCommand: NSCreateCommand, UTMScriptable {
         case configurationNotFound
         case nameNotSpecified
         case architectureNotSpecified
+        case osNotSpecified
+        case osNotSupported
+        case ipswNotSpecified
+        case ipswNotSupported
         
         var errorDescription: String? {
             switch self {
@@ -140,6 +166,10 @@ class UTMScriptingCreateCommand: NSCreateCommand, UTMScriptable {
             case .configurationNotFound: return NSLocalizedString("A valid configuration must be specified.", comment: "UTMScriptingAppDelegate")
             case .nameNotSpecified: return NSLocalizedString("No name specified in the configuration.", comment: "UTMScriptingAppDelegate")
             case .architectureNotSpecified: return NSLocalizedString("No architecture specified in the configuration.", comment: "UTMScriptingAppDelegate")
+            case .osNotSpecified: return NSLocalizedString("No operating system specified in the configuration.", comment: "UTMScriptingAppDelegate")
+            case .osNotSupported: return NSLocalizedString("This operating system is not supported on your machine", comment: "UTMScriptingAppDelegate")
+            case .ipswNotSpecified: return NSLocalizedString("No ipsw file specified in the configuration for macos vm", comment: "UTMScriptingAppDelegate")
+            case .ipswNotSupported: return NSLocalizedString("Your machine does not support running this IPSW.", comment: "UTMScriptingAppDelegate")
             }
         }
     }