David/file upload

.env.sample

@@ -6,11 +6,12 @@ EMAIL_PASS=
 EMAIL_SERVICE=
 EMAIL_USER=
 GDRIVE_CLIENT_EMAIL=
-GDRIVE_CLIENT_ID=
+GDRIVE_PROJECT_ID=
 GDRIVE_PRIVATE_KEY=
 GDRIVE_PROJECT_ID=
 POSTGRES_DATABASE=
 POSTGRES_PASSWORD=
 POSTGRES_PORT=
 POSTGRES_URL_NON_POOLING=
-POSTGRES_USER=
+POSTGRES_USER=
+VERCEL=

.eslintrc.json

@@ -1,12 +1,6 @@
 {
-  "extends": [
-    "next",
-    "next/core-web-vitals",
-    "plugin:prettier/recommended"
-  ],
-  "plugins": [
-    "prettier"
-  ],
+  "extends": ["next", "next/core-web-vitals", "plugin:prettier/recommended"],
+  "plugins": ["prettier"],
   "rules": {
     "prettier/prettier": [
       "error",
@@ -16,4 +10,4 @@
     ],
     "no-unused-vars": "error"
   }
-}
+}

app/api/deleteFile/route.ts

@@ -0,0 +1,53 @@
+import { google } from 'googleapis';
+import { NextRequest, NextResponse } from 'next/server';
+
+export async function DELETE(req: NextRequest) {
+  const { searchParams } = new URL(req.url);
+  const fileId = searchParams.get('fileId');
+
+  if (!fileId) {
+    return NextResponse.json(
+      { message: 'File ID is required' },
+      { status: 400 }
+    );
+  }
+  try {
+    const private_key = process.env.GDRIVE_PRIVATE_KEY?.replace(/\\n/g, '\n');
+
+    if (!private_key) {
+      return NextResponse.json(
+        { error: 'Missing Google Drive private key' },
+        { status: 500 }
+      );
+    }
+
+    const auth = new google.auth.GoogleAuth({
+      projectId: process.env.GDRIVE_PROJECT_ID,
+      scopes: 'https://www.googleapis.com/auth/drive',
+      credentials: {
+        type: 'service_account',
+        client_id: process.env.GDRIVE_PROJECT_ID,
+        client_email: process.env.GDRIVE_CLIENT_EMAIL,
+        private_key: private_key,
+      },
+    });
+
+    const drive = google.drive({ version: 'v3', auth });
+
+    // Delete the file
+    await drive.files.delete({
+      fileId: fileId,
+    });
+
+    return NextResponse.json(
+      { message: 'File deleted successfully' },
+      { status: 200 }
+    );
+  } catch (error) {
+    console.error('Error deleting file:', error);
+    return NextResponse.json(
+      { message: 'Error deleting file', error: (error as Error).message },
+      { status: 500 }
+    );
+  }
+}

app/api/searchDrive/route.ts

@@ -0,0 +1,54 @@
+import { NextRequest, NextResponse } from 'next/server';
+import { google } from 'googleapis';
+
+export async function GET(req: NextRequest) {
+  const newHeaders = new Headers(req.headers);
+  newHeaders.set(
+    'Cache-Control',
+    'no-cache, no-store, max-age=0, must-revalidate'
+  );
+
+  const private_key = process.env.GDRIVE_PRIVATE_KEY?.replace(/\\n/g, '\n');
+
+  if (!private_key) {
+    return NextResponse.json(
+      { error: 'Missing Google Drive private key' },
+      { status: 500 }
+    );
+  }
+
+  const auth = new google.auth.GoogleAuth({
+    projectId: process.env.GDRIVE_PROJECTID,
+    scopes: 'https://www.googleapis.com/auth/drive',
+    credentials: {
+      type: 'service_account',
+      client_id: process.env.GDRIVE_PROJECT_ID,
+      client_email: process.env.GDRIVE_CLIENT_EMAIL,
+      private_key: private_key,
+    },
+  });
+  const drive = google.drive({ version: 'v3', auth });
+
+  try {
+    const res = await drive.files.list({
+      q: "'1schkzvm_b46UGovHpQ2uH-X-nJtlm32_' in parents",
+      fields: 'nextPageToken, files(id, name)',
+      spaces: 'drive',
+    });
+
+    const files = res.data.files;
+    return NextResponse.json(
+      { results: { files } },
+      {
+        status: 200,
+        headers: newHeaders,
+      }
+    );
+  } catch (err) {
+    console.error('Error querying Google Drive:', err);
+    return NextResponse.json(
+      { error: 'Internal Server Error' },
+      { status: 500 }
+    );
+  }
+}

app/api/uploadFile/route.ts

@@ -0,0 +1,80 @@
+import { google } from 'googleapis';
+import { NextRequest, NextResponse } from 'next/server';
+import { Readable } from 'node:stream';
+
+export async function POST(req: NextRequest) {
+  const formData = await req.formData();
+
+  const file: File | null = formData.get('file') as File | null;
+  let fileBuffer: Buffer;
+
+  if (file instanceof File) {
+    const arrayBuffer = await file.arrayBuffer();
+    fileBuffer = Buffer.from(arrayBuffer);
+  } else {
+    console.error('No file was provided in the form data');
+    return NextResponse.json(
+      { error: 'No file was provided' },
+      { status: 400 }
+    );
+  }
+
+  const filename: string | null = formData.get('fileName') as string | null;
+
+  if (!filename) {
+    console.error('No filename was provided in the form data');
+    return NextResponse.json(
+      { error: 'No filename was provided' },
+      { status: 400 }
+    );
+  }
+
+  const private_key = process.env.GDRIVE_PRIVATE_KEY?.replace(/\\n/g, '\n');
+
+  if (!private_key) {
+    return NextResponse.json(
+      { error: 'Missing Google Drive private key' },
+      { status: 500 }
+    );
+  }
+
+  const auth = new google.auth.GoogleAuth({
+    projectId: process.env.GDRIVE_PROJECTID,
+    scopes: 'https://www.googleapis.com/auth/drive',
+    credentials: {
+      type: 'service_account',
+      client_id: process.env.GDRIVE_PROJECT_ID,
+      client_email: process.env.GDRIVE_CLIENT_EMAIL,
+      private_key: private_key,
+    },
+  });
+
+  const drive = google.drive({ version: 'v3', auth });
+
+  try {
+    await drive.files.create({
+      requestBody: {
+        name: filename,
+        mimeType: 'application/pdf',
+        parents: ['1schkzvm_b46UGovHpQ2uH-X-nJtlm32_'],
+        driveId: '1schkzvm_b46UGovHpQ2uH-X-nJtlm32_',
+      },
+      media: {
+        mimeType: 'application/pdf',
+        body: Readable.from(fileBuffer),
+      },
+      supportsAllDrives: true,
+    });
+
+    return NextResponse.json(
+      { message: 'File uploaded successfully' },
+      { status: 200 }
+    );
+  } catch (error) {
+    console.error('Error uploading file:', error);
+    return NextResponse.json(
+      { message: 'Error uploading file', error: (error as Error).message },
+      { status: 500 }
+    );
+  }
+}

docker-compose.yml

@@ -30,4 +30,4 @@ services:
       - ./db-init:/docker-entrypoint-initdb.d
 
 volumes:
-  postgres_data:
+  postgres_data: