win arm whl fix

Author: bewithgaurav

setup.py

@@ -2,19 +2,56 @@
 import sys
 from setuptools import setup, find_packages
 from setuptools.dist import Distribution
+from setuptools.command.bdist_wheel import bdist_wheel
 
 # Custom distribution to force platform-specific wheel
 class BinaryDistribution(Distribution):
     def has_ext_modules(self):
         return True
 
+# Custom bdist_wheel command to override platform tag
+class CustomBdistWheel(bdist_wheel):
+    def finalize_options(self):
+        # Call the original finalize_options first to initialize self.bdist_dir
+        bdist_wheel.finalize_options(self)
+        
+        # Override the platform tag with our custom one based on ARCHITECTURE env var
+        if sys.platform.startswith('win'):
+            # Strip quotes if present
+            arch = os.environ.get('ARCHITECTURE', 'x64')
+            if isinstance(arch, str):
+                arch = arch.strip('"\'')
+                
+            print(f"Architecture from environment: '{arch}'")
+            
+            if arch in ['x86', 'win32']:
+                self.plat_name = "win32"
+                platform_dir = "win32"
+            elif arch == 'arm64':
+                self.plat_name = "win_arm64"
+                platform_dir = "win_arm64"
+            else:  # Default to x64/amd64
+                self.plat_name = "win_amd64"
+                platform_dir = "win_amd64"
+            
+            # Override the plat_name for the wheel
+            print(f"Setting wheel platform tag to: {self.plat_name}")
+            
+            # Force platform-specific paths if bdist_dir is already set
+            if self.bdist_dir and "win-amd64" in self.bdist_dir:
+                self.bdist_dir = self.bdist_dir.replace("win-amd64", f"win-{platform_dir}")
+                print(f"Using build directory: {self.bdist_dir}")
+
 # Find all packages in the current directory
 packages = find_packages()
 
 # Determine the architecture and platform tag for the wheel
 if sys.platform.startswith('win'):
     # Get architecture from environment variable or default to x64
     arch = os.environ.get('ARCHITECTURE', 'x64')
+    # Strip quotes if present
+    if isinstance(arch, str):
+        arch = arch.strip('"\'')
 
     # Normalize architecture values
     if arch in ['x86', 'win32']:
@@ -68,4 +105,8 @@ def has_ext_modules(self):
             'libs/*/vcredist/*', 'libs/*/vcredist/**/*',  # Exclude vcredist directories, added here since `'libs/*' is already included`
         ],
     },
+    # Register custom commands
+    cmdclass={
+        'bdist_wheel': CustomBdistWheel,
+    },
 )