You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
2.9 KiB
107 lines
2.9 KiB
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 getFileAbosluteDir = (filePath: string):string => {
|
|
const parts = filePath.split('/');
|
|
parts.pop();
|
|
return parts.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);
|
|
}
|
|
}
|
|
};
|
|
|
|
export const getPluginWorkDir = (args: IpluginInputArgs):string => {
|
|
const pluginWorkDir = `${args.workDir}/${new Date().getTime()}`;
|
|
args.deps.fsextra.ensureDirSync(pluginWorkDir);
|
|
return pluginWorkDir;
|
|
};
|