forked from darkreader/darkreader
-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy paththrottle.ts
More file actions
84 lines (71 loc) · 2.02 KB
/
Copy paththrottle.ts
File metadata and controls
84 lines (71 loc) · 2.02 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
export function throttle<T extends(...args: any[]) => any>(callback: T): T & {cancel: () => void} {
let pending = false;
let frameId: number | null = null;
let lastArgs: any[];
const throttled: T = ((...args: any[]) => {
lastArgs = args;
if (frameId) {
pending = true;
} else {
callback(...lastArgs);
frameId = requestAnimationFrame(() => {
frameId = null;
if (pending) {
callback(...lastArgs);
pending = false;
}
});
}
}) as any;
const cancel = () => {
// TODO: reove cast once types are updated
cancelAnimationFrame(frameId!);
pending = false;
frameId = null;
};
return Object.assign(throttled, {cancel});
}
type Task = () => void;
declare const __TEST__: boolean;
interface AsyncTaskQueue {
add: (task: Task) => void;
cancel: () => void;
}
export function createAsyncTasksQueue(): AsyncTaskQueue {
const tasks: Task[] = [];
let frameId: number | null = null;
function runTasks() {
let task: Task | undefined;
while ((task = tasks.shift())) {
task();
}
frameId = null;
if (__TEST__) {
document.dispatchEvent(new CustomEvent('__darkreader__test__asyncQueueComplete'));
}
}
function add(task: Task) {
tasks.push(task);
if (!frameId) {
frameId = requestAnimationFrame(runTasks);
}
}
function cancel() {
tasks.splice(0);
// TODO: reove cast once types are updated
cancelAnimationFrame(frameId!);
frameId = null;
}
return {add, cancel};
}
const delayTokens = new Set<symbol>;
export function requestAnimationFrameOnce(token: symbol, callback: () => void): void {
if (delayTokens.has(token)) {
return;
}
delayTokens.add(token);
requestAnimationFrame(() => {
delayTokens.delete(token);
callback();
});
}