@medplum/agent 2.0.29 → 2.0.31

package/installer.nsi

@@ -0,0 +1,199 @@
+# Medplum Agent Installer Builder
+# For use with NSIS 3.0+
+# See: https://nsis.sourceforge.io/
+
+!define COMPANY_NAME             "Medplum"
+!define APP_NAME                 "Medplum Agent"
+!define SERVICE_NAME             "MedplumAgent"
+!define INSTALLER_FILE_NAME      "medplum-agent-installer.exe"
+!define DEFAULT_BASE_URL         "https://api.medplum.com/"
+
+Name                             "${APP_NAME}"
+OutFile                          "${INSTALLER_FILE_NAME}"
+VIProductVersion                 "1.0.0.0"
+VIAddVersionKey ProductName      "${APP_NAME}"
+VIAddVersionKey Comments         "${APP_NAME}"
+VIAddVersionKey CompanyName      "${COMPANY_NAME}"
+VIAddVersionKey LegalCopyright   "${COMPANY_NAME}"
+VIAddVersionKey FileDescription  "${APP_NAME}"
+VIAddVersionKey FileVersion      1
+VIAddVersionKey ProductVersion   1
+VIAddVersionKey InternalName     "${APP_NAME}"
+VIAddVersionKey LegalTrademarks  "${COMPANY_NAME}"
+VIAddVersionKey OriginalFilename "${INSTALLER_FILE_NAME}"
+
+InstallDir "$PROGRAMFILES64\${APP_NAME}"
+
+!include "nsDialogs.nsh"
+
+RequestExecutionLevel admin
+
+Var WelcomeDialog
+Var WelcomeLabel
+Var baseUrl
+Var clientId
+Var clientSecret
+Var agentId
+
+Page custom WelcomePage
+Page custom InputPage InputPageLeave
+Page instfiles
+
+# The WelcomePage is a simple static screen that displays a friendly message.
+Function WelcomePage
+    nsDialogs::Create 1018
+    Pop $WelcomeDialog
+
+    ${If} $WelcomeDialog == error
+        Abort
+    ${EndIf}
+
+    ${NSD_CreateLabel} 0 0 100% 50u "Welcome to the ${APP_NAME} Installer!$\r$\n$\r$\nClick next to continue."
+    Pop $WelcomeLabel
+
+    nsDialogs::Show
+FunctionEnd
+
+# The InputPage captures all of the user input for the agent.
+Function InputPage
+    nsDialogs::Create 1018
+    Pop $0
+
+    StrCpy $baseUrl "${DEFAULT_BASE_URL}"
+    ${NSD_CreateLabel} 0 0 30% 12u "Base URL:"
+    Pop $R0
+    ${NSD_CreateText} 35% 0 65% 12u $baseUrl
+    Pop $R1
+
+    ${NSD_CreateLabel} 0 15u 30% 12u "Client ID:"
+    Pop $R2
+    ${NSD_CreateText} 35% 15u 65% 12u $clientId
+    Pop $R3
+
+    ${NSD_CreateLabel} 0 30u 30% 12u "Client Secret:"
+    Pop $R4
+    ${NSD_CreateText} 35% 30u 65% 12u $clientSecret
+    Pop $R5
+
+    ${NSD_CreateLabel} 0 45u 30% 12u "Agent ID:"
+    Pop $R6
+    ${NSD_CreateText} 35% 45u 65% 12u $agentId
+    Pop $R7
+
+    ${NSD_SetFocus} $R3
+    nsDialogs::Show
+FunctionEnd
+
+Function InputPageLeave
+    ${NSD_GetText} $R1 $baseUrl
+    ${NSD_GetText} $R3 $clientId
+    ${NSD_GetText} $R5 $clientSecret
+    ${NSD_GetText} $R7 $agentId
+FunctionEnd
+
+# Do the actual installation.
+# Install all of the files.
+# Install the Windows Service.
+Section
+    DetailPrint "${APP_NAME}"
+
+    # Call userInfo plugin to get user info.  The plugin puts the result in the stack
+    userInfo::getAccountType
+
+    # Pop the result from the stack into $0
+    Pop $0
+
+    # Compare the result with the string "Admin" to see if the user is admin.
+    # If match, jump 3 lines down.
+    strCmp $0 "Admin" +3
+
+    # If there is not a match, print message and return
+    DetailPrint "User is not admin: $0"
+    return
+
+    # Otherwise, confirm and return
+    DetailPrint "User is admin"
+
+    # Print user input
+    DetailPrint "Base URL: $baseUrl"
+    DetailPrint "Client ID: $clientId"
+    DetailPrint "Client Secret: $clientSecret"
+    DetailPrint "Agent ID: $agentId"
+
+    # Copy the service files to the root directory
+    SetOutPath "$INSTDIR"
+    File ..\..\node_modules\node-windows\bin\winsw\winsw.exe
+    File dist\medplum-agent-win-x64.exe
+    File README.md
+
+    # Create the winsw.xml config file
+    # See config file format: https://github.com/winsw/winsw/blob/v3/docs/xml-config-file.md
+    FileOpen $9 winsw.xml w
+    FileWrite $9 "<service>$\r$\n"
+    FileWrite $9 "<id>${SERVICE_NAME}</id>$\r$\n"
+    FileWrite $9 "<name>${APP_NAME}</name>$\r$\n"
+    FileWrite $9 "<description>Securely connects local devices to ${COMPANY_NAME} cloud</description>$\r$\n"
+    FileWrite $9 "<executable>$INSTDIR\medplum-agent-win-x64.exe</executable>$\r$\n"
+    FileWrite $9 "<arguments>$\"$baseUrl$\" $\"$clientId$\" $\"$clientSecret$\" $\"$agentId$\"</arguments>$\r$\n"
+    FileWrite $9 "<startmode>Automatic</startmode>$\r$\n"
+    FileWrite $9 "</service>$\r$\n"
+    FileClose $9
+
+    # Install the service
+    DetailPrint "Installing service..."
+    StrCpy $0 "winsw.exe install"
+    #DetailPrint "$0"
+    ExecWait $0 $1
+    DetailPrint "Install returned $1"
+
+    # Start the service
+    DetailPrint "Starting service..."
+    StrCpy $0 "winsw.exe start"
+    #DetailPrint "$0"
+    ExecWait $0 $1
+    DetailPrint "Start service returned $1"
+
+    # Create the uninstaller
+    DetailPrint "Creating the uninstaller..."
+    SetOutPath $INSTDIR
+    WriteUninstaller "$INSTDIR\uninstall.exe"
+
+    # Register the uninstaller
+    DetailPrint "Registering the uninstaller..."
+    WriteRegStr HKLM "SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\${SERVICE_NAME}" "DisplayName" "${APP_NAME}"
+    WriteRegStr HKLM "SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\${SERVICE_NAME}" "UninstallString" "$\"$INSTDIR\uninstall.exe$\""
+    WriteRegStr HKLM "SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\${SERVICE_NAME}" "QuietUninstallString" "$\"$INSTDIR\uninstall.exe$\" /S"
+    DetailPrint "Uninstaller complete"
+
+    # Create Start menu shortcuts
+    DetailPrint "$SMPROGRAMS\${APP_NAME}\${APP_NAME} Uninstall.lnk"
+    CreateDirectory "$SMPROGRAMS\${APP_NAME}"
+    CreateShortCut "$SMPROGRAMS\${APP_NAME}\${APP_NAME} Uninstall.lnk" "$INSTDIR\uninstall.exe"
+
+# default section end
+SectionEnd
+
+# Start the uninstaller
+Section Uninstall
+
+    # Uninstall the service
+    DetailPrint "Uninstalling service..."
+    SetOutPath "$INSTDIR"
+    StrCpy $0 "winsw.exe uninstall"
+    #DetailPrint "$0"
+    ExecWait $0 $1
+    DetailPrint "Uninstall returned $1"
+
+    # Get out of the service directory so we can delete it
+    SetOutPath "$PROGRAMFILES64"
+
+    # Uninstall the Start menu shortcuts
+    RMDir /r /REBOOTOK "$SMPROGRAMS\${APP_NAME}"
+
+    # Delete the files
+    RMDir /r /REBOOTOK "$INSTDIR"
+
+    # Unregister the program
+    DeleteRegKey HKLM "SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\${SERVICE_NAME}"
+
+SectionEnd

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@medplum/agent",
-  "version": "2.0.29",
+  "version": "2.0.31",
   "description": "Medplum Agent",
   "author": "Medplum <hello@medplum.com>",
   "license": "Apache-2.0",
@@ -17,17 +17,22 @@
     "clean": "rimraf dist",
     "build": "npm run clean && tsc && node esbuild.mjs",
     "test": "jest",
-    "package": "pkg ./dist/cjs/index.cjs --targets node18-win-x64 --output dist/medplum-agent-win-x64.exe"
+    "agent": "ts-node src/main.ts",
+    "package": "pkg ./dist/cjs/index.cjs --targets node18-win-x64 --output dist/medplum-agent-win-x64.exe --options no-warnings",
+    "installer": "makensis installer.nsi"
   },
   "dependencies": {
     "@medplum/core": "*",
     "@medplum/hl7": "*",
-    "node-windows": "1.0.0-beta.8"
+    "node-windows": "1.0.0-beta.8",
+    "ws": "8.13.0"
   },
   "devDependencies": {
     "@medplum/fhirtypes": "*",
     "@medplum/mock": "*",
     "@types/node-windows": "0.1.2",
+    "@types/ws": "8.5.5",
+    "mock-socket": "9.2.1",
     "pkg": "5.8.1"
   }
 }

package/src/__mocks__/ws.ts

@@ -0,0 +1 @@
+export { WebSocket as default } from 'mock-socket';

package/src/main.test.ts

@@ -1,13 +1,15 @@
-import { allOk, Hl7Message } from '@medplum/core';
-import { Bot, Resource } from '@medplum/fhirtypes';
+import { allOk, createReference, Hl7Message } from '@medplum/core';
+import { Agent, Bot, Endpoint, Resource } from '@medplum/fhirtypes';
 import { Hl7Client } from '@medplum/hl7';
 import { MockClient } from '@medplum/mock';
+import { Server } from 'mock-socket';
 import { App } from './main';
 
 jest.mock('node-windows');
 
 const medplum = new MockClient();
 let bot: Bot;
+let endpoint: Endpoint;
 
 describe('Agent', () => {
   beforeAll(async () => {
@@ -18,17 +20,73 @@ describe('Agent', () => {
     });
 
     bot = await medplum.createResource<Bot>({ resourceType: 'Bot' });
+
+    endpoint = await medplum.createResource<Endpoint>({
+      resourceType: 'Endpoint',
+      address: 'mllp://0.0.0.0:56000',
+    });
   });
 
   test('Runs successfully', async () => {
-    const app = new App(medplum, bot);
-    app.start();
+    const agent = await medplum.createResource<Agent>({
+      resourceType: 'Agent',
+      channel: [
+        {
+          endpoint: createReference(endpoint),
+          targetReference: createReference(bot),
+        },
+      ],
+    });
+
+    const app = new App(medplum, agent.id as string);
+    await app.start();
+    app.stop();
     app.stop();
   });
 
   test('Send and receive', async () => {
-    const app = new App(medplum, bot);
-    app.start();
+    const mockServer = new Server('wss://example.com/ws/agent');
+
+    mockServer.on('connection', (socket) => {
+      socket.on('message', (data) => {
+        const command = JSON.parse((data as Buffer).toString('utf8'));
+        if (command.type === 'connect') {
+          socket.send(
+            Buffer.from(
+              JSON.stringify({
+                type: 'connected',
+              })
+            )
+          );
+        }
+
+        if (command.type === 'transmit') {
+          const hl7Message = Hl7Message.parse(command.message);
+          const ackMessage = hl7Message.buildAck();
+          socket.send(
+            Buffer.from(
+              JSON.stringify({
+                type: 'transmit',
+                message: ackMessage.toString(),
+              })
+            )
+          );
+        }
+      });
+    });
+
+    const agent = await medplum.createResource<Agent>({
+      resourceType: 'Agent',
+      channel: [
+        {
+          endpoint: createReference(endpoint),
+          targetReference: createReference(bot),
+        },
+      ],
+    });
+
+    const app = new App(medplum, agent.id as string);
+    await app.start();
 
     const client = new Hl7Client({
       host: 'localhost',
@@ -46,8 +104,12 @@ describe('Agent', () => {
       )
     );
     expect(response).toBeDefined();
+    expect(response.header.getComponent(9, 1)).toBe('ACK');
+    expect(response.segments).toHaveLength(2);
+    expect(response.segments[1].name).toBe('MSA');
 
     client.close();
     app.stop();
+    mockServer.stop();
   });
 });

package/src/main.ts

@@ -1,61 +1,188 @@
-import { ContentType, MedplumClient, normalizeErrorString } from '@medplum/core';
-import { Bot } from '@medplum/fhirtypes';
-import { Hl7MessageEvent, Hl7Server } from '@medplum/hl7';
+import { Hl7Message, MedplumClient, resolveId } from '@medplum/core';
+import { AgentChannel, Bot, Endpoint, Reference } from '@medplum/fhirtypes';
+import { Hl7Connection, Hl7MessageEvent, Hl7Server } from '@medplum/hl7';
 import { EventLogger } from 'node-windows';
-
-const log = new EventLogger({
-  source: 'MedplumService',
-  eventLog: 'SYSTEM',
-});
+import WebSocket from 'ws';
 
 export class App {
   readonly log: EventLogger;
-  readonly server: Hl7Server;
+  readonly channels: AgentHl7Channel[];
 
-  constructor(readonly medplum: MedplumClient, readonly bot: Bot) {
-    this.log = new EventLogger({
-      source: 'MedplumService',
-      eventLog: 'SYSTEM',
-    });
+  constructor(
+    readonly medplum: MedplumClient,
+    readonly agentId: string
+  ) {
+    this.log = {
+      info: console.log,
+      warn: console.warn,
+      error: console.error,
+    } as EventLogger;
 
-    this.server = new Hl7Server();
-    this.server.addEventListener('message', (event) => this.handler(event));
+    this.channels = [];
   }
 
-  start(): void {
+  async start(): Promise<void> {
     this.log.info('Medplum service starting...');
-    this.server.start(56000);
+
+    const agent = await this.medplum.readResource('Agent', this.agentId);
+
+    for (const definition of agent.channel as AgentChannel[]) {
+      const endpoint = await this.medplum.readReference(definition.endpoint as Reference<Endpoint>);
+      const channel = new AgentHl7Channel(this, definition, endpoint);
+      channel.start();
+      this.channels.push(channel);
+    }
+
     this.log.info('Medplum service started successfully');
   }
 
   stop(): void {
     this.log.info('Medplum service stopping...');
-    this.server.stop();
+    this.channels.forEach((channel) => channel.stop());
     this.log.info('Medplum service stopped successfully');
   }
+}
+
+export class AgentHl7Channel {
+  readonly server: Hl7Server;
+  readonly connections: AgentHl7ChannelConnection[] = [];
+
+  constructor(
+    readonly app: App,
+    readonly definition: AgentChannel,
+    readonly endpoint: Endpoint
+  ) {
+    this.server = new Hl7Server((connection) => {
+      this.app.log.info('HL7 connection established');
+      this.connections.push(new AgentHl7ChannelConnection(this, connection));
+    });
+  }
+
+  start(): void {
+    const address = new URL(this.endpoint.address as string);
+    this.app.log.info(`Channel starting on ${address}`);
+    this.server.start(parseInt(address.port, 10));
+    this.app.log.info('Channel started successfully');
+  }
+
+  stop(): void {
+    this.app.log.info('Channel stopping...');
+    for (const connection of this.connections) {
+      connection.close();
+    }
+    this.server.stop();
+    this.app.log.info('Channel stopped successfully');
+  }
+}
+
+export class AgentHl7ChannelConnection {
+  readonly webSocket: WebSocket;
+  readonly webSocketQueue: Hl7Message[] = [];
+  readonly hl7ConnectionQueue: Hl7Message[] = [];
+  live = false;
+
+  constructor(
+    readonly channel: AgentHl7Channel,
+    readonly hl7Connection: Hl7Connection
+  ) {
+    const app = channel.app;
+    const medplum = app.medplum;
+
+    // Add listener immediately to handle incoming messages
+    this.hl7Connection.addEventListener('message', (event) => this.handler(event));
+
+    const webSocketUrl = new URL(medplum.getBaseUrl());
+    webSocketUrl.protocol = webSocketUrl.protocol === 'https:' ? 'wss:' : 'ws:';
+    webSocketUrl.pathname = '/ws/agent';
+    console.log('Connecting to WebSocket:', webSocketUrl.href);
+
+    this.webSocket = new WebSocket(webSocketUrl);
+    this.webSocket.binaryType = 'nodebuffer';
+    this.webSocket.addEventListener('error', console.error);
+    this.webSocket.addEventListener('open', () => {
+      this.webSocket.send(
+        JSON.stringify({
+          type: 'connect',
+          accessToken: medplum.getAccessToken(),
+          botId: resolveId(channel.definition.targetReference as Reference<Bot>),
+        })
+      );
+    });
+
+    this.webSocket.addEventListener('message', (e) => {
+      try {
+        const data = e.data as Buffer;
+        const str = data.toString('utf8');
+        console.log('Received from WebSocket:', str.replaceAll('\r', '\n'));
+        const command = JSON.parse(str);
+        switch (command.type) {
+          case 'connected':
+            this.live = true;
+            this.trySendToWebSocket();
+            break;
+          case 'transmit':
+            this.hl7ConnectionQueue.push(Hl7Message.parse(command.message));
+            this.trySendToHl7Connection();
+            break;
+        }
+      } catch (err) {
+        console.log('WebSocket error', err);
+      }
+    });
+  }
 
   private async handler(event: Hl7MessageEvent): Promise<void> {
     try {
       console.log('Received:');
       console.log(event.message.toString().replaceAll('\r', '\n'));
-
-      await this.medplum.post(
-        this.medplum.fhirUrl('Bot', this.bot.id as string, '$execute'),
-        event.message.toString(),
-        ContentType.HL7_V2
-      );
-
-      const ack = event.message.buildAck();
-      console.log('Response:');
-      console.log(ack.toString().replaceAll('\r', '\n'));
-      event.send(ack);
+      this.webSocketQueue.push(event.message);
+      this.trySendToWebSocket();
     } catch (err) {
       console.log('HL7 error', err);
-      log.error(normalizeErrorString(err));
     }
   }
+
+  private trySendToWebSocket(): void {
+    if (this.live) {
+      while (this.webSocketQueue.length > 0) {
+        const msg = this.webSocketQueue.shift();
+        if (msg) {
+          this.webSocket.send(
+            JSON.stringify({
+              type: 'transmit',
+              message: msg.toString(),
+            })
+          );
+        }
+      }
+    }
+  }
+
+  private trySendToHl7Connection(): void {
+    while (this.hl7ConnectionQueue.length > 0) {
+      const msg = this.hl7ConnectionQueue.shift();
+      if (msg) {
+        this.hl7Connection.send(msg);
+      }
+    }
+  }
+
+  close(): void {
+    this.hl7Connection.close();
+    this.webSocket.close();
+  }
 }
 
 if (typeof require !== 'undefined' && require.main === module) {
-  new App(new MedplumClient(), { resourceType: 'Bot', id: '00000000-00000000-00000000-00000000' }).start();
+  if (process.argv.length < 6) {
+    console.log('Usage: node medplum-agent.js <baseUrl> <clientId> <clientSecret> <agentId>');
+    process.exit(1);
+  }
+
+  const [_node, _script, baseUrl, clientId, clientSecret, agentId] = process.argv;
+  const medplum = new MedplumClient({ baseUrl, clientId });
+  medplum
+    .startClientLogin(clientId, clientSecret)
+    .then(() => new App(medplum, agentId).start())
+    .catch(console.error);
 }