daedalOS/components/system/Files/FileEntry/functions.ts

import type { FSModule } from "browserfs/dist/node/core/FS";
import extensions from "components/system/Files/FileEntry/extensions";
import type { FileInfo } from "components/system/Files/FileEntry/useFileInfo";
import ini from "ini";
import { parseBuffer } from "music-metadata-browser";
import { IMAGE_FILE_EXTENSIONS, SHORTCUT_EXTENSION } from "utils/constants";
import { bufferToUrl } from "utils/functions";

const getIconByFileExtension = (extension: string): string =>
  `/icons/${extensions[extension]?.icon || "unknown"}.png`;

const getProcessByFileExtension = (extension: string): string => {
  const [defaultProcess = ""] = extensions[extension]?.process || [];

  return defaultProcess;
};

const getShortcutInfo = (contents: Buffer) => {
  const {
    InternetShortcut: { BaseURL: pid = "", IconFile: icon = "", URL: url = "" },
  } = ini.parse(contents.toString());

  return { icon, pid, url };
};

export const getInfoWithoutExtension = (
  fs: FSModule,
  path: string,
  callback: React.Dispatch<React.SetStateAction<FileInfo>>
): void =>
  fs.stat(path, (_error, stats) => {
    const isDirectory = stats ? stats.isDirectory() : false;

    callback({
      icon: `/icons/${isDirectory ? "folder.png" : "unknown.png"}`,
      pid: isDirectory ? "FileExplorer" : "",
      url: path,
    });
  });

export const getInfoWithExtension = (
  fs: FSModule,
  path: string,
  extension: string,
  callback: React.Dispatch<React.SetStateAction<FileInfo>>
): void => {
  const getInfoByFileExtension = (icon?: string) =>
    callback({
      icon: icon || getIconByFileExtension(extension),
      pid: getProcessByFileExtension(extension),
      url: path,
    });

  if (extension === SHORTCUT_EXTENSION) {
    fs.readFile(path, (error, contents = Buffer.from("")) => {
      if (error) {
        getInfoByFileExtension();
      } else {
        callback(getShortcutInfo(contents));
      }
    });
  } else if (IMAGE_FILE_EXTENSIONS.has(extension)) {
    fs.readFile(path, (error, contents = Buffer.from("")) =>
      getInfoByFileExtension(error ? "/icons/photo.png" : bufferToUrl(contents))
    );
  } else if (extension === ".mp3") {
    fs.readFile(path, (error, contents = Buffer.from("")) =>
      parseBuffer(contents).then(
        ({ common: { picture: [picture] = [] } = {} }) =>
          getInfoByFileExtension(
            !error && picture ? bufferToUrl(picture.data) : "/icons/music.png"
          )
      )
    );
  } else {
    getInfoByFileExtension();
  }
};