Support Windows ARM Python versions

Note that only Python >=3.11.0 supports Windows ARM.

Author: nre-ableton

src/com/ableton/Pyenv.groovy

@@ -57,6 +57,17 @@ class Pyenv implements Serializable {
       }
     }
 
+    // For Windows ARM, see if there is a ARM binary that can be used to avoid running
+    // Python under PRISM.
+    if (script.env.OS == 'Windows_NT' &&
+      // If Java is an x86 binary on ARM64 running under PRISM, env.PROCESSOR_ARCHITECTURE
+      // will report AMD64, so we need to use env.PROCESSOR_IDENTIFIER instead.
+      script.env.PROCESSOR_IDENTIFIER?.startsWith('ARMv8') &&
+      versionSupported(trimmedPythonVersion + '-arm')
+    ) {
+      trimmedPythonVersion += '-arm'
+    }
+
     VirtualEnv venv = new VirtualEnv(script, randomSeed)
     script.withEnv(["PYENV_VERSION=${trimmedPythonVersion}"]) {
       try {

test/com/ableton/PyenvTest.groovy

@@ -160,6 +160,35 @@ class PyenvTest extends BasePipelineTest {
     shMocks.each { mock -> assertCallStackContains(mock[0]) }
   }
 
+  @Test
+  void createVirtualEnvWindowsArm() {
+    String pythonVersion = '1.2.3'
+    String pyenvRoot = 'C:\\mock\\pyenv\\root'
+    String posixPyenvRoot = '/c/mock/pyenv/root'
+    String shPyenvRoot = 'C:/mock/pyenv/root'
+    List shMocks = [
+      new Tuple(installCommands(
+        pyenvRoot: shPyenvRoot,
+        pythonVersion: pythonVersion + '-arm',
+        isUnix: false,
+        posixPyenvRoot: posixPyenvRoot,
+      ), '', 0),
+      new Tuple("${shPyenvRoot}/bin/pyenv install --list", '''Available versions:
+  1.2.3
+  1.2.3-arm
+''', 0),
+    ]
+    shMocks.each { mock -> helper.addShMock(mock[0], mock[1], mock[2]) }
+    helper.registerAllowedMethod('fileExists', [String]) { return true }
+    script.env['OS'] = 'Windows_NT'
+    script.env['PROCESSOR_IDENTIFIER'] = 'ARMv8 (64-bit) Family 8 Model AC4 Revision 0'
+
+    Object venv = new Pyenv(script, pyenvRoot).createVirtualEnv(pythonVersion, 1)
+
+    assertEquals("/workspace/.venv/${TEST_RANDOM_NAME}" as String, venv.venvRootDir)
+    shMocks.each { mock -> assertCallStackContains(mock[0]) }
+  }
+
   @Test
   void createVirtualEnvUnsupportedPythonVersion() {
     String pythonVersion = '6.6.6'