@crawlee/playwright 4.0.0-beta.13 → 4.0.0-beta.130

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (33) hide show
  1. package/README.md +17 -13
  2. package/index.d.ts +2 -2
  3. package/index.js +1 -1
  4. package/internals/adaptive-playwright-crawler.d.ts +128 -59
  5. package/internals/adaptive-playwright-crawler.js +317 -230
  6. package/internals/enqueue-links/click-elements.d.ts +37 -55
  7. package/internals/enqueue-links/click-elements.js +63 -55
  8. package/internals/playwright-browser-pool.d.ts +71 -0
  9. package/internals/playwright-browser-pool.js +61 -0
  10. package/internals/playwright-crawler.d.ts +193 -124
  11. package/internals/playwright-crawler.js +63 -62
  12. package/internals/playwright-launcher.d.ts +28 -18
  13. package/internals/playwright-launcher.js +19 -18
  14. package/internals/utils/playwright-utils.d.ts +61 -24
  15. package/internals/utils/playwright-utils.js +145 -95
  16. package/internals/utils/rendering-type-prediction.d.ts +28 -13
  17. package/internals/utils/rendering-type-prediction.js +87 -29
  18. package/package.json +18 -14
  19. package/index.d.ts.map +0 -1
  20. package/index.js.map +0 -1
  21. package/internals/adaptive-playwright-crawler.d.ts.map +0 -1
  22. package/internals/adaptive-playwright-crawler.js.map +0 -1
  23. package/internals/enqueue-links/click-elements.d.ts.map +0 -1
  24. package/internals/enqueue-links/click-elements.js.map +0 -1
  25. package/internals/playwright-crawler.d.ts.map +0 -1
  26. package/internals/playwright-crawler.js.map +0 -1
  27. package/internals/playwright-launcher.d.ts.map +0 -1
  28. package/internals/playwright-launcher.js.map +0 -1
  29. package/internals/utils/playwright-utils.d.ts.map +0 -1
  30. package/internals/utils/playwright-utils.js.map +0 -1
  31. package/internals/utils/rendering-type-prediction.d.ts.map +0 -1
  32. package/internals/utils/rendering-type-prediction.js.map +0 -1
  33. package/tsconfig.build.tsbuildinfo +0 -1
@@ -2,46 +2,28 @@ import { isDeepStrictEqual } from 'node:util';
2
2
  import { BasicCrawler } from '@crawlee/basic';
3
3
  import { extractUrlsFromPage } from '@crawlee/browser';
4
4
  import { CheerioCrawler } from '@crawlee/cheerio';
5
- import { Configuration, RequestHandlerError, RequestHandlerResult, Router, Statistics, withCheckedStorageAccess, } from '@crawlee/core';
6
- import { extractUrlsFromCheerio } from '@crawlee/utils';
5
+ import { createStorageTransaction, EnqueueStrategy, OwnedOrInjected, parseArgument, RequestHandlerError, resolveBaseUrlForEnqueueLinksFiltering, Router, Statistics, } from '@crawlee/core';
6
+ import { extractUrlsFromCheerio } from '@crawlee/utils/internal';
7
+ import { z } from 'zod';
7
8
  import { addTimeoutToPromise } from '@apify/timeout';
8
9
  import { PlaywrightCrawler } from './playwright-crawler.js';
9
- import { RenderingTypePredictor } from './utils/rendering-type-prediction.js';
10
- class AdaptivePlaywrightCrawlerStatistics extends Statistics {
11
- state = null; // this needs to be assigned for a valid override, but the initialization is done by a reset() call from the parent constructor
12
- constructor(options = {}) {
13
- super(options);
14
- this.reset();
15
- }
16
- reset() {
17
- super.reset();
18
- this.state.httpOnlyRequestHandlerRuns = 0;
19
- this.state.browserRequestHandlerRuns = 0;
20
- this.state.renderingTypeMispredictions = 0;
21
- }
22
- async _maybeLoadStatistics() {
23
- await super._maybeLoadStatistics();
24
- const savedState = await this.keyValueStore?.getValue(this.persistStateKey);
25
- if (!savedState) {
26
- return;
27
- }
28
- this.state.httpOnlyRequestHandlerRuns = savedState.httpOnlyRequestHandlerRuns;
29
- this.state.browserRequestHandlerRuns = savedState.browserRequestHandlerRuns;
30
- this.state.renderingTypeMispredictions = savedState.renderingTypeMispredictions;
31
- }
32
- trackHttpOnlyRequestHandlerRun() {
33
- this.state.httpOnlyRequestHandlerRuns ??= 0;
34
- this.state.httpOnlyRequestHandlerRuns += 1;
35
- }
36
- trackBrowserRequestHandlerRun() {
37
- this.state.browserRequestHandlerRuns ??= 0;
38
- this.state.browserRequestHandlerRuns += 1;
39
- }
40
- trackRenderingTypeMisprediction() {
41
- this.state.renderingTypeMispredictions ??= 0;
42
- this.state.renderingTypeMispredictions += 1;
43
- }
44
- }
10
+ import { RenderingTypePredictor, } from './utils/rendering-type-prediction.js';
11
+ const adaptiveStatisticStateSchema = z.object({
12
+ /** How many requests were handled by the HTTP-only request handler. */
13
+ httpOnlyRequestHandlerRuns: z.number().default(0),
14
+ /** How many requests were handled in a browser. */
15
+ browserRequestHandlerRuns: z.number().default(0),
16
+ /** How many times the HTTP-only handler produced a result the `resultChecker` rejected. */
17
+ renderingTypeMispredictions: z.number().default(0),
18
+ });
19
+ /**
20
+ * The {@link AdaptivePlaywrightCrawlerStatisticState} fields as a {@link Statistics} state extension, defaults
21
+ * and all. A {@link Statistics} instance to be injected into an {@link AdaptivePlaywrightCrawler} has to carry
22
+ * them - `deserialize.extend()` your own fields onto this one and pass the result as `stateExtension`.
23
+ */
24
+ export const adaptivePlaywrightCrawlerStatisticState = {
25
+ deserialize: adaptiveStatisticStateSchema,
26
+ };
45
27
  const proxyLogMethods = [
46
28
  'error',
47
29
  'exception',
@@ -82,42 +64,80 @@ const proxyLogMethods = [
82
64
  * @experimental
83
65
  */
84
66
  export class AdaptivePlaywrightCrawler extends BasicCrawler {
85
- config;
86
- renderingTypePredictor;
87
- resultChecker;
88
- resultComparator;
89
- preventDirectStorageAccess;
90
- staticContextPipeline;
91
- browserContextPipeline;
92
- individualRequestHandlerTimeoutMillis;
93
- resultObjects = new WeakMap();
94
- teardownHooks = [];
95
- constructor(options = {}, config = Configuration.getGlobalConfig()) {
96
- const { requestHandler, renderingTypeDetectionRatio = 0.1, renderingTypePredictor, resultChecker, resultComparator, statisticsOptions, preventDirectStorageAccess = true, requestHandlerTimeoutSecs = 60, errorHandler, failedRequestHandler, preNavigationHooks, postNavigationHooks, extendContext, contextPipelineBuilder, ...rest } = options;
67
+ #renderingTypePredictor;
68
+ #resultChecker;
69
+ #shouldPropagateError;
70
+ #resultComparator;
71
+ #staticContextPipeline;
72
+ #browserContextPipeline;
73
+ #individualRequestHandlerTimeoutMillis;
74
+ /**
75
+ * The write policy of the per-attempt transactions. Defaults the request queue to `deferred`:
76
+ * a discarded attempt's enqueues must never reach the queue.
77
+ */
78
+ #attemptWritePolicy;
79
+ #teardownHooks = [];
80
+ constructor(options = {}) {
81
+ const { requestHandler, renderingTypeDetectionRatio = 0.1, renderingTypePredictor, resultChecker, shouldPropagateError, resultComparator, statistics, requestHandlerTimeoutSecs = 60, errorHandler, failedRequestHandler, preNavigationHooks = [], postNavigationHooks = [], extendContext, contextPipelineBuilder, transactionalStorage, ...rest } = options;
82
+ // The user's value is replaced by `false` in the `super` call below — validate it separately,
83
+ // wrapped in an object so the error still names the field.
84
+ parseArgument({ transactionalStorage }, z.object({ transactionalStorage: BasicCrawler.optionsShape.transactionalStorage }), 'AdaptivePlaywrightCrawlerOptions');
85
+ // The extra fields are only tracked if the injected instance was built with them - the types enforce that,
86
+ // but plain JS callers would otherwise silently increment `undefined` into a sticky `NaN`. Extend
87
+ // `adaptivePlaywrightCrawlerStatisticState` to satisfy this.
88
+ if (statistics !== undefined) {
89
+ parseArgument(statistics.state, z.object({
90
+ httpOnlyRequestHandlerRuns: z.number(),
91
+ browserRequestHandlerRuns: z.number(),
92
+ renderingTypeMispredictions: z.number(),
93
+ }), 'statistics.state');
94
+ }
95
+ // Per-attempt buffering is load-bearing here: the handler runs up to twice per request and the
96
+ // losing attempt's writes must be discardable.
97
+ if (transactionalStorage === false) {
98
+ throw new Error('AdaptivePlaywrightCrawler requires transactional storage - it runs the request handler ' +
99
+ 'multiple times per request and must be able to discard the storage writes of losing ' +
100
+ 'attempts. `transactionalStorage: false` is therefore not supported; a write policy ' +
101
+ 'object is accepted and forwarded to the per-attempt transactions.');
102
+ }
97
103
  super({
98
104
  ...rest,
99
- // Pass error handlers to the "main" crawler - we only pluck them from `rest` so that they don't go to the sub crawlers
100
105
  errorHandler,
101
106
  failedRequestHandler,
102
- // Same for request handler
103
107
  requestHandler,
104
- // The builder intentionally returns null so that it crashes the crawler when it tries to use this instead of one of two the specialized context pipelines
105
- // (that would be a logical error in this class)
106
- contextPipelineBuilder: () => null,
107
- }, config);
108
- this.config = config;
109
- this.individualRequestHandlerTimeoutMillis = requestHandlerTimeoutSecs * 1000;
110
- this.renderingTypePredictor =
111
- renderingTypePredictor ?? new RenderingTypePredictor({ detectionRatio: renderingTypeDetectionRatio });
112
- this.resultChecker = resultChecker ?? (() => true);
108
+ requestHandlerTimeoutSecs,
109
+ // The base would build a `Statistics` without the adaptive fields, so provide a default that has them.
110
+ // The cast covers a `StatisticStateExtension` that adds further fields - those can only come from an
111
+ // injected instance, in which case this default is never built.
112
+ statistics: statistics ??
113
+ new Statistics({
114
+ logMessage: `${AdaptivePlaywrightCrawler.name} request statistics:`,
115
+ stateExtension: adaptivePlaywrightCrawlerStatisticState,
116
+ }),
117
+ contextPipelineBuilder: contextPipelineBuilder ?? (() => this.buildContextPipeline()),
118
+ // The base crawler must not wrap requests in a transaction of its own - this crawler opens
119
+ // one per request handler attempt in `crawlOne` instead, forwarding the write policy of the
120
+ // user-facing option (validated above) to those.
121
+ transactionalStorage: false,
122
+ });
123
+ this.#individualRequestHandlerTimeoutMillis = requestHandlerTimeoutSecs * 1000;
124
+ // `renderingTypeDetectionRatio` only configures the default predictor - an injected one brings its own
125
+ // detection ratio (and its own state), so the option is ignored in that case.
126
+ this.#renderingTypePredictor = OwnedOrInjected.resolve(renderingTypePredictor, () => new RenderingTypePredictor({ detectionRatio: renderingTypeDetectionRatio }));
127
+ this.#attemptWritePolicy = {
128
+ requestQueue: 'deferred',
129
+ ...(typeof transactionalStorage === 'object' ? transactionalStorage : {}),
130
+ };
131
+ this.#resultChecker = resultChecker ?? (() => true);
132
+ this.#shouldPropagateError = shouldPropagateError ?? (() => false);
113
133
  if (resultComparator !== undefined) {
114
- this.resultComparator = resultComparator;
134
+ this.#resultComparator = resultComparator;
115
135
  }
116
136
  else if (resultChecker !== undefined) {
117
- this.resultComparator = (resultA, resultB) => this.resultChecker(resultA) && this.resultChecker(resultB);
137
+ this.#resultComparator = (resultA, resultB) => this.#resultChecker(resultA) && this.#resultChecker(resultB);
118
138
  }
119
139
  else {
120
- this.resultComparator = (resultA, resultB) => {
140
+ this.#resultComparator = (resultA, resultB) => {
121
141
  return (resultA.datasetItems.length === resultB.datasetItems.length &&
122
142
  resultA.datasetItems.every((itemA, i) => {
123
143
  const itemB = resultB.datasetItems[i];
@@ -125,96 +145,94 @@ export class AdaptivePlaywrightCrawler extends BasicCrawler {
125
145
  }));
126
146
  };
127
147
  }
148
+ // `extendContext` is forwarded to the inner crawlers, which run it *before* navigation (see
149
+ // `BasicCrawler`), keeping the behavior consistent with the non-adaptive crawlers: the
150
+ // extension is visible to the pre/post-navigation hooks and the request handler, but cannot
151
+ // access navigation-dependent members (`page`, `response`, `$`, ...).
152
+ //
153
+ // The adaptive hooks target a subset context (`AdaptiveHookContext`); the casts to the inner
154
+ // crawlers' `PlaywrightHook` type relax that nominal difference. The `ContextPipeline` merges
155
+ // each hook's overrides at runtime regardless of the static type.
128
156
  const staticCrawler = new CheerioCrawler({
129
157
  ...rest,
130
- useSessionPool: false,
131
- statisticsOptions: {
132
- persistenceOptions: { enable: false },
133
- },
134
- preNavigationHooks: [
135
- async (context) => {
136
- for (const hook of preNavigationHooks ?? []) {
137
- await hook(context, undefined);
138
- }
139
- },
140
- ],
141
- postNavigationHooks: [
142
- async (context) => {
143
- for (const hook of postNavigationHooks ?? []) {
144
- await hook(context, undefined);
145
- }
146
- },
147
- ],
148
- }, config);
158
+ statistics: new Statistics({ persistenceOptions: { enable: false } }),
159
+ preNavigationHooks,
160
+ postNavigationHooks,
161
+ extendContext,
162
+ });
149
163
  const browserCrawler = new PlaywrightCrawler({
150
164
  ...rest,
151
- useSessionPool: false,
152
- statisticsOptions: {
153
- persistenceOptions: { enable: false },
154
- },
155
- preNavigationHooks: [
156
- async (context, gotoOptions) => {
157
- for (const hook of preNavigationHooks ?? []) {
158
- await hook(context, gotoOptions);
159
- }
160
- },
161
- ],
162
- postNavigationHooks: [
163
- async (context, gotoOptions) => {
164
- for (const hook of postNavigationHooks ?? []) {
165
- await hook(context, gotoOptions);
166
- }
167
- },
168
- ],
169
- }, config);
170
- this.teardownHooks.push(browserCrawler.teardown.bind(browserCrawler));
171
- this.staticContextPipeline = staticCrawler.contextPipeline
172
- .compose({
165
+ statistics: new Statistics({ persistenceOptions: { enable: false } }),
166
+ preNavigationHooks: preNavigationHooks,
167
+ postNavigationHooks: postNavigationHooks,
168
+ extendContext,
169
+ });
170
+ this.#teardownHooks.push(browserCrawler.teardown.bind(browserCrawler));
171
+ this.#staticContextPipeline = staticCrawler.contextPipeline.compose({
173
172
  action: this.adaptCheerioContext.bind(this),
174
- })
175
- .compose({
176
- action: async (context) => extendContext ? await extendContext(context) : context,
177
173
  });
178
- this.browserContextPipeline = browserCrawler.contextPipeline
179
- .compose({
174
+ this.#browserContextPipeline = browserCrawler.contextPipeline.compose({
180
175
  action: this.adaptPlaywrightContext.bind(this),
181
- })
182
- .compose({
183
- action: async (context) => extendContext ? await extendContext(context) : context,
184
176
  });
185
- this.stats = new AdaptivePlaywrightCrawlerStatistics({
186
- logMessage: `${this.log.getOptions().prefix} request statistics:`,
187
- config,
188
- ...statisticsOptions,
177
+ }
178
+ async init() {
179
+ // Only the predictor we built ourselves is ours to initialize - an injected one is borrowed, so its
180
+ // lifecycle (including restoring persisted state) stays with whoever created it.
181
+ await this.#renderingTypePredictor.ifOwned((predictor) => predictor.initialize());
182
+ return await super.init();
183
+ }
184
+ buildContextPipeline() {
185
+ const errorMessage = (prop) => `The \`${prop}\` property is not available on the outer context pipeline of AdaptivePlaywrightCrawler - it is provided by the inner (static/browser) pipelines`;
186
+ return super.buildContextPipeline().compose({
187
+ action: async ({ request }) => ({
188
+ get request() {
189
+ return request;
190
+ },
191
+ get response() {
192
+ throw new Error(errorMessage('response'));
193
+ },
194
+ get page() {
195
+ throw new Error(errorMessage('page'));
196
+ },
197
+ get querySelector() {
198
+ throw new Error(errorMessage('querySelector'));
199
+ },
200
+ get querySelectorAll() {
201
+ throw new Error(errorMessage('querySelectorAll'));
202
+ },
203
+ get waitForSelector() {
204
+ throw new Error(errorMessage('waitForSelector'));
205
+ },
206
+ get parseWithCheerio() {
207
+ throw new Error(errorMessage('parseWithCheerio'));
208
+ },
209
+ get enqueueLinks() {
210
+ throw new Error(errorMessage('enqueueLinks'));
211
+ },
212
+ }),
189
213
  });
190
- this.preventDirectStorageAccess = preventDirectStorageAccess;
191
214
  }
192
215
  async adaptCheerioContext(cheerioContext) {
193
- const enqueueLinks = this.resultObjects.get(cheerioContext)?.enqueueLinks;
194
- if (enqueueLinks === undefined) {
195
- throw new Error('Logical error - `this.resultObjects` does not contain the result object');
196
- }
197
216
  return {
198
217
  get page() {
199
218
  throw new Error('Page object was used in HTTP-only request handler');
200
219
  },
201
220
  async querySelector(selector) {
221
+ return cheerioContext.$(selector).first();
222
+ },
223
+ async querySelectorAll(selector) {
202
224
  return cheerioContext.$(selector);
203
225
  },
204
- async enqueueLinks(options = {}) {
205
- const urls = options.urls ??
206
- extractUrlsFromCheerio(cheerioContext.$, options.selector, options.baseUrl ?? cheerioContext.request.loadedUrl);
207
- await enqueueLinks({ ...options, urls });
226
+ enqueueLinks: async (options = {}) => {
227
+ const urls = extractUrlsFromCheerio(cheerioContext.$, options.selector, options.baseUrl ?? cheerioContext.request.loadedUrl);
228
+ return (await this.enqueueLinks(urls, options, cheerioContext.request));
208
229
  },
209
230
  response: cheerioContext.response,
210
231
  };
211
232
  }
212
233
  async adaptPlaywrightContext(playwrightContext) {
234
+ // Capture the original response to avoid infinite recursion when the getter is copied to the context
213
235
  const originalResponse = playwrightContext.response;
214
- const enqueueLinks = this.resultObjects.get(playwrightContext)?.enqueueLinks;
215
- if (enqueueLinks === undefined) {
216
- throw new Error('Logical error - `this.resultObjects` does not contain the result object');
217
- }
218
236
  return {
219
237
  response: new Response(Uint8Array.from(await originalResponse.body()), {
220
238
  headers: originalResponse.headers(),
@@ -222,52 +240,67 @@ export class AdaptivePlaywrightCrawler extends BasicCrawler {
222
240
  statusText: originalResponse.statusText(),
223
241
  }),
224
242
  async querySelector(selector, timeoutMs = 5000) {
243
+ const locator = playwrightContext.page.locator(selector).first();
244
+ await locator.waitFor({ timeout: timeoutMs, state: 'attached' });
245
+ const $ = await playwrightContext.parseWithCheerio();
246
+ return $(selector).first();
247
+ },
248
+ async querySelectorAll(selector, timeoutMs = 5000) {
225
249
  const locator = playwrightContext.page.locator(selector).first();
226
250
  await locator.waitFor({ timeout: timeoutMs, state: 'attached' });
227
251
  const $ = await playwrightContext.parseWithCheerio();
228
252
  return $(selector);
229
253
  },
230
- async enqueueLinks(options = {}, timeoutMs = 5000) {
254
+ enqueueLinks: async (options = {}, timeoutMs = 5000) => {
255
+ // TODO consider using `context.parseWithCheerio` to make this universal and avoid code duplication
231
256
  const selector = options.selector ?? 'a';
232
257
  const locator = playwrightContext.page.locator(selector).first();
233
258
  await locator.waitFor({ timeout: timeoutMs, state: 'attached' });
234
- // TODO consider using `context.parseWithCheerio` to make this universal and avoid code duplication
235
- const urls = options.urls ??
236
- (await extractUrlsFromPage(playwrightContext.page, selector, options.baseUrl ?? playwrightContext.request.loadedUrl));
237
- await enqueueLinks({ ...options, urls });
259
+ const urls = await extractUrlsFromPage(playwrightContext.page, selector, options.baseUrl ?? playwrightContext.request.loadedUrl);
260
+ return (await this.enqueueLinks(urls, options, playwrightContext.request));
238
261
  },
239
262
  };
240
263
  }
241
- async crawlOne(renderingType, context, useStateFunction) {
242
- const result = new RequestHandlerResult(this.config, AdaptivePlaywrightCrawler.CRAWLEE_STATE_KEY);
264
+ /**
265
+ * Runs one request handler attempt inside its own {@link StorageTransaction}, wrapping the inner
266
+ * (static or browser) context pipeline. The transaction is pushed to `transactions` *at creation
267
+ * time, before the `try`* - the `ok: false` branch of the returned {@link Result} carries no
268
+ * result, and failed attempts are routine here. The caller owns the outcome and disposal.
269
+ */
270
+ async crawlOne(renderingType, context, useStateFunction, transactions) {
271
+ const transaction = createStorageTransaction({
272
+ policy: this.#attemptWritePolicy,
273
+ commitTimeoutMillis: this.internalTimeoutMillis,
274
+ });
275
+ transactions.push(transaction);
243
276
  const logs = [];
244
277
  const deferredCleanup = [];
245
- const resultBoundContextHelpers = {
246
- addRequests: result.addRequests,
247
- pushData: result.pushData,
248
- useState: this.allowStorageAccess(useStateFunction),
249
- getKeyValueStore: this.allowStorageAccess(result.getKeyValueStore),
250
- enqueueLinks: result.enqueueLinks,
278
+ const attemptBoundContextHelpers = {
279
+ useState: useStateFunction,
251
280
  log: this.createLogProxy(context.log, logs),
252
281
  registerDeferredCleanup: (cleanup) => deferredCleanup.push(cleanup),
253
282
  };
254
- const subCrawlerContext = { ...context, ...resultBoundContextHelpers };
255
- this.resultObjects.set(subCrawlerContext, result);
283
+ const subCrawlerContext = Object.defineProperties({}, Object.getOwnPropertyDescriptors(context));
284
+ // Mark attempt-bound helpers as non-configurable so they survive the sub-crawler context pipeline
285
+ // (which would otherwise override them with the sub-crawler's own versions, losing the binding).
286
+ for (const [key, descriptor] of Object.entries(Object.getOwnPropertyDescriptors(attemptBoundContextHelpers))) {
287
+ Object.defineProperty(subCrawlerContext, key, { ...descriptor, configurable: false });
288
+ }
256
289
  try {
257
290
  const callAdaptiveRequestHandler = async () => {
258
291
  if (renderingType === 'static') {
259
- await this.staticContextPipeline.call(subCrawlerContext, async (finalContext) => await this.requestHandler(finalContext));
292
+ await this.#staticContextPipeline.call(subCrawlerContext, this.requestHandler.bind(this));
260
293
  }
261
294
  else if (renderingType === 'clientOnly') {
262
- await this.browserContextPipeline.call(subCrawlerContext, async (finalContext) => await this.requestHandler(finalContext));
295
+ await this.#browserContextPipeline.call(subCrawlerContext, this.requestHandler.bind(this));
263
296
  }
264
297
  };
265
- await addTimeoutToPromise(async () => withCheckedStorageAccess(() => {
266
- if (this.preventDirectStorageAccess) {
267
- throw new Error('Directly accessing storage in a request handler is not allowed in AdaptivePlaywrightCrawler');
268
- }
269
- }, callAdaptiveRequestHandler), this.individualRequestHandlerTimeoutMillis, 'Request handler timed out');
270
- return { result, ok: true, logs };
298
+ // this crawler overrides `runRequestHandler` and times each rendering-type run itself, so it has
299
+ // to resolve any per-route override too - otherwise routes would be silently ignored here
300
+ const routeTimeoutSecs = this.requestHandler.getTimeoutSecs?.(context.request.label);
301
+ const timeoutMillis = routeTimeoutSecs === undefined ? this.#individualRequestHandlerTimeoutMillis : routeTimeoutSecs * 1000;
302
+ await addTimeoutToPromise(async () => transaction.run(callAdaptiveRequestHandler), timeoutMillis, 'Request handler timed out');
303
+ return { result: transaction, ok: true, logs };
271
304
  }
272
305
  catch (error) {
273
306
  return { error, ok: false, logs };
@@ -277,89 +310,116 @@ export class AdaptivePlaywrightCrawler extends BasicCrawler {
277
310
  }
278
311
  }
279
312
  async runRequestHandler(crawlingContext) {
280
- const renderingTypePrediction = this.renderingTypePredictor.predict(crawlingContext.request);
313
+ const renderingTypePrediction = this.#renderingTypePredictor.value.predict(crawlingContext.request);
281
314
  const shouldDetectRenderingType = Math.random() < renderingTypePrediction.detectionProbabilityRecommendation;
282
315
  if (!shouldDetectRenderingType) {
283
316
  crawlingContext.log.debug(`Predicted rendering type ${renderingTypePrediction.renderingType} for ${crawlingContext.request.url}`);
284
317
  }
285
- if (renderingTypePrediction.renderingType === 'static' && !shouldDetectRenderingType) {
286
- crawlingContext.log.debug(`Running HTTP-only request handler for ${crawlingContext.request.url}`);
287
- this.stats.trackHttpOnlyRequestHandlerRun();
288
- const plainHTTPRun = await this.crawlOne('static', crawlingContext, crawlingContext.useState);
289
- if (plainHTTPRun.ok && this.resultChecker(plainHTTPRun.result)) {
290
- crawlingContext.log.debug(`HTTP-only request handler succeeded for ${crawlingContext.request.url}`);
291
- plainHTTPRun.logs?.forEach(([log, method, ...args]) => log[method](...args));
292
- await this.commitResult(crawlingContext, plainHTTPRun.result);
293
- return;
294
- }
295
- // Execution will "fall through" and try running the request handler in a browser
296
- if (!plainHTTPRun.ok) {
297
- const actualError = plainHTTPRun.error instanceof RequestHandlerError
298
- ? plainHTTPRun.error.cause
299
- : plainHTTPRun.error;
300
- crawlingContext.log.exception(actualError, `HTTP-only request handler failed for ${crawlingContext.request.url}`);
301
- }
302
- else {
303
- crawlingContext.log.warning(`HTTP-only request handler returned a suspicious result for ${crawlingContext.request.url}`);
304
- this.stats.trackRenderingTypeMisprediction();
305
- }
306
- }
307
- crawlingContext.log.debug(`Running browser request handler for ${crawlingContext.request.url}`);
308
- this.stats.trackBrowserRequestHandlerRun();
309
- // Run the request handler in a browser. The copy of the crawler state is kept so that we can perform
310
- // a rendering type detection if necessary. Without this measure, the HTTP request handler would run
311
- // under different conditions, which could change its behavior. Changes done to the crawler state by
312
- // the HTTP request handler will not be committed to the actual storage.
313
- const stateTracker = {
314
- stateCopy: null,
315
- async getLiveState(defaultValue = {}) {
316
- const state = await crawlingContext.useState(defaultValue);
317
- if (this.stateCopy === null) {
318
- this.stateCopy = JSON.parse(JSON.stringify(state));
319
- }
320
- return state;
321
- },
322
- async getStateCopy(defaultValue = {}) {
323
- if (this.stateCopy === null) {
324
- return defaultValue;
318
+ // Every transaction created for this request - up to two, since the static-then-browser
319
+ // fall-through and the browser-then-detection pair are mutually exclusive. Disposed in the
320
+ // `finally` below, not earlier: the comparators read the journals after `crawlOne` returns.
321
+ const transactions = [];
322
+ try {
323
+ if (renderingTypePrediction.renderingType === 'static' && !shouldDetectRenderingType) {
324
+ crawlingContext.log.debug(`Running HTTP-only request handler for ${crawlingContext.request.url}`);
325
+ this.statistics.state.httpOnlyRequestHandlerRuns++;
326
+ const plainHTTPRun = await this.crawlOne('static', crawlingContext, crawlingContext.useState, transactions);
327
+ if (plainHTTPRun.ok && this.#resultChecker(plainHTTPRun.result)) {
328
+ crawlingContext.log.debug(`HTTP-only request handler succeeded for ${crawlingContext.request.url}`);
329
+ plainHTTPRun.logs?.forEach(([log, method, ...args]) => log[method](...args));
330
+ await plainHTTPRun.result.commit();
331
+ return;
325
332
  }
326
- return this.stateCopy;
327
- },
328
- };
329
- const browserRun = await this.crawlOne('clientOnly', crawlingContext, stateTracker.getLiveState.bind(stateTracker));
330
- if (!browserRun.ok) {
331
- throw browserRun.error;
332
- }
333
- await this.commitResult(crawlingContext, browserRun.result);
334
- if (shouldDetectRenderingType) {
335
- crawlingContext.log.debug(`Detecting rendering type for ${crawlingContext.request.url}`);
336
- const plainHTTPRun = await this.crawlOne('static', crawlingContext, stateTracker.getStateCopy.bind(stateTracker));
337
- const detectionResult = (() => {
333
+ // Execution will "fall through" and try running the request handler in a browser
338
334
  if (!plainHTTPRun.ok) {
339
- return 'clientOnly';
335
+ const actualError = plainHTTPRun.error instanceof RequestHandlerError
336
+ ? plainHTTPRun.error.cause
337
+ : plainHTTPRun.error;
338
+ if (await this.#shouldPropagateError(actualError, crawlingContext)) {
339
+ throw actualError;
340
+ }
341
+ crawlingContext.log.exception(actualError, `HTTP-only request handler failed for ${crawlingContext.request.url}`);
342
+ }
343
+ else {
344
+ crawlingContext.log.warning(`HTTP-only request handler returned a suspicious result for ${crawlingContext.request.url}`);
345
+ this.statistics.state.renderingTypeMispredictions++;
340
346
  }
341
- if (this.resultComparator(plainHTTPRun.result, browserRun.result)) {
342
- return 'static';
347
+ }
348
+ crawlingContext.log.debug(`Running browser request handler for ${crawlingContext.request.url}`);
349
+ this.statistics.state.browserRequestHandlerRuns++;
350
+ // Run the request handler in a browser. The copy of the crawler state is kept so that we can perform
351
+ // a rendering type detection if necessary. Without this measure, the HTTP request handler would run
352
+ // under different conditions, which could change its behavior. Changes done to the crawler state by
353
+ // the HTTP request handler will not be committed to the actual storage.
354
+ const stateTracker = {
355
+ stateCopy: null,
356
+ async getLiveState(defaultValue = {}) {
357
+ const state = await crawlingContext.useState(defaultValue);
358
+ if (this.stateCopy === null) {
359
+ this.stateCopy = JSON.parse(JSON.stringify(state));
360
+ }
361
+ return state;
362
+ },
363
+ async getStateCopy(defaultValue = {}) {
364
+ if (this.stateCopy === null) {
365
+ return defaultValue;
366
+ }
367
+ return this.stateCopy;
368
+ },
369
+ };
370
+ const browserRun = await this.crawlOne('clientOnly', crawlingContext, stateTracker.getLiveState.bind(stateTracker), transactions);
371
+ if (!browserRun.ok) {
372
+ throw browserRun.error;
373
+ }
374
+ browserRun.logs?.forEach(([log, method, ...args]) => log[method](...args));
375
+ await browserRun.result.commit();
376
+ if (shouldDetectRenderingType) {
377
+ crawlingContext.log.debug(`Detecting rendering type for ${crawlingContext.request.url}`);
378
+ // The detection attempt's transaction is never committed - its writes exist only for the
379
+ // result comparison.
380
+ const plainHTTPRun = await this.crawlOne('static', crawlingContext, stateTracker.getStateCopy.bind(stateTracker), transactions);
381
+ const detectionResult = (() => {
382
+ if (!plainHTTPRun.ok) {
383
+ return 'clientOnly';
384
+ }
385
+ const comparisonResult = this.#resultComparator(plainHTTPRun.result, browserRun.result);
386
+ if (comparisonResult === true || comparisonResult === 'equal') {
387
+ return 'static';
388
+ }
389
+ if (comparisonResult === false || comparisonResult === 'different') {
390
+ return 'clientOnly';
391
+ }
392
+ return undefined;
393
+ })();
394
+ crawlingContext.log.debug(`Detected rendering type ${detectionResult} for ${crawlingContext.request.url}`);
395
+ if (detectionResult !== undefined) {
396
+ this.#renderingTypePredictor.value.storeResult(crawlingContext.request, detectionResult);
343
397
  }
344
- return 'clientOnly';
345
- })();
346
- crawlingContext.log.debug(`Detected rendering type ${detectionResult} for ${crawlingContext.request.url}`);
347
- this.renderingTypePredictor.storeResult(crawlingContext.request, detectionResult);
398
+ }
399
+ }
400
+ finally {
401
+ // A still-open transaction here belongs to a discarded attempt - roll it back, then release.
402
+ for (const transaction of transactions) {
403
+ transaction.rollback();
404
+ transaction.dispose();
405
+ }
348
406
  }
349
407
  }
350
- async commitResult(crawlingContext, { calls, keyValueStoreChanges }) {
351
- await Promise.all([
352
- ...calls.pushData.map(async (params) => crawlingContext.pushData(...params)),
353
- ...calls.enqueueLinks.map(async (params) => await crawlingContext.enqueueLinks(...params)),
354
- ...calls.addRequests.map(async (params) => crawlingContext.addRequests(...params)),
355
- ...Object.entries(keyValueStoreChanges).map(async ([storeIdOrName, changes]) => {
356
- const store = await crawlingContext.getKeyValueStore(storeIdOrName);
357
- await Promise.all(Object.entries(changes).map(async ([key, { changedValue, options }]) => store.setValue(key, changedValue, options)));
358
- }),
359
- ]);
360
- }
361
- allowStorageAccess(func) {
362
- return async (...args) => withCheckedStorageAccess(() => { }, async () => func(...args));
408
+ async enqueueLinks(urls, options, request) {
409
+ const baseUrl = resolveBaseUrlForEnqueueLinksFiltering({
410
+ enqueueStrategy: options?.strategy,
411
+ finalRequestUrl: request.loadedUrl,
412
+ originalRequestUrl: request.url,
413
+ userProvidedBaseUrl: options?.baseUrl,
414
+ });
415
+ const requestsWithDepth = this.addCrawlDepthRequestGenerator(urls, request.crawlDepth + 1);
416
+ // The per-attempt transaction buffers these (the queue policy defaults to `deferred` here),
417
+ // so a discarded attempt's enqueues never reach the queue.
418
+ return await this.addRequests(requestsWithDepth, {
419
+ ...options,
420
+ baseUrl,
421
+ strategy: options.strategy ?? EnqueueStrategy.SameHostname,
422
+ });
363
423
  }
364
424
  createLogProxy(log, logs) {
365
425
  return new Proxy(log, {
@@ -375,12 +435,39 @@ export class AdaptivePlaywrightCrawler extends BasicCrawler {
375
435
  }
376
436
  async teardown() {
377
437
  await super.teardown();
378
- for (const hook of this.teardownHooks) {
438
+ for (const hook of this.#teardownHooks) {
379
439
  await hook();
380
440
  }
381
441
  }
382
442
  }
383
- export function createAdaptivePlaywrightRouter(routes) {
384
- return Router.create(routes);
443
+ export function createAdaptivePlaywrightRouter(routesOrSchemas) {
444
+ return Router.create(routesOrSchemas);
445
+ }
446
+ /**
447
+ * An opt-in {@link AdaptivePlaywrightCrawlerOptions.resultComparator|`resultComparator`} that considers two
448
+ * request handler results equal only if *all* of their observable effects match - the pushed dataset items, the
449
+ * enqueued requests, and the key-value store changes. This is stricter than the default comparator, which only
450
+ * compares dataset items.
451
+ *
452
+ * **Beware:** enqueued URLs are compared exactly. The same page rendered in a browser and via plain HTTP often
453
+ * yields links that differ only in tracking query parameters, for example:
454
+ * - `https://sdk.apify.com/docs/guides/getting-started`
455
+ * - `https://sdk.apify.com/docs/guides/getting-started?__hsfp=1136113150&__hssc=7591405.1.173549427712`
456
+ *
457
+ * Such links are treated as *different*, which will make the crawler favor browser rendering for those pages.
458
+ *
459
+ * **Example usage:**
460
+ * ```ts
461
+ * const crawler = new AdaptivePlaywrightCrawler({
462
+ * resultComparator: fullResultComparator,
463
+ * async requestHandler({ pushData, enqueueLinks }) {
464
+ * // ...
465
+ * },
466
+ * });
467
+ * ```
468
+ */
469
+ export function fullResultComparator(resultA, resultB) {
470
+ return (isDeepStrictEqual(resultA.datasetItems, resultB.datasetItems) &&
471
+ isDeepStrictEqual(resultA.enqueuedUrls, resultB.enqueuedUrls) &&
472
+ isDeepStrictEqual(resultA.keyValueStoreChanges, resultB.keyValueStoreChanges));
385
473
  }
386
- //# sourceMappingURL=adaptive-playwright-crawler.js.map