Fix issues with rails/actioncable in the server sdk (#14714)

* Fix issues with long running asynchronous calls in the server sdk

Author: adolfo-pd

packages/sdk/CHANGELOG.md

@@ -1,6 +1,15 @@
 <!-- markdownlint-disable MD024 -->
 # Changelog
 
+## [1.0.7] - 2024-11-21
+
+### Changed
+
+- The backend client now correctly uses asynchronous messaging to handle long running
+  operations.
+- Updated the backend command line tool to respect the `ENVIRONMENT` env variable
+  if set.
+
 ## [1.0.6] - 2024-11-20
 
 ### Changed

packages/sdk/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pipedream/sdk",
-  "version": "1.0.6",
+  "version": "1.0.7",
   "description": "Pipedream SDK",
   "main": "dist/server/server/index.js",
   "module": "dist/server/server/index.js",

packages/sdk/src/server/async.ts

@@ -4,6 +4,11 @@ import type { AsyncResponseManagerOpts } from "../shared/async";
 import { adapters } from "@rails/actioncable";
 import * as WS from "ws";
 
+declare global {
+  function addEventListener(type: string, listener: () => void): void;
+  function removeEventListener(type: string, listener: () => void): void;
+}
+
 export type ServerAsyncResponseManagerOpts = {
   apiHost: string;
   getOauthToken: () => Promise<AccessToken> | AccessToken;
@@ -16,12 +21,20 @@ export class ServerAsyncResponseManager extends AsyncResponseManager {
   constructor(opts: ServerAsyncResponseManagerOpts) {
     super();
     this.serverOpts = opts;
+    // eslint-disable-next-line @typescript-eslint/no-empty-function
+    global.addEventListener = () => {};
+    // eslint-disable-next-line @typescript-eslint/no-empty-function
+    global.removeEventListener = () => {};
     if (typeof adapters.WebSocket === "undefined")
-      adapters.WebSocket == WS;
+      adapters.WebSocket = WS as unknown as WebSocket;
   }
 
   protected override async getOpts(): Promise<AsyncResponseManagerOpts> {
-    const token = await this.serverOpts.getOauthToken();
+    const oauthToken = await this.serverOpts.getOauthToken();
+    if (!oauthToken?.token?.access_token) {
+      throw new Error("Invalid OAuth token structure");
+    }
+    const token = oauthToken.token.access_token;
     const projectId = await this.serverOpts.getProjectId();
     return {
       url: `wss://${this.serverOpts.apiHost}/websocket?oauth_token=${token}`,

packages/sdk/src/server/cli.ts

@@ -2,7 +2,7 @@ import { createBackendClient } from "./index";
 import { program } from "commander";
 
 const {
-  CLIENT_ID, CLIENT_SECRET, PROJECT_ID, API_HOST,
+  CLIENT_ID, CLIENT_SECRET, PROJECT_ID, API_HOST, ENVIRONMENT,
 } = process.env;
 
 if (!CLIENT_ID || !CLIENT_SECRET || !PROJECT_ID) {
@@ -17,6 +17,9 @@ const client = createBackendClient({
   },
   projectId: PROJECT_ID,
   apiHost: API_HOST,
+  environment: ENVIRONMENT === "production"
+    ? "production"
+    : "development",
 });
 
 program