From c09226c93d8275c49e0c75d5680afdb853650e04 Mon Sep 17 00:00:00 2001
From: Erick Zhao <erick@hotmail.ca>
Date: Wed, 11 Dec 2024 12:09:05 -0800
Subject: [PATCH] fix(flat): don't use pkgbuild for MAS distributions

---
 src/flat.ts | 61 ++++++++++++++++++++++++++++++++++++-----------------
 1 file changed, 42 insertions(+), 19 deletions(-)

diff --git a/src/flat.ts b/src/flat.ts
index aed3ccf..51ffaa9 100644
--- a/src/flat.ts
+++ b/src/flat.ts
@@ -49,27 +49,44 @@ async function validateFlatOpts (opts: FlatOptions): Promise<ValidatedFlatOption
 
 /**
  * This function returns a promise flattening the application.
- * @function
- * @param {Object} opts - Options.
- * @returns {Promise} Promise.
+ * @param opts - Options for building the .pkg archive
  */
 async function buildApplicationPkg (opts: ValidatedFlatOptions, identity: Identity) {
-  const componentPkgPath = path.join(path.dirname(opts.app), path.basename(opts.app, '.app') + '-component.pkg');
-  const pkgbuildArgs = ['--install-location', opts.install, '--component', opts.app, componentPkgPath];
-  if (opts.scripts) {
-    pkgbuildArgs.unshift('--scripts', opts.scripts);
-  }
-  debugLog('Building component package... ' + opts.app);
-  await execFileAsync('pkgbuild', pkgbuildArgs);
+  if (opts.platform === 'mas') {
+    const args = ['--component', opts.app, opts.install, '--sign', identity.name, opts.pkg];
+    if (opts.keychain) {
+      args.unshift('--keychain', opts.keychain);
+    }
 
-  const args = ['--package', componentPkgPath, opts.install, '--sign', identity.name, opts.pkg];
-  if (opts.keychain) {
-    args.unshift('--keychain', opts.keychain);
-  }
+    debugLog('Flattening Mac App Store package... ' + opts.app);
+    await execFileAsync('productbuild', args);
+  } else {
+    const componentPkgPath = path.join(
+      path.dirname(opts.app),
+      path.basename(opts.app, '.app') + '-component.pkg'
+    );
+    const pkgbuildArgs = [
+      '--install-location',
+      opts.install,
+      '--component',
+      opts.app,
+      componentPkgPath
+    ];
+    if (opts.scripts) {
+      pkgbuildArgs.unshift('--scripts', opts.scripts);
+    }
+    debugLog('Building component package... ' + opts.app);
+    await execFileAsync('pkgbuild', pkgbuildArgs);
 
-  debugLog('Flattening... ' + opts.app);
-  await execFileAsync('productbuild', args);
-  await execFileAsync('rm', [componentPkgPath]);
+    const args = ['--package', componentPkgPath, opts.install, '--sign', identity.name, opts.pkg];
+    if (opts.keychain) {
+      args.unshift('--keychain', opts.keychain);
+    }
+
+    debugLog('Flattening OS X Installer package... ' + opts.app);
+    await execFileAsync('productbuild', args);
+    await execFileAsync('rm', [componentPkgPath]);
+  }
 }
 
 /**
@@ -89,7 +106,10 @@ export async function buildPkg (_opts: FlatOptions) {
     if (validatedOptions.identityValidation === false) {
       // Do nothing
     } else {
-      identities = await findIdentities(validatedOptions.keychain || null, validatedOptions.identity);
+      identities = await findIdentities(
+        validatedOptions.keychain || null,
+        validatedOptions.identity
+      );
     }
   } else {
     debugWarn('No `identity` passed in arguments...');
@@ -105,7 +125,10 @@ export async function buildPkg (_opts: FlatOptions) {
       debugLog(
         'Finding `Developer ID Application` certificate for distribution outside the Mac App Store...'
       );
-      identities = await findIdentities(validatedOptions.keychain || null, 'Developer ID Installer:');
+      identities = await findIdentities(
+        validatedOptions.keychain || null,
+        'Developer ID Installer:'
+      );
     }
   }