Tdarr_Plugins/FlowPluginsTs/FlowHelpers/1.0.0/fileUtils.ts

import { promises as fs } from 'fs';
import { IpluginInputArgs } from './interfaces/interfaces';

export const getContainer = (filePath: string): string => {
  const parts = filePath.split('.');
  return parts[parts.length - 1];
};

export const getFileName = (filePath: string): string => {
  const parts = filePath.split('/');
  const fileNameAndContainer = parts[parts.length - 1];
  const parts2 = fileNameAndContainer.split('.');
  parts2.pop();
  return parts2.join('.');
};

export const getFfType = (codecType: string): string => (codecType === 'video' ? 'v' : 'a');

export const getSubStem = ({
  inputPathStem,
  inputPath,
}: {
  inputPathStem: string,
  inputPath: string,
}): string => {
  const subStem = inputPath.substring(inputPathStem.length);
  const parts = subStem.split('/');
  parts.pop();

  return parts.join('/');
};

const getFileSize = async (file:string):Promise<number> => {
  const stats = await fs.stat(file);
  const { size } = stats;
  return size;
};

export const moveFileAndValidate = async ({
  inputPath,
  outputPath,
  args,
}: {
  inputPath: string,
  outputPath: string,
  args: IpluginInputArgs
}):Promise<void> => {
  const inputSize = await getFileSize(inputPath);

  args.jobLog(`Attempt 1: Moving file from ${inputPath} to ${outputPath}`);

  const res1 = await new Promise((resolve) => {
    args.deps.gracefulfs.rename(inputPath, outputPath, (err: Error) => {
      if (err) {
        args.jobLog(`Failed to move file from ${inputPath} to ${outputPath}`);
        args.jobLog(JSON.stringify(err));
        resolve(false);
      } else {
        resolve(true);
      }
    });
  });

  let outputSize = 0;
  try {
    outputSize = await getFileSize(outputPath);
  } catch (err) {
    args.jobLog(JSON.stringify(err));
  }

  if (!res1 || inputSize !== outputSize) {
    args.jobLog(`Attempt 1  failed: Moving file from ${inputPath} to ${outputPath}`);
    args.jobLog(`Attempt 2: Moving file from ${inputPath} to ${outputPath}`);

    const res2 = await new Promise((resolve) => {
      args.deps.mvdir(inputPath, outputPath, { overwrite: true })
        .then(() => {
          resolve(true);
        }).catch((err: Error) => {
          args.jobLog(`Failed to move file from ${inputPath} to ${outputPath}`);
          args.jobLog(JSON.stringify(err));
          resolve(false);
        });
    });

    outputSize = await getFileSize(outputPath);

    if (!res2 || inputSize !== outputSize) {
      const errMessage = `Failed to move file from ${inputPath} to ${outputPath}, check errors above`;
      args.jobLog(errMessage);
      throw new Error(errMessage);
    }
  }
};