google-drive-mock 1.0.9 → 1.0.11

package/AGENTS.md

@@ -10,5 +10,6 @@ After making changes to the codebase or to the tests, always run the following c
 
 ## Remember
 
-Whenever the `npm test:real` fails, fix the tests to match the "real" backend.
-Ensure that the behavior of the mock is exactly equal to the "real" backend.
+- Whenever the `npm test:real` fails, fix the tests to match the "real" backend.
+- Ensure that the behavior of the mock is exactly equal to the "real" backend.
+- The header field "If-None-Match" does not work in google drive. Do never use it or assume it works.

package/dist/mappers.js

@@ -30,6 +30,8 @@ function toV2File(file) {
             isRoot: false // Mock simplification
         })),
         version: file.version,
+        fileSize: file.size,
+        md5Checksum: file.md5Checksum,
         downloadUrl: `http://localhost/drive/v2/files/${file.id}?alt=media`
     };
 }

package/dist/routes/v2.js

@@ -331,8 +331,19 @@ const createV2Router = (config) => {
     // V2 Upload (POST)
     app.post('/upload/drive/v2/files', (req, res) => {
         const uploadType = req.query.uploadType;
+        // V2 behavior: If-None-Match on POST seems to check against the collection or just strictly fail if an entity exists contextually.
+        // Tests show it returns 412 Precondition Failed when '*' is used.
+        const ifNoneMatch = req.headers['if-none-match'];
+        if (ifNoneMatch) {
+            res.status(412).json({ error: { code: 412, message: "Precondition Failed" } });
+            return;
+        }
         if (uploadType === 'media') {
-            const rawBody = req.body;
+            let rawBody = req.body;
+            // Handle edge case where express.json() parses empty body as {}
+            if (req.headers['content-length'] === '0' && JSON.stringify(rawBody) === '{}') {
+                rawBody = '';
+            }
             // For simple upload, metadata is default
             const name = "Untitled";
             const newFile = store_1.driveStore.createFile({
@@ -401,7 +412,11 @@ const createV2Router = (config) => {
         }
         const uploadType = req.query.uploadType;
         if (uploadType === 'media') {
-            const rawBody = req.body;
+            let rawBody = req.body;
+            // Handle edge case where express.json() parses empty body as {}
+            if (req.headers['content-length'] === '0' && JSON.stringify(rawBody) === '{}') {
+                rawBody = '';
+            }
             const updatedFile = store_1.driveStore.updateFile(fileId, {
                 content: rawBody,
                 modifiedTime: new Date().toISOString()

package/dist/routes/v3.js

@@ -20,54 +20,67 @@ const createV3Router = () => {
         const orderBy = req.query.orderBy;
         if (q) {
             // Enhanced query parser for Mock
-            const parts = q.split(' and ').map(p => p.trim());
+            const orParts = q.split(' or ');
             files = files.filter(file => {
-                return parts.every(part => {
-                    var _a, _b, _c, _d, _e, _f, _g, _h;
-                    // name = '...'
-                    if (part.startsWith("name = '")) {
-                        const name = (_a = part.match(/name = '(.*)'/)) === null || _a === void 0 ? void 0 : _a[1];
-                        return file.name === name;
-                    }
-                    // name contains '...'
-                    if (part.startsWith("name contains '")) {
-                        const token = (_b = part.match(/name contains '(.*)'/)) === null || _b === void 0 ? void 0 : _b[1];
-                        return token && file.name.includes(token);
-                    }
-                    // 'ID' in parents
-                    if (part.includes(" in parents")) {
-                        const parentId = (_c = part.match(/'(.*)' in parents/)) === null || _c === void 0 ? void 0 : _c[1];
-                        return parentId && ((_d = file.parents) === null || _d === void 0 ? void 0 : _d.includes(parentId));
-                    }
-                    // trashed = ...
-                    if (part === "trashed = false") {
-                        return file.trashed !== true;
-                    }
-                    if (part === "trashed = true") {
-                        return file.trashed === true;
-                    }
-                    // mimeType = '...'
-                    if (part.startsWith("mimeType = '")) {
-                        const mime = (_e = part.match(/mimeType = '(.*)'/)) === null || _e === void 0 ? void 0 : _e[1];
-                        return file.mimeType === mime;
-                    }
-                    // mimeType != '...'
-                    if (part.startsWith("mimeType != '")) {
-                        const mime = (_f = part.match(/mimeType != '(.*)'/)) === null || _f === void 0 ? void 0 : _f[1];
-                        return file.mimeType !== mime;
-                    }
-                    // modifiedTime > '...'
-                    if (part.startsWith("modifiedTime > '")) {
-                        const timeStr = (_g = part.match(/modifiedTime > '(.*)'/)) === null || _g === void 0 ? void 0 : _g[1];
-                        return timeStr && new Date(file.modifiedTime) > new Date(timeStr);
-                    }
-                    // modifiedTime < '...'
-                    if (part.startsWith("modifiedTime < '")) {
-                        const timeStr = (_h = part.match(/modifiedTime < '(.*)'/)) === null || _h === void 0 ? void 0 : _h[1];
-                        return timeStr && new Date(file.modifiedTime) < new Date(timeStr);
-                    }
-                    // Ignore unknown filters for now
-                    return true;
+                return orParts.some(orPart => {
+                    const andParts = orPart.split(' and ').map(p => p.trim());
+                    return andParts.every(part => {
+                        var _a, _b, _c, _d, _e, _f, _g, _h;
+                        // name = '...'
+                        if (part.startsWith("name = '")) {
+                            const name = (_a = part.match(/name = '(.*)'/)) === null || _a === void 0 ? void 0 : _a[1];
+                            // Handle escaped quotes if simple match fails, but simple regex here might be enough for now
+                            // The user does name.replace("'", "\\'") but the regex (.*) is greedy and might consume escaped quotes correctly-ish
+                            // but we are stripping the outer quotes.
+                            // If user sends: name = 'foo\'bar', regex captures: foo\'bar
+                            // We need to unescape? The user sends escaped string for the Query Parser.
+                            // Drive API expects the string inside quotes to be the value.
+                            // If `name = 'a\'b'`, the value is `a'b`.
+                            // JSON.parse(`"${name}"`) might decode it if we treat it as JSON string? Use simple replace for now.
+                            const finalName = name ? name.replace(/\\'/g, "'") : name;
+                            return file.name === finalName;
+                        }
+                        // name contains '...'
+                        if (part.startsWith("name contains '")) {
+                            const token = (_b = part.match(/name contains '(.*)'/)) === null || _b === void 0 ? void 0 : _b[1];
+                            const finalToken = token ? token.replace(/\\'/g, "'") : token;
+                            return finalToken && file.name.includes(finalToken);
+                        }
+                        // 'ID' in parents
+                        if (part.includes(" in parents")) {
+                            const parentId = (_c = part.match(/'(.*)' in parents/)) === null || _c === void 0 ? void 0 : _c[1];
+                            return parentId && ((_d = file.parents) === null || _d === void 0 ? void 0 : _d.includes(parentId));
+                        }
+                        // trashed = ...
+                        if (part === "trashed = false") {
+                            return file.trashed !== true;
+                        }
+                        if (part === "trashed = true") {
+                            return file.trashed === true;
+                        }
+                        // mimeType = '...'
+                        if (part.startsWith("mimeType = '")) {
+                            const mime = (_e = part.match(/mimeType = '(.*)'/)) === null || _e === void 0 ? void 0 : _e[1];
+                            return file.mimeType === mime;
+                        }
+                        // mimeType != '...'
+                        if (part.startsWith("mimeType != '")) {
+                            const mime = (_f = part.match(/mimeType != '(.*)'/)) === null || _f === void 0 ? void 0 : _f[1];
+                            return file.mimeType !== mime;
+                        }
+                        // modifiedTime > '...'
+                        if (part.startsWith("modifiedTime > '")) {
+                            const timeStr = (_g = part.match(/modifiedTime > '(.*)'/)) === null || _g === void 0 ? void 0 : _g[1];
+                            return timeStr && new Date(file.modifiedTime) > new Date(timeStr);
+                        }
+                        // modifiedTime < '...'
+                        if (part.startsWith("modifiedTime < '")) {
+                            const timeStr = (_h = part.match(/modifiedTime < '(.*)'/)) === null || _h === void 0 ? void 0 : _h[1];
+                            return timeStr && new Date(file.modifiedTime) < new Date(timeStr);
+                        }
+                        // Ignore unknown filters for now
+                        return true;
+                    });
                 });
             });
         }
@@ -134,8 +147,23 @@ const createV3Router = () => {
     // Upload Files Route
     app.post('/upload/drive/v3/files', (req, res) => {
         const uploadType = req.query.uploadType;
-        if (uploadType !== 'multipart') {
-            res.status(400).json({ error: { code: 400, message: "Only uploadType=multipart is supported in this mock route" } });
+        if (uploadType !== 'multipart' && uploadType !== 'media') {
+            res.status(400).json({ error: { code: 400, message: "Only uploadType=multipart or uploadType=media is supported in this mock route" } });
+            return;
+        }
+        if (uploadType === 'media') {
+            const rawBody = req.body;
+            // Handle edge case where express.json() parses empty body as {}
+            if (req.headers['content-length'] === '0' && JSON.stringify(rawBody) === '{}') {
+                // Empty body
+            }
+            const newFile = store_1.driveStore.createFile({
+                name: "Untitled",
+                mimeType: req.headers['content-type'] || "application/octet-stream",
+                parents: [],
+                content: typeof rawBody === 'string' ? rawBody : JSON.stringify(rawBody) // Handle body if parsed
+            });
+            res.status(200).json(newFile);
             return;
         }
         const contentTypeHeader = req.headers['content-type'];
@@ -196,22 +224,6 @@ const createV3Router = () => {
         catch (_b) {
             content = contentPart.body;
         }
-        const existing = store_1.driveStore.listFiles().find(f => {
-            if (f.name !== metadata.name)
-                return false;
-            // Filter trashed?
-            if (f.trashed)
-                return false;
-            const newParents = metadata.parents || [];
-            const existingParents = f.parents || [];
-            if (newParents.length === 0 && existingParents.length === 0)
-                return true;
-            return newParents.some((p) => existingParents.includes(p));
-        });
-        if (existing) {
-            res.status(409).json({ error: { code: 409, message: "Conflict: File with same name already exists" } });
-            return;
-        }
         const newFile = store_1.driveStore.createFile(Object.assign(Object.assign({}, metadata), { content: content }));
         res.status(200).json(newFile);
     });
@@ -238,8 +250,68 @@ const createV3Router = () => {
             res.status(200).json(updatedFile);
             return;
         }
-        // Add multipart support if needed, but media is primary for now
-        res.status(400).json({ error: { code: 400, message: "Only uploadType=media is currently supported for V3 PATCH upload" } });
+        const contentTypeHeader = req.headers['content-type'];
+        const contentType = Array.isArray(contentTypeHeader) ? contentTypeHeader[0] : contentTypeHeader;
+        // Check for multipart
+        if (contentType && contentType.includes('multipart/related')) {
+            const boundaryMatch = contentType.match(/boundary=(.+)/);
+            if (!boundaryMatch) {
+                res.status(400).json({ error: { code: 400, message: "Multipart boundary missing" } });
+                return;
+            }
+            let boundary = boundaryMatch[1];
+            if (boundary.startsWith('"') && boundary.endsWith('"')) {
+                boundary = boundary.substring(1, boundary.length - 1);
+            }
+            const rawBody = req.body;
+            if (typeof rawBody !== 'string') {
+                res.status(400).json({ error: { code: 400, message: "Body parsing failed" } });
+                return;
+            }
+            const parts = rawBody.split(`--${boundary}`);
+            const validParts = parts.filter(p => p.trim() !== '' && p.trim() !== '--');
+            if (validParts.length < 2) {
+                res.status(400).json({ error: { code: 400, message: "Invalid multipart body: expected at least metadata and content" } });
+                return;
+            }
+            const parsePart = (rawPart) => {
+                const splitIndex = rawPart.indexOf('\r\n\r\n');
+                if (splitIndex === -1)
+                    return null;
+                const headers = rawPart.substring(0, splitIndex).trim();
+                const body = rawPart.substring(splitIndex + 4);
+                return {
+                    headers,
+                    body: body.replace(/\r\n$/, '') // Remove trailing CRLF from part body
+                };
+            };
+            const metadataPart = parsePart(validParts[0]);
+            const contentPart = parsePart(validParts[1]);
+            if (!metadataPart || !contentPart) {
+                res.status(400).json({ error: { code: 400, message: "Failed to parse parts" } });
+                return;
+            }
+            let metadata;
+            try {
+                metadata = JSON.parse(metadataPart.body);
+            }
+            catch (_a) {
+                res.status(400).json({ error: { code: 400, message: "Invalid JSON in metadata part" } });
+                return;
+            }
+            let content;
+            try {
+                content = JSON.parse(contentPart.body);
+            }
+            catch (_b) {
+                content = contentPart.body;
+            }
+            // Perform update
+            const updatedFile = store_1.driveStore.updateFile(fileId, Object.assign(Object.assign({}, metadata), { content: content, modifiedTime: new Date().toISOString() }));
+            res.status(200).json(updatedFile);
+            return;
+        }
+        res.status(400).json({ error: { code: 400, message: "Only uploadType=media or multipart/related is supported for V3 PATCH upload" } });
     });
     // Files: Create (Standard)
     app.post('/drive/v3/files', (req, res) => {

package/dist/store.d.ts

@@ -9,6 +9,8 @@ export interface DriveFile {
     trashed: boolean;
     createdTime: string;
     modifiedTime: string;
+    size: string;
+    md5Checksum: string;
     [key: string]: unknown;
 }
 export interface DriveChange {
@@ -38,6 +40,7 @@ export declare class DriveStore {
     private files;
     private changes;
     constructor();
+    private calculateStats;
     createFile(file: Partial<DriveFile> & {
         name: string;
     }): DriveFile;

package/dist/store.js

@@ -1,18 +1,71 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.driveStore = exports.DriveStore = void 0;
+const crypto = __importStar(require("crypto"));
 class DriveStore {
     constructor() {
         this.files = new Map();
         this.changes = [];
     }
+    calculateStats(content) {
+        let buffer;
+        if (typeof content === 'string') {
+            buffer = Buffer.from(content);
+        }
+        else if (content === undefined || content === null) {
+            buffer = Buffer.from('');
+        }
+        else {
+            buffer = Buffer.from(JSON.stringify(content));
+        }
+        return {
+            size: String(buffer.length),
+            md5Checksum: crypto.createHash('md5').update(buffer).digest('hex')
+        };
+    }
     createFile(file) {
         if (!file.name) {
             throw new Error("File name is required");
         }
         const id = file.id || Math.random().toString(36).substring(7);
         const now = new Date().toISOString();
-        const newFile = Object.assign(Object.assign({ kind: "drive#file", mimeType: "application/octet-stream", trashed: false, createdTime: now, modifiedTime: now }, file), { id, version: 1, etag: "1" });
+        const stats = this.calculateStats(file.content);
+        const newFile = Object.assign(Object.assign({ kind: "drive#file", mimeType: "application/octet-stream", trashed: false, createdTime: now, modifiedTime: now }, file), { id, version: 1, etag: "1", 
+            // Ensure calculated stats override provided ones
+            size: stats.size, md5Checksum: stats.md5Checksum });
         this.files.set(id, newFile);
         this.addChange(newFile);
         return newFile;
@@ -21,9 +74,14 @@ class DriveStore {
         const file = this.files.get(id);
         if (!file)
             return null;
+        // If content is being updated, recalculate stats
+        let statsUpdates = {};
+        if (updates.content !== undefined) {
+            statsUpdates = this.calculateStats(updates.content);
+        }
         // Merge updates and increment version
         const newVersion = file.version + 1;
-        const updatedFile = Object.assign(Object.assign(Object.assign({}, file), updates), { version: newVersion, etag: String(newVersion), modifiedTime: new Date().toISOString() });
+        const updatedFile = Object.assign(Object.assign(Object.assign(Object.assign({}, file), updates), statsUpdates), { version: newVersion, etag: String(newVersion), modifiedTime: new Date().toISOString() });
         this.files.set(id, updatedFile);
         this.addChange(updatedFile);
         return updatedFile;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "google-drive-mock",
-  "version": "1.0.9",
+  "version": "1.0.11",
   "description": "Mock-Server that simulates being google-drive. Used for testing.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -12,7 +12,7 @@
     "test:slow": "LATENCY=20 vitest run",
     "test:browser": "start-server-and-test dev http://localhost:3000 'TEST_TARGET=mock BROWSER_ENABLED=true vitest run --browser --no-file-parallelism'",
     "test:browser:real": "TEST_TARGET=real BROWSER_ENABLED=true vitest run --browser",
-    "test:real": "TEST_TARGET=real vitest run",
+    "test:real": "ts-node scripts/check-token.ts && TEST_TARGET=real vitest run",
     "example:login": "ts-node examples/serve-login.ts",
     "lint": "eslint .",
     "lint:fix": "eslint . --fix"

package/scripts/check-token.ts

@@ -0,0 +1,75 @@
+import * as fs from 'fs';
+import * as path from 'path';
+import * as https from 'https';
+
+// Load .ENV manually to avoid devDependency issues if dotenv isn't available in this context, 
+// though project seems to use it.
+const envPath = path.resolve(__dirname, '../.ENV');
+if (fs.existsSync(envPath)) {
+    const envContent = fs.readFileSync(envPath, 'utf8');
+    envContent.split('\n').forEach(line => {
+        const match = line.match(/^([^=]+)=(.*)$/);
+        if (match) {
+            const key = match[1].trim();
+            const value = match[2].trim().replace(/^['"]|['"]$/g, ''); // strip quotes
+            if (!process.env[key]) {
+                process.env[key] = value;
+            }
+        }
+    });
+}
+
+const token = process.env.GDRIVE_TOKEN;
+
+if (!token) {
+    console.error('❌ Error: GDRIVE_TOKEN not found in environment or .ENV file.');
+    process.exit(1);
+}
+
+// Simple check only if running real tests (though script is likely invoked specifically for that)
+// The user asked to run this BEFORE test:real.
+
+console.log('🔄 Verifying GDRIVE_TOKEN...');
+
+const options = {
+    hostname: 'www.googleapis.com',
+    path: '/drive/v3/about?fields=user',
+    method: 'GET',
+    headers: {
+        'Authorization': `Bearer ${token}`,
+        'User-Agent': 'node-script'
+    }
+};
+
+const req = https.request(options, (res) => {
+    let data = '';
+
+    res.on('data', (chunk) => {
+        data += chunk;
+    });
+
+    res.on('end', () => {
+        if (res.statusCode === 200) {
+            try {
+                const body = JSON.parse(data);
+                console.log(`✅ Token is valid. User: ${body.user?.emailAddress || 'Unknown'}`);
+                process.exit(0);
+            } catch (e: unknown) {
+                const msg = e instanceof Error ? e.message : String(e);
+                console.error('❌ Error parsing response:', msg);
+                process.exit(1);
+            }
+        } else {
+            console.error(`❌ Token verification failed. Tell the human to update the .ENV file with a valid token. Status: ${res.statusCode}`);
+            console.error('Response:', data);
+            process.exit(1);
+        }
+    });
+});
+
+req.on('error', (e) => {
+    console.error(`❌ Request error: ${e.message}`);
+    process.exit(1);
+});
+
+req.end();

package/src/mappers.ts

@@ -28,6 +28,8 @@ export function toV2File(file: DriveFile): Record<string, unknown> {
             isRoot: false // Mock simplification
         })),
         version: file.version,
+        fileSize: file.size,
+        md5Checksum: file.md5Checksum,
         downloadUrl: `http://localhost/drive/v2/files/${file.id}?alt=media`
     };
 }

package/src/routes/v2.ts

@@ -375,8 +375,21 @@ export const createV2Router = (config: AppConfig) => {
     app.post('/upload/drive/v2/files', (req: Request, res: Response) => {
         const uploadType = req.query.uploadType as string;
 
+        // V2 behavior: If-None-Match on POST seems to check against the collection or just strictly fail if an entity exists contextually.
+        // Tests show it returns 412 Precondition Failed when '*' is used.
+        const ifNoneMatch = req.headers['if-none-match'];
+        if (ifNoneMatch) {
+            res.status(412).json({ error: { code: 412, message: "Precondition Failed" } });
+            return;
+        }
+
         if (uploadType === 'media') {
-            const rawBody = req.body;
+            let rawBody = req.body;
+            // Handle edge case where express.json() parses empty body as {}
+            if (req.headers['content-length'] === '0' && JSON.stringify(rawBody) === '{}') {
+                rawBody = '';
+            }
+
             // For simple upload, metadata is default
             const name = "Untitled";
 
@@ -466,7 +479,12 @@ export const createV2Router = (config: AppConfig) => {
         const uploadType = req.query.uploadType as string;
 
         if (uploadType === 'media') {
-            const rawBody = req.body;
+            let rawBody = req.body;
+            // Handle edge case where express.json() parses empty body as {}
+            if (req.headers['content-length'] === '0' && JSON.stringify(rawBody) === '{}') {
+                rawBody = '';
+            }
+
             const updatedFile = driveStore.updateFile(fileId, {
                 content: rawBody,
                 modifiedTime: new Date().toISOString()