Mar 21, 2025 · Mar 3, 2025 · Mar 3, 2025 · Mar 4, 2025 · Mar 17, 2025 · Mar 17, 2025
diff --git a/src/inbox.ts b/src/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"

 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(
    private readonly restClient: Api,
    private readonly storage: Storage,
  ) {
    // const url = this.restClient.getAxiosInstance().defaults.baseURL
    const token = this.restClient.getAxiosInstance().defaults.headers.common["Coder-Session-Token"] as
      | string
      | undefined
    // const inboxUrl = new URL(`${url}/api/v2/notifications/watch`);
    const inboxUrl = new URL(`ws://localhost:8080`)

    this.storage.writeToCoderOutputChannel("Listening to Coder Inbox")

    // We're gonna connect over WebSocket so replace the scheme.
    if (inboxUrl.protocol === "https") {
      inboxUrl.protocol = "wss"
    } else if (inboxUrl.protocol === "http") {
      inboxUrl.protocol = "ws"
    }

    this.socket = new WebSocket(inboxUrl, {
      headers: {
        "Coder-Session-Token": token,
      },
    })

    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)
  }
 }
diff --git a/src/remote.ts b/src/remote.ts
 import { Commands } from "./commands"
 import { featureSetForVersion, FeatureSet } from "./featureSet"
 import { getHeaderCommand } from "./headers"
 import { Inbox } from "./inbox"
 import { SSHConfig, SSHValues, mergeSSHConfigValues } from "./sshConfig"
 import { computeSSHProperties, sshSupportsSetEnv } from "./sshSupport"
 import { Storage } from "./storage"
    disposables.push(monitor)
    disposables.push(monitor.onChange.event((w) => (this.commands.workspace = w)))

    // Watch coder inbox for messages
    const inbox = new Inbox(workspaceRestClient, this.storage)
    disposables.push(inbox)

    // Wait for the agent to connect.
    if (agent.status === "connecting") {
      this.storage.writeToCoderOutputChannel(`Waiting for ${workspaceName}/${agent.name}...`)
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,89 @@
		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"
DanielleMaywood marked this conversation as resolved. OutdatedShow resolvedHide resolved

		type InboxMessage = {
		unread_count: number
		notification: {
		id: string
		user_id: string
		template_id: string
		targets: string[]
		title: string
		content: string
		actions: {
		[key: string]: string
		}
DanielleMaywood marked this conversation as resolved. OutdatedShow resolvedHide resolved
		read_at: string
		created_at: string
		}
		}

		const TemplateWorkspaceOutOfMemory = "a9d027b4-ac49-4fb1-9f6d-45af15f64e7a"
		const TemplateWorkspaceOutOfDisk = "f047f6a3-5713-40f7-85aa-0394cce9fa3a"
DanielleMaywood marked this conversation as resolved. OutdatedShow resolvedHide resolved

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

		constructor(
		private readonly restClient: Api,
		private readonly storage: Storage,
DanielleMaywood marked this conversation as resolved. OutdatedShow resolvedHide resolved
		) {
		// const url = this.restClient.getAxiosInstance().defaults.baseURL
		const token = this.restClient.getAxiosInstance().defaults.headers.common["Coder-Session-Token"] as
		\| string
		\| undefined
		// const inboxUrl = new URL(`${url}/api/v2/notifications/watch`);
		const inboxUrl = new URL(`ws://localhost:8080`)

		this.storage.writeToCoderOutputChannel("Listening to Coder Inbox")

		// We're gonna connect over WebSocket so replace the scheme.
		if (inboxUrl.protocol === "https") {
		inboxUrl.protocol = "wss"
		} else if (inboxUrl.protocol === "http") {
		inboxUrl.protocol = "ws"
		}

		this.socket = new WebSocket(inboxUrl, {
		headers: {
		"Coder-Session-Token": token,
		},
		})

		this.socket.on("error", (error) => {
		this.notifyError(error)
		})
DanielleMaywood marked this conversation as resolved. OutdatedShow resolvedHide resolved

		this.socket.on("message", (data) => {
Parkreiner marked this conversation as resolved. OutdatedShow resolvedHide resolved
		try {
		const inboxMessage = JSON.parse(data.toString()) as InboxMessage
Parkreiner marked this conversation as resolved. OutdatedShow resolvedHide resolved

		if (
		inboxMessage.notification.template_id === TemplateWorkspaceOutOfDisk \|\|
		inboxMessage.notification.template_id === TemplateWorkspaceOutOfMemory
		) {
		vscode.window.showWarningMessage(inboxMessage.notification.title)
Parkreiner marked this conversation as resolved. OutdatedShow resolvedHide resolved
		}
		} catch (error) {
		this.notifyError(error)
		}
Parkreiner marked this conversation as resolved. Show resolvedHide resolved
		})
		}

		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)
		}
		}
Original file line number	Diff line number	Diff line change
Expand Up		@@ -15,6 +15,7 @@ import * as cli from "./cliManager"
		import { Commands } from "./commands"
		import { featureSetForVersion, FeatureSet } from "./featureSet"
		import { getHeaderCommand } from "./headers"
		import { Inbox } from "./inbox"
		import { SSHConfig, SSHValues, mergeSSHConfigValues } from "./sshConfig"
		import { computeSSHProperties, sshSupportsSetEnv } from "./sshSupport"
		import { Storage } from "./storage"
Expand DownExpand Up		@@ -403,6 +404,10 @@ export class Remote {
		disposables.push(monitor)
		disposables.push(monitor.onChange.event((w) => (this.commands.workspace = w)))

		// Watch coder inbox for messages
		const inbox = new Inbox(workspaceRestClient, this.storage)
		disposables.push(inbox)

		// Wait for the agent to connect.
		if (agent.status === "connecting") {
		this.storage.writeToCoderOutputChannel(`Waiting for ${workspaceName}/${agent.name}...`)
Expand Down