bertui 0.4.1 → 0.4.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bertui",
-  "version": "0.4.1",
+  "version": "0.4.4",
   "description": "Lightning-fast React dev server powered by Bun and Elysia",
   "type": "module",
   "main": "./index.js",
@@ -44,6 +44,9 @@
     "url": "https://github.com/BunElysiaReact/BERTUI.git"
   },
   "dependencies": {
+    "@jsquash/jpeg": "^1.6.0",
+    "@jsquash/png": "^3.1.1",
+    "@jsquash/webp": "^1.5.0",
     "elysia": "^1.0.0",
     "ernest-logger": "latest",
     "lightningcss": "^1.30.2"

package/src/build/image-optimizer.js

@@ -1,18 +1,52 @@
-// bertui/src/build/image-optimizer.js
-import { join, extname, basename, dirname } from 'path';
-import { existsSync, mkdirSync, readdirSync, statSync, cpSync } from 'fs';
+// bertui/src/build/image-optimizer.js - WASM-POWERED VERSION 🚀
+import { join, extname } from 'path';
+import { existsSync, mkdirSync, readdirSync, cpSync } from 'fs';
 import logger from '../logger/logger.js';
 
 /**
- * Optimize images using Bun's native image processing
- * This is FAST because it uses native code under the hood
+ * 🎯 WASM-powered image optimization using @jsquash
+ * Zero OS dependencies, pure JavaScript, blazing fast!
+ */
+
+// Lazy-load WASM modules (only when needed)
+let pngEncode, pngDecode;
+let jpegEncode, jpegDecode;
+let webpEncode, webpDecode;
+
+async function initializePNG() {
+  if (!pngEncode) {
+    const { encode, decode } = await import('@jsquash/png');
+    pngEncode = encode;
+    pngDecode = decode;
+  }
+}
+
+async function initializeJPEG() {
+  if (!jpegEncode) {
+    const { encode, decode } = await import('@jsquash/jpeg');
+    jpegEncode = encode;
+    jpegDecode = decode;
+  }
+}
+
+async function initializeWebP() {
+  if (!webpEncode) {
+    const { encode, decode } = await import('@jsquash/webp');
+    webpEncode = encode;
+    webpDecode = decode;
+  }
+}
+
+/**
+ * Optimize images using WASM-powered codecs
+ * This is FAST and has ZERO OS dependencies! 🚀
  */
 export async function optimizeImages(srcDir, outDir) {
-  const imageExtensions = ['.png', '.jpg', '.jpeg', '.webp', '.gif'];
+  const imageExtensions = ['.png', '.jpg', '.jpeg', '.webp', '.gif', '.svg'];
   let optimized = 0;
   let totalSaved = 0;
 
-  logger.info('🖼️  Optimizing images...');
+  logger.info('🖼️  Optimizing images with WASM codecs...');
 
   async function processDirectory(dir, targetDir) {
     const entries = readdirSync(dir, { withFileTypes: true });
@@ -35,10 +69,13 @@ export async function optimizeImages(srcDir, outDir) {
             if (result) {
               optimized++;
               totalSaved += result.saved;
-              logger.debug(`Optimized: ${entry.name} (saved ${(result.saved / 1024).toFixed(2)}KB)`);
+              const savedPercent = ((result.saved / result.originalSize) * 100).toFixed(1);
+              logger.debug(
+                `✨ ${entry.name}: ${(result.originalSize / 1024).toFixed(1)}KB → ${(result.newSize / 1024).toFixed(1)}KB (-${savedPercent}%)`
+              );
             }
           } catch (error) {
-            logger.warn(`Failed to optimize ${entry.name}: ${error.message}`);
+            logger.warn(`⚠️  Failed to optimize ${entry.name}: ${error.message}`);
             // Fallback: just copy the file
             cpSync(srcPath, destPath);
           }
@@ -50,15 +87,16 @@ export async function optimizeImages(srcDir, outDir) {
   await processDirectory(srcDir, outDir);
 
   if (optimized > 0) {
-    logger.success(`Optimized ${optimized} images (saved ${(totalSaved / 1024).toFixed(2)}KB)`);
+    logger.success(
+      `✅ Optimized ${optimized} images (saved ${(totalSaved / 1024).toFixed(2)}KB total)`
+    );
   }
 
   return { optimized, saved: totalSaved };
 }
 
 /**
- * Optimize a single image using Bun's native capabilities
- * Falls back to direct copy if optimization fails
+ * Optimize a single image using WASM codecs
  */
 async function optimizeImage(srcPath, destPath) {
   const ext = extname(srcPath).toLowerCase();
@@ -66,118 +104,80 @@ async function optimizeImage(srcPath, destPath) {
   const originalSize = originalFile.size;
 
   try {
-    // Read the image
-    const imageBuffer = await originalFile.arrayBuffer();
-
-    // For PNG/JPEG, we can optimize
-    if (ext === '.png' || ext === '.jpg' || ext === '.jpeg') {
-      // Use Bun's native image optimization
-      // This is fast because it uses native C libraries
-      const optimized = await optimizeWithBun(imageBuffer, ext);
-      
-      if (optimized && optimized.byteLength < originalSize) {
-        await Bun.write(destPath, optimized);
-        const saved = originalSize - optimized.byteLength;
-        return { saved, originalSize, newSize: optimized.byteLength };
-      }
+    // For SVG and GIF, just copy (no optimization needed/supported)
+    if (ext === '.svg' || ext === '.gif') {
+      cpSync(srcPath, destPath);
+      return null;
     }
 
-    // For other formats or if optimization didn't help, just copy
-    cpSync(srcPath, destPath);
-    return null;
-  } catch (error) {
-    // If anything fails, just copy the original
-    cpSync(srcPath, destPath);
-    return null;
-  }
-}
+    // Read the original image
+    const originalBuffer = await originalFile.arrayBuffer();
 
-/**
- * Optimize using Bun's native capabilities
- * This is a placeholder - Bun doesn't have built-in image optimization yet
- * We'll use a fast external library via Bun's FFI or shell commands
- */
-async function optimizeWithBun(buffer, ext) {
-  try {
-    // For now, we'll use oxipng and mozjpeg via shell commands
-    // These are the FASTEST available options (Rust-based)
-    const tempInput = `/tmp/bertui_input_${Date.now()}${ext}`;
-    const tempOutput = `/tmp/bertui_output_${Date.now()}${ext}`;
-
-    await Bun.write(tempInput, buffer);
+    let optimizedBuffer;
 
     if (ext === '.png') {
-      // Use oxipng (Rust-based, ultra-fast)
-      const proc = Bun.spawn(['oxipng', '-o', '2', '--strip', 'safe', tempInput, '-o', tempOutput], {
-        stdout: 'ignore',
-        stderr: 'ignore'
+      await initializePNG();
+      
+      // Decode → Re-encode with compression
+      const imageData = await pngDecode(originalBuffer);
+      
+      // Encode with oxipng-level compression (quality 85, similar to oxipng -o 2)
+      optimizedBuffer = await pngEncode(imageData, { 
+        quality: 85,
+        effort: 2 // 0-10, higher = better compression but slower
       });
-      await proc.exited;
-
-      if (existsSync(tempOutput)) {
-        const optimized = await Bun.file(tempOutput).arrayBuffer();
-        // Cleanup
-        Bun.spawn(['rm', tempInput, tempOutput]);
-        return optimized;
-      }
+      
     } else if (ext === '.jpg' || ext === '.jpeg') {
-      // Use mozjpeg (fastest JPEG optimizer)
-      const proc = Bun.spawn(['cjpeg', '-quality', '85', '-optimize', '-outfile', tempOutput, tempInput], {
-        stdout: 'ignore',
-        stderr: 'ignore'
-      });
-      await proc.exited;
-
-      if (existsSync(tempOutput)) {
-        const optimized = await Bun.file(tempOutput).arrayBuffer();
-        // Cleanup
-        Bun.spawn(['rm', tempInput, tempOutput]);
-        return optimized;
-      }
+      await initializeJPEG();
+      
+      // Decode → Re-encode with quality 85 (mozjpeg-like quality)
+      const imageData = await jpegDecode(originalBuffer);
+      optimizedBuffer = await jpegEncode(imageData, { quality: 85 });
+      
+    } else if (ext === '.webp') {
+      await initializeWebP();
+      
+      // WebP optimization
+      const imageData = await webpDecode(originalBuffer);
+      optimizedBuffer = await webpEncode(imageData, { quality: 85 });
     }
 
-    // Cleanup on failure
-    if (existsSync(tempInput)) Bun.spawn(['rm', tempInput]);
-    if (existsSync(tempOutput)) Bun.spawn(['rm', tempOutput]);
+    // Only save if we actually reduced the size
+    if (optimizedBuffer && optimizedBuffer.byteLength < originalSize) {
+      await Bun.write(destPath, optimizedBuffer);
+      const saved = originalSize - optimizedBuffer.byteLength;
+      return { saved, originalSize, newSize: optimizedBuffer.byteLength };
+    }
 
+    // If optimization didn't help, just copy the original
+    cpSync(srcPath, destPath);
     return null;
+    
   } catch (error) {
+    // If anything fails, just copy the original
+    logger.warn(`Optimization failed for ${srcPath.split('/').pop()}, copying original`);
+    cpSync(srcPath, destPath);
     return null;
   }
 }
 
 /**
- * Check if optimization tools are installed
+ * Check if optimization is available (always true with WASM! 🎉)
  */
 export async function checkOptimizationTools() {
-  const tools = [];
-
   try {
-    const oxipng = Bun.spawn(['which', 'oxipng'], { stdout: 'pipe' });
-    await oxipng.exited;
-    if (oxipng.exitCode === 0) {
-      tools.push('oxipng');
-    }
-  } catch (e) {}
-
-  try {
-    const cjpeg = Bun.spawn(['which', 'cjpeg'], { stdout: 'pipe' });
-    await cjpeg.exited;
-    if (cjpeg.exitCode === 0) {
-      tools.push('mozjpeg');
-    }
-  } catch (e) {}
-
-  if (tools.length === 0) {
-    logger.warn('⚠️  No image optimization tools found. Install for better performance:');
-    logger.warn('   macOS:   brew install oxipng mozjpeg');
-    logger.warn('   Ubuntu:  apt install oxipng mozjpeg');
-    logger.warn('   Images will be copied without optimization.');
-  } else {
-    logger.success(`Found optimization tools: ${tools.join(', ')}`);
+    // Try to import the WASM modules
+    await import('@jsquash/png');
+    await import('@jsquash/jpeg');
+    await import('@jsquash/webp');
+    
+    logger.success('✅ WASM image optimization available');
+    logger.info('📦 Using @jsquash (zero OS dependencies!)');
+    return ['png', 'jpeg', 'webp'];
+  } catch (error) {
+    logger.error('❌ WASM codecs not installed. Run: bun add @jsquash/png @jsquash/jpeg @jsquash/webp');
+    return [];
   }
-
-  return tools;
 }
 
 /**
@@ -198,7 +198,7 @@ export function copyImages(srcDir, outDir) {
         if (!existsSync(destPath)) {
           mkdirSync(destPath, { recursive: true });
         }
-        processDirectory(srcPath, destPath);
+        processDirectory(srcPath, targetDir);
       } else if (entry.isFile()) {
         const ext = extname(entry.name).toLowerCase();
 
@@ -211,6 +211,6 @@ export function copyImages(srcDir, outDir) {
   }
 
   processDirectory(srcDir, outDir);
-  logger.info(`Copied ${copied} images without optimization`);
+  logger.info(`📋 Copied ${copied} images without optimization`);
   return copied;
 }

package/src/build.js

@@ -1,8 +1,10 @@
+// src/build.js - FIXED VERSION
 import { join, relative, basename, extname, dirname } from 'path';
 import { existsSync, mkdirSync, rmSync, cpSync, readdirSync, statSync } from 'fs';
 import logger from './logger/logger.js';
 import { buildCSS } from './build/css-builder.js';
 import { loadEnvVariables, replaceEnvInCode } from './utils/env.js';
+import { optimizeImages, checkOptimizationTools, copyImages } from './build/image-optimizer.js';
 
 export async function buildProduction(options = {}) {
   const root = options.root || process.cwd();
@@ -38,11 +40,14 @@ export async function buildProduction(options = {}) {
     logger.info('Step 2: Building CSS with Lightning CSS...');
     await buildAllCSS(root, outDir);
     
-    // ✅ FIX: Copy all static assets from src/ and public/
-    logger.info('Step 3: Copying static assets...');
-    await copyAllStaticAssets(root, outDir);
+    logger.info('Step 3: Checking image optimization tools...');
+    const optimizationTools = await checkOptimizationTools();
     
-    logger.info('Step 4: Bundling JavaScript with Bun...');
+    logger.info('Step 4: Copying and optimizing static assets...');
+    // ✅ FIX 1: Copy images from BOTH src/images/ and public/
+    await copyAllStaticAssets(root, outDir, optimizationTools.length > 0);
+    
+    logger.info('Step 5: Bundling JavaScript with Bun...');
     const buildEntry = join(buildDir, 'main.js');
     
     if (!existsSync(buildEntry)) {
@@ -65,9 +70,6 @@ export async function buildProduction(options = {}) {
       external: ['react', 'react-dom', 'react-dom/client', 'react/jsx-runtime'],
       define: {
         'process.env.NODE_ENV': '"production"',
-        'process.env.PUBLIC_APP_NAME': JSON.stringify(envVars.PUBLIC_APP_NAME || 'BertUI App'),
-        'process.env.PUBLIC_API_URL': JSON.stringify(envVars.PUBLIC_API_URL || ''),
-        'process.env.PUBLIC_USERNAME': JSON.stringify(envVars.PUBLIC_USERNAME || ''),
         ...Object.fromEntries(
           Object.entries(envVars).map(([key, value]) => [
             `process.env.${key}`,
@@ -85,7 +87,8 @@ export async function buildProduction(options = {}) {
     
     logger.success('JavaScript bundled with tree-shaking');
     
-    logger.info('Step 5: Generating SEO-optimized HTML files...');
+    logger.info('Step 6: Generating SEO-optimized HTML files...');
+    // ✅ FIX 2: Generate HTML for ALL routes including index.html
     await generateProductionHTML(root, outDir, result, routes);
     
     rmSync(buildDir, { recursive: true });
@@ -121,76 +124,45 @@ export async function buildProduction(options = {}) {
   }
 }
 
-// ✅ NEW FUNCTION: Copy all static assets
-async function copyAllStaticAssets(root, outDir) {
+// ✅ FIX 3: Enhanced asset copying with proper directory structure
+async function copyAllStaticAssets(root, outDir, optimize = true) {
   const publicDir = join(root, 'public');
-  const srcDir = join(root, 'src');
+  const srcImagesDir = join(root, 'src', 'images');
   
   let assetsCopied = 0;
+  let assetsOptimized = 0;
   
-  // Copy from public/
-  if (existsSync(publicDir)) {
-    assetsCopied += await copyStaticAssetsFromDir(publicDir, outDir, 'public');
-  }
+  // Create images directory in dist/
+  const distImagesDir = join(outDir, 'images');
+  mkdirSync(distImagesDir, { recursive: true });
   
-  // Copy static assets from src/ (images, fonts, etc.)
-  if (existsSync(srcDir)) {
-    const assetsOutDir = join(outDir, 'assets');
-    mkdirSync(assetsOutDir, { recursive: true });
-    assetsCopied += await copyStaticAssetsFromDir(srcDir, assetsOutDir, 'src', true);
+  // Copy from public/ to root of dist/
+  if (existsSync(publicDir)) {
+    logger.info('Copying public/ assets...');
+    if (optimize) {
+      const result = await optimizeImages(publicDir, outDir);
+      assetsOptimized += result.optimized;
+    } else {
+      assetsCopied += copyImages(publicDir, outDir);
+    }
   }
   
-  logger.success(`Copied ${assetsCopied} static assets`);
-}
-
-// ✅ NEW FUNCTION: Recursively copy static assets
-async function copyStaticAssetsFromDir(sourceDir, targetDir, label, skipStyles = false) {
-  const staticExtensions = [
-    '.png', '.jpg', '.jpeg', '.gif', '.svg', '.webp', '.avif', // Images
-    '.woff', '.woff2', '.ttf', '.otf', '.eot', // Fonts
-    '.mp4', '.webm', '.ogg', '.mp3', '.wav', // Media
-    '.pdf', '.zip', '.json', '.xml', '.txt' // Documents
-  ];
-  
-  let copiedCount = 0;
-  
-  function copyRecursive(dir, targetBase) {
-    const entries = readdirSync(dir, { withFileTypes: true });
-    
-    for (const entry of entries) {
-      const srcPath = join(dir, entry.name);
-      const relativePath = relative(sourceDir, srcPath);
-      const destPath = join(targetBase, relativePath);
-      
-      if (entry.isDirectory()) {
-        // Skip node_modules, .bertui, etc.
-        if (entry.name === 'node_modules' || entry.name.startsWith('.')) {
-          continue;
-        }
-        
-        // Skip styles directory if requested
-        if (skipStyles && entry.name === 'styles') {
-          continue;
-        }
-        
-        mkdirSync(destPath, { recursive: true });
-        copyRecursive(srcPath, targetBase);
-      } else if (entry.isFile()) {
-        const ext = extname(entry.name);
-        
-        // Copy static assets only
-        if (staticExtensions.includes(ext.toLowerCase())) {
-          mkdirSync(dirname(destPath), { recursive: true });
-          cpSync(srcPath, destPath);
-          logger.debug(`Copied ${label}/${relativePath}`);
-          copiedCount++;
-        }
-      }
+  // ✅ FIX: Copy from src/images/ to dist/images/
+  if (existsSync(srcImagesDir)) {
+    logger.info('Copying src/images/ to dist/images/...');
+    if (optimize) {
+      const result = await optimizeImages(srcImagesDir, distImagesDir);
+      assetsOptimized += result.optimized;
+    } else {
+      assetsCopied += copyImages(srcImagesDir, distImagesDir);
     }
   }
   
-  copyRecursive(sourceDir, targetDir);
-  return copiedCount;
+  if (optimize && assetsOptimized > 0) {
+    logger.success(`🎨 Optimized ${assetsOptimized} images with WASM codecs`);
+  } else {
+    logger.success(`📋 Copied ${assetsCopied} static assets`);
+  }
 }
 
 async function buildAllCSS(root, outDir) {
@@ -447,7 +419,6 @@ async function compileBuildDirectory(srcDir, buildDir, root, envVars) {
         code = replaceEnvInCode(code, envVars);
         code = fixBuildImports(code, srcPath, outPath, root);
         
-        // ✅ FIX: Add React import if needed
         if (usesJSX(code) && !code.includes('import React')) {
           code = `import React from 'react';\n${code}`;
         }
@@ -486,7 +457,6 @@ async function compileBuildFile(srcPath, buildDir, filename, root, envVars) {
     
     let compiled = await transpiler.transform(code);
     
-    // ✅ FIX: Add React import if needed
     if (usesJSX(compiled) && !compiled.includes('import React')) {
       compiled = `import React from 'react';\n${compiled}`;
     }
@@ -582,6 +552,7 @@ function extractMetaFromSource(code) {
   }
 }
 
+// ✅ FIX 4: Generate proper HTML files with correct meta tags
 async function generateProductionHTML(root, outDir, buildResult, routes) {
   const mainBundle = buildResult.outputs.find(o => 
     o.path.includes('main') && o.kind === 'entry-point'
@@ -604,18 +575,15 @@ async function generateProductionHTML(root, outDir, buildResult, routes) {
     ).join('\n');
   }
   
+  // ✅ Load config for default meta
   const { loadConfig } = await import('./config/loadConfig.js');
   const config = await loadConfig(root);
   const defaultMeta = config.meta || {};
   
   logger.info('Generating SEO-optimized HTML files...');
   
+  // ✅ FIX: Generate HTML for ALL routes (including dynamic as fallback)
   for (const route of routes) {
-    if (route.type === 'dynamic') {
-      logger.info(`Skipping dynamic route: ${route.route}`);
-      continue;
-    }
-    
     const sourceCode = await Bun.file(route.path).text();
     const pageMeta = extractMetaFromSource(sourceCode);
     const meta = { ...defaultMeta, ...pageMeta };
@@ -624,7 +592,24 @@ async function generateProductionHTML(root, outDir, buildResult, routes) {
       logger.info(`Extracted meta for ${route.route}: ${JSON.stringify(pageMeta)}`);
     }
     
-    const html = `<!DOCTYPE html>
+    const html = generateHTML(meta, route, bundlePath, userStylesheets);
+    
+    let htmlPath;
+    if (route.route === '/') {
+      htmlPath = join(outDir, 'index.html');
+    } else {
+      const routeDir = join(outDir, route.route);
+      mkdirSync(routeDir, { recursive: true });
+      htmlPath = join(routeDir, 'index.html');
+    }
+    
+    await Bun.write(htmlPath, html);
+    logger.success(`Generated ${route.route === '/' ? 'index.html' : route.route + '/index.html'}`);
+  }
+}
+
+function generateHTML(meta, route, bundlePath, userStylesheets) {
+  return `<!DOCTYPE html>
 <html lang="${meta.lang || 'en'}">
 <head>
   <meta charset="UTF-8">
@@ -668,17 +653,4 @@ ${userStylesheets}
   <script type="module" src="/${bundlePath}"></script>
 </body>
 </html>`;
-    
-    let htmlPath;
-    if (route.route === '/') {
-      htmlPath = join(outDir, 'index.html');
-    } else {
-      const routeDir = join(outDir, route.route);
-      mkdirSync(routeDir, { recursive: true });
-      htmlPath = join(routeDir, 'index.html');
-    }
-    
-    await Bun.write(htmlPath, html);
-    logger.success(`Generated ${route.route === '/' ? 'index.html' : route.route + '/index.html'} with meta: ${meta.title || 'default'}`);
-  }
 }

package/src/server/dev-server.js

@@ -1,4 +1,4 @@
-// src/server/dev-server.js - FIXED VERSION
+// src/server/dev-server.js - FIXED IMAGE SERVING
 import { Elysia } from 'elysia';
 import { watch } from 'fs';
 import { join, extname } from 'path';
@@ -13,6 +13,7 @@ export async function startDevServer(options = {}) {
   const compiledDir = join(root, '.bertui', 'compiled');
   const stylesDir = join(root, '.bertui', 'styles');
   const srcDir = join(root, 'src');
+  const publicDir = join(root, 'public');
   
   const config = await loadConfig(root);
   
@@ -30,10 +31,10 @@ export async function startDevServer(options = {}) {
       return serveHTML(root, hasRouter, config);
     })
     
-    // ✅ NEW: Serve images from src/images/
+    // ✅ FIX: Serve images from src/images/ (CRITICAL)
     .get('/images/*', async ({ params, set }) => {
-      const imagesDir = join(srcDir, 'images');
-      const filepath = join(imagesDir, params['*']);
+      const srcImagesDir = join(srcDir, 'images');
+      const filepath = join(srcImagesDir, params['*']);
       const file = Bun.file(filepath);
       
       if (!await file.exists()) {
@@ -47,12 +48,27 @@ export async function startDevServer(options = {}) {
       return new Response(file, {
         headers: { 
           'Content-Type': contentType,
-          'Cache-Control': 'public, max-age=31536000'
+          'Cache-Control': 'no-cache' // Dev server = no cache
         }
       });
     })
     
-    // ✅ NEW: Serve any static asset from src/ (fonts, videos, etc.)
+    // ✅ Serve from public/ directory
+    .get('/public/*', async ({ params, set }) => {
+      const filepath = join(publicDir, params['*']);
+      const file = Bun.file(filepath);
+      
+      if (!await file.exists()) {
+        set.status = 404;
+        return 'File not found';
+      }
+      
+      return new Response(file, {
+        headers: { 'Cache-Control': 'no-cache' }
+      });
+    })
+    
+    // ✅ Generic asset serving
     .get('/assets/*', async ({ params, set }) => {
       const filepath = join(srcDir, params['*']);
       const file = Bun.file(filepath);
@@ -68,7 +84,7 @@ export async function startDevServer(options = {}) {
       return new Response(file, {
         headers: { 
           'Content-Type': contentType,
-          'Cache-Control': 'public, max-age=31536000'
+          'Cache-Control': 'no-cache'
         }
       });
     })
@@ -108,16 +124,6 @@ export async function startDevServer(options = {}) {
           }
         }
         
-        if (path.startsWith('public/')) {
-          const publicDir = join(root, 'public');
-          const filepath = join(publicDir, path.replace('public/', ''));
-          const file = Bun.file(filepath);
-          
-          if (await file.exists()) {
-            return new Response(file);
-          }
-        }
-        
         set.status = 404;
         return 'File not found';
       }
@@ -436,19 +442,6 @@ ws.onclose = () => {
       });
     })
     
-    .get('/public/*', async ({ params, set }) => {
-      const publicDir = join(root, 'public');
-      const filepath = join(publicDir, params['*']);
-      const file = Bun.file(filepath);
-      
-      if (!await file.exists()) {
-        set.status = 404;
-        return 'File not found';
-      }
-      
-      return new Response(file);
-    })
-    
     .listen(port);
   
   if (!app.server) {
@@ -458,7 +451,8 @@ ws.onclose = () => {
   
   logger.success(`🚀 Server running at http://localhost:${port}`);
   logger.info(`📁 Serving: ${root}`);
-  logger.info(`🖼️  Images available at: /images/*`);
+  logger.info(`🖼️  Images: /images/* → src/images/`);
+  logger.info(`📦 Public: /public/* → public/`);
   
   setupWatcher(root, compiledDir, clients, async () => {
     hasRouter = existsSync(join(compiledDir, 'router.js'));
@@ -536,7 +530,6 @@ ${userStylesheets}
   });
 }
 
-// ✅ NEW: Helper for image content types
 function getImageContentType(ext) {
   const types = {
     '.jpg': 'image/jpeg',
@@ -594,14 +587,12 @@ function setupWatcher(root, compiledDir, clients, onRecompile) {
     if (!filename) return;
     
     const ext = extname(filename);
-    
-    // ✅ Watch image changes too
     const watchedExtensions = ['.js', '.jsx', '.ts', '.tsx', '.css', '.png', '.jpg', '.jpeg', '.gif', '.svg', '.webp', '.avif'];
     
     if (watchedExtensions.includes(ext)) {
       logger.info(`📝 File changed: ${filename}`);
       
-      // For images, just reload without recompiling
+      // For images, just reload
       if (['.png', '.jpg', '.jpeg', '.gif', '.svg', '.webp', '.avif'].includes(ext)) {
         for (const client of clients) {
           try {