Mirror/SponsorBlock
1
0
Fork 0
mirror of https://github.com/ajayyy/SponsorBlock.git synced 2025-12-08 20:47:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

Move skip rule code to new file

Browse Source
This commit is contained in:
Ajay
2025-06-06 23:39:51 -04:00
parent 1b5dc38750
commit dc5bb5b8aa
6 changed files with 150 additions and 144 deletions
Download Patch File Download Diff File Expand all files Collapse all files

105
src/utils/segmentData.ts
View File

@@ -1,13 +1,12 @@
import { DataCache } from "../../maze-utils/src/cache";
import { getHash, HashedValue } from "../../maze-utils/src/hash";
import Config, { AdvancedSkipRule, SkipRuleAttribute, SkipRuleOperator } from "../config";
import Config, { } from "../config";
import * as CompileConfig from "../../config.json";
import { ActionType, ActionTypes, CategorySelection, CategorySkipOption, SponsorSourceType, SponsorTime, VideoID } from "../types";
import { ActionType, ActionTypes, CategorySkipOption, SponsorSourceType, SponsorTime, VideoID } from "../types";
import { getHashParams } from "./pageUtils";
import { asyncRequestToServer } from "./requests";
import { extensionUserAgent } from "../../maze-utils/src";
import { VideoLabelsCacheData } from "./videoLabels";
import { getVideoDuration } from "../../maze-utils/src/video";
import { getCategorySelection } from "./skipRule";
const segmentDataCache = new DataCache<VideoID, SegmentResponse>(() => {
return {
@@ -105,102 +104,4 @@ function getEnabledActionTypes(forceFullVideo = false): ActionType[] {
}
return actionTypes;
}
export function getCategorySelection(segment: SponsorTime | VideoLabelsCacheData): CategorySelection {
for (const ruleSet of Config.local.skipRules) {
if (ruleSet.rules.every((rule) => isSkipRulePassing(segment, rule))) {
return { name: segment.category, option: ruleSet.skipOption } as CategorySelection;
}
}
for (const selection of Config.config.categorySelections) {
if (selection.name === segment.category) {
return selection;
}
}
return { name: segment.category, option: CategorySkipOption.Disabled} as CategorySelection;
}
function getSkipRuleValue(segment: SponsorTime | VideoLabelsCacheData, rule: AdvancedSkipRule): string | number | undefined {
switch (rule.attribute) {
case SkipRuleAttribute.StartTime:
return (segment as SponsorTime).segment?.[0];
case SkipRuleAttribute.EndTime:
return (segment as SponsorTime).segment?.[1];
case SkipRuleAttribute.Duration:
return (segment as SponsorTime).segment?.[1] - (segment as SponsorTime).segment?.[0];
case SkipRuleAttribute.StartTimePercent: {
const startTime = (segment as SponsorTime).segment?.[0];
if (startTime === undefined) return undefined;
return startTime / getVideoDuration() * 100;
}
case SkipRuleAttribute.EndTimePercent: {
const endTime = (segment as SponsorTime).segment?.[1];
if (endTime === undefined) return undefined;
return endTime / getVideoDuration() * 100;
}
case SkipRuleAttribute.DurationPercent: {
const startTime = (segment as SponsorTime).segment?.[0];
const endTime = (segment as SponsorTime).segment?.[1];
if (startTime === undefined || endTime === undefined) return undefined;
return (endTime - startTime) / getVideoDuration() * 100;
}
case SkipRuleAttribute.Category:
return segment.category;
case SkipRuleAttribute.Description:
return (segment as SponsorTime).description || "";
case SkipRuleAttribute.Source:
switch ((segment as SponsorTime).source) {
case SponsorSourceType.Local:
return "local";
case SponsorSourceType.YouTube:
return "youtube";
case SponsorSourceType.Autogenerated:
return "autogenerated";
case SponsorSourceType.Server:
return "server";
}
break;
default:
return undefined;
}
}
function isSkipRulePassing(segment: SponsorTime | VideoLabelsCacheData, rule: AdvancedSkipRule): boolean {
const value = getSkipRuleValue(segment, rule);
switch (rule.operator) {
case SkipRuleOperator.Less:
return typeof value === "number" && value < (rule.value as number);
case SkipRuleOperator.LessOrEqual:
return typeof value === "number" && value <= (rule.value as number);
case SkipRuleOperator.Greater:
return typeof value === "number" && value > (rule.value as number);
case SkipRuleOperator.GreaterOrEqual:
return typeof value === "number" && value >= (rule.value as number);
case SkipRuleOperator.Equal:
return value === rule.value;
case SkipRuleOperator.NotEqual:
return value !== rule.value;
case SkipRuleOperator.Contains:
return String(value).toLocaleLowerCase().includes(String(rule.value).toLocaleLowerCase());
case SkipRuleOperator.Regex:
return new RegExp(rule.value as string).test(String(value));
default:
return false;
}
}
export function getCategoryDefaultSelection(category: string): CategorySelection {
for (const selection of Config.config.categorySelections) {
if (selection.name === category) {
return selection;
}
}
return { name: category, option: CategorySkipOption.Disabled} as CategorySelection;
}