Dynamic template function args and TTL option for request chaining (#266)

src-web/hooks/useHttpAuthenticationConfig.ts

@@ -18,7 +18,7 @@ import { activeWorkspaceIdAtom } from './useActiveWorkspace';
 export function useHttpAuthenticationConfig(
   authName: string | null,
   values: Record<string, JsonPrimitive>,
-  request: HttpRequest | GrpcRequest | WebsocketRequest | Folder | Workspace,
+  model: HttpRequest | GrpcRequest | WebsocketRequest | Folder | Workspace,
 ) {
   const workspaceId = useAtomValue(activeWorkspaceIdAtom);
   const environmentId = useAtomValue(activeEnvironmentIdAtom);
@@ -37,7 +37,7 @@ export function useHttpAuthenticationConfig(
   return useQuery({
     queryKey: [
       'http_authentication_config',
-      request,
+      model,
       authName,
       values,
       responseKey,
@@ -53,7 +53,7 @@ export function useHttpAuthenticationConfig(
         {
           authName,
           values,
-          request,
+          model,
           environmentId,
         },
       );

src-web/hooks/useTemplateFunctionConfig.ts

@@ -0,0 +1,60 @@
+import { useQuery } from '@tanstack/react-query';
+import type {
+  Folder,
+  GrpcRequest,
+  HttpRequest,
+  WebsocketRequest,
+  Workspace,
+} from '@yaakapp-internal/models';
+import { httpResponsesAtom } from '@yaakapp-internal/models';
+import type { GetTemplateFunctionConfigResponse, JsonPrimitive } from '@yaakapp-internal/plugins';
+import { useAtomValue } from 'jotai';
+import { md5 } from 'js-md5';
+import { invokeCmd } from '../lib/tauri';
+import { activeEnvironmentIdAtom } from './useActiveEnvironment';
+import { activeWorkspaceIdAtom } from './useActiveWorkspace';
+
+export function useTemplateFunctionConfig(
+  functionName: string | null,
+  values: Record<string, JsonPrimitive>,
+  model: HttpRequest | GrpcRequest | WebsocketRequest | Folder | Workspace,
+) {
+  const workspaceId = useAtomValue(activeWorkspaceIdAtom);
+  const environmentId = useAtomValue(activeEnvironmentIdAtom);
+  const responses = useAtomValue(httpResponsesAtom);
+
+  // Some auth handlers like OAuth 2.0 show the current token after a successful request. To
+  // handle that, we'll force the auth to re-fetch after each new response closes
+  const responseKey = md5(
+    responses
+      .filter((r) => r.state === 'closed')
+      .map((r) => r.id)
+      .join(':'),
+  );
+
+  return useQuery({
+    queryKey: [
+      'template_function_config',
+      model,
+      functionName,
+      values,
+      responseKey,
+      workspaceId,
+      environmentId,
+    ],
+    placeholderData: (prev) => prev, // Keep previous data on refetch
+    queryFn: async () => {
+      if (functionName == null) return null;
+      const config = await invokeCmd<GetTemplateFunctionConfigResponse>(
+        'cmd_template_function_config',
+        {
+          functionName: functionName,
+          values,
+          model,
+          environmentId,
+        },
+      );
+      return config.function;
+    },
+  });
+}

src-web/hooks/useTemplateFunctions.ts

@@ -1,6 +1,9 @@
 import { useQuery } from '@tanstack/react-query';
-import type { GetTemplateFunctionsResponse, TemplateFunction } from '@yaakapp-internal/plugins';
-import { atom, useAtomValue , useSetAtom } from 'jotai';
+import type {
+  GetTemplateFunctionSummaryResponse,
+  TemplateFunction,
+} from '@yaakapp-internal/plugins';
+import { atom, useAtomValue, useSetAtom } from 'jotai';
 import { useMemo, useState } from 'react';
 import type { TwigCompletionOption } from '../components/core/Editor/twig/completion';
 import { invokeCmd } from '../lib/tauri';
@@ -55,7 +58,9 @@ export function useSubscribeTemplateFunctions() {
     refetchInterval: numFns > 0 ? Infinity : 1000,
     refetchOnMount: true,
     queryFn: async () => {
-      const result = await invokeCmd<GetTemplateFunctionsResponse[]>('cmd_template_functions');
+      const result = await invokeCmd<GetTemplateFunctionSummaryResponse[]>(
+        'cmd_template_function_summaries',
+      );
       setNumFns(result.length);
       const functions = result.flatMap((r) => r.functions) ?? [];
       setAtom(functions);