download file at /drive/:id instead of with ?id

Nirmalraj10567 · Jun 27, 2020 · 0b7b37c · 0b7b37c
1 parent c96401f
commit 0b7b37c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -40,7 +40,7 @@ server.use("/api/v1/drive/folder", async (req, res) => {
   res.send(await getFiles(folderId));
 });
 
-server.use("/api/v1/drive/file/:slug", sendFileStream);
+server.use("/api/v1/drive/file/:id", sendFileStream);
 
 server.use("/api/v1/drive/getAuthURL", (req, res) => {
   const CLIENT_ID = req.query.clientId;

diff --git a/utils/gdrive.js b/utils/gdrive.js
@@ -209,7 +209,8 @@ function getFiles(folderId) {
 }
 
 function sendFileStream(req, res) {
-  const fileId = req.query.id;
+  const fileId = req.params.id || req.query.id;
+  if (!fileId) res.sendStatus(404);
   drive.files.get(
     {
       fileId,