ehbp 0.0.3 → 0.0.4

package/dist/esm/test/streaming.test.js

@@ -0,0 +1,71 @@
+import { describe, it } from 'node:test';
+import assert from 'node:assert';
+describe('Streaming', () => {
+    it('should handle streaming responses', async () => {
+        // Create a mock streaming response
+        const mockStreamData = 'Number: 1\nNumber: 2\nNumber: 3\n';
+        const mockResponse = new Response(mockStreamData, {
+            status: 200,
+            headers: {
+                'Content-Type': 'text/plain',
+                'Ehbp-Encapsulated-Key': 'abcd1234' // Mock encapsulated key
+            }
+        });
+        // Test that we can read from a stream
+        const reader = mockResponse.body?.getReader();
+        assert(reader, 'Response should have a readable stream');
+        const decoder = new TextDecoder();
+        let receivedData = '';
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            const text = decoder.decode(value, { stream: true });
+            receivedData += text;
+        }
+        assert(receivedData === mockStreamData, 'Should receive all stream data');
+    });
+    it('should handle empty streams', async () => {
+        const emptyResponse = new Response('', {
+            status: 200,
+            headers: {
+                'Ehbp-Encapsulated-Key': 'abcd1234'
+            }
+        });
+        const reader = emptyResponse.body?.getReader();
+        assert(reader, 'Response should have a readable stream');
+        const { done } = await reader.read();
+        assert(done, 'Empty stream should be done immediately');
+    });
+    it('should handle chunked data correctly', async () => {
+        // Simulate chunked data
+        const chunks = ['Hello', ' ', 'World', '!'];
+        const stream = new ReadableStream({
+            start(controller) {
+                chunks.forEach(chunk => {
+                    controller.enqueue(new TextEncoder().encode(chunk));
+                });
+                controller.close();
+            }
+        });
+        const response = new Response(stream, {
+            status: 200,
+            headers: {
+                'Ehbp-Encapsulated-Key': 'abcd1234'
+            }
+        });
+        const reader = response.body?.getReader();
+        assert(reader, 'Response should have a readable stream');
+        const decoder = new TextDecoder();
+        let receivedData = '';
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            const text = decoder.decode(value, { stream: true });
+            receivedData += text;
+        }
+        assert(receivedData === 'Hello World!', 'Should receive all chunks correctly');
+    });
+});
+//# sourceMappingURL=streaming.test.js.map

package/dist/esm/test/streaming.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming.test.js","sourceRoot":"","sources":["../../../src/test/streaming.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,MAAM,MAAM,aAAa,CAAC;AAEjC,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IAEzB,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,mCAAmC;QACnC,MAAM,cAAc,GAAG,mCAAmC,CAAC;QAC3D,MAAM,YAAY,GAAG,IAAI,QAAQ,CAAC,cAAc,EAAE;YAChD,MAAM,EAAE,GAAG;YACX,OAAO,EAAE;gBACP,cAAc,EAAE,YAAY;gBAC5B,uBAAuB,EAAE,UAAU,CAAC,wBAAwB;aAC7D;SACF,CAAC,CAAC;QAEH,sCAAsC;QACtC,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;QAC9C,MAAM,CAAC,MAAM,EAAE,wCAAwC,CAAC,CAAC;QAEzD,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,YAAY,GAAG,EAAE,CAAC;QAEtB,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAEhB,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YACrD,YAAY,IAAI,IAAI,CAAC;QACvB,CAAC;QAED,MAAM,CAAC,YAAY,KAAK,cAAc,EAAE,gCAAgC,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,aAAa,GAAG,IAAI,QAAQ,CAAC,EAAE,EAAE;YACrC,MAAM,EAAE,GAAG;YACX,OAAO,EAAE;gBACP,uBAAuB,EAAE,UAAU;aACpC;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;QAC/C,MAAM,CAAC,MAAM,EAAE,wCAAwC,CAAC,CAAC;QAEzD,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QACrC,MAAM,CAAC,IAAI,EAAE,yCAAyC,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,wBAAwB;QACxB,MAAM,MAAM,GAAG,CAAC,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;QAC5C,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;YAChC,KAAK,CAAC,UAAU;gBACd,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;oBACrB,UAAU,CAAC,OAAO,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;gBACtD,CAAC,CAAC,CAAC;gBACH,UAAU,CAAC,KAAK,EAAE,CAAC;YACrB,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,MAAM,EAAE;YACpC,MAAM,EAAE,GAAG;YACX,OAAO,EAAE;gBACP,uBAAuB,EAAE,UAAU;aACpC;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;QAC1C,MAAM,CAAC,MAAM,EAAE,wCAAwC,CAAC,CAAC;QAEzD,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,YAAY,GAAG,EAAE,CAAC;QAEtB,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAEhB,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YACrD,YAAY,IAAI,IAAI,CAAC;QACvB,CAAC;QAED,MAAM,CAAC,YAAY,KAAK,cAAc,EAAE,qCAAqC,CAAC,CAAC;IACjF,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ehbp",
-  "version": "0.0.3",
+  "version": "0.0.4",
   "description": "JavaScript client for Encrypted HTTP Body Protocol (EHBP)",
   "main": "./dist/cjs/index.js",
   "module": "./dist/esm/index.js",
@@ -48,5 +48,10 @@
   },
   "engines": {
     "node": ">=20.0.0"
-  }
+  },
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ]
 }

package/build-browser.js

@@ -1,54 +0,0 @@
-#!/usr/bin/env node
-
-/**
- * Build script to create a browser-compatible bundle
- */
-
-import { build } from 'esbuild';
-import { readFileSync, writeFileSync } from 'fs';
-import { join, dirname } from 'path';
-import { fileURLToPath } from 'url';
-
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = dirname(__filename);
-
-async function buildBrowser() {
-  console.log('Building browser bundle...');
-  
-  try {
-    // Build the main bundle
-    await build({
-      entryPoints: ['src/index.ts'],
-      bundle: true,
-      outfile: 'dist/browser.js',
-      format: 'esm',
-      target: 'es2020',
-      platform: 'browser',
-      external: [],
-      sourcemap: true,
-      minify: false,
-      define: {
-        'process.env.NODE_ENV': '"production"'
-      }
-    });
-
-    // Create a simple wrapper that exports everything
-    const wrapper = `
-// Browser-compatible wrapper for EHBP client
-export * from './browser.js';
-`;
-
-    writeFileSync(join(__dirname, 'dist', 'index.js'), wrapper);
-    
-    console.log('Browser bundle created successfully!');
-    console.log('Files created:');
-    console.log('  - dist/browser.js (main bundle)');
-    console.log('  - dist/index.js (wrapper)');
-    
-  } catch (error) {
-    console.error('Build failed:', error);
-    process.exit(1);
-  }
-}
-
-buildBrowser();

package/chat.html

@@ -1,285 +0,0 @@
-<!DOCTYPE html>
-<html lang="en">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>EHBP Chat Interface</title>
-    <style>
-        body {
-            font-family: Arial, sans-serif;
-            max-width: 800px;
-            margin: 0 auto;
-            padding: 20px;
-            background-color: #f5f5f5;
-        }
-        .container {
-            background: white;
-            padding: 0;
-            height: 95vh;
-            display: flex;
-            flex-direction: column;
-            border-radius: 0;
-        }
-        h1 {
-            color: #333;
-            text-align: center;
-            margin: 15px 0 10px 0;
-            padding: 0 20px;
-        }
-        .chat-container {
-            flex: 1;
-            display: flex;
-            flex-direction: column;
-            overflow: hidden;
-        }
-        .messages {
-            flex: 1;
-            padding: 20px;
-            overflow-y: auto;
-        }
-        .message {
-            margin-bottom: 12px;
-            padding: 8px 12px;
-            border-radius: 4px;
-            max-width: 80%;
-        }
-        .message.user {
-            background-color: #007bff;
-            color: white;
-            margin-left: auto;
-            text-align: right;
-            border: none;
-        }
-        .message.assistant {
-            background-color: #f8f9fa;
-            color: #333;
-            margin-right: auto;
-            border: none;
-        }
-        .message.system {
-            background-color: #fff3cd;
-            color: #856404;
-            margin: 0 auto;
-            text-align: center;
-            font-style: italic;
-            border: none;
-        }
-        .input-area {
-            display: flex;
-            align-items: center;
-            padding: 15px 20px;
-            gap: 8px;
-            border-top: 1px solid #eee;
-            background-color: white;
-        }
-        .input-area input {
-            flex: 1;
-            padding: 8px 10px;
-            border: 1px solid #ddd;
-            border-radius: 4px;
-            height: 32px;
-            font-family: inherit;
-            font-size: 13px;
-            outline: none;
-            box-sizing: border-box;
-        }
-        .input-area input:focus {
-            border-color: #007bff;
-        }
-        .input-area button {
-            padding: 8px 16px;
-            background-color: #007bff;
-            color: white;
-            border: none;
-            border-radius: 4px;
-            cursor: pointer;
-            font-size: 13px;
-            font-weight: 500;
-            min-width: 60px;
-        }
-        .input-area button:hover {
-            background-color: #0056b3;
-        }
-        .input-area button:disabled {
-            background-color: #6c757d;
-            cursor: not-allowed;
-        }
-        .typing-indicator {
-            padding: 10px;
-            font-style: italic;
-            color: #666;
-            display: none;
-        }
-    </style>
-</head>
-<body>
-    <div class="container">
-        <h1>EHBP Chat Interface</h1>
-        <div class="chat-container">
-            <div id="messages" class="messages"></div>
-            <div class="typing-indicator" id="typingIndicator">Assistant is typing...</div>
-            <div class="input-area">
-                <input type="text" id="messageInput" placeholder="Type your message here..." onkeydown="handleKeyDown(event)" />
-                <button id="sendBtn" onclick="sendMessage()">Send</button>
-            </div>
-        </div>
-    </div>
-
-    <script type="module">
-        import { Identity, createTransport } from './dist/browser.js';
-
-        const serverUrl = 'http://localhost:8443';
-
-        let transport = null;
-        let clientIdentity = null;
-        let conversationHistory = [];
-
-        async function initializeClient() {
-            try {
-                clientIdentity = await Identity.generate();
-                transport = await createTransport(serverUrl, clientIdentity);
-                return true;
-            } catch (error) {
-                console.error('Failed to connect:', error.message);
-                return false;
-            }
-        }
-
-        function addMessage(role, content) {
-            const messagesContainer = document.getElementById('messages');
-            const messageDiv = document.createElement('div');
-            messageDiv.className = `message ${role}`;
-            messageDiv.textContent = content;
-            messagesContainer.appendChild(messageDiv);
-            messagesContainer.scrollTop = messagesContainer.scrollHeight;
-
-            if (role !== 'system') {
-                conversationHistory.push({ role, content });
-            }
-        }
-
-        function showTypingIndicator() {
-            document.getElementById('typingIndicator').style.display = 'block';
-            const messagesContainer = document.getElementById('messages');
-            messagesContainer.scrollTop = messagesContainer.scrollHeight;
-        }
-
-        function hideTypingIndicator() {
-            document.getElementById('typingIndicator').style.display = 'none';
-        }
-
-        async function sendMessage() {
-            const messageInput = document.getElementById('messageInput');
-            const sendBtn = document.getElementById('sendBtn');
-            const message = messageInput.value.trim();
-
-            if (!message) return;
-
-            messageInput.disabled = true;
-            sendBtn.disabled = true;
-
-            try {
-                if (!transport) {
-                    const initialized = await initializeClient();
-                    if (!initialized) return;
-                }
-
-                addMessage('user', message);
-                messageInput.value = '';
-                showTypingIndicator();
-
-                const response = await transport.request(`${serverUrl}/v1/chat/completions`, {
-                    method: 'POST',
-                    headers: { 'Content-Type': 'application/json' },
-                    body: JSON.stringify({
-                        model: "qwen:0.5b",
-                        messages: conversationHistory,
-                        stream: true
-                    })
-                });
-
-                hideTypingIndicator();
-                if (!response.ok) throw new Error(`HTTP ${response.status}: ${response.statusText}`);
-                await handleStreamingResponse(response);
-            } catch (error) {
-                hideTypingIndicator();
-                addMessage('system', `Error: ${error.message}`);
-                console.error('Chat error:', error);
-            } finally {
-                messageInput.disabled = false;
-                sendBtn.disabled = false;
-                messageInput.focus();
-            }
-        }
-
-        async function handleStreamingResponse(response) {
-            const reader = response.body?.getReader();
-            if (!reader) {
-                throw new Error('No readable stream available');
-            }
-
-            const decoder = new TextDecoder();
-            let assistantMessage = '';
-            let messageDiv = null;
-
-            try {
-                while (true) {
-                    const { done, value } = await reader.read();
-                    if (done) break;
-
-                    const text = decoder.decode(value, { stream: true });
-                    const lines = text.split('\n');
-
-                    for (const line of lines) {
-                        if (line.trim() === '') continue;
-                        if (line.startsWith('data: ')) {
-                            const data = line.slice(6);
-                            if (data === '[DONE]') continue;
-
-                            try {
-                                const parsed = JSON.parse(data);
-                                const delta = parsed.choices?.[0]?.delta;
-                                if (delta?.content) {
-                                    assistantMessage += delta.content;
-                                    if (!messageDiv) {
-                                        const messagesContainer = document.getElementById('messages');
-                                        messageDiv = document.createElement('div');
-                                        messageDiv.className = 'message assistant';
-                                        messagesContainer.appendChild(messageDiv);
-                                    }
-                                    messageDiv.textContent = assistantMessage;
-                                    document.getElementById('messages').scrollTop = document.getElementById('messages').scrollHeight;
-                                }
-                            } catch (parseError) {
-                                console.warn('Failed to parse streaming data:', parseError);
-                            }
-                        }
-                    }
-                }
-
-                if (assistantMessage) {
-                    conversationHistory.push({ role: 'assistant', content: assistantMessage });
-                }
-
-            } finally {
-                reader.releaseLock();
-            }
-        }
-
-
-        function handleKeyDown(event) {
-            if (event.key === 'Enter' && !event.shiftKey) {
-                event.preventDefault();
-                sendMessage();
-            }
-        }
-
-        window.sendMessage = sendMessage;
-        window.handleKeyDown = handleKeyDown;
-
-        document.addEventListener('DOMContentLoaded', () => {
-            document.getElementById('messageInput').focus();
-        });
-    </script>
-</body>
-</html>

package/src/client.ts

@@ -1,181 +0,0 @@
-import { Identity } from './identity.js';
-import { PROTOCOL } from './protocol.js';
-
-/**
- * HTTP transport for EHBP
- */
-export class Transport {
-  private clientIdentity: Identity;
-  private serverHost: string;
-  private serverPublicKey: CryptoKey;
-
-  constructor(clientIdentity: Identity, serverHost: string, serverPublicKey: CryptoKey) {
-    this.clientIdentity = clientIdentity;
-    this.serverHost = serverHost;
-    this.serverPublicKey = serverPublicKey;
-  }
-
-  /**
-   * Create a new transport by fetching server public key
-   */
-  static async create(serverURL: string, clientIdentity: Identity): Promise<Transport> {
-    const url = new URL(serverURL);
-    const serverHost = url.host;
-
-    // Fetch server public key
-    const keysURL = new URL(PROTOCOL.KEYS_PATH, serverURL);
-    const response = await fetch(keysURL.toString());
-
-    if (!response.ok) {
-      throw new Error(`Failed to get server public key: ${response.status}`);
-    }
-
-    const contentType = response.headers.get('content-type');
-    if (contentType !== PROTOCOL.KEYS_MEDIA_TYPE) {
-      throw new Error(`Invalid content type: ${contentType}`);
-    }
-
-    const keysData = new Uint8Array(await response.arrayBuffer());
-    const serverIdentity = await Identity.unmarshalPublicConfig(keysData);
-    const serverPublicKey = serverIdentity.getPublicKey();
-
-    return new Transport(clientIdentity, serverHost, serverPublicKey);
-  }
-
-  /**
-   * Get the server public key
-   */
-  getServerPublicKey(): CryptoKey {
-    return this.serverPublicKey;
-  }
-
-  /**
-   * Get the server public key as hex string
-   */
-  async getServerPublicKeyHex(): Promise<string> {
-    const exported = await crypto.subtle.exportKey('raw', this.serverPublicKey);
-    const keyBytes = new Uint8Array(exported);
-    return Array.from(keyBytes)
-      .map(b => b.toString(16).padStart(2, '0'))
-      .join('');
-  }
-
-  /**
-   * Get the client public key
-   */
-  getClientPublicKey(): CryptoKey {
-    return this.clientIdentity.getPublicKey();
-  }
-
-  /**
-   * Make an encrypted HTTP request
-   */
-  async request(input: RequestInfo | URL, init?: RequestInit): Promise<Response> {
-    // Extract body from init or original request before creating Request object
-    let requestBody: BodyInit | null = null;
-    
-    if (input instanceof Request) {
-      // If input is a Request, extract its body
-      if (input.body) {
-        requestBody = await input.arrayBuffer();
-      }
-    } else {
-      // If input is URL/string, get body from init
-      requestBody = init?.body || null;
-    }
-
-    // Create the URL with correct host
-    let url: URL;
-    let method: string;
-    let headers: HeadersInit;
-
-    if (input instanceof Request) {
-      url = new URL(input.url);
-      method = input.method;
-      headers = input.headers;
-    } else {
-      url = new URL(input);
-      method = init?.method || 'GET';
-      headers = init?.headers || {};
-    }
-
-    url.host = this.serverHost;
-    
-    let request = new Request(url.toString(), {
-      method,
-      headers,
-      body: requestBody,
-      duplex: 'half'
-    } as RequestInit);
-
-    // Encrypt request body if present (check the original requestBody, not request.body)
-    if (requestBody !== null && requestBody !== undefined) {
-      request = await this.clientIdentity.encryptRequest(request, this.serverPublicKey);
-    } else {
-      // No body, just set client public key header
-      const headers = new Headers(request.headers);
-      headers.set(PROTOCOL.CLIENT_PUBLIC_KEY_HEADER, await this.clientIdentity.getPublicKeyHex());
-      request = new Request(request.url, {
-        method: request.method,
-        headers,
-        body: null
-      });
-    }
-
-    // Make the request
-    const response = await fetch(request);
-
-    if (!response.ok) {
-      console.warn(`Server returned non-OK status: ${response.status}`);
-    }
-
-    // Check for encapsulated key header
-    const encapKeyHeader = response.headers.get(PROTOCOL.ENCAPSULATED_KEY_HEADER);
-    if (!encapKeyHeader) {
-      throw new Error(`Missing ${PROTOCOL.ENCAPSULATED_KEY_HEADER} encapsulated key header`);
-    }
-
-    // Decode encapsulated key
-    const serverEncapKey = new Uint8Array(
-      encapKeyHeader.match(/.{2}/g)!.map(byte => parseInt(byte, 16))
-    );
-
-    // Decrypt response
-    return await this.clientIdentity.decryptResponse(response, serverEncapKey);
-  }
-
-  /**
-   * Convenience method for GET requests
-   */
-  async get(url: string | URL, init?: RequestInit): Promise<Response> {
-    return this.request(url, { ...init, method: 'GET' });
-  }
-
-  /**
-   * Convenience method for POST requests
-   */
-  async post(url: string | URL, body?: BodyInit, init?: RequestInit): Promise<Response> {
-    return this.request(url, { ...init, method: 'POST', body });
-  }
-
-  /**
-   * Convenience method for PUT requests
-   */
-  async put(url: string | URL, body?: BodyInit, init?: RequestInit): Promise<Response> {
-    return this.request(url, { ...init, method: 'PUT', body });
-  }
-
-  /**
-   * Convenience method for DELETE requests
-   */
-  async delete(url: string | URL, init?: RequestInit): Promise<Response> {
-    return this.request(url, { ...init, method: 'DELETE' });
-  }
-}
-
-/**
- * Create a new transport instance
- */
-export async function createTransport(serverURL: string, clientIdentity: Identity): Promise<Transport> {
-  return Transport.create(serverURL, clientIdentity);
-}