more-compute 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

frontend/lib/websocket-native.ts

@@ -0,0 +1,202 @@
+import { Cell, ExecutionResult } from '@/types/notebook';
+
+export class WebSocketService {
+  private ws: WebSocket | null = null;
+  private listeners: Map<string, Function[]> = new Map();
+  private reconnectAttempts = 0;
+  private maxReconnectAttempts = 5;
+  private url: string = '';
+
+  connect(url: string = 'ws://localhost:8888/ws'): Promise<void> {
+    this.url = url;
+    return new Promise((resolve, reject) => {
+      try {
+        this.ws = new WebSocket(this.url);
+        
+        this.ws.onopen = () => {
+          console.log('Connected to server');
+          this.reconnectAttempts = 0;
+          this.emit('connect', null);
+          resolve();
+        };
+
+        this.ws.onerror = (error) => {
+          console.error('WebSocket error:', error);
+          this.emit('disconnect', null);
+          reject(error);
+        };
+
+        this.ws.onclose = (ev) => {
+          console.log('Disconnected from server', { code: ev.code, reason: ev.reason, wasClean: ev.wasClean });
+          this.emit('disconnect', null);
+          this.handleDisconnect();
+        };
+
+        this.ws.onmessage = (event) => {
+          try {
+            const data = JSON.parse(event.data);
+            console.log('[WS RECV]', data);
+            this.handleMessage(data);
+          } catch (error) {
+            console.error('Failed to parse message:', error);
+          }
+        };
+      } catch (error) {
+        reject(error);
+      }
+    });
+  }
+
+  private handleMessage(data: any) {
+    const messageType = data.type;
+    const messageData = data.data;
+  
+    // Map FastAPI message types to our expected events
+    switch (messageType) {
+      case 'notebook_data':
+        this.emit('notebook_loaded', messageData);
+        break;
+      case 'notebook_updated':
+        this.emit('notebook_updated', messageData);
+        break;
+      case 'execution_start':
+        this.emit('execution_start', messageData);
+        break;
+      case 'stream_output':
+        this.emit('stream_output', messageData);
+        break;
+      case 'execution_result':
+        this.emit('execution_result', messageData);
+        break;
+      case 'execution_complete': // This may be sent by some legacy paths
+        this.emit('execution_complete', messageData);
+        break;
+      case 'execution_error':
+        this.emit('execution_error', messageData);
+        break;
+      case 'execution_interrupted':
+        this.emit('execution_interrupted', messageData);
+        break;
+      case 'cell_updated': // Note: This is now handled by notebook_updated
+        this.emit('cell_updated', messageData);
+        break;
+      case 'cell_added':
+        this.emit('cell_added', messageData);
+        break;
+      case 'cell_deleted':
+        this.emit('cell_deleted', messageData);
+        break;
+      case 'kernel_status':
+        this.emit('kernel_status', messageData);
+        break;
+      case 'packages_updated':
+        // Notify listeners and also broadcast a DOM event for decoupled UIs
+        this.emit('packages_updated', messageData);
+        try {
+          window.dispatchEvent(new CustomEvent('mc:packages-updated', { detail: messageData }));
+        } catch {}
+        break;
+      case 'error':
+        this.emit('error', messageData);
+        break;
+      default:
+        console.warn('Unknown message type:', messageType);
+    }
+  }
+
+  private handleDisconnect() {
+    this.emit('disconnect', null);
+    // Attempt to reconnect
+    if (this.reconnectAttempts < this.maxReconnectAttempts) {
+      this.reconnectAttempts++;
+      console.log(`Attempting to reconnect (${this.reconnectAttempts}/${this.maxReconnectAttempts})...`);
+      setTimeout(() => {
+        this.connect(this.url).catch(console.error);
+      }, 1000 * this.reconnectAttempts);
+    }
+  }
+
+  on(event: string, callback: Function) {
+    if (!this.listeners.has(event)) {
+      this.listeners.set(event, []);
+    }
+    this.listeners.get(event)!.push(callback);
+  }
+
+  off(event: string, callback: Function) {
+    const callbacks = this.listeners.get(event);
+    if (callbacks) {
+      const index = callbacks.indexOf(callback);
+      if (index > -1) {
+        callbacks.splice(index, 1);
+      }
+    }
+  }
+
+  private emit(event: string, data: any) {
+    const callbacks = this.listeners.get(event);
+    if (callbacks) {
+      callbacks.forEach(callback => callback(data));
+    }
+  }
+
+  private send(type: string, data: any = {}) {
+    if (this.ws && this.ws.readyState === WebSocket.OPEN) {
+      this.ws.send(JSON.stringify({ type, data }));
+    } else {
+      console.error('WebSocket is not connected');
+    }
+  }
+
+  // Notebook operations
+  loadNotebook(notebookName: string) {
+    this.send('load_notebook', { notebook_name: notebookName });
+  }
+
+  saveNotebook() {
+    this.send('save_notebook');
+  }
+
+  // Cell operations
+  addCell(index: number, cellType: 'code' | 'markdown', source: string = '') {
+    this.send('add_cell', {
+      index,
+      cell_type: cellType,
+      source,
+    });
+  }
+
+  deleteCell(cellIndex: number) {
+    this.send('delete_cell', { cell_index: cellIndex });
+  }
+
+  updateCell(cellIndex: number, source: string) {
+    this.send('update_cell', {
+      cell_index: cellIndex,
+      source,
+    });
+  }
+
+  executeCell(cellIndex: number, source: string) {
+    this.send('execute_cell', {
+      cell_index: cellIndex,
+      source,
+    });
+  }
+
+  // Kernel operations
+  resetKernel() {
+    this.send('reset_kernel');
+  }
+
+  interruptKernel(cellIndex?: number) {
+    this.send('interrupt_kernel', { cell_index: cellIndex });
+  }
+
+  disconnect() {
+    if (this.ws) {
+      this.ws.close();
+      this.ws = null;
+    }
+  }
+}

frontend/lib/websocket.ts

@@ -0,0 +1,134 @@
+import { io, Socket } from 'socket.io-client';
+import { Cell, ExecutionResult } from '@/types/notebook';
+
+export class WebSocketService {
+  private socket: Socket | null = null;
+  private listeners: Map<string, Function[]> = new Map();
+
+  connect(url: string = 'ws://localhost:8000'): Promise<void> {
+    return new Promise((resolve, reject) => {
+      // For development, connect directly to the backend WebSocket
+      const wsUrl = process.env.NODE_ENV === 'production' 
+        ? '/ws' 
+        : 'ws://localhost:8000/ws';
+      
+      // Use native WebSocket for FastAPI compatibility
+      const ws = new WebSocket(wsUrl);
+      
+      // Wrap WebSocket in Socket.IO-like interface
+      this.socket = this.createSocketWrapper(ws);
+
+      this.socket.on('connect', () => {
+        console.log('Connected to server');
+        resolve();
+      });
+
+      this.socket.on('connect_error', (error) => {
+        console.error('Connection error:', error);
+        reject(error);
+      });
+
+      this.socket.on('disconnect', () => {
+        console.log('Disconnected from server');
+      });
+
+      // Set up event forwarding
+      this.setupEventForwarding();
+    });
+  }
+
+  private setupEventForwarding() {
+    if (!this.socket) return;
+
+    // Forward common events
+    const events = [
+      'notebook_loaded',
+      'cell_added', 
+      'cell_deleted',
+      'cell_updated',
+      'execution_result',
+      'kernel_status',
+      'error',
+    ];
+
+    events.forEach(event => {
+      this.socket!.on(event, (data) => {
+        this.emit(event, data);
+      });
+    });
+  }
+
+  on(event: string, callback: Function) {
+    if (!this.listeners.has(event)) {
+      this.listeners.set(event, []);
+    }
+    this.listeners.get(event)!.push(callback);
+  }
+
+  off(event: string, callback: Function) {
+    const callbacks = this.listeners.get(event);
+    if (callbacks) {
+      const index = callbacks.indexOf(callback);
+      if (index > -1) {
+        callbacks.splice(index, 1);
+      }
+    }
+  }
+
+  private emit(event: string, data: any) {
+    const callbacks = this.listeners.get(event);
+    if (callbacks) {
+      callbacks.forEach(callback => callback(data));
+    }
+  }
+
+  // Notebook operations
+  loadNotebook(notebookName: string) {
+    this.socket?.emit('load_notebook', { notebook_name: notebookName });
+  }
+
+  saveNotebook() {
+    this.socket?.emit('save_notebook');
+  }
+
+  // Cell operations
+  addCell(index: number, cellType: 'code' | 'markdown', source: string = '') {
+    this.socket?.emit('add_cell', {
+      index,
+      cell_type: cellType,
+      source,
+    });
+  }
+
+  deleteCell(cellIndex: number) {
+    this.socket?.emit('delete_cell', { cell_index: cellIndex });
+  }
+
+  updateCell(cellIndex: number, source: string) {
+    this.socket?.emit('update_cell', {
+      cell_index: cellIndex,
+      source,
+    });
+  }
+
+  executeCell(cellIndex: number, source: string) {
+    this.socket?.emit('execute_cell', {
+      cell_index: cellIndex,
+      source,
+    });
+  }
+
+  // Kernel operations
+  resetKernel() {
+    this.socket?.emit('reset_kernel');
+  }
+
+  interruptKernel() {
+    this.socket?.emit('interrupt_kernel');
+  }
+
+  disconnect() {
+    this.socket?.disconnect();
+    this.socket = null;
+  }
+}

frontend/next-env.d.ts

@@ -0,0 +1,6 @@
+/// <reference types="next" />
+/// <reference types="next/image-types/global" />
+/// <reference path="./.next/types/routes.d.ts" />
+
+// NOTE: This file should not be edited
+// see https://nextjs.org/docs/app/api-reference/config/typescript for more information.

frontend/next.config.mjs

@@ -0,0 +1,17 @@
+/** @type {import('next').NextConfig} */
+const nextConfig = {
+  async rewrites() {
+    return [
+      {
+        source: '/ws/:path*',
+        destination: 'http://localhost:8000/ws/:path*',
+      },
+      {
+        source: '/api/:path*',
+        destination: 'http://localhost:8000/api/:path*',
+      },
+    ];
+  },
+};
+
+export default nextConfig;

frontend/next.config.ts

@@ -0,0 +1,7 @@
+import type { NextConfig } from "next";
+
+const nextConfig: NextConfig = {
+  /* config options here */
+};
+
+export default nextConfig;