feat: Python and Java support with URL routing and HF Spaces Dockerfile

Dockerfile

@@ -0,0 +1,38 @@
+FROM node:20-bookworm
+
+# Install Python and Java
+RUN apt-get update && apt-get install -y \
+    python3 \
+    openjdk-17-jdk \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set working directory
+WORKDIR /app
+
+# Copy root package files
+COPY package.json package-lock.json ./
+
+# Copy workspace package files
+COPY packages/shared/package.json ./packages/shared/
+COPY packages/server/package.json ./packages/server/
+COPY packages/client/package.json ./packages/client/
+
+# Install dependencies
+RUN npm ci
+
+# Copy full source
+COPY . .
+
+# Build packages
+RUN npm run build
+
+# Set environment variables for Hugging Face Spaces
+ENV NODE_ENV=production
+ENV PORT=7860
+ENV CLIENT_URL=http://localhost:7860
+
+# Expose the HF port
+EXPOSE 7860
+
+# Start the server
+CMD ["npm", "run", "start", "-w", "packages/server"]

packages/client/src/App.tsx

@@ -1,13 +1,17 @@
-import { BrowserRouter, Routes, Route } from 'react-router-dom';
+import { BrowserRouter, Routes, Route, Navigate } from 'react-router-dom';
 import LandingPage from './components/landing/LandingPage';
 import IDEShell from './components/layout/IDEShell';
 
+import EnvironmentSelector from './components/ide/EnvironmentSelector';
+
 function App() {
   return (
     <BrowserRouter>
       <Routes>
         <Route path="/" element={<LandingPage />} />
-        <Route path="/ide" element={<IDEShell />} />
+        <Route path="/ide" element={<EnvironmentSelector />} />
+        <Route path="/ide/:env" element={<IDEShell />} />
+        <Route path="*" element={<Navigate to="/" replace />} />
       </Routes>
     </BrowserRouter>
   );

packages/client/src/components/ide/EnvironmentSelector.tsx

@@ -0,0 +1,52 @@
+import React from 'react';
+import { useNavigate } from 'react-router-dom';
+import { Monitor, Code2, Terminal } from 'lucide-react';
+
+export default function EnvironmentSelector() {
+  const navigate = useNavigate();
+
+  return (
+    <div className="min-h-screen flex flex-col items-center justify-center bg-zinc-950 text-zinc-100 p-4">
+      <div className="max-w-3xl w-full">
+        <h1 className="text-3xl font-bold text-center mb-2">Select Environment</h1>
+        <p className="text-center text-zinc-400 mb-10">
+          Choose your project type to launch the IDE
+        </p>
+        
+        <div className="grid grid-cols-1 sm:grid-cols-3 gap-6">
+          <EnvCard 
+            title="Web Development" 
+            description="HTML, CSS, JS/TS, React with Live Preview" 
+            icon={<Monitor className="w-10 h-10 mb-4 text-blue-400" />}
+            onClick={() => navigate('/ide/web')}
+          />
+          <EnvCard 
+            title="Java" 
+            description="Java JDK 17 with Console Output" 
+            icon={<Code2 className="w-10 h-10 mb-4 text-orange-400" />}
+            onClick={() => navigate('/ide/java')}
+          />
+          <EnvCard 
+            title="Python" 
+            description="Python 3 Environment with Console Output" 
+            icon={<Terminal className="w-10 h-10 mb-4 text-green-400" />}
+            onClick={() => navigate('/ide/python')}
+          />
+        </div>
+      </div>
+    </div>
+  );
+}
+
+function EnvCard({ title, description, icon, onClick }: { title: string, description: string, icon: React.ReactNode, onClick: () => void }) {
+  return (
+    <div 
+      onClick={onClick}
+      className="flex flex-col items-center justify-center p-8 bg-zinc-900 border border-zinc-800 hover:border-zinc-700 hover:bg-zinc-800/80 cursor-pointer rounded-xl transition-all duration-200 shadow-sm"
+    >
+      {icon}
+      <h3 className="text-xl font-medium mb-3">{title}</h3>
+      <p className="text-sm text-center text-zinc-400 leading-relaxed">{description}</p>
+    </div>
+  );
+}

packages/client/src/components/layout/IDEShell.tsx

@@ -1,17 +1,27 @@
 import { useState, useEffect } from 'react';
+import { useParams, useNavigate } from 'react-router-dom';
 import TopBar from './TopBar';
 import StatusBar from './StatusBar';
 import PanelLayout from './PanelLayout';
 import { useFileStore } from '@/stores/fileStore';
 import { useEditorStore } from '@/stores/editorStore';
+import { useEnvStore, Environment } from '@/stores/envStore';
 
 export default function IDEShell() {
+  const { env } = useParams<{ env: string }>();
+  const navigate = useNavigate();
+
   const [layout, setLayout] = useState<'editor' | 'split' | 'preview'>('split');
   const initialize = useFileStore((s) => s.initialize);
   const initialized = useFileStore((s) => s.initialized);
   const files = useFileStore((s) => s.files);
+  const createFile = useFileStore((s) => s.createFile);
+  const deleteFile = useFileStore((s) => s.deleteFile);
   const addFile = useEditorStore((s) => s.addFile);
+  const removeFile = useEditorStore((s) => s.removeFile);
   const openFiles = useEditorStore((s) => s.openFiles);
+  const environment = useEnvStore((s) => s.environment);
+  const setEnvironment = useEnvStore((s) => s.setEnvironment);
   const [hasAutoOpened, setHasAutoOpened] = useState(false);
 
   useEffect(() => {
@@ -19,13 +29,56 @@ export default function IDEShell() {
   }, [initialize]);
 
   useEffect(() => {
-    if (initialized && !hasAutoOpened) {
-      if (Object.keys(openFiles).length === 0 && files['index.html']) {
-        addFile('index.html', files['index.html'], 'html');
+    if (env && !['web', 'java', 'python'].includes(env)) {
+      navigate('/ide', { replace: true });
+    }
+  }, [env, navigate]);
+
+  useEffect(() => {
+    if (!initialized || !env || !['web', 'java', 'python'].includes(env)) return;
+
+    const targetEnv = env as Environment;
+    
+    // If URL environment differs from store, clean up and scaffold
+    if (environment !== targetEnv) {
+      setEnvironment(targetEnv);
+      
+      // Cleanup old files
+      Object.keys(files).forEach(f => {
+        if (f === 'README.md') return;
+        const isJava = f.endsWith('.java');
+        const isPython = f.endsWith('.py');
+        const isWeb = ['.html', '.css', '.js', '.ts', '.tsx', '.jsx'].some(ext => f.endsWith(ext));
+        
+        if (targetEnv === 'java' && !isJava) { deleteFile(f); removeFile(f); }
+        if (targetEnv === 'python' && !isPython) { deleteFile(f); removeFile(f); }
+        if (targetEnv === 'web' && !isWeb) { deleteFile(f); removeFile(f); }
+      });
+
+      // Scaffold files
+      if (targetEnv === 'web') {
+        const content = `<!DOCTYPE html>\n<html lang="en">\n<head>\n  <meta charset="UTF-8">\n  <meta name="viewport" content="width=device-width, initial-scale=1.0">\n  <title>Document</title>\n  <style>\n    body { font-family: sans-serif; display: flex; justify-content: center; align-items: center; height: 100vh; margin: 0; }\n  </style>\n</head>\n<body>\n  <h1>Hello Web!</h1>\n</body>\n</html>`;
+        if (!files['index.html']) createFile('index.html', content);
+      } else if (targetEnv === 'java') {
+        const content = `public class Main {\n    public static void main(String[] args) {\n        System.out.println("Hello, Java!");\n    }\n}`;
+        if (!files['Main.java']) createFile('Main.java', content);
+      } else if (targetEnv === 'python') {
+        const content = `print("Hello, Python!")`;
+        if (!files['main.py']) createFile('main.py', content);
+      }
+    }
+  }, [env, environment, initialized]); // Missing files to avoid loop on scaffolding
+
+  useEffect(() => {
+    if (initialized && environment === env && !hasAutoOpened) {
+      if (Object.keys(openFiles).length === 0) {
+        if (environment === 'web' && files['index.html']) addFile('index.html', files['index.html'], 'html');
+        else if (environment === 'java' && files['Main.java']) addFile('Main.java', files['Main.java'], 'java');
+        else if (environment === 'python' && files['main.py']) addFile('main.py', files['main.py'], 'python');
       }
       setHasAutoOpened(true);
     }
-  }, [initialized, hasAutoOpened, files, openFiles, addFile]);
+  }, [initialized, environment, env, hasAutoOpened, files, openFiles, addFile]);
 
   if (!initialized) {
     return (
@@ -37,9 +90,9 @@ export default function IDEShell() {
 
   return (
     <div className="h-screen flex flex-col bg-background overflow-hidden">
-      <TopBar layout={layout} onLayoutChange={setLayout} />
-      <PanelLayout layout={layout} />
-      <StatusBar />
+        <TopBar layout={layout} onLayoutChange={setLayout} />
+        <PanelLayout layout={layout} />
+        <StatusBar />
     </div>
   );
 }

packages/client/src/components/layout/PanelLayout.tsx

@@ -2,7 +2,9 @@ import { Panel, PanelGroup, PanelResizeHandle } from 'react-resizable-panels';
 import Sidebar from './Sidebar';
 import MultiFileEditor from '@/components/editor/MultiFileEditor';
 import LivePreview from '@/components/preview/LivePreview';
+import TerminalPanel from '@/components/terminal/TerminalPanel';
 import AIChatPanel from '@/components/ai/AIChatPanel';
+import { useEnvStore } from '@/stores/envStore';
 
 interface PanelLayoutProps {
   layout: 'editor' | 'split' | 'preview';
@@ -19,6 +21,15 @@ function ResizeHandle({ direction = 'vertical' }: { direction?: 'vertical' | 'ho
 }
 
 export default function PanelLayout({ layout }: PanelLayoutProps) {
+  const environment = useEnvStore((s) => s.environment);
+
+  const renderSecondaryPanel = () => {
+    if (environment === 'java' || environment === 'python') {
+      return <TerminalPanel />;
+    }
+    return <LivePreview />;
+  };
+
   return (
     <PanelGroup direction="horizontal" className="flex-1 min-h-0">
       {/* Left: Sidebar */}
@@ -31,7 +42,7 @@ export default function PanelLayout({ layout }: PanelLayoutProps) {
       {/* Center: Editor + Preview */}
       <Panel defaultSize={52} minSize={30} className="flex flex-col overflow-hidden">
         {layout === 'editor' && <MultiFileEditor />}
-        {layout === 'preview' && <LivePreview />}
+        {layout === 'preview' && renderSecondaryPanel()}
         {layout === 'split' && (
           <PanelGroup direction="horizontal">
             <Panel defaultSize={50}>
@@ -39,7 +50,7 @@ export default function PanelLayout({ layout }: PanelLayoutProps) {
             </Panel>
             <ResizeHandle direction="vertical" />
             <Panel defaultSize={50}>
-              <LivePreview />
+              {renderSecondaryPanel()}
             </Panel>
           </PanelGroup>
         )}

packages/client/src/components/layout/TopBar.tsx

@@ -1,8 +1,10 @@
 import { useState } from 'react';
+import { useNavigate } from 'react-router-dom';
 import { Menu, Play, Shield, Github, Monitor, Columns, Eye } from 'lucide-react';
 import { Button } from '@/components/ui/button';
 import { useAgentReview } from '@/hooks/useAgentReview';
 import RepoImporter from '@/components/github/RepoImporter';
+import { useEnvStore } from '@/stores/envStore';
 
 interface TopBarProps {
   layout: 'editor' | 'split' | 'preview';
@@ -10,8 +12,16 @@ interface TopBarProps {
 }
 
 export default function TopBar({ layout, onLayoutChange }: TopBarProps) {
+  const navigate = useNavigate();
   const { reviewWorkspace, isReviewing } = useAgentReview();
   const [showImporter, setShowImporter] = useState(false);
+  const environment = useEnvStore((s) => s.environment);
+
+  // Define global window method or a custom hook to trigger execution that terminal listens to?
+  // We'll dispatch a custom event for now that terminal/execution handler can listen to.
+  const executeCode = () => {
+    window.dispatchEvent(new CustomEvent('asipilot:execute'));
+  };
 
   return (
     <>
@@ -45,6 +55,17 @@ export default function TopBar({ layout, onLayoutChange }: TopBarProps) {
           ))}
         </div>
 
+        {(environment === 'java' || environment === 'python') && (
+          <Button
+            size="sm"
+            onClick={executeCode}
+            className="gap-1.5 text-xs bg-green-600 hover:bg-green-700 text-white border-0 shadow-sm transition-all"
+          >
+            <Play className="w-3.5 h-3.5 fill-current" />
+            Run
+          </Button>
+        )}
+
         <Button
           size="sm"
           variant="ghost"
@@ -65,6 +86,15 @@ export default function TopBar({ layout, onLayoutChange }: TopBarProps) {
           <Github className="w-3.5 h-3.5" />
           Import
         </Button>
+
+        <Button
+          size="sm"
+          variant="ghost"
+          onClick={() => navigate('/ide')}
+          className="gap-1.5 text-xs text-muted-foreground hover:text-foreground ml-2"
+        >
+          Change Environment
+        </Button>
       </div>
 
       <RepoImporter open={showImporter} onClose={() => setShowImporter(false)} />

packages/client/src/components/terminal/TerminalPanel.tsx

@@ -1,11 +1,19 @@
 import { useRef, useEffect } from 'react';
 import { Terminal } from 'xterm';
 import { FitAddon } from 'xterm-addon-fit';
+import { getSocket } from '@/services/socket';
+import { WS_EVENTS } from '@asipilot/shared';
+import { useEditorStore } from '@/stores/editorStore';
+import { useEnvStore } from '@/stores/envStore';
 import 'xterm/css/xterm.css';
 
 export default function TerminalPanel() {
   const containerRef = useRef<HTMLDivElement>(null);
   const terminalRef = useRef<Terminal>();
+  
+  const activeFilePath = useEditorStore((s) => s.activeFilePath);
+  const openFiles = useEditorStore((s) => s.openFiles);
+  const environment = useEnvStore((s) => s.environment);
 
   useEffect(() => {
     if (!containerRef.current) return;
@@ -41,28 +49,6 @@ export default function TerminalPanel() {
     terminal.writeln('');
     terminal.write('\x1b[32m❯\x1b[0m ');
 
-    // Simple echo
-    let currentLine = '';
-    terminal.onKey(({ key, domEvent }) => {
-      const char = key;
-      if (domEvent.keyCode === 13) {
-        terminal.writeln('');
-        if (currentLine.trim()) {
-          terminal.writeln(`\x1b[90m$ ${currentLine}\x1b[0m`);
-        }
-        currentLine = '';
-        terminal.write('\x1b[32m❯\x1b[0m ');
-      } else if (domEvent.keyCode === 8) {
-        if (currentLine.length > 0) {
-          currentLine = currentLine.slice(0, -1);
-          terminal.write('\b \b');
-        }
-      } else if (char.length === 1 && !domEvent.ctrlKey && !domEvent.altKey) {
-        currentLine += char;
-        terminal.write(char);
-      }
-    });
-
     terminalRef.current = terminal;
 
     const resizeObserver = new ResizeObserver(() => fitAddon.fit());
@@ -74,5 +60,54 @@ export default function TerminalPanel() {
     };
   }, []);
 
+  useEffect(() => {
+    const handleExecute = () => {
+      if (!activeFilePath) return;
+      const file = openFiles[activeFilePath];
+      if (!file || !environment) return;
+
+      terminalRef.current?.clear();
+      terminalRef.current?.writeln(`\x1b[33mRunning ${file.path}...\x1b[0m\n`);
+      
+      const socket = getSocket();
+      socket.emit(WS_EVENTS.EXECUTE_REQUEST, {
+        language: environment,
+        content: file.content
+      });
+    };
+
+    window.addEventListener('asipilot:execute', handleExecute);
+    return () => window.removeEventListener('asipilot:execute', handleExecute);
+  }, [activeFilePath, openFiles, environment]);
+
+  useEffect(() => {
+    const socket = getSocket();
+    
+    const onToken = (data: { token: string; isError?: boolean }) => {
+      const color = data.isError ? '\x1b[31m' : '\x1b[0m';
+      const text = data.token.replace(/\n/g, '\r\n');
+      terminalRef.current?.write(`${color}${text}\x1b[0m`);
+    };
+    
+    const onComplete = () => {
+      terminalRef.current?.writeln('\n\n\x1b[32m❯ Execution finished.\x1b[0m ');
+    };
+    
+    const onError = (data: { error: string }) => {
+      terminalRef.current?.writeln(`\r\n\x1b[31m[Error] ${data.error}\x1b[0m\r\n`);
+      terminalRef.current?.writeln('\x1b[32m❯\x1b[0m ');
+    };
+
+    socket.on(WS_EVENTS.EXECUTE_TOKEN, onToken);
+    socket.on(WS_EVENTS.EXECUTE_COMPLETE, onComplete);
+    socket.on(WS_EVENTS.EXECUTE_ERROR, onError);
+
+    return () => {
+      socket.off(WS_EVENTS.EXECUTE_TOKEN, onToken);
+      socket.off(WS_EVENTS.EXECUTE_COMPLETE, onComplete);
+      socket.off(WS_EVENTS.EXECUTE_ERROR, onError);
+    };
+  }, []);
+
   return <div ref={containerRef} className="h-full w-full" />;
 }

packages/client/src/hooks/useCodeCompletion.ts

@@ -35,7 +35,7 @@ export function useCodeCompletion() {
           .filter((f) => f !== filePath)
           .slice(0, 3)
           .map((f) => {
-            const file = editorState.openFiles.get(f);
+            const file = editorState.openFiles[f];
             return file ? { path: f, content: file.content } : null;
           })
           .filter(Boolean);

packages/client/src/stores/envStore.ts

@@ -0,0 +1,24 @@
+import { create } from 'zustand';
+import { persist, createJSONStorage } from 'zustand/middleware';
+
+export type Environment = 'web' | 'java' | 'python' | null;
+
+interface EnvState {
+  environment: Environment;
+  setEnvironment: (env: Environment) => void;
+  resetEnvironment: () => void;
+}
+
+export const useEnvStore = create<EnvState>()(
+  persist(
+    (set) => ({
+      environment: null,
+      setEnvironment: (env) => set({ environment: env }),
+      resetEnvironment: () => set({ environment: null }),
+    }),
+    {
+      name: 'env-storage',
+      storage: createJSONStorage(() => localStorage),
+    }
+  )
+);

packages/client/vite.config.d.ts

@@ -0,0 +1,2 @@
+declare const _default: import("vite").UserConfig;
+export default _default;

packages/client/vite.config.js

@@ -0,0 +1,24 @@
+import { defineConfig } from 'vite';
+import react from '@vitejs/plugin-react';
+import path from 'path';
+export default defineConfig({
+    plugins: [react()],
+    resolve: {
+        alias: {
+            '@': path.resolve(__dirname, './src'),
+        },
+    },
+    server: {
+        port: 5173,
+        proxy: {
+            '/api': {
+                target: 'http://localhost:3001',
+                changeOrigin: true,
+            },
+            '/socket.io': {
+                target: 'http://localhost:3001',
+                ws: true,
+            },
+        },
+    },
+});

packages/server/src/agents/orchestrator.ts

@@ -17,7 +17,7 @@ export class AgentOrchestrator extends EventEmitter {
 
   constructor() {
     super();
-    this.agents = new Map([
+    this.agents = new Map<AgentType, BaseAgent>([
       ['security', new SecurityAgent()],
       ['performance', new PerformanceAgent()],
       ['style', new StyleAgent()],

packages/server/src/index.ts

@@ -2,6 +2,8 @@ import express from 'express';
 import { createServer } from 'http';
 import { Server as SocketServer } from 'socket.io';
 import helmet from 'helmet';
+import path from 'path';
+import { fileURLToPath } from 'url';
 import { config } from './config/env.js';
 import { corsMiddleware } from './middleware/cors.middleware.js';
 import { errorHandler } from './middleware/error-handler.middleware.js';
@@ -37,6 +39,21 @@ app.use('/api/v1', apiRoutes);
 // Error handler
 app.use(errorHandler);
 
+// Serve static frontend in production
+if (config.NODE_ENV === 'production') {
+  const __filename = fileURLToPath(import.meta.url);
+  const __dirname = path.dirname(__filename);
+  
+  // In dist/index.js, __dirname is packages/server/dist
+  // So we go up 3 levels: dist -> server -> packages -> then into client/dist
+  const clientPath = path.join(__dirname, '../../client/dist');
+  app.use(express.static(clientPath));
+  
+  app.get('*', (req, res) => {
+    res.sendFile(path.join(clientPath, 'index.html'));
+  });
+}
+
 // WebSocket
 setupWebSocketHandlers(io);
 

packages/server/src/middleware/auth.middleware.ts

@@ -6,7 +6,7 @@ import { config } from '../config/env.js';
  * If AUTH_API_KEY is set in env, requires it in the Authorization header.
  */
 export function authMiddleware(req: Request, res: Response, next: NextFunction) {
-  const apiKey = (config as Record<string, string>)['AUTH_API_KEY'];
+  const apiKey = (config as unknown as Record<string, string>)['AUTH_API_KEY'];
   if (!apiKey) return next();
 
   const authHeader = req.headers.authorization;

packages/server/src/middleware/error-handler.middleware.ts

@@ -9,8 +9,8 @@ export function errorHandler(err: Error, req: Request, res: Response, _next: Nex
     stack: err.stack,
   });
 
-  const statusCode = (err as Record<string, unknown>).statusCode as number || 500;
-  const code = (err as Record<string, unknown>).code as string || 'INTERNAL_ERROR';
+  const statusCode = (err as unknown as Record<string, unknown>).statusCode as number || 500;
+  const code = (err as unknown as Record<string, unknown>).code as string || 'INTERNAL_ERROR';
 
   res.status(statusCode).json({
     success: false,

packages/server/src/services/cache.service.ts

@@ -12,8 +12,8 @@ class CacheService {
       this.redis = new Redis(config.REDIS_URL, {
         maxRetriesPerRequest: 3,
         retryStrategy: (times: number) => {
-          if (times > 5) return null;
-          return Math.min(times * 200, 2000);
+          // Do not retry. Fail immediately to gracefully disable caching and prevent log spam.
+          return null;
         },
         lazyConnect: true,
       });

packages/server/src/websocket/events.ts

@@ -11,4 +11,8 @@ export const WS_EVENTS = {
   CHAT_COMPLETE: 'chat:complete',
   CHAT_STOP: 'chat:stop',
   CHAT_ERROR: 'chat:error',
+  EXECUTE_REQUEST: 'execute:request',
+  EXECUTE_TOKEN: 'execute:token',
+  EXECUTE_COMPLETE: 'execute:complete',
+  EXECUTE_ERROR: 'execute:error',
 } as const;

packages/server/src/websocket/execution.ts

@@ -0,0 +1,104 @@
+import { Socket } from 'socket.io';
+import { spawn, exec } from 'child_process';
+import { promises as fs } from 'fs';
+import path from 'path';
+import os from 'os';
+import { WS_EVENTS } from '@asipilot/shared';
+import { logger } from '../utils/logger.js';
+import type { WSExecuteRequest } from '@asipilot/shared';
+
+// Time limit for execution to prevent infinite loops (10 seconds)
+const EXECUTION_TIMEOUT_MS = 10000;
+
+export async function handleCodeExecution(socket: Socket, data: WSExecuteRequest) {
+  const { language, content } = data;
+  let tempDir: string | null = null;
+  
+  try {
+    tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'asipilot-exec-'));
+    
+    if (language === 'python') {
+      if (!tempDir) throw new Error("Temp dir not created");
+      const dir = tempDir;
+      const filePath = path.join(dir, 'main.py');
+      await fs.writeFile(filePath, content, 'utf-8');
+      
+      await runProcess(socket, 'python3', [filePath]);
+    } else if (language === 'java') {
+      if (!tempDir) throw new Error("Temp dir not created");
+      const dir = tempDir;
+      const filePath = path.join(dir, 'Main.java');
+      await fs.writeFile(filePath, content, 'utf-8');
+      
+      // Compile Java First
+      await new Promise<void>((resolve, reject) => {
+        exec(`javac Main.java`, { cwd: dir, timeout: 5000 }, (error: Error | null, stdout: string, stderr: string) => {
+          if (error) {
+            socket.emit(WS_EVENTS.EXECUTE_TOKEN, { token: stderr || error.message, isError: true });
+            reject(new Error('Compilation Failed'));
+          } else {
+            resolve();
+          }
+        });
+      });
+      
+      // Run Java class
+      await runProcess(socket, 'java', ['Main'], tempDir);
+    } else {
+      socket.emit(WS_EVENTS.EXECUTE_TOKEN, { token: `Unsupported language: ${language}`, isError: true });
+      socket.emit(WS_EVENTS.EXECUTE_COMPLETE, {});
+    }
+  } catch (err) {
+    logger.error('Execution setup error', err);
+    socket.emit(WS_EVENTS.EXECUTE_ERROR, { error: (err as Error).message });
+    socket.emit(WS_EVENTS.EXECUTE_COMPLETE, {});
+  } finally {
+    // Cleanup
+    if (tempDir) {
+      try {
+        await fs.rm(tempDir, { recursive: true, force: true });
+      } catch (e) {
+        logger.error(`Failed to cleanup temp dir ${tempDir}`, e);
+      }
+    }
+  }
+}
+
+function runProcess(socket: Socket, command: string, args: string[], cwd?: string): Promise<void> {
+  return new Promise((resolve, reject) => {
+    const child = spawn(command, args, { cwd });
+    
+    let isFinished = false;
+
+    const timeoutPath = setTimeout(() => {
+      if (!isFinished) {
+        child.kill('SIGTERM');
+        socket.emit(WS_EVENTS.EXECUTE_TOKEN, { token: '\n[Execution Terminated: Timeout (10s)]', isError: true });
+      }
+    }, EXECUTION_TIMEOUT_MS);
+
+    child.stdout.on('data', (data) => {
+      socket.emit(WS_EVENTS.EXECUTE_TOKEN, { token: data.toString() });
+    });
+
+    child.stderr.on('data', (data) => {
+      socket.emit(WS_EVENTS.EXECUTE_TOKEN, { token: data.toString(), isError: true });
+    });
+
+    child.on('close', (code) => {
+      isFinished = true;
+      clearTimeout(timeoutPath);
+      socket.emit(WS_EVENTS.EXECUTE_TOKEN, { token: `\n[Process exited with code ${code}]` });
+      socket.emit(WS_EVENTS.EXECUTE_COMPLETE, {});
+      resolve();
+    });
+
+    child.on('error', (err) => {
+      isFinished = true;
+      clearTimeout(timeoutPath);
+      socket.emit(WS_EVENTS.EXECUTE_TOKEN, { token: `\n[Process error: ${err.message}]`, isError: true });
+      socket.emit(WS_EVENTS.EXECUTE_COMPLETE, {});
+      reject(err);
+    });
+  });
+}

packages/server/src/websocket/handler.ts

@@ -3,7 +3,8 @@ import { WS_EVENTS } from './events.js';
 import { asi1 } from '../services/asi1-client.js';
 import { orchestrator } from '../agents/orchestrator.js';
 import { logger } from '../utils/logger.js';
-import type { WSCompletionRequest, WSReviewRequest, WSChatMessage, AgentType } from '@asipilot/shared';
+import { handleCodeExecution } from './execution.js';
+import type { WSCompletionRequest, WSReviewRequest, WSChatMessage, AgentType, WSExecuteRequest } from '@asipilot/shared';
 
 const activeAbortControllers = new Map<string, AbortController>();
 
@@ -65,6 +66,11 @@ export function setupWebSocketHandlers(io: SocketServer) {
       }
     });
 
+    // Code execution
+    socket.on(WS_EVENTS.EXECUTE_REQUEST, async (data: WSExecuteRequest) => {
+      await handleCodeExecution(socket, data);
+    });
+
     // Chat with streaming
     socket.on(WS_EVENTS.CHAT_MESSAGE, async (data: WSChatMessage) => {
       const abortController = new AbortController();

packages/shared/package.json

@@ -2,10 +2,11 @@
   "name": "@asipilot/shared",
   "version": "1.0.0",
   "private": true,
-  "main": "./src/index.ts",
-  "types": "./src/index.ts",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
   "exports": {
-    ".": "./src/index.ts"
+    ".": "./dist/index.js"
   },
   "scripts": {
     "build": "tsc",

packages/shared/src/constants.ts

@@ -25,6 +25,12 @@ export const WS_EVENTS = {
   REVIEW_FILE: 'review:file',
   REVIEW_FILE_RESULT: 'review:file:result',
 
+  // Execution
+  EXECUTE_REQUEST: 'execute:request',
+  EXECUTE_TOKEN: 'execute:token',
+  EXECUTE_COMPLETE: 'execute:complete',
+  EXECUTE_ERROR: 'execute:error',
+
   // Chat
   CHAT_MESSAGE: 'chat:message',
   CHAT_TOKEN: 'chat:token',
@@ -37,7 +43,7 @@ export const WS_EVENTS = {
 export const SUPPORTED_LANGUAGES = [
   'html', 'css', 'scss', 'less', 'javascript', 'typescript',
   'javascriptreact', 'typescriptreact', 'vue', 'svelte',
-  'json', 'markdown', 'yaml', 'xml', 'svg',
+  'json', 'markdown', 'yaml', 'xml', 'svg', 'java', 'python'
 ] as const;
 
 // File Extension to Language Mapping
@@ -64,6 +70,8 @@ export const EXTENSION_MAP: Record<string, string> = {
   '.yml': 'yaml',
   '.xml': 'xml',
   '.svg': 'svg',
+  '.java': 'java',
+  '.py': 'python',
   '.env': 'plaintext',
   '.gitignore': 'plaintext',
   '.prettierrc': 'json',

packages/shared/src/index.ts

@@ -1,3 +1,3 @@
-export * from './types';
-export * from './constants';
-export * from './utils';
+export * from './types.js';
+export * from './constants.js';
+export * from './utils.js';

packages/shared/src/types.ts

@@ -240,6 +240,11 @@ export interface WSChatMessage {
   history: ASI1Message[];
 }
 
+export interface WSExecuteRequest {
+  language: string;
+  content: string;
+}
+
 // ===== API Response Wrappers =====
 
 export interface ApiResponse<T> {

packages/shared/src/utils.ts

@@ -1,4 +1,4 @@
-import { EXTENSION_MAP, EXCLUDED_DIRECTORIES, FRONTEND_EXTENSIONS } from './constants';
+import { EXTENSION_MAP, EXCLUDED_DIRECTORIES, FRONTEND_EXTENSIONS } from './constants.js';
 
 /**
  * Estimate token count from text (~4 chars per token heuristic).