external/SponsorBlockServer
1
0
Fork 0
mirror of https://github.com/ajayyy/SponsorBlockServer.git synced 2024-09-19 20:51:19 +02:00
Code Issues Projects Releases Packages Wiki Activity

Store video info from submissions

Browse source
This commit is contained in:
Ajay Ramachandran 2021-10-17 14:42:48 -04:00
parent bda2ff4d23
commit 450f4a2d44
4 changed files with 32 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
databases/_upgrade_sponsorTimes_25.sql Normal file
View file

@ -0,0 +1,13 @@
BEGIN TRANSACTION;
CREATE TABLE IF NOT EXISTS "videoInfo" (
"videoID" TEXT PRIMARY KEY NOT NULL,
"channelID" TEXT NOT NULL,
"title" TEXT NOT NULL,
"published" REAL NOT NULL,
"genreUrl" REAL NOT NULL
);
UPDATE "config" SET value = 25 WHERE key = 'version';
COMMIT;

5
src/routes/postSkipSegments.ts
View file

@ -651,6 +651,11 @@ export async function postSkipSegments(req: Request, res: Response): Promise<Res
//add to private db as well
await privateDB.prepare("run", `INSERT INTO "sponsorTimes" VALUES(?, ?, ?, ?)`, [videoID, hashedIP, timeSubmitted, service]);
await db.prepare("run", `INSERT INTO "videoInfo" ("videoID", "channelID", "title", "published", "genreUrl")
SELECT ?, ?, ?, ?, ?
WHERE NOT EXISTS (SELECT 1 FROM "videoInfo" WHERE "videoID" = ?)`, [
videoID, apiVideoInfo?.data?.authorId || "", apiVideoInfo?.data?.title || "", apiVideoInfo?.data?.published || 0, apiVideoInfo?.data?.genreUrl || "", videoID]);
// Clear redis cache for this video
QueryCacher.clearVideoCache({
videoID,

12
test/cases/postSkipSegments.ts
View file

@ -35,6 +35,7 @@ describe("postSkipSegments", () => {
const queryDatabase = (videoID: string) => db.prepare("get", `SELECT "startTime", "endTime", "locked", "category" FROM "sponsorTimes" WHERE "videoID" = ?`, [videoID]);
const queryDatabaseActionType = (videoID: string) => db.prepare("get", `SELECT "startTime", "endTime", "locked", "category", "actionType" FROM "sponsorTimes" WHERE "videoID" = ?`, [videoID]);
const queryDatabaseDuration = (videoID: string) => db.prepare("get", `SELECT "startTime", "endTime", "locked", "category", "videoDuration" FROM "sponsorTimes" WHERE "videoID" = ?`, [videoID]);
const queryDatabaseVideoInfo = (videoID: string) => db.prepare("get", `SELECT * FROM "videoInfo" WHERE "videoID" = ?`, [videoID]);
const endpoint = "/api/skipSegments";
const postSkipSegmentJSON = (data: Record<string, any>) => client({
@ -107,6 +108,17 @@ describe("postSkipSegments", () => {
category: "sponsor",
};
assert.ok(partialDeepEquals(row, expected));
const videoInfo = await queryDatabaseVideoInfo(videoID);
const expectedVideoInfo = {
videoID,
title: "Example Title",
channelID: "ExampleChannel",
published: 123,
genreUrl: ""
};
assert.ok(partialDeepEquals(videoInfo, expectedVideoInfo));
done();
})
.catch(err => done(err));

2
test/youtubeMock.ts
View file

@ -35,6 +35,8 @@ export class YouTubeApiMock {
err: null,
data: {
title: "Example Title",
authorId: "ExampleChannel",
published: 123,
lengthSeconds: 4980,
videoThumbnails: [
{