Enables cross-platform DLL build of the shell extension (for MSI installs)

Author: zooba

_msbuild.py

@@ -148,6 +148,31 @@ def launcher_exe(name, platform, windowed=False):
     )
 
 
+def pyshellext(ext='.exe', **props):
+    link_opts = ItemDefinition(
+        'Link',
+        AdditionalDependencies=Prepend('RuntimeObject.lib;'),
+    )
+    if ext != '.exe':
+        link_opts.options['ModuleDefinitionFile'] = '$(SourceRootDir)src\\pyshellext\\pyshellext.def'
+
+
+    return CProject(f"pyshellext{ext.rpartition('.')[0]}",
+        VersionInfo(
+            FileDescription='Python shell extension',
+            OriginalFilename=f'pyshellext{ext}',
+        ),
+        ItemDefinition('ClCompile', LanguageStandard='stdcpp20'),
+        link_opts,
+        Manifest('default.manifest'),
+        CSourceFile('shellext.cpp'),
+        ResourceFile('pyshellext.rc'),
+        source='src/pyshellext',
+        StaticLibcppLinkage=True,
+        **props,
+    )
+
+
 PACKAGE = Package('python-manager',
     PyprojectTomlFile('pyproject.toml'),
     # MSIX manifest
@@ -206,23 +231,9 @@ def launcher_exe(name, platform, windowed=False):
     main_exe("python3"),
     mainw_exe("pythonw3"),
 
-    CProject("pyshellext",
-        VersionInfo(
-            FileDescription="Python shell extension",
-            OriginalFilename="pyshellext.exe",
-        ),
-        Property('StaticLibcppLinkage', 'true'),
-        ItemDefinition('ClCompile', LanguageStandard='stdcpp20'),
-        ItemDefinition('Link',
-            AdditionalDependencies=Prepend("RuntimeObject.lib;"),
-            SubSystem='WINDOWS',
-        ),
-        Manifest('default.manifest'),
-        CSourceFile('shellext.cpp'),
-        ResourceFile('pyshellext.rc'),
-        source='src/pyshellext',
-        ConfigurationType='Application',
-    ),
+    pyshellext(".exe", ConfigurationType="Application"),
+    pyshellext("-64.dll", Platform="x64"),
+    pyshellext("-arm64.dll", Platform="ARM64"),
 )
 
 

make-msix.py

@@ -53,10 +53,19 @@
                   "/mf", "appx"])
 
 # Clean up non-shipping files from LAYOUT
+preserved = [
+    *LAYOUT.glob("pyshellext*.dll"),
+]
+
+for f in preserved:
+    print("Preserving", f, "as", TEMP / f.name)
+    copyfile(f, TEMP / f.name)
+
 unlink(
     *LAYOUT.rglob("*.pdb"),
     *LAYOUT.rglob("*.pyc"),
     *LAYOUT.rglob("__pycache__"),
+    *preserved,
 )
 
 # Package into DIST
@@ -122,3 +131,9 @@ def patch_appx(source):
     with zipfile.ZipFile(DIST_MSIXUPLOAD, "w") as zf:
         zf.write(DIST_STORE_MSIX, arcname=DIST_STORE_MSIX.name)
         zf.write(DIST_APPXSYM, arcname=DIST_APPXSYM.name)
+
+
+for f in preserved:
+    print("Restoring", f, "from", TEMP / f.name)
+    copyfile(TEMP / f.name, f)
+    unlink(TEMP / f.name)

src/pymanager/msi.wxs

@@ -27,6 +27,17 @@
       <ComponentRef Id="ProductComponent" />
       <ComponentRef Id="RuntimeComponent" />
       <ComponentRef Id="TemplatesComponent" />
+      <!-- TODO: Platform-specific components for pyshellext.
+      Ensure only one is installed based on the _real_ processor type!
+
+      <File Source="pyshellext-64.dll">
+        <Class Id="{C7E29CB0-9691-4DE8-B72B-6719DDC0B4A1}" Advertise="no" Context="InProcServer32" />
+      </File>
+      <File Source="pyshellext-arm64.dll">
+        <Class Id="{C7E29CB0-9691-4DE8-B72B-6719DDC0B4A1}" Advertise="no" Context="InProcServer32" />
+      </File>
+
+      -->
     </Feature>
 
     <util:BroadcastEnvironmentChange />
@@ -44,18 +55,6 @@
       <File Source="vcruntime140.dll" />
       <File Source="vcruntime140_1.dll" />
 
-      <!--
-      When installed via MSIX, this extension must be out of proc.
-      Apparently, when installed via MSI, it must be in proc. Which means we'd
-      need to compile DLLs for each platform just for the MSI, as well as the
-      EXE that we use for regular installs.
-      Right now, not worth it.
-
-      <File Source="pyshellext.exe">
-        <Class Id="{C7E29CB0-9691-4DE8-B72B-6719DDC0B4A1}" Advertise="no" Context="LocalServer32" />
-      </File>
-      -->
-
       <File Source="version.txt" />
       <Environment Id="PATH" Action="set" Name="PATH" Part="last" System="yes" Value="[INSTALLFOLDER]" />
 

src/pyshellext/pyshellext.def

@@ -0,0 +1,3 @@
+EXPORTS
+  DllGetClassObject PRIVATE
+  DllCanUnloadNow PRIVATE

src/pyshellext/shellext.cpp

@@ -316,11 +316,12 @@ class DECLSPEC_UUID(CLSID_COMMAND_ENUMERATOR) CommandEnumerator
 
 
 class DECLSPEC_UUID(CLSID_IDLE_COMMAND) IdleCommand
-    : public RuntimeClass<RuntimeClassFlags<ClassicCom>, IExplorerCommand>
+    : public RuntimeClass<RuntimeClassFlags<ClassicCom>, IExplorerCommand, IObjectWithSite>
 {
     std::vector<IdleData> idles;
     std::wstring iconPath;
     std::wstring title;
+    ComPtr<IUnknown> _site;
 public:
     IdleCommand() : title(L"Edit in &IDLE")
     {
@@ -432,12 +433,28 @@ class DECLSPEC_UUID(CLSID_IDLE_COMMAND) IdleCommand
         ).Detach();
         return S_OK;
     }
+
+    IFACEMETHODIMP GetSite(REFIID riid, void **ppvSite)
+    {
+        if (_site) {
+            return _site->QueryInterface(riid, ppvSite);
+        }
+        *ppvSite = NULL;
+        return E_FAIL;
+    }
+
+    IFACEMETHODIMP SetSite(IUnknown *pSite)
+    {
+        _site = pSite;
+        return S_OK;
+    }
 };
 
 
 CoCreatableClass(IdleCommand);
 
 #ifdef PYSHELLEXT_TEST
+
 IExplorerCommand *MakeLaunchCommand(std::wstring title, std::wstring exe, std::wstring idle)
 {
     IdleData data = { .title = title, .exe = exe, .idle = idle };
@@ -449,10 +466,34 @@ IExplorerCommand *MakeIdleCommand(HKEY hive, LPCWSTR root)
 {
     return Make<IdleCommand>(hive, root).Detach();
 }
-#endif
 
+#elif defined(_WINDLL)
+
+#pragma comment(linker, "/export:DllGetClassObject")
+#pragma comment(linker, "/export:DllCanUnloadNow")
+
+STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, _COM_Outptr_ void** ppv)
+{
+    return Module<InProc>::GetModule().GetClassObject(rclsid, riid, ppv);
+}
+
+
+STDAPI DllCanUnloadNow()
+{
+    return Module<InProc>::GetModule().Terminate() ? S_OK : S_FALSE;
+}
+
+STDAPI_(BOOL) DllMain(_In_opt_ HINSTANCE hinst, DWORD reason, _In_opt_ void*)
+{
+    if (reason == DLL_PROCESS_ATTACH) {
+        hModule = hinst;
+        DisableThreadLibraryCalls(hinst);
+    }
+    return TRUE;
+}
+
+#else
 
-#ifndef PYSHELLEXT_TEST
 class OutOfProcModule : public Module<OutOfProc, OutOfProcModule>
 { };
 
@@ -475,4 +516,5 @@ int WINAPI wWinMain(
     CoUninitialize();
     return 0;
 }
+
 #endif