feat: add an ability to trim cache for Puppeteer (#10199)

test/installation/assets/puppeteer/installCanary.js

@@ -0,0 +1,34 @@
+/**
+ * Copyright 2023 Google Inc. All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import {
+  Browser,
+  detectBrowserPlatform,
+  install,
+  resolveBuildId,
+} from '@puppeteer/browsers';
+
+(async () => {
+  await install({
+    cacheDir: process.env['PUPPETEER_CACHE_DIR'],
+    browser: Browser.CHROME,
+    buildId: await resolveBuildId(
+      Browser.CHROME,
+      detectBrowserPlatform(),
+      'canary'
+    ),
+  });
+})();

test/installation/assets/puppeteer/trimCache.js

@@ -0,0 +1,21 @@
+/**
+ * Copyright 2023 Google Inc. All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import puppeteer from 'puppeteer';
+
+(async () => {
+  await puppeteer.trimCache();
+})();

test/installation/src/puppeteer.spec.ts

@@ -15,6 +15,7 @@
  */
 
 import assert from 'assert';
+import {readdirSync} from 'fs';
 import {readdir} from 'fs/promises';
 import {join} from 'path';
 
@@ -64,3 +65,38 @@ describe('`puppeteer` with PUPPETEER_DOWNLOAD_PATH', () => {
     await this.runScript(script, 'mjs');
   });
 });
+
+describe('`puppeteer` clears cache', () => {
+  configureSandbox({
+    dependencies: ['@puppeteer/browsers', 'puppeteer-core', 'puppeteer'],
+    env: cwd => {
+      return {
+        PUPPETEER_CACHE_DIR: join(cwd, '.cache', 'puppeteer'),
+      };
+    },
+  });
+
+  it('evaluates', async function () {
+    assert.equal(
+      readdirSync(join(this.sandbox, '.cache', 'puppeteer', 'chrome')).length,
+      1
+    );
+
+    await this.runScript(
+      await readAsset('puppeteer', 'installCanary.js'),
+      'mjs'
+    );
+
+    assert.equal(
+      readdirSync(join(this.sandbox, '.cache', 'puppeteer', 'chrome')).length,
+      2
+    );
+
+    await this.runScript(await readAsset('puppeteer', 'trimCache.js'), 'mjs');
+
+    assert.equal(
+      readdirSync(join(this.sandbox, '.cache', 'puppeteer', 'chrome')).length,
+      1
+    );
+  });
+});