inbox.ts

import { Api } from "coder/site/src/api/api"
import * as vscode from "vscode"
import { WebSocket } from "ws"
import { errToStr } from "./api-helper"
import { Storage } from "./storage"
import { ProxyAgent } from "proxy-agent"

type InboxMessage = {
  unread_count: number
  notification: {
    id: string
    user_id: string
    template_id: string
    targets: string[]
    title: string
    content: string
    actions: {
      [key: string]: string
    }
    read_at: string
    created_at: string
  }
}

const TemplateWorkspaceOutOfMemory = "a9d027b4-ac49-4fb1-9f6d-45af15f64e7a"
const TemplateWorkspaceOutOfDisk = "f047f6a3-5713-40f7-85aa-0394cce9fa3a"

export class Inbox implements vscode.Disposable {
  private disposed = false
  private socket: WebSocket

  constructor(
    httpAgent: ProxyAgent,
    restClient: Api,
    private readonly storage: Storage,
  ) {
    const baseUrlRaw = restClient.getAxiosInstance().defaults.baseURL
    if (!baseUrlRaw) {
      throw new Error("No base URL set on REST client")
    }

    const baseUrl = new URL(baseUrlRaw)
    const socketProto = baseUrl.protocol === "https:" ? "wss:" : "ws:"
    const socketUrlRaw = `${socketProto}//${baseUrl.host}/api/v2/notifications/watch`

    this.socket = new WebSocket(new URL(socketUrlRaw), {
      followRedirects: true,
      agent: httpAgent,
      headers: {
        "Coder-Session-Token": restClient.getAxiosInstance().defaults.headers.common["Coder-Session-Token"] as
          | string
          | undefined,
      },
    })

    this.socket.on("open", () => {
      this.storage.writeToCoderOutputChannel("Listening to Coder Inbox")
    })

    this.socket.on("error", (error) => {
      this.notifyError(error)
    })

    this.socket.on("message", (data) => {
      try {
        const inboxMessage = JSON.parse(data.toString()) as InboxMessage

        if (
          inboxMessage.notification.template_id === TemplateWorkspaceOutOfDisk ||
          inboxMessage.notification.template_id === TemplateWorkspaceOutOfMemory
        ) {
          vscode.window.showWarningMessage(inboxMessage.notification.title)
        }
      } catch (error) {
        this.notifyError(error)
      }
    })
  }

  dispose() {
    if (!this.disposed) {
      this.storage.writeToCoderOutputChannel("No longer listening to Coder Inbox")
      this.socket.close()
      this.disposed = true
    }
  }

  private notifyError(error: unknown) {
    const message = errToStr(error, "Got empty error while monitoring Coder Inbox")
    this.storage.writeToCoderOutputChannel(message)
  }
}