google-drive-mock 1.0.7 → 1.0.8

package/dist/index.js

@@ -45,7 +45,7 @@ const createApp = (config = {}) => {
         next();
     }));
     app.use(express_1.default.json());
-    app.use(express_1.default.text({ type: ['multipart/mixed', 'multipart/related'] }));
+    app.use(express_1.default.text({ type: ['multipart/mixed', 'multipart/related', 'text/*', 'application/xml'] }));
     // Batch Route
     app.post('/batch', batch_1.handleBatchRequest);
     app.post('/batch/drive/v3', batch_1.handleBatchRequest);

package/dist/routes/v2.js

@@ -17,6 +17,19 @@ const createV2Router = (config) => {
         const newFile = store_1.driveStore.createFile(Object.assign(Object.assign({}, fileData), { name: name, mimeType: fileData.mimeType || "application/octet-stream", parents: fileData.parents || [] }));
         res.status(200).json((0, mappers_1.toV2File)(newFile));
     });
+    // V2 Generate IDs (Must come before /:fileId)
+    app.get('/drive/v2/files/generateIds', (req, res) => {
+        const count = parseInt(req.query.maxResults) || 10;
+        const ids = [];
+        for (let i = 0; i < count; i++) {
+            ids.push(Math.random().toString(36).substring(2, 15));
+        }
+        res.json({
+            kind: "drive#generatedIds",
+            ids: ids,
+            space: req.query.space || 'drive'
+        });
+    });
     // V2 Files: Get
     app.get('/drive/v2/files/:fileId', (req, res) => {
         const fileId = req.params.fileId;
@@ -33,6 +46,9 @@ const createV2Router = (config) => {
             res.setHeader('ETag', file.etag);
         }
         if (req.query.alt === 'media') {
+            if (file.mimeType) {
+                res.setHeader('Content-Type', file.mimeType);
+            }
             if (file.content === undefined) {
                 res.send("");
                 return;
@@ -47,6 +63,43 @@ const createV2Router = (config) => {
         }
         res.json((0, mappers_1.toV2File)(file));
     });
+    // V2 Export
+    app.get('/drive/v2/files/:fileId/export', (req, res) => {
+        const fileId = req.params.fileId;
+        if (typeof fileId !== 'string') {
+            res.status(400).send("Invalid file ID");
+            return;
+        }
+        const file = store_1.driveStore.getFile(fileId);
+        if (!file) {
+            res.status(404).json({ error: { code: 404, message: "File not found" } });
+            return;
+        }
+        // Mock export: just return content. Real API validates mimeType compatibility.
+        res.send(file.content || "");
+    });
+    // V2 Watch
+    app.post('/drive/v2/files/:fileId/watch', (req, res) => {
+        const fileId = req.params.fileId;
+        if (typeof fileId !== 'string') {
+            res.status(400).send("Invalid file ID");
+            return;
+        }
+        const file = store_1.driveStore.getFile(fileId);
+        if (!file) {
+            res.status(404).json({ error: { code: 404, message: "File not found" } });
+            return;
+        }
+        // Mock Channel response
+        res.json({
+            kind: "api#channel",
+            id: req.body.id || Math.random().toString(36).substring(7),
+            resourceId: fileId,
+            resourceUri: `${config.apiEndpoint}/drive/v2/files/${fileId}`,
+            token: req.body.token,
+            expiration: Date.now() + 3600000 // 1 hour
+        });
+    });
     // V2 Files: Update (PUT)
     app.put('/drive/v2/files/:fileId', (req, res) => {
         const fileId = req.params.fileId;
@@ -69,6 +122,25 @@ const createV2Router = (config) => {
                 return;
             }
         }
+        // Handle addParents/removeParents
+        let parents = existingFile.parents || [];
+        const addParents = req.query.addParents;
+        const removeParents = req.query.removeParents;
+        if (addParents) {
+            const toAdd = addParents.split(',');
+            toAdd.forEach(id => {
+                if (!parents.includes(id))
+                    parents.push(id);
+            });
+        }
+        if (removeParents) {
+            const toRemove = removeParents.split(',');
+            parents = parents.filter(id => !toRemove.includes(id));
+        }
+        // Merge parents into updates if they were modified
+        if (addParents || removeParents) {
+            updates.parents = parents;
+        }
         const updatedFile = store_1.driveStore.updateFile(fileId, updates);
         res.json((0, mappers_1.toV2File)(updatedFile));
     });
@@ -94,6 +166,25 @@ const createV2Router = (config) => {
                 return;
             }
         }
+        // Handle addParents/removeParents
+        let parents = existingFile.parents || [];
+        const addParents = req.query.addParents;
+        const removeParents = req.query.removeParents;
+        if (addParents) {
+            const toAdd = addParents.split(',');
+            toAdd.forEach(id => {
+                if (!parents.includes(id))
+                    parents.push(id);
+            });
+        }
+        if (removeParents) {
+            const toRemove = removeParents.split(',');
+            parents = parents.filter(id => !toRemove.includes(id));
+        }
+        // Merge parents into updates if they were modified
+        if (addParents || removeParents) {
+            updates.parents = parents;
+        }
         const updatedFile = store_1.driveStore.updateFile(fileId, updates);
         res.json((0, mappers_1.toV2File)(updatedFile));
     });
@@ -192,38 +283,12 @@ const createV2Router = (config) => {
             startPageToken: token
         });
     });
-    // V2 Upload (Multipart)
-    app.post('/upload/drive/v2/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" } });
-            return;
-        }
-        const contentTypeHeader = req.headers['content-type'];
-        const contentType = Array.isArray(contentTypeHeader) ? contentTypeHeader[0] : contentTypeHeader;
-        if (!contentType || !contentType.includes('multipart/related')) {
-            res.status(400).json({ error: { code: 400, message: "Content-Type must be multipart/related" } });
-            return;
-        }
-        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;
-        }
+    // Helper for multipart parsing
+    const parseMultipart = (rawBody, boundary) => {
         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" } });
-            return;
+            return null;
         }
         const parsePart = (rawPart) => {
             let splitIndex = rawPart.indexOf('\r\n\r\n');
@@ -233,7 +298,6 @@ const createV2Router = (config) => {
                 separatorLength = 2;
             }
             if (splitIndex === -1) {
-                console.log('V2 Upload: Could not find header/body separator in part', rawPart.substring(0, 50));
                 return null;
             }
             const headers = rawPart.substring(0, splitIndex).trim();
@@ -246,16 +310,14 @@ const createV2Router = (config) => {
         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;
+            return null;
         }
         let metadata;
         try {
             metadata = JSON.parse(metadataPart.body);
         }
         catch (_a) {
-            res.status(400).json({ error: { code: 400, message: "Invalid JSON in metadata part" } });
-            return;
+            return null;
         }
         let content;
         try {
@@ -264,10 +326,114 @@ const createV2Router = (config) => {
         catch (_b) {
             content = contentPart.body;
         }
-        const fileData = (0, mappers_1.fromV2Update)(metadata);
-        const name = fileData.name || metadata.title || "Untitled";
-        const newFile = store_1.driveStore.createFile(Object.assign(Object.assign({}, fileData), { name: name, mimeType: fileData.mimeType || "application/octet-stream", parents: fileData.parents || [], content: content }));
-        res.status(200).json((0, mappers_1.toV2File)(newFile));
+        return { metadata, content };
+    };
+    // V2 Upload (POST)
+    app.post('/upload/drive/v2/files', (req, res) => {
+        const uploadType = req.query.uploadType;
+        if (uploadType === 'media') {
+            const rawBody = req.body;
+            // For simple upload, metadata is default
+            const name = "Untitled";
+            const newFile = store_1.driveStore.createFile({
+                name: name,
+                mimeType: req.headers['content-type'] || "application/octet-stream",
+                parents: [],
+                content: rawBody
+            });
+            res.status(200).json((0, mappers_1.toV2File)(newFile));
+            return;
+        }
+        if (uploadType === 'multipart') {
+            const contentTypeHeader = req.headers['content-type'];
+            const contentType = Array.isArray(contentTypeHeader) ? contentTypeHeader[0] : contentTypeHeader;
+            if (!contentType || !contentType.includes('multipart/related')) {
+                res.status(400).json({ error: { code: 400, message: "Content-Type must be multipart/related" } });
+                return;
+            }
+            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 parsed = parseMultipart(rawBody, boundary);
+            if (!parsed) {
+                res.status(400).json({ error: { code: 400, message: "Invalid multipart body" } });
+                return;
+            }
+            const { metadata, content } = parsed;
+            const fileData = (0, mappers_1.fromV2Update)(metadata);
+            const name = fileData.name || metadata.title || "Untitled";
+            const newFile = store_1.driveStore.createFile(Object.assign(Object.assign({}, fileData), { name: name, mimeType: fileData.mimeType || "application/octet-stream", parents: fileData.parents || [], content: content }));
+            res.status(200).json((0, mappers_1.toV2File)(newFile));
+            return;
+        }
+        res.status(400).json({ error: { code: 400, message: "Invalid uploadType" } });
+    });
+    // V2 Upload (PUT)
+    app.put('/upload/drive/v2/files/:fileId', (req, res) => {
+        const fileId = req.params.fileId;
+        if (typeof fileId !== 'string') {
+            res.status(400).send("Invalid file ID");
+            return;
+        }
+        const existingFile = store_1.driveStore.getFile(fileId);
+        if (!existingFile) {
+            res.status(404).json({ error: { code: 404, message: "File not found" } });
+            return;
+        }
+        const uploadType = req.query.uploadType;
+        if (uploadType === 'media') {
+            const rawBody = req.body;
+            const updatedFile = store_1.driveStore.updateFile(fileId, {
+                content: rawBody,
+                modifiedTime: new Date().toISOString()
+            });
+            res.status(200).json((0, mappers_1.toV2File)(updatedFile));
+            return;
+        }
+        if (uploadType === 'multipart') {
+            const contentTypeHeader = req.headers['content-type'];
+            const contentType = Array.isArray(contentTypeHeader) ? contentTypeHeader[0] : contentTypeHeader;
+            if (!contentType || !contentType.includes('multipart/related')) {
+                res.status(400).json({ error: { code: 400, message: "Content-Type must be multipart/related" } });
+                return;
+            }
+            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 parsed = parseMultipart(rawBody, boundary);
+            if (!parsed) {
+                res.status(400).json({ error: { code: 400, message: "Invalid multipart body" } });
+                return;
+            }
+            const { metadata, content } = parsed;
+            const fileData = (0, mappers_1.fromV2Update)(metadata);
+            const updatedFile = store_1.driveStore.updateFile(fileId, Object.assign(Object.assign({}, fileData), { content: content, modifiedTime: new Date().toISOString() }));
+            res.status(200).json((0, mappers_1.toV2File)(updatedFile));
+            return;
+        }
+        res.status(400).json({ error: { code: 400, message: "Invalid uploadType" } });
     });
     // V2 Trash
     app.post('/drive/v2/files/:fileId/trash', (req, res) => {

package/dist/routes/v3.js

@@ -215,6 +215,32 @@ const createV3Router = () => {
         const newFile = store_1.driveStore.createFile(Object.assign(Object.assign({}, metadata), { content: content }));
         res.status(200).json(newFile);
     });
+    // Upload Files: Update (PATCH)
+    app.patch('/upload/drive/v3/files/:fileId', (req, res) => {
+        const fileId = req.params.fileId;
+        if (typeof fileId !== 'string') {
+            res.status(400).send("Invalid file ID");
+            return;
+        }
+        const existingFile = store_1.driveStore.getFile(fileId);
+        if (!existingFile) {
+            res.status(404).json({ error: { code: 404, message: "File not found" } });
+            return;
+        }
+        const uploadType = req.query.uploadType;
+        if (uploadType === 'media') {
+            const rawBody = req.body;
+            // V3 update content via media upload
+            const updatedFile = store_1.driveStore.updateFile(fileId, {
+                content: rawBody,
+                modifiedTime: new Date().toISOString()
+            });
+            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" } });
+    });
     // Files: Create (Standard)
     app.post('/drive/v3/files', (req, res) => {
         const body = req.body || {};
@@ -239,6 +265,22 @@ const createV3Router = () => {
             res.status(400).json({ error: { code: 400, message: "Invalid field selection: etag" } });
             return;
         }
+        if (req.query.alt === 'media') {
+            if (file.mimeType) {
+                res.setHeader('Content-Type', file.mimeType);
+            }
+            if (file.content === undefined) {
+                res.send("");
+                return;
+            }
+            if (typeof file.content === 'object') {
+                res.json(file.content);
+            }
+            else {
+                res.send(file.content);
+            }
+            return;
+        }
         res.json(file);
     });
     // Files: Update

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "google-drive-mock",
-  "version": "1.0.7",
+  "version": "1.0.8",
   "description": "Mock-Server that simulates being google-drive. Used for testing.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/index.ts

@@ -36,7 +36,7 @@ const createApp = (config: AppConfig = {}) => {
     });
 
     app.use(express.json());
-    app.use(express.text({ type: ['multipart/mixed', 'multipart/related'] }));
+    app.use(express.text({ type: ['multipart/mixed', 'multipart/related', 'text/*', 'application/xml'] }));
 
     // Batch Route
     app.post('/batch', handleBatchRequest);