nitrostack 1.0.65 → 1.0.66

package/src/studio/app/page.tsx

@@ -0,0 +1,351 @@
+'use client';
+
+import { useEffect, useState } from 'react';
+import { useStudioStore } from '@/lib/store';
+import { api } from '@/lib/api';
+import { ToolCard } from '@/components/ToolCard';
+import { WidgetRenderer } from '@/components/WidgetRenderer';
+import type { Tool } from '@/lib/types';
+import { Wrench, RefreshCw, X, Play, AlertCircle } from 'lucide-react';
+
+export default function ToolsPage() {
+  const { tools, setTools, loading, setLoading, connection, setConnection, jwtToken, apiKey, oauthState } = useStudioStore();
+  const [searchQuery, setSearchQuery] = useState('');
+  const [selectedTool, setSelectedTool] = useState<Tool | null>(null);
+  const [toolArgs, setToolArgs] = useState<Record<string, any>>({});
+  const [toolResult, setToolResult] = useState<any>(null);
+  const [executingTool, setExecutingTool] = useState(false);
+  
+  // Get effective token - check both jwtToken and OAuth token
+  const effectiveToken = jwtToken || oauthState?.currentToken;
+
+  // Initialize MCP, load tools and check connection on mount
+  useEffect(() => {
+    const init = async () => {
+      await api.initialize();
+      await loadTools();
+      await checkConnection();
+    };
+    init();
+  }, []);
+
+  const checkConnection = async () => {
+    try {
+      const health = await api.checkConnection();
+      setConnection({
+        connected: health.connected,
+        status: health.connected ? 'connected' : 'disconnected',
+      });
+    } catch (error) {
+      setConnection({ connected: false, status: 'disconnected' });
+    }
+  };
+
+  const loadTools = async () => {
+    setLoading('tools', true);
+    try {
+      const data = await api.getTools();
+      setTools(data.tools || []);
+    } catch (error) {
+      console.error('Failed to load tools:', error);
+    } finally {
+      setLoading('tools', false);
+    }
+  };
+
+  const handleExecuteTool = (tool: Tool) => {
+    setSelectedTool(tool);
+    setToolArgs({});
+    setToolResult(null);
+  };
+
+  const handleSubmitTool = async (e: React.FormEvent) => {
+    e.preventDefault();
+    if (!selectedTool) return;
+
+    setExecutingTool(true);
+    setToolResult(null);
+
+    try {
+      // Pass JWT token and API key if available (check both jwtToken and OAuth token)
+      const result = await api.callTool(selectedTool.name, toolArgs, effectiveToken || undefined, apiKey || undefined);
+      setToolResult(result);
+
+      // Extract JWT token from ANY tool response (not just 'login')
+      // Check if result contains a token field at any level
+      if (result.content) {
+        try {
+          const content = result.content[0]?.text;
+          if (content) {
+            const parsed = JSON.parse(content);
+            // Check for token in multiple possible locations
+            const token = parsed.token || parsed.access_token || parsed.jwt || parsed.data?.token;
+            if (token) {
+              console.log('🔐 Token received from tool, saving to global state');
+              useStudioStore.getState().setJwtToken(token);
+            }
+          }
+        } catch (e) {
+          // Ignore parsing errors
+        }
+      }
+    } catch (error) {
+      console.error('Tool execution failed:', error);
+      setToolResult({ error: 'Tool execution failed' });
+    } finally {
+      setExecutingTool(false);
+    }
+  };
+
+  const filteredTools = tools.filter((tool) =>
+    tool.name.toLowerCase().includes(searchQuery.toLowerCase())
+  );
+
+  return (
+    <>
+      <div className="fixed inset-0 flex flex-col bg-background" style={{ left: 'var(--sidebar-width, 15rem)' }}>
+        {/* Sticky Header */}
+        <div className="sticky top-0 z-10 border-b border-border/50 px-6 py-3 flex items-center justify-between bg-card/80 backdrop-blur-md shadow-sm">
+          <div className="flex items-center gap-3">
+            <div className="w-8 h-8 rounded-lg bg-gradient-to-br from-primary to-amber-500 flex items-center justify-center shadow-md">
+              <Wrench className="w-5 h-5 text-white" strokeWidth={2.5} />
+            </div>
+            <div>
+              <h1 className="text-lg font-bold text-foreground">Tools</h1>
+            </div>
+          </div>
+          <button onClick={loadTools} className="btn btn-primary text-sm px-4 py-2 gap-2">
+            <RefreshCw className="w-4 h-4" />
+            Refresh
+          </button>
+        </div>
+
+        {/* Content - ONLY this scrolls */}
+        <div className="flex-1 overflow-y-auto overflow-x-hidden">
+          <div className="max-w-7xl mx-auto px-6 py-6">
+            {/* Search */}
+            <div className="mb-6">
+              <input
+                type="text"
+                placeholder="Search tools..."
+                value={searchQuery}
+                onChange={(e) => setSearchQuery(e.target.value)}
+                className="input w-full"
+              />
+            </div>
+
+            {/* Tools Grid */}
+            {loading.tools ? (
+              <div className="grid grid-cols-1 md:grid-cols-2 lg:grid-cols-3 gap-6">
+                {[1, 2, 3].map((i) => (
+                  <div key={i} className="card skeleton h-64"></div>
+                ))}
+              </div>
+            ) : filteredTools.length === 0 ? (
+              <div className="empty-state">
+                <AlertCircle className="empty-state-icon" />
+                <p className="empty-state-title">
+                  {searchQuery ? 'No tools found matching your search' : 'No tools available'}
+                </p>
+                <p className="empty-state-description">
+                  {searchQuery ? 'Try a different search term' : 'No MCP tools have been registered'}
+                </p>
+              </div>
+            ) : (
+              <div className="grid grid-cols-1 md:grid-cols-2 lg:grid-cols-3 gap-6">
+                {filteredTools.map((tool) => (
+                  <ToolCard key={tool.name} tool={tool} onExecute={handleExecuteTool} />
+                ))}
+              </div>
+            )}
+          </div>
+        </div>
+      </div>
+
+      {/* Tool Executor Modal */}
+      {selectedTool && (
+          <div
+            className="fixed inset-0 z-50 flex items-center justify-center animate-fade-in"
+            style={{ backgroundColor: 'rgba(0, 0, 0, 0.85)' }}
+            onClick={() => setSelectedTool(null)}
+          >
+            <div
+              className="w-[700px] max-h-[90vh] overflow-auto rounded-2xl border border-border p-6 bg-card shadow-2xl animate-scale-in"
+              onClick={(e) => e.stopPropagation()}
+            >
+              <div className="flex items-center justify-between mb-4">
+                <div className="flex items-center gap-3">
+                  <div className="w-10 h-10 rounded-lg bg-primary/10 flex items-center justify-center">
+                    <Wrench className="w-5 h-5 text-primary" />
+                  </div>
+                  <h2 className="text-xl font-bold text-foreground">{selectedTool.name}</h2>
+                </div>
+                <button
+                  onClick={() => setSelectedTool(null)}
+                  className="btn btn-ghost w-10 h-10 p-0"
+                >
+                  <X className="w-5 h-5" />
+                </button>
+              </div>
+
+              <p className="text-sm text-muted-foreground mb-6">
+                {selectedTool.description || 'No description'}
+              </p>
+
+              <form onSubmit={handleSubmitTool}>
+                {/* Generate form inputs from schema */}
+                {selectedTool.inputSchema?.properties && typeof selectedTool.inputSchema.properties === 'object' && Object.keys(selectedTool.inputSchema.properties).length > 0 ? (
+                  <>
+                    {Object.entries(selectedTool.inputSchema.properties).map(([key, prop]: [string, any]) => {
+                      const isRequired = selectedTool.inputSchema?.required?.includes(key);
+                      
+                      // Handle different input types
+                      if (prop.enum) {
+                        // Enum/Select field
+                        return (
+                          <div key={key} className="mb-4">
+                            <label className="block text-sm font-medium text-foreground mb-2">
+                              {prop.title || key}
+                              {isRequired && <span className="text-destructive ml-1">*</span>}
+                            </label>
+                            <select
+                              className="input"
+                              value={toolArgs[key] || prop.default || ''}
+                              onChange={(e) => setToolArgs({ ...toolArgs, [key]: e.target.value })}
+                              required={isRequired}
+                            >
+                              <option value="">Select...</option>
+                              {prop.enum.map((val: any) => (
+                                <option key={val} value={val}>
+                                  {val}
+                                </option>
+                              ))}
+                            </select>
+                            {prop.description && (
+                              <p className="text-xs text-muted-foreground mt-1">{prop.description}</p>
+                            )}
+                          </div>
+                        );
+                      } else if (prop.type === 'boolean') {
+                        // Checkbox field
+                        return (
+                          <div key={key} className="mb-4">
+                            <label className="flex items-center gap-2 cursor-pointer">
+                              <input
+                                type="checkbox"
+                                className="w-4 h-4"
+                                checked={toolArgs[key] || false}
+                                onChange={(e) => setToolArgs({ ...toolArgs, [key]: e.target.checked })}
+                              />
+                              <span className="text-sm font-medium text-foreground">
+                                {prop.title || key}
+                                {isRequired && <span className="text-destructive ml-1">*</span>}
+                              </span>
+                            </label>
+                            {prop.description && (
+                              <p className="text-xs text-muted-foreground mt-1 ml-6">{prop.description}</p>
+                            )}
+                          </div>
+                        );
+                      } else {
+                        // Text/Number field
+                        return (
+                          <div key={key} className="mb-4">
+                            <label className="block text-sm font-medium text-foreground mb-2">
+                              {prop.title || key}
+                              {isRequired && <span className="text-destructive ml-1">*</span>}
+                            </label>
+                            <input
+                              type={prop.type === 'number' || prop.type === 'integer' ? 'number' : 'text'}
+                              className="input"
+                              value={toolArgs[key] || prop.default || ''}
+                              onChange={(e) => {
+                                const value = prop.type === 'number' || prop.type === 'integer' 
+                                  ? (e.target.value ? Number(e.target.value) : '') 
+                                  : e.target.value;
+                                setToolArgs({ ...toolArgs, [key]: value });
+                              }}
+                              required={isRequired}
+                              placeholder={prop.description}
+                              min={prop.minimum}
+                              max={prop.maximum}
+                            />
+                            {prop.description && (
+                              <p className="text-xs text-muted-foreground mt-1">{prop.description}</p>
+                            )}
+                          </div>
+                        );
+                      }
+                    })}
+                  </>
+                ) : (
+                  <div className="mb-4 p-4 bg-muted/30 rounded-lg border border-border">
+                    <p className="text-sm text-muted-foreground">No input required</p>
+                  </div>
+                )}
+
+                <button
+                  type="submit"
+                  className="btn btn-primary w-full gap-2"
+                  disabled={executingTool}
+                >
+                  <Play className="w-4 h-4" />
+                  {executingTool ? 'Executing...' : 'Execute Tool'}
+                </button>
+              </form>
+
+              {/* Result */}
+              {toolResult && (
+                <div className="mt-6 space-y-4">
+                  <div>
+                    <h3 className="font-semibold text-foreground mb-3">Result:</h3>
+                    <pre className="bg-muted/30 p-4 rounded-lg text-sm overflow-auto max-h-64 text-foreground font-mono border border-border">
+                      {JSON.stringify(toolResult, null, 2)}
+                    </pre>
+                  </div>
+                  
+                  {/* Widget UI Rendering */}
+                  {(() => {
+                    // Get widget URI from multiple sources (same as ToolCard)
+                    const widgetUri = 
+                      selectedTool.widget?.route || 
+                      selectedTool.outputTemplate || 
+                      selectedTool._meta?.['ui/template'] || 
+                      selectedTool._meta?.['openai/outputTemplate'];
+                    
+                    return widgetUri && toolResult ? (
+                      <div>
+                        <h3 className="font-semibold text-foreground mb-3">UI Widget:</h3>
+                        <div className="border border-border rounded-lg overflow-hidden h-64 bg-background shadow-inner">
+                          <WidgetRenderer
+                            uri={widgetUri}
+                            data={(() => {
+                            // Try to parse JSON from content[0].text, otherwise use raw result
+                            if (toolResult.content?.[0]?.text) {
+                              try {
+                                const parsed = JSON.parse(toolResult.content[0].text);
+                                // Unwrap if response was wrapped by TransformInterceptor
+                                if (parsed.success !== undefined && parsed.data !== undefined) {
+                                  return parsed.data;
+                                }
+                                return parsed;
+                              } catch {
+                                return { message: toolResult.content[0].text };
+                              }
+                            }
+                            return toolResult;
+                          })()}
+                            className="w-full h-full"
+                          />
+                        </div>
+                      </div>
+                    ) : null;
+                  })()}
+                </div>
+              )}
+            </div>
+          </div>
+        )}
+    </>
+  );
+}