next-gdrive-index/src/app/api/raw/[...rest]/route.ts

import { type NextRequest, NextResponse } from "next/server";

import { encryptionService } from "~/lib/utils.server";

import { GetFile } from "~/actions/files";
import { ValidatePaths } from "~/actions/paths";

export const dynamic = "force-dynamic";

export async function GET(request: NextRequest, { params }: { params: Promise<{ rest: string[] }> }) {
  const { rest } = await params;
  const ALLOWED_TYPES = ["image/*", "video/*", "audio/*"]; // Based from mime types
  const paths = rest.map((path) => {
    if (path.startsWith("/")) return decodeURIComponent(path.slice(1));
    return decodeURIComponent(path);
  });

  try {
    const validatedPaths = await ValidatePaths(paths);
    if (!validatedPaths.success) {
      throw new Error(`[404] ${validatedPaths.message}`, {
        cause: validatedPaths.error,
      });
    }

    const currentFile = validatedPaths.data.pop();
    if (!currentFile) {
      throw new Error("[404] File not found", {
        cause: "Failed to get current file",
      });
    }
    if (ALLOWED_TYPES.every((type) => !new RegExp(type.replace("*", ".*")).test(currentFile.mimeType))) {
      throw new Error("[400] Invalid file type", {
        cause: "Raw link only available for video, image, and audio files",
      });
    }

    const fileMeta = await GetFile(currentFile.id);
    if (!fileMeta.success) {
      throw new Error(`[500] ${fileMeta.message}`, {
        cause: fileMeta.error,
      });
    }
    if (!fileMeta.data?.encryptedWebContentLink) {
      throw new Error("[500] No download link found", {
        cause: "No download link found",
      });
    }

    const decryptedLink = await encryptionService.decrypt(fileMeta.data.encryptedWebContentLink);
    return new NextResponse(null, {
      status: 302,
      headers: {
        Location: decryptedLink,
      },
    });
  } catch (error) {
    const e = error as Error;
    const message = e.message.replace(/\[.*\]/, "").trim();
    const status = /\[.*\]/.exec(e.message)?.[0].replace(/\[|\]/g, "").trim() ?? 500;

    return NextResponse.json(
      {
        scope: "api/raw",
        message,
        cause: e.cause ?? "Unknown",
      },
      {
        status: Number(status),
      },
    );
  }
}