feat(publisher-gcs): add Google Cloud Storage publisher

package.json

@@ -39,6 +39,7 @@
     "@electron/get": "^2.0.0",
     "@electron/osx-sign": "^1.0.5",
     "@electron/rebuild": "^3.2.10",
+    "@google-cloud/storage": "^7.1.0",
     "@malept/cross-spawn-promise": "^2.0.0",
     "@octokit/core": "^3.2.4",
     "@octokit/plugin-retry": "^3.0.9",

packages/publisher/gcs/package.json

@@ -0,0 +1,31 @@
+{
+  "name": "@electron-forge/publisher-gcs",
+  "version": "6.4.2",
+  "description": "Google Cloud Storage publisher for Electron Forge",
+  "repository": "https://github.com/electron/forge",
+  "author": "Evgeny Vlasenko",
+  "license": "MIT",
+  "main": "dist/PublisherGCS.js",
+  "typings": "dist/PublisherGCS.d.ts",
+  "devDependencies": {
+    "chai": "^4.3.3",
+    "mocha": "^9.0.1"
+  },
+  "engines": {
+    "node": ">= 14.17.5"
+  },
+  "dependencies": {
+    "@electron-forge/publisher-base": "6.4.2",
+    "@electron-forge/shared-types": "6.4.2",
+    "@google-cloud/storage": "^7.1.0",
+    "debug": "^4.3.1",
+    "google-auth-library": "^9.0.0"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "files": [
+    "dist",
+    "src"
+  ]
+}

packages/publisher/gcs/src/Config.ts

@@ -0,0 +1,59 @@
+import { PredefinedAcl } from '@google-cloud/storage';
+
+export interface PublisherGCSConfig {
+  /**
+   * The path to the file that is either:
+   * - the JSON file that contains your Google service account credentials, or
+   * - the PEM/PKCS #12-formatted file that contains the private key.
+   *
+   * Defaults to the value in the `GOOGLE_APPLICATION_CREDENTIALS` environment variable.
+   */
+  keyFilename?: string;
+  /**
+   * The Google Cloud project ID.
+   *
+   * Defaults to the value in the `GOOGLE_CLOUD_PROJECT` environment variable.
+   * */
+  projectId?: string;
+  /**
+   * The email for your Google service account, *required* when using a PEM/PKCS #12-formatted
+   * file in the [[keyFilename]] option.
+   *
+   * Defaults to the value in the `GOOGLE_CLOUD_CLIENT_EMAIL` environment variable.
+   */
+  clientEmail?: string;
+  /**
+   * The private key for your Google service account.
+   *
+   * Defaults to the value in the `GOOGLE_CLOUD_PRIVATE_KEY` environment variable.
+   */
+  privateKey?: string;
+  /**
+   * The name of the Google Cloud Storage bucket where artifacts are uploaded.
+   */
+  bucket?: string;
+  /**
+   * The key prefix where artifacts are uploaded, e.g., `my/prefix`.
+   *
+   * Defaults to the application `version` specified in the app's `package.json`.
+   */
+  folder?: string;
+  /**
+   * Apply a predefined set of access controls to this object.
+   */
+  predefinedAcl?: PredefinedAcl;
+  /**
+   * Whether to make uploaded artifacts public to the internet.
+   * Alias for config.predefinedAcl = 'publicRead'.
+   */
+  public?: boolean;
+  /**
+   * Whether to make uploaded artifacts private.
+   * Alias for config.predefinedAcl = 'private'.
+   */
+  private?: boolean;
+  /**
+   * Custom function to provide the key to upload a given file to
+   */
+  keyResolver?: (fileName: string, platform: string, arch: string) => string;
+}

packages/publisher/gcs/src/PublisherGCS.ts

@@ -0,0 +1,98 @@
+import path from 'path';
+
+import { PublisherBase, PublisherOptions } from '@electron-forge/publisher-base';
+import { Storage } from '@google-cloud/storage';
+import debug from 'debug';
+import { CredentialBody } from 'google-auth-library';
+
+import { PublisherGCSConfig } from './Config';
+
+const d = debug('electron-forge:publish:gcs');
+
+type GCSArtifact = {
+  path: string;
+  keyPrefix: string;
+  platform: string;
+  arch: string;
+};
+
+export default class PublisherGCS extends PublisherBase<PublisherGCSConfig> {
+  name = 'gcs';
+
+  private GCSKeySafe = (key: string) => {
+    return key.replace(/@/g, '_').replace(/\//g, '_');
+  };
+
+  async publish({ makeResults, setStatusLine }: PublisherOptions): Promise<void> {
+    const artifacts: GCSArtifact[] = [];
+
+    if (!this.config.bucket) {
+      throw new Error('In order to publish to Google Cloud Storage you must set the "bucket" property in your Forge config.');
+    }
+
+    for (const makeResult of makeResults) {
+      artifacts.push(
+        ...makeResult.artifacts.map((artifact) => ({
+          path: artifact,
+          keyPrefix: this.config.folder || this.GCSKeySafe(makeResult.packageJSON.name),
+          platform: makeResult.platform,
+          arch: makeResult.arch,
+        }))
+      );
+    }
+
+    const storage = new Storage({
+      keyFilename: this.config.keyFilename,
+      credentials: this.generateCredentials(),
+      projectId: this.config.projectId,
+    });
+
+    const bucket = storage.bucket(this.config.bucket);
+
+    d('creating Google Cloud Storage client with options:', this.config);
+
+    let uploaded = 0;
+    const updateStatusLine = () => setStatusLine(`Uploading distributable (${uploaded}/${artifacts.length})`);
+
+    updateStatusLine();
+    await Promise.all(
+      artifacts.map(async (artifact) => {
+        d('uploading:', artifact.path);
+
+        await bucket.upload(artifact.path, {
+          gzip: true,
+          destination: this.keyForArtifact(artifact),
+          predefinedAcl: this.config.predefinedAcl,
+          public: this.config.public,
+          private: this.config.private,
+        });
+
+        uploaded += 1;
+        updateStatusLine();
+      })
+    );
+  }
+
+  keyForArtifact(artifact: GCSArtifact): string {
+    if (this.config.keyResolver) {
+      return this.config.keyResolver(path.basename(artifact.path), artifact.platform, artifact.arch);
+    }
+
+    return `${artifact.keyPrefix}/${artifact.platform}/${artifact.arch}/${path.basename(artifact.path)}`;
+  }
+
+  generateCredentials(): CredentialBody | undefined {
+    const clientEmail = this.config.clientEmail;
+    const privateKey = this.config.privateKey;
+
+    if (clientEmail && privateKey) {
+      return {
+        client_email: clientEmail,
+        private_key: privateKey,
+      };
+    }
+    return undefined;
+  }
+}
+
+export { PublisherGCS, PublisherGCSConfig };