@platformatic/watt-extra 0.1.0

package/plugins/update.js

@@ -0,0 +1,128 @@
+import WebSocket from 'ws'
+import { once } from 'node:events'
+import { setTimeout as sleep } from 'node:timers/promises'
+
+function createWebSocketUrl (httpUrl, path) {
+  const url = new URL(httpUrl)
+  url.protocol = url.protocol.replace('http', 'ws')
+  const basePath = url.pathname.endsWith('/') ? url.pathname : `${url.pathname}/`
+  url.pathname = `${basePath}${path}`
+  return url.toString()
+}
+
+async function updatePlugin (app) {
+  const reconnectInterval = app.env.PLT_UPDATES_RECONNECT_INTERVAL_SEC * 1000
+
+  let socket = null
+
+  async function processMessage (data) {
+    try {
+      const message = JSON.parse(data)
+      const { topic, type, command } = message
+
+      // Handle trigger-flamegraph command from ICC
+      if (command === 'trigger-flamegraph') {
+        app.log.info({ command }, 'Received trigger-flamegraph command from ICC')
+        await app.sendFlamegraphs()
+        return
+      }
+
+      // Handle updates websocket format: { type: '...', topic: '...', data: {...} }
+      if (!topic || !type) {
+        app.log.warn({ message }, 'Received invalid message from updates websocket')
+        return
+      }
+
+      if (type === 'config-updated') {
+        app.log.info({ topic, type }, 'Received config update from updates websocket')
+        await app.updateConfig(message)
+      } else {
+        app.log.info({ topic, type }, 'Received message, not handled type')
+      }
+    } catch (err) {
+      app.log.error(err, 'Error processing message from updates websocket')
+    }
+  }
+
+  async function connectToUpdates () {
+    const applicationId = app.instanceConfig?.applicationId
+    if (!applicationId) {
+      app.log.warn('No application ID found, cannot connect to updates websocket')
+      return null
+    }
+
+    const iccUrl = app.env.PLT_ICC_URL
+    if (!iccUrl) {
+      app.log.warn('No PLT_ICC_URL found in environment, cannot connect to updates websocket')
+      return null
+    }
+
+    const wsUrl = createWebSocketUrl(iccUrl, `api/updates/applications/${applicationId}`)
+    app.log.info(`Connecting to updates websocket at ${wsUrl}`)
+
+    try {
+      const headers = await app.getAuthorizationHeader()
+
+      socket = new WebSocket(wsUrl, { headers })
+      await once(socket, 'open')
+
+      app.log.info('Connected to updates websocket')
+      // Subscribing, if subscription fails we throw, so the caller can retry
+      const subscribeMsg = JSON.stringify({ command: 'subscribe', topic: '/config' })
+      socket.send(subscribeMsg)
+
+      const command = await once(socket, 'message')
+      const message = JSON.parse(command[0])
+      if (message?.command !== 'ack') {
+        app.log.error({ message }, 'Subscription updates failed')
+        throw new Error('Subscription updates failed')
+      }
+      app.log.info('Received subscription acknowledgment from updates websocket')
+
+      // listen for subsequent messages
+      socket.on('message', processMessage)
+
+      socket.on('error', (err) => {
+        app.log.error(err, 'Error in updates websocket connection')
+        reconnectToUpdates()
+      })
+
+      socket.on('close', (code, reason) => {
+        app.log.info({ code, reason: reason.toString() }, 'Updates websocket connection closed')
+        reconnectToUpdates()
+      })
+    } catch (err) {
+      app.log.error(err, 'Failed to connect and subscribe to updates websocket')
+      reconnectToUpdates()
+    }
+  }
+
+  let isReconnecting = false
+  let isClosing = false
+
+  async function reconnectToUpdates () {
+    if (isReconnecting || isClosing) return
+    isReconnecting = true
+
+    await sleep(reconnectInterval)
+
+    isReconnecting = false
+    app.log.info('Reconnecting to updates websocket')
+    await connectToUpdates()
+  }
+
+  app.updateConfig = async (message) => {
+    await app.wattpro.applyIccConfigUpdates(message.data)
+  }
+
+  app.connectToUpdates = connectToUpdates
+  app.closeUpdates = async () => {
+    isClosing = true
+    if (socket) {
+      socket.close()
+      socket = null
+    }
+  }
+}
+
+export default updatePlugin

package/renovate.json

@@ -0,0 +1,6 @@
+{
+  "$schema": "https://docs.renovatebot.com/renovate-schema.json",
+  "extends": [
+    "config:recommended"
+  ]
+}