Add API KEY to mobula (#3439)

smartcontractkit · Sep 19, 2024 · 40e20f2 · 40e20f2
1 parent e93bc76
commit 40e20f2
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 2 deletions.
diff --git a/.changeset/seven-parrots-move.md b/.changeset/seven-parrots-move.md
@@ -0,0 +1,5 @@
+---
+'@chainlink/mobula-state-adapter': minor
+---
+
+Add API Key
diff --git a/packages/sources/mobula-state/src/config/index.ts b/packages/sources/mobula-state/src/config/index.ts
@@ -6,4 +6,10 @@ export const config = new AdapterConfig({
     type: 'string',
     default: 'wss://feed.zobula.xyz',
   },
+  API_KEY: {
+    description: 'An API key for Data Provider',
+    type: 'string',
+    required: true,
+    sensitive: true,
+  },
 })
diff --git a/packages/sources/mobula-state/src/transport/price.ts b/packages/sources/mobula-state/src/transport/price.ts
@@ -19,8 +19,10 @@ export type WsTransportTypes = BaseEndpointTypes & {
 export const wsTransport = new WebSocketTransport<WsTransportTypes>({
   url: (context) => context.adapterSettings.WS_API_ENDPOINT,
   handlers: {
-    open: (connection) => {
-      connection.send(JSON.stringify({ type: 'feed' }))
+    open: (connection, context) => {
+      connection.send(
+        JSON.stringify({ type: 'feed', authorization: context.adapterSettings.API_KEY }),
+      )
     },
     message(message) {
       if (!message.price) {

diff --git a/packages/sources/mobula-state/test/integration/adapter.test.ts b/packages/sources/mobula-state/test/integration/adapter.test.ts
@@ -24,6 +24,7 @@ describe('websocket', () => {
   beforeAll(async () => {
     oldEnv = JSON.parse(JSON.stringify(process.env))
     process.env['WS_API_ENDPOINT'] = wsEndpoint
+    process.env['API_KEY'] = 'mock-api-key'
     mockWebSocketProvider(WebSocketClassProvider)
     mockWsServer = mockWebsocketServer(wsEndpoint)