diff --git a/skills/gemini-web-generate/scripts/src/anti-ban.ts b/skills/gemini-web-generate/scripts/src/anti-ban.js similarity index 74% rename from skills/gemini-web-generate/scripts/src/anti-ban.ts rename to skills/gemini-web-generate/scripts/src/anti-ban.js index d3ef3bb..42bfc0c 100644 --- a/skills/gemini-web-generate/scripts/src/anti-ban.ts +++ b/skills/gemini-web-generate/scripts/src/anti-ban.js @@ -6,50 +6,41 @@ * - Each wait adds 0-jitterMax seconds of random jitter */ -export interface AntiBanConfig { - baseDelay: number; // seconds - maxDelay: number; // seconds - jitterMax: number; // seconds -} - -const DEFAULT_CONFIG: AntiBanConfig = { +const DEFAULT_CONFIG = { baseDelay: Number(process.env.GEMINI_BASE_DELAY) || 15, maxDelay: Number(process.env.GEMINI_MAX_DELAY) || 120, jitterMax: 5, }; export class AntiBanController { - private currentDelay: number; - private config: AntiBanConfig; - retryCount = 0; - rateLimitEvents = 0; - - constructor(config?: Partial) { + constructor(config) { this.config = { ...DEFAULT_CONFIG, ...config }; this.currentDelay = this.config.baseDelay; + this.retryCount = 0; + this.rateLimitEvents = 0; } /** Call after a successful generation */ - onSuccess(): void { + onSuccess() { this.currentDelay = Math.max(this.config.baseDelay * 0.5, this.currentDelay * 0.7); } /** Call after rate limit / timeout / error */ - onRateLimit(): void { + onRateLimit() { this.rateLimitEvents++; this.currentDelay = Math.min(this.config.maxDelay, this.currentDelay * 2); } /** Wait between requests with jitter */ - async wait(): Promise { + async wait() { const jitter = Math.random() * this.config.jitterMax; const totalMs = (this.currentDelay + jitter) * 1000; - const log = (await import('./browser.js')).log; + const { log } = await import('./browser.js'); log(`[anti-ban] waiting ${(totalMs / 1000).toFixed(1)}s (base=${this.currentDelay.toFixed(1)}s, jitter=${jitter.toFixed(1)}s)`); await new Promise((r) => setTimeout(r, totalMs)); } - get delay(): number { + get delay() { return this.currentDelay; }