sandbox-web-ide/backend/server/src/fileoperations.ts

import * as dotenv from "dotenv";
import {
  R2FileBody,
  R2Files,
  Sandbox,
  TFile,
  TFileData,
  TFolder,
} from "./types";

dotenv.config();

export const getSandboxFiles = async (id: string) => {
  const res = await fetch(
    `${process.env.STORAGE_WORKER_URL}/api?sandboxId=${id}`,
    {
      headers: {
        Authorization: `${process.env.WORKERS_KEY}`,
      },
    }
  );
  const data: R2Files = await res.json();

  const paths = data.objects.map((obj) => obj.key);
  const processedFiles = await processFiles(paths, id);
  return processedFiles;
};

export const getFolder = async (folderId: string) => {
  const res = await fetch(
    `${process.env.STORAGE_WORKER_URL}/api?folderId=${folderId}`,
    {
      headers: {
        Authorization: `${process.env.WORKERS_KEY}`,
      },
    }
  );
  const data: R2Files = await res.json();

  return data.objects.map((obj) => obj.key);
};

const processFiles = async (paths: string[], id: string) => {
  const root: TFolder = { id: "/", type: "folder", name: "/", children: [] };
  const fileData: TFileData[] = [];

  paths.forEach((path) => {
    const allParts = path.split("/");
    if (allParts[1] !== id) {
      return;
    }

    const parts = allParts.slice(2);
    let current: TFolder = root;

    for (let i = 0; i < parts.length; i++) {
      const part = parts[i];
      const isFile = i === parts.length - 1 && part.length;
      const existing = current.children.find((child) => child.name === part);

      if (existing) {
        if (!isFile) {
          current = existing as TFolder;
        }
      } else {
        if (isFile) {
          const file: TFile = { id: path, type: "file", name: part };
          current.children.push(file);
          fileData.push({ id: path, data: "" });
        } else {
          const folder: TFolder = {
            // id: path, // todo: wrong id. for example, folder "src" ID is: projects/a7vgttfqbgy403ratp7du3ln/src/App.css
            id: `projects/${id}/${parts.slice(0, i + 1).join("/")}`,
            type: "folder",
            name: part,
            children: [],
          };
          current.children.push(folder);
          current = folder;
        }
      }
    }
  });

  await Promise.all(
    fileData.map(async (file) => {
      const data = await fetchFileContent(file.id);
      file.data = data;
    })
  );

  return {
    files: root.children,
    fileData,
  };
};

const fetchFileContent = async (fileId: string): Promise<string> => {
  try {
    const fileRes = await fetch(
      `${process.env.STORAGE_WORKER_URL}/api?fileId=${fileId}`,
      {
        headers: {
          Authorization: `${process.env.WORKERS_KEY}`,
        },
      }
    );
    return await fileRes.text();
  } catch (error) {
    console.error("ERROR fetching file:", error);
    return "";
  }
};

export const createFile = async (fileId: string) => {
  const res = await fetch(`${process.env.STORAGE_WORKER_URL}/api`, {
    method: "POST",
    headers: {
      "Content-Type": "application/json",
      Authorization: `${process.env.WORKERS_KEY}`,
    },
    body: JSON.stringify({ fileId }),
  });
  return res.ok;
};

export const renameFile = async (
  fileId: string,
  newFileId: string,
  data: string
) => {
  const res = await fetch(`${process.env.STORAGE_WORKER_URL}/api/rename`, {
    method: "POST",
    headers: {
      "Content-Type": "application/json",
      Authorization: `${process.env.WORKERS_KEY}`,
    },
    body: JSON.stringify({ fileId, newFileId, data }),
  });
  return res.ok;
};

export const saveFile = async (fileId: string, data: string) => {
  const res = await fetch(`${process.env.STORAGE_WORKER_URL}/api/save`, {
    method: "POST",
    headers: {
      "Content-Type": "application/json",
      Authorization: `${process.env.WORKERS_KEY}`,
    },
    body: JSON.stringify({ fileId, data }),
  });
  return res.ok;
};

export const deleteFile = async (fileId: string) => {
  const res = await fetch(`${process.env.STORAGE_WORKER_URL}/api`, {
    method: "DELETE",
    headers: {
      "Content-Type": "application/json",
      Authorization: `${process.env.WORKERS_KEY}`,
    },
    body: JSON.stringify({ fileId }),
  });
  return res.ok;
};

export const getProjectSize = async (id: string) => {
  const res = await fetch(
    `${process.env.STORAGE_WORKER_URL}/api/size?sandboxId=${id}`,
    {
      headers: {
        Authorization: `${process.env.WORKERS_KEY}`,
      },
    }
  );
  return (await res.json()).size;
};