d1s 0.2.5 → 0.2.8

package/src/app/api/pull-remote/confirm/route.js

@@ -12,6 +12,7 @@ import {
 } from '@/lib/sql-operations';
 import { getCurrentSchema, analyzeDependencies } from '@/lib/schema-operations';
 import { initMigrationsTable, recordMigration, getRemoteMigrations } from '@/lib/migrations.js';
+import logger from '@/lib/logger';
 
 /**
  * Applies the pull migrations by:
@@ -52,33 +53,33 @@ async function applyPullMigrations(projectDir, accountId, apiToken, databaseName
     await Promise.all(stagedFilesList.map(file => unlink(path.join(stagingDir, file))));
 
     // --- STEP 1: Fetch remote migration metadata ---
-    // console.log('Fetching remote migration metadata...');
+    logger.debug('Fetching remote migration metadata...');
     const remoteMigrations = await getRemoteMigrations(projectDir, databaseName, env);
-    // console.log('Remote migration metadata:', remoteMigrations);
+    logger.debug('Remote migration metadata:', remoteMigrations);
 
     // --- STEP 2: For each remote migration, fetch its file content ---
-    // console.log('Fetching migration file content from remote...');
+    logger.debug('Fetching migration file content from remote...');
     const stagedFiles = new Map();
     for (const migration of remoteMigrations) {
       try {
         const migrationFileName = migration.name; // Use the remote migration's name.
         const content = await getMigrationFileContent(projectDir, databaseName, migrationFileName, env);
         stagedFiles.set(migrationFileName, content);
-        // console.log(`Staged migration file: ${migrationFileName}`);
+        logger.debug(`Staged migration file: ${migrationFileName}`);
         await writeFile(path.join(stagingDir, migrationFileName), content, 'utf8');
       } catch (fileError) {
-        console.error(`Error fetching content for migration ${migration.name}:`, fileError);
+        logger.error(`Error fetching content for migration ${migration.name}:`, fileError);
         throw new Error(`Failed to fetch migration file content for ${migration.name}`);
       }
     }
 
     // --- STEP 3: Fetch local schema for destructive update ---
-    // console.log('Fetching local schema for destructive update...');
+    logger.debug('Fetching local schema for destructive update...');
     const currentSchema = await getCurrentSchema(projectDir, databaseName, env, false);
     const tableSchema = currentSchema.filter(obj =>
       obj.type === 'table' && obj.name && obj.name.toLowerCase() !== 'if'
     );
-    // console.log('Local table schema:', tableSchema);
+    logger.debug('Local table schema:', tableSchema);
     const dropOrder = tableSchema.length > 0
       ? analyzeDependencies(
           tableSchema.map(obj => ({
@@ -88,18 +89,18 @@ async function applyPullMigrations(projectDir, accountId, apiToken, databaseName
           }))
         )
       : [];
-    // console.log('Computed drop order:', dropOrder);
+    logger.debug('Computed drop order:', dropOrder);
 
     // --- STEP 4: Destructive update (all-or-nothing) ---
-    // console.log('Beginning destructive update...');
+    logger.debug('Beginning destructive update...');
     try {
-      // console.log('Disabling foreign keys...');
+      logger.debug('Disabling foreign keys...');
       await executeSQL(projectDir, databaseName, 'PRAGMA foreign_keys=OFF;', env, false);
       
       // Drop triggers.
       const triggers = currentSchema.filter(obj => obj.type === 'trigger');
       for (const trigger of triggers) {
-        // console.log(`Dropping trigger: ${trigger.name}`);
+        logger.debug(`Dropping trigger: ${trigger.name}`);
         await executeSQL(
           projectDir,
           databaseName,
@@ -111,7 +112,7 @@ async function applyPullMigrations(projectDir, accountId, apiToken, databaseName
       // Drop indexes.
       const indexes = currentSchema.filter(obj => obj.type === 'index');
       for (const index of indexes) {
-        // console.log(`Dropping index: ${index.name}`);
+        logger.debug(`Dropping index: ${index.name}`);
         await executeSQL(
           projectDir,
           databaseName,
@@ -122,7 +123,7 @@ async function applyPullMigrations(projectDir, accountId, apiToken, databaseName
       }
       // Drop tables in reverse dependency order.
       for (const tableName of dropOrder.reverse()) {
-        // console.log(`Dropping table: ${tableName}`);
+        logger.debug(`Dropping table: ${tableName}`);
         await executeSQL(
           projectDir,
           databaseName,
@@ -131,7 +132,7 @@ async function applyPullMigrations(projectDir, accountId, apiToken, databaseName
           false
         );
       }
-      // console.log('Re-enabling foreign keys...');
+      logger.debug('Re-enabling foreign keys...');
       await executeSQL(projectDir, databaseName, 'PRAGMA foreign_keys=ON;', env, false);
     } catch (dropError) {
       throw new Error(`Error dropping existing objects: ${dropError.message}`);
@@ -139,7 +140,7 @@ async function applyPullMigrations(projectDir, accountId, apiToken, databaseName
 
     // --- STEP 5: Reset the local migrations table ---
     try {
-      // console.log('Resetting local migrations table...');
+      logger.debug('Resetting local migrations table...');
       await initMigrationsTable(projectDir, databaseName, 'local', true);
     } catch (initError) {
       throw new Error(`Error resetting migrations table: ${initError.message}`);
@@ -152,7 +153,7 @@ async function applyPullMigrations(projectDir, accountId, apiToken, databaseName
     try {
       for (const [filename, content] of stagedFiles.entries()) {
         currentFilename = filename;
-        // console.log(`Applying migration: ${filename}`);
+        logger.debug(`Applying migration: ${filename}`);
         // Write the migration file to the main migrations folder.
         await writeFile(path.join(migrationsDir, filename), content, 'utf8');
 
@@ -162,7 +163,7 @@ async function applyPullMigrations(projectDir, accountId, apiToken, databaseName
           .update(content)
           .digest('hex');
 
-        // console.log(`Recording migration: ${filename} with checksum ${checksum}`);
+        logger.debug(`Recording migration: ${filename} with checksum ${checksum}`);
         await recordMigration(
           projectDir,
           databaseName,
@@ -181,26 +182,26 @@ async function applyPullMigrations(projectDir, accountId, apiToken, databaseName
     }
 
     // --- STEP 7: Replace main migrations folder with staged files ---
-    // console.log('Replacing main migrations folder with staged files...');
+    logger.debug('Replacing main migrations folder with staged files...');
     const mainFiles = await readdir(migrationsDir).catch(() => []);
     if (mainFiles.length) {
       for (const file of mainFiles) {
-        // console.log(`Deleting local migration file: ${file}`);
+        logger.debug(`Deleting local migration file: ${file}`);
         await unlink(path.join(migrationsDir, file));
       }
     }
     for (const [filename, content] of stagedFiles.entries()) {
-      // console.log(`Copying staged file ${filename} to main migrations folder`);
+      logger.debug(`Copying staged file ${filename} to main migrations folder`);
       await writeFile(path.join(migrationsDir, filename), content, 'utf8');
     }
 
-    // console.log('Pull migration applied successfully.');
+    logger.debug('Pull migration applied successfully.');
     return {
       appliedMigrations: Array.from(stagedFiles.keys()),
       details: appliedPreview
     };
   } catch (error) {
-    console.error('Pull migration error:', error);
+    logger.error('Pull migration error:', error);
     throw error;
   }
 }
@@ -251,7 +252,7 @@ export async function POST(req) {
       details: result
     });
   } catch (error) {
-    console.error('Pull remote confirm error:', error);
+    logger.error('Pull remote confirm error:', error);
     return NextResponse.json({
       success: false,
       error: {

package/src/app/api/pull-remote/preview/route.js

@@ -2,9 +2,10 @@
 
 import { NextResponse } from 'next/server';
 import { findWranglerConfig } from '@/lib/find-wrangler-config.js';
-import { ErrorTypes,} from '@/lib/sql-operations';
+import { ErrorTypes } from '@/lib/sql-operations';
 import { getCurrentSchema, analyzeDependencies } from '@/lib/schema-operations';
 import { getRemoteMigrations } from '@/lib/migrations.js';
+import logger from '@/lib/logger';
 
 /**
  * Generates a preview for pulling remote migrations.
@@ -64,7 +65,7 @@ async function generatePullPreview(projectDir, accountId, apiToken, databaseName
 
     return preview;
   } catch (error) {
-    console.error('Pull preview generation error:', error);
+    logger.error('Pull preview generation error:', error);
     throw error;
   }
 }
@@ -123,7 +124,7 @@ export async function POST(req) {
       details: preview
     });
   } catch (error) {
-    console.error('Pull remote preview error:', error);
+    logger.error('Pull remote preview error:', error);
     return NextResponse.json({
       success: false,
       error: {

package/src/app/api/push-remote/preview/route.js

@@ -6,6 +6,7 @@ import path from 'path';
 import { findWranglerConfig } from '@/lib/find-wrangler-config.js';
 import { executeSQL, ErrorTypes } from '@/lib/sql-operations';
 import { validateMigrationFile } from '@/lib/migrations.js';
+import logger from '@/lib/logger';
 
 /**
  * Pushes migration preview to remote.
@@ -100,7 +101,7 @@ async function pushToRemote(projectDir, accountId, apiToken, databaseName, previ
       }
     }
 
-    // console.log("Push Preview Result (server):", results);
+    logger.debug("Push Preview Result (server):", results);
     return { success: true, preview: true, applied: results.applied.length, skipped: results.skipped.length, details: results };
   } catch (error) {
     throw {
@@ -157,7 +158,7 @@ export async function POST(req) {
       details: result
     });
   } catch (error) {
-    console.error('Push to remote preview error:', error);
+    logger.error('Push to remote preview error:', error);
     return NextResponse.json({
       success: false,
       error: {

package/src/app/api/query/route.js

@@ -11,15 +11,19 @@ import {
 import { handleSchemaChange } from '@/lib/migrations';
 import { isSchemaChange } from '@/lib/sql-operations';
 import { parseSQLError } from '@/lib/parse-sql-error';
+import logger from '@/lib/logger';
 
 /**
  * API POST handler to execute SQL queries against the configured D1 database.
  */
 export async function POST(req) {
+  logger.logRequest('POST', '/api/query');
+
   try {
     // Ensure that the project directory is set.
     const projectDir = process.env.D1_PROJECT_DIR;
     if (!projectDir) {
+      logger.error('Project directory not configured');
       return NextResponse.json(
         { error: 'Project directory not configured' },
         { status: 500 }
@@ -28,7 +32,10 @@ export async function POST(req) {
 
     // Extract the SQL query from the request body.
     const { query } = await req.json();
-    // console.log('Received query:', query);
+
+    if (logger.shouldLog('debug')) {
+      logger.debug('Received query:', query);
+    }
 
     if (!query?.trim()) {
       return NextResponse.json(
@@ -40,6 +47,7 @@ export async function POST(req) {
     // Retrieve the Wrangler configuration using the centralized helper.
     const wranglerResult = await findWranglerConfig(projectDir);
     if (!wranglerResult || !wranglerResult.config) {
+      logger.error('No Wrangler configuration file found');
       return NextResponse.json(
         { success: false, error: { message: 'No Wrangler configuration file found' } },
         { status: 404 }
@@ -94,7 +102,7 @@ export async function POST(req) {
       // Process schema changes and capture migrations.
       // The handleSchemaChange helper is responsible for executing the DDL statements.
       const migrations = await handleSchemaChange(projectDir, databaseName, statements);
-      console.log('Schema changes processed:', migrations);
+      // console.log('Schema changes processed:', migrations);
       
       result = {
         success: true,
@@ -127,7 +135,7 @@ export async function POST(req) {
     return NextResponse.json(result);
 
   } catch (error) {
-    console.error('Query execution error:', error);
+    logger.logError(error, 'query-route');
     const parsedError = parseSQLError(error);
     
     const statusCode = 

package/src/app/api/tables/[name]/route.js

@@ -5,6 +5,7 @@ import { promisify } from 'util';
 import { exec } from 'child_process';
 import { findWranglerConfig } from '@/lib/find-wrangler-config.js';
 import { dbQueue } from '@/lib/database-queue';
+import logger from '@/lib/logger';
 
 const execAsync = promisify(exec);
 
@@ -35,7 +36,7 @@ async function executeD1Query(command, projectDir, databaseName) {
     const result = JSON.parse(stdout);
     return result[0]?.results || [];
   } catch (error) {
-    console.error('Query execution error:', error);
+    logger.error('Query execution error:', error);
     throw error;
   } finally {
     isProcessing = false;
@@ -145,7 +146,7 @@ export async function GET(request, context) {
       });
 
     } catch (error) {
-      console.error('Error handling request:', error);
+      logger.error('Error handling request:', error);
       return NextResponse.json({
         success: false,
         error: {

package/src/app/api/tables/route.js

@@ -4,6 +4,7 @@ import { NextResponse } from 'next/server';
 import { promisify } from 'util';
 import { exec } from 'child_process';
 import { findWranglerConfig } from '@/lib/find-wrangler-config.js';
+import logger from '@/lib/logger';
 
 const execAsync = promisify(exec);
 
@@ -75,11 +76,13 @@ async function retryWithTimeout(fn, retries = 3, timeout = 1000) {
  * It uses the Wrangler configuration (via findWranglerConfig) to determine the D1 database.
  */
 export async function GET() {
+  logger.logRequest('GET', '/api/tables');
+
   try {
     const projectDir = process.env.D1_PROJECT_DIR;
 
     if (!projectDir) {
-      console.error('D1_PROJECT_DIR environment variable is not set');
+      logger.error('D1_PROJECT_DIR environment variable is not set');
       return NextResponse.json(
         { error: 'Project directory not configured' },
         { status: 500 }
@@ -89,6 +92,7 @@ export async function GET() {
     // Use the centralized helper to get Wrangler configuration
     const wranglerResult = await findWranglerConfig(projectDir);
     if (!wranglerResult || !wranglerResult.config) {
+      logger.error('No Wrangler configuration file found');
       return NextResponse.json(
         { error: 'No Wrangler configuration file found in project directory' },
         { status: 404 }
@@ -138,7 +142,7 @@ export async function GET() {
     });
 
   } catch (error) {
-    console.error('Error fetching tables:', error);
+    logger.logError(error, 'tables-route');
     
     // Enhanced error response for troubleshooting
     const errorResponse = {

package/src/app/api/update/route.js

@@ -0,0 +1,11 @@
+// app/api/update/route.js
+import { NextResponse } from 'next/server';
+import updateNotifier from 'update-notifier';
+import pkg from '../../../../package.json' assert { type: 'json' };
+
+export async function GET() {
+  // Check for updates (you might want to cache this result to avoid too many requests)
+  const notifier = updateNotifier({ pkg, updateCheckInterval: 1000 * 60 * 60 * 24 });
+  const update = notifier.update;
+  return NextResponse.json({ update });
+}

package/src/app/page.jsx

@@ -1,10 +1,12 @@
 // app/page.js
 'use client'
 
+import { useState } from 'react';
 import { AppSidebar } from '../components/sidebar';
 import { SQLEditor } from '../components/sql-editor';
 import { TableView } from '../components/table-view';
-import { useState } from 'react';
+import UpdateBanner from '@/components/update-banner';
+
 
 export default function StudioPage() {
   const [currentView, setCurrentView] = useState('sql-editor');
@@ -17,12 +19,16 @@ export default function StudioPage() {
 
   return (
     <div className="min-h-screen bg-background">
+      {/* notification banner at the top of the page */}
+      <UpdateBanner />
+      {/* sidebar */}
       <AppSidebar 
         className="w-64 border-r" 
         currentView={currentView}
         onNavigate={setCurrentView}
         onTableSelect={handleTableSelect}
       />
+      {/* main content */}
       <main className="ml-64 h-screen">
         {currentView === 'sql-editor' && <SQLEditor />}
         {currentView === 'table-view' && selectedTable && (

package/src/components/update-banner.jsx

@@ -0,0 +1,64 @@
+// components/update-banner.jsx
+
+'use client'
+
+import { useEffect, useState } from 'react';
+import axios from 'axios';
+
+export default function UpdateBanner() {
+  const [updateInfo, setUpdateInfo] = useState(null);
+  const [visible, setVisible] = useState(true);
+
+  // On component mount, check if the banner was previously dismissed
+  useEffect(() => {
+    const isDismissed = localStorage.getItem('updateBannerDismissed');
+    if (isDismissed === 'true') {
+      setVisible(false);
+    }
+  }, []);
+
+  // Fetch update info from /api/update
+  useEffect(() => {
+    axios
+      .get('/api/update')
+      .then(({ data }) => {
+        // If there's an update object returned
+        if (data?.update) {
+          setUpdateInfo(data.update);
+          // If user has not dismissed the banner, ensure it's visible
+          // so they see the new version if it differs from the old one
+          setVisible(true);
+        }
+      })
+      .catch(console.error);
+  }, []);
+
+  // If no update or the banner was dismissed, don't render
+  if (!updateInfo || !visible) return null;
+
+  // Close handler that writes to localStorage
+  const handleClose = () => {
+    localStorage.setItem('updateBannerDismissed', 'true');
+    setVisible(false);
+  };
+
+  return (
+    <div className="bg-gray-200 w-full px-3 py-3 flex items-center justify-between">
+      <div className="text-gray-800">
+        <strong>Update available!</strong> You are on version {updateInfo.current}; 
+        version {updateInfo.latest} is available. Run{' '}
+        <code className="bg-white text-sm px-1 py-0.5 rounded">
+          npm install -g d1s@latest
+        </code>{' '}
+        to update.
+      </div>
+      <button
+        onClick={handleClose}
+        className="ml-4 text-gray-500 hover:text-gray-700"
+        aria-label="Close update banner"
+      >
+        &times;
+      </button>
+    </div>
+  );
+}

package/src/lib/database-queue.js

@@ -1,33 +1,37 @@
 // utils/database-queue.js
+
+import logger from '@/lib/logger.js';
+
 class DatabaseQueue {
-   constructor() {
-     this.queue = [];
-     this.isProcessing = false;
-   }
+  constructor() {
+    this.queue = [];
+    this.isProcessing = false;
+  }
+
+  async add(operation) {
+    return new Promise((resolve, reject) => {
+      this.queue.push({ operation, resolve, reject });
+      this.processNext();
+    });
+  }
+
+  async processNext() {
+    if (this.isProcessing || this.queue.length === 0) return;
+    
+    this.isProcessing = true;
+    const { operation, resolve, reject } = this.queue.shift();
+    
+    try {
+      const result = await operation();
+      resolve(result);
+    } catch (error) {
+      logger.error('Error processing database queue operation:', error);
+      reject(error);
+    } finally {
+      this.isProcessing = false;
+      this.processNext();
+    }
+  }
+}
  
-   async add(operation) {
-     return new Promise((resolve, reject) => {
-       this.queue.push({ operation, resolve, reject });
-       this.processNext();
-     });
-   }
- 
-   async processNext() {
-     if (this.isProcessing || this.queue.length === 0) return;
-     
-     this.isProcessing = true;
-     const { operation, resolve, reject } = this.queue.shift();
-     
-     try {
-       const result = await operation();
-       resolve(result);
-     } catch (error) {
-       reject(error);
-     } finally {
-       this.isProcessing = false;
-       this.processNext();
-     }
-   }
- }
- 
- export const dbQueue = new DatabaseQueue();
+export const dbQueue = new DatabaseQueue();

package/src/lib/find-wrangler-config.js

@@ -1,12 +1,15 @@
-// lib/find-wrangler-config.js
 import { readFile } from 'fs/promises';
 import { findUp } from 'find-up';
 import { parse as parseToml } from '@iarna/toml';
 import stripJsonComments from 'strip-json-comments';
 import logger from './logger.js';
 
+let cachedConfig = null;
+let cachedProjectPath = null;
+
 /**
  * Finds and parses the Wrangler configuration file.
+ * This function caches the result so that the configuration is read and logged only once per project path.
  * 
  * @param {string} userProjectPath - The path where to look for the config file
  * @returns {Promise<{config: object, projectPath: string}|null>}
@@ -17,8 +20,10 @@ export async function findWranglerConfig(userProjectPath) {
       throw new Error('Project path is required and must be a string');
     }
 
-    // Uncomment the following line to log the search path if needed:
-    // logger.debug(`Looking for wrangler configuration in: ${userProjectPath}`);
+    // Return the cached config if available for the same project path.
+    if (cachedConfig && cachedProjectPath === userProjectPath) {
+      return cachedConfig;
+    }
 
     // Define the possible configuration file names in priority order.
     const possibleFiles = ['wrangler.toml', 'wrangler.json', 'wrangler.jsonc'];
@@ -56,13 +61,14 @@ export async function findWranglerConfig(userProjectPath) {
       config = JSON.parse(configContent);
     }
 
-    return {
-      config,
-      projectPath: userProjectPath
-    };
+    // Cache the result.
+    cachedConfig = { config, projectPath: userProjectPath };
+    cachedProjectPath = userProjectPath;
+
+    return cachedConfig;
 
   } catch (error) {
     logger.error(`Error reading Wrangler configuration: ${error.message}`);
     throw error;
   }
-}
+}

package/src/lib/logger.js

@@ -2,18 +2,27 @@ import winston from 'winston';
 
 const { combine, timestamp, printf, colorize } = winston.format;
 
+const isDevelopment = process.env.NODE_ENV === 'development' || process.env.NODE_ENV === undefined;
+const isProduction = process.env.NODE_ENV === 'production';
+const isDebug = process.env.DEBUG === 'true';
+
 const logFormat = printf(({ level, message, timestamp }) => {
   return `${timestamp} ${level}: ${message}`;
 });
 
 const logger = winston.createLogger({
-  level: process.env.NODE_ENV === 'development' ? 'debug' : 'info',
+  level: isProduction ? 'warn' : (isDebug ? 'debug' : 'info'),
   format: combine(
     timestamp({ format: 'YYYY-MM-DD HH:mm:ss' }),
     colorize(),
     logFormat
   ),
-  transports: [new winston.transports.Console()]
+  transports: [
+    new winston.transports.Console({
+      level: isProduction ? 'warn' : (isDebug ? 'debug' : 'info'),
+      silent: process.env.NODE_ENV === 'test'
+    })
+  ]
 });
 
 export default logger;

package/src/lib/migrations.js

@@ -16,6 +16,7 @@ import {
   tableExists
 } from './sql-operations.js';
 import { analyzeDependencies } from '@/lib/schema-operations';
+import logger from './logger.js';
 
 const execAsync = promisify(exec);
 
@@ -67,7 +68,7 @@ export async function initMigrationsTable(projectDir, databaseName, environment
       );
     }
   } catch (error) {
-    console.error(`Failed to initialize migrations table in ${environment}:`, error);
+    logger.error(`Failed to initialize migrations table in ${environment}:`, error);
     throw error;
   }
 }
@@ -173,7 +174,6 @@ export async function getRemoteMigrations(projectDir, databaseName, env) {
  * @param {Object} migrationData { name, checksum }
  * @param {('local'|'remote')} environment 
  */
-// Update tracking function to handle both cases
 export async function trackMigration(projectDir, databaseName, { name, checksum }, environment = 'local') {
   const flag = environment === 'remote' ? '--remote' : '--local';
   
@@ -215,7 +215,7 @@ export async function trackMigration(projectDir, databaseName, { name, checksum
       { cwd: projectDir }
     );
   } catch (error) {
-    console.error('Failed to track migration:', error);
+    logger.error('Failed to track migration:', error);
     throw error;
   }
 }
@@ -273,7 +273,7 @@ export async function compareMigrations(projectDir, databaseName) {
       count: migrationMap.size
     };
   } catch (error) {
-    console.error('Failed to compare migrations:', error);
+    logger.error('Failed to compare migrations:', error);
     throw error;
   }
 }

package/src/lib/schema-operations.js

@@ -1,5 +1,7 @@
 // schema-operations.js
+
 import { executeSQL, ErrorTypes } from './sql-operations.js';
+import logger from './logger.js';
 
 /**
  * Retrieves the current database schema (tables, indexes, triggers),
@@ -24,6 +26,7 @@ export async function getCurrentSchema(projectDir, databaseName, env = {}, remot
     const parsed = JSON.parse(result);
     return parsed[0]?.results || [];
   } catch (error) {
+    logger.error('Failed to get current schema:', error);
     throw {
       type: ErrorTypes.EXECUTION_ERROR,
       message: 'Failed to get current schema',
@@ -100,4 +103,4 @@ export function orderTables(dependencies) {
     }
   }
   return ordered;
-}
+}