Hacky server streaming done

src-web/hooks/useAlert.ts

@@ -1,3 +1,4 @@
+import { useCallback } from 'react';
 import type { DialogProps } from '../components/core/Dialog';
 import { useDialog } from '../components/DialogContext';
 import type { AlertProps } from './Alert';
@@ -5,20 +6,16 @@ import { Alert } from './Alert';
 
 export function useAlert() {
   const dialog = useDialog();
-  return ({
-    id,
-    title,
-    body,
-  }: {
-    id: string;
-    title: DialogProps['title'];
-    body: AlertProps['body'];
-  }) =>
-    dialog.show({
-      id,
-      title,
-      hideX: true,
-      size: 'sm',
-      render: ({ hide }) => Alert({ onHide: hide, body }),
-    });
+  return useCallback(
+    ({ id, title, body }: { id: string; title: DialogProps['title']; body: AlertProps['body'] }) =>
+      dialog.show({
+        id,
+        title,
+        hideX: true,
+        size: 'sm',
+        render: ({ hide }) => Alert({ onHide: hide, body }),
+      }),
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+    [],
+  );
 }

src-web/hooks/useGrpc.ts

@@ -1,13 +1,30 @@
 import { useMutation, useQuery } from '@tanstack/react-query';
 import { invoke } from '@tauri-apps/api';
+import { useState } from 'react';
+import { useListenToTauriEvent } from './useListenToTauriEvent';
+
+interface ReflectResponseService {
+  name: string;
+  methods: { name: string; schema: string; serverStreaming: boolean; clientStreaming: boolean }[];
+}
+
+interface Message {
+  message: string;
+  time: Date;
+}
 
 export function useGrpc(url: string | null) {
-  const callUnary = useMutation<
-    string,
-    unknown,
-    { service: string; method: string; message: string }
-  >({
-    mutationKey: ['grpc_call_reflect', url],
+  const [messages, setMessages] = useState<Message[]>([]);
+  useListenToTauriEvent<string>(
+    'grpc_message',
+    (event) => {
+      console.log('GOT MESSAGE', event);
+      setMessages((prev) => [...prev, { message: event.payload, time: new Date() }]);
+    },
+    [],
+  );
+  const unary = useMutation<string, string, { service: string; method: string; message: string }>({
+    mutationKey: ['grpc_unary', url],
     mutationFn: async ({ service, method, message }) => {
       if (url === null) throw new Error('No URL provided');
       return (await invoke('grpc_call_unary', {
@@ -19,17 +36,36 @@ export function useGrpc(url: string | null) {
     },
   });
 
-  const reflect = useQuery<string | null>({
+  const serverStreaming = useMutation<
+    string,
+    string,
+    { service: string; method: string; message: string }
+  >({
+    mutationKey: ['grpc_server_streaming', url],
+    mutationFn: async ({ service, method, message }) => {
+      if (url === null) throw new Error('No URL provided');
+      return (await invoke('grpc_server_streaming', {
+        endpoint: url,
+        service,
+        method,
+        message,
+      })) as string;
+    },
+  });
+
+  const reflect = useQuery<ReflectResponseService[]>({
     queryKey: ['grpc_reflect', url ?? ''],
     queryFn: async () => {
-      if (url === null) return null;
+      if (url === null) return [];
       console.log('GETTING SCHEMA', url);
-      return (await invoke('grpc_reflect', { endpoint: url })) as string;
+      return (await invoke('grpc_reflect', { endpoint: url })) as ReflectResponseService[];
     },
   });
 
   return {
-    callUnary,
+    unary,
+    serverStreaming,
     schema: reflect.data,
+    messages,
   };
 }

src-web/hooks/useHotKey.ts

@@ -5,44 +5,47 @@ import { debounce } from '../lib/debounce';
 import { useOsInfo } from './useOsInfo';
 
 export type HotkeyAction =
-  | 'request.send'
+  | 'environmentEditor.toggle'
+  | 'grpc.send'
+  | 'hotkeys.showHelp'
   | 'request.create'
   | 'request.duplicate'
-  | 'sidebar.toggle'
-  | 'sidebar.focus'
-  | 'urlBar.focus'
-  | 'environmentEditor.toggle'
-  | 'hotkeys.showHelp'
-  | 'requestSwitcher.prev'
+  | 'request.send'
   | 'requestSwitcher.next'
-  | 'settings.show';
+  | 'requestSwitcher.prev'
+  | 'settings.show'
+  | 'sidebar.focus'
+  | 'sidebar.toggle'
+  | 'urlBar.focus';
 
 const hotkeys: Record<HotkeyAction, string[]> = {
-  'request.send': ['CmdCtrl+Enter', 'CmdCtrl+r'],
+  'environmentEditor.toggle': ['CmdCtrl+Shift+e'],
+  'grpc.send': ['CmdCtrl+Enter', 'CmdCtrl+r'],
+  'hotkeys.showHelp': ['CmdCtrl+Shift+/'],
   'request.create': ['CmdCtrl+n'],
   'request.duplicate': ['CmdCtrl+d'],
-  'sidebar.toggle': ['CmdCtrl+b'],
-  'sidebar.focus': ['CmdCtrl+1'],
-  'urlBar.focus': ['CmdCtrl+l'],
-  'environmentEditor.toggle': ['CmdCtrl+Shift+e'],
-  'hotkeys.showHelp': ['CmdCtrl+Shift+/'],
-  'settings.show': ['CmdCtrl+,'],
-  'requestSwitcher.prev': ['Control+Tab'],
+  'request.send': ['CmdCtrl+Enter', 'CmdCtrl+r'],
   'requestSwitcher.next': ['Control+Shift+Tab'],
+  'requestSwitcher.prev': ['Control+Tab'],
+  'settings.show': ['CmdCtrl+,'],
+  'sidebar.focus': ['CmdCtrl+1'],
+  'sidebar.toggle': ['CmdCtrl+b'],
+  'urlBar.focus': ['CmdCtrl+l'],
 };
 
 const hotkeyLabels: Record<HotkeyAction, string> = {
-  'request.send': 'Send Request',
+  'environmentEditor.toggle': 'Edit Environments',
+  'grpc.send': 'Send Message',
+  'hotkeys.showHelp': 'Show Keyboard Shortcuts',
   'request.create': 'New Request',
   'request.duplicate': 'Duplicate Request',
-  'sidebar.toggle': 'Toggle Sidebar',
-  'sidebar.focus': 'Focus Sidebar',
-  'urlBar.focus': 'Focus URL',
-  'environmentEditor.toggle': 'Edit Environments',
-  'hotkeys.showHelp': 'Show Keyboard Shortcuts',
-  'requestSwitcher.prev': 'Go To Next Request',
+  'request.send': 'Send Request',
   'requestSwitcher.next': 'Go To Previous Request',
+  'requestSwitcher.prev': 'Go To Next Request',
   'settings.show': 'Open Settings',
+  'sidebar.focus': 'Focus Sidebar',
+  'sidebar.toggle': 'Toggle Sidebar',
+  'urlBar.focus': 'Focus URL',
 };
 
 export const hotkeyActions: HotkeyAction[] = Object.keys(hotkeys) as (keyof typeof hotkeys)[];