alikia2x/OpenRewind
1
1
Fork 0
Code Issues Releases 1 Activity

fix: correctly handle the situation where the screenshot file does not exist

Browse Source
This commit is contained in:
alikia2x (寒寒) 2024-12-29 23:00:57 +08:00
parent d2899d4b50
commit 71d00103dd
Signed by: alikia2x
GPG Key ID: 56209E0CCD8420C6
1 changed files with 25 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
src/electron/backend/encoding.ts
View File

@ -29,13 +29,23 @@ export function checkFramesForEncoding(db: Database) {
const buffer: Frame[] = []; const buffer: Frame[] = [];
if (frames.length < MIN_FRAMES_TO_ENCODE) return;
for (let i = 1; i < frames.length; i++) { for (let i = 1; i < frames.length; i++) {
const frame = frames[i]; const frame = frames[i];
const lastFrame = frames[i - 1]; const lastFrame = frames[i - 1];
const currentFrameSize = sizeOf(join(getScreenshotsDir(), frame.imgFilename)); const framePath = join(getScreenshotsDir(), frame.imgFilename);
const lastFrameSize = sizeOf(join(getScreenshotsDir(), lastFrame.imgFilename)); const lastFramePath = join(getScreenshotsDir(), lastFrame.imgFilename);
if (!fs.existsSync(framePath)) {
console.warn("File not exist:", frame.imgFilename);
deleteFrameFromDB(db, frame.id)
continue;
}
if (!fs.existsSync(lastFramePath)) {
console.warn("File not exist:", lastFrame.imgFilename);
deleteFrameFromDB(db, lastFrame.id)
continue;
}
const currentFrameSize = sizeOf(framePath);
const lastFrameSize = sizeOf(lastFramePath);
const twoFramesHaveSameSize = const twoFramesHaveSameSize =
currentFrameSize.width === lastFrameSize.width && currentFrameSize.width === lastFrameSize.width &&
currentFrameSize.height === lastFrameSize.height; currentFrameSize.height === lastFrameSize.height;
@ -93,14 +103,23 @@ function deleteNonExistentScreenshots(db: Database) {
for (const filename of filesInDir) { for (const filename of filesInDir) {
if (!dbFileSet.has(filename)) { if (!dbFileSet.has(filename)) {
fs.unlinkSync(path.join(screenshotDir, filename)); //fs.unlinkSync(path.join(screenshotDir, filename));
console.log("[dry-run] delete:", filename);
} }
} }
} }
export async function deleteUnnecessaryScreenshots(db: Database) { export async function deleteUnnecessaryScreenshots(db: Database) {
deleteEncodedScreenshots(db); deleteEncodedScreenshots(db);
deleteNonExistentScreenshots(db); //deleteNonExistentScreenshots(db);
}
export function deleteFrameFromDB(db: Database, id: number) {
const deleteStmt = db.prepare(`
DELETE FROM frame WHERE id = ?;
`);
deleteStmt.run(id);
console.log(`Deleted frame ${id} from database`);
} }
function getTasksPerforming() { function getTasksPerforming() {