connect to nats with client token

frontend/interactEM/src/client/utils.ts

@@ -0,0 +1,22 @@
+import { Auth, Client } from "@hey-api/client-axios"
+
+export const getTokenFromClient = async (
+  client: Client,
+): Promise<string | undefined> => {
+  const auth = client.getConfig().auth
+
+  if (!auth) {
+    return undefined
+  }
+
+  if (typeof auth === "string") {
+    return auth
+  }
+
+  const defaultAuth: Auth = {
+    type: "http",
+    scheme: "bearer",
+  }
+
+  return await auth(defaultAuth)
+}

frontend/interactEM/src/nats/NatsContext.tsx

@@ -9,6 +9,8 @@ import {
 } from "@nats-io/jetstream"
 import { Kvm } from "@nats-io/kv"
 import config from "../config"
+import { client } from "../client"
+import { getTokenFromClient } from "../client/utils"
 
 interface NatsContextType {
   natsConnection: NatsConnection | null
@@ -20,6 +22,15 @@ interface NatsContextType {
 
 const NatsContext = createContext<NatsContextType | undefined>(undefined)
 
+const getConnectionId = () => {
+  let id = sessionStorage.getItem("interactEM-connection-id")
+  if (!id) {
+    id = `interactEM-${Date.now()}-${Math.random().toString(36).substring(2, 15)}`
+    sessionStorage.setItem("interactEM-connection-id", id)
+  }
+  return id
+}
+
 export const useNats = (): NatsContextType => {
   const context = useContext(NatsContext)
   if (!context) {
@@ -45,7 +56,17 @@ export const NatsProvider: React.FC<{ children: React.ReactNode }> = ({
   useEffect(() => {
     const setupNatsConnection = async () => {
       try {
-        const nc = await wsconnect({ servers: [config.NATS_SERVER_URL] })
+        const token = await getTokenFromClient(client)
+        if (!token) {
+          console.error("Failed to get token from client")
+          return
+        }
+        console.log("Connecting to NATS with token:", token)
+        const nc = await wsconnect({
+          servers: [config.NATS_SERVER_URL],
+          name: getConnectionId(),
+          token: token,
+        })
         setNatsConnection(nc)
         setIsConnected(true)