Websocket Support (#159)

src-tauri/yaak-ws/index.ts

@@ -0,0 +1,77 @@
+import { invoke } from '@tauri-apps/api/core';
+import { WebsocketConnection, WebsocketEvent, WebsocketRequest } from '@yaakapp-internal/models';
+
+export function upsertWebsocketRequest(
+  request: WebsocketRequest | Partial<Omit<WebsocketRequest, 'id'>>,
+) {
+  return invoke('plugin:yaak-ws|upsert_request', {
+    request,
+  }) as Promise<WebsocketRequest>;
+}
+
+export function deleteWebsocketRequest(requestId: string) {
+  return invoke('plugin:yaak-ws|delete_request', {
+    requestId,
+  });
+}
+
+export function deleteWebsocketConnection(connectionId: string) {
+  return invoke('plugin:yaak-ws|delete_connection', {
+    connectionId,
+  });
+}
+
+export function deleteWebsocketConnections(requestId: string) {
+  return invoke('plugin:yaak-ws|delete_connections', {
+    requestId,
+  });
+}
+
+export function listWebsocketRequests({ workspaceId }: { workspaceId: string }) {
+  return invoke('plugin:yaak-ws|list_requests', { workspaceId }) as Promise<WebsocketRequest[]>;
+}
+
+export function listWebsocketEvents({ connectionId }: { connectionId: string }) {
+  return invoke('plugin:yaak-ws|list_events', { connectionId }) as Promise<WebsocketEvent[]>;
+}
+
+export function listWebsocketConnections({ workspaceId }: { workspaceId: string }) {
+  return invoke('plugin:yaak-ws|list_connections', { workspaceId }) as Promise<
+    WebsocketConnection[]
+  >;
+}
+
+export function connectWebsocket({
+  requestId,
+  environmentId,
+  cookieJarId,
+}: {
+  requestId: string;
+  environmentId: string | null;
+  cookieJarId: string | null;
+}) {
+  return invoke('plugin:yaak-ws|connect', {
+    requestId,
+    environmentId,
+    cookieJarId,
+  }) as Promise<WebsocketConnection>;
+}
+
+export function closeWebsocket({ connectionId }: { connectionId: string }) {
+  return invoke('plugin:yaak-ws|close', {
+    connectionId,
+  });
+}
+
+export function sendWebsocket({
+  connectionId,
+  environmentId,
+}: {
+  connectionId: string;
+  environmentId: string | null;
+}) {
+  return invoke('plugin:yaak-ws|send', {
+    connectionId,
+    environmentId,
+  });
+}