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

import {
  getIconByFileExtension,
  getProcessByFileExtension
} from 'components/system/Files/FileEntry/functions';
import { useFileSystem } from 'contexts/fileSystem';
import ini from 'ini';
import { extname } from 'path';
import { useEffect, useState } from 'react';
import { IMAGE_FILE_EXTENSIONS, SHORTCUT_EXTENSION } from 'utils/constants';
import { bufferToUrl } from 'utils/functions';

type FileInfo = {
  icon: string;
  pid: string;
  url: string;
};

const useFileInfo = (path: string): FileInfo => {
  const [info, setInfo] = useState<FileInfo>({
    icon: '',
    pid: '',
    url: ''
  });
  const { fs } = useFileSystem();

  useEffect(() => {
    if (fs) {
      const extension = extname(path).toLowerCase();

      if (!extension) {
        fs.stat(path, (_error, stats) => {
          const isDirectory = stats ? stats.isDirectory() : false;

          setInfo({
            icon: `/icons/${isDirectory ? 'folder.png' : 'unknown.png'}`,
            pid: isDirectory ? 'FileExplorer' : '',
            url: path
          });
        });
      } else {
        const getInfoByFileExtension = (icon?: string) =>
          setInfo({
            icon: icon || getIconByFileExtension(extension),
            pid: getProcessByFileExtension(extension),
            url: path
          });

        if (extension === SHORTCUT_EXTENSION) {
          fs.readFile(path, (error, contents = Buffer.from('')) => {
            if (error) {
              getInfoByFileExtension();
            } else {
              const {
                InternetShortcut: {
                  BaseURL: pid = '',
                  IconFile: icon = '',
                  URL: url = ''
                }
              } = ini.parse(contents.toString());

              setInfo({ icon, pid, url });
            }
          });
        } else if (IMAGE_FILE_EXTENSIONS.includes(extension)) {
          fs.readFile(path, (error, contents = Buffer.from('')) =>
            getInfoByFileExtension(
              error ? '/icons/photo.png' : bufferToUrl(contents)
            )
          );
        } else {
          getInfoByFileExtension();
        }
      }
    }
  }, [fs, path]);

  return info;
};

export default useFileInfo;