feat(plugin-lighthouse): add onlyGroups support

Author: hanna-skryl

packages/plugin-lighthouse/src/lib/binding.ts

@@ -10,9 +10,11 @@ import {
   singleQuote,
 } from '@code-pushup/utils';
 import {
+  LIGHTHOUSE_GROUP_SLUGS,
   LIGHTHOUSE_PLUGIN_SLUG,
   LIGHTHOUSE_PLUGIN_TITLE,
 } from './constants.js';
+import type { LighthouseGroupSlug } from './types.js';
 
 const { name: PACKAGE_NAME } = createRequire(import.meta.url)(
   '../../package.json',
@@ -21,6 +23,13 @@ const { name: PACKAGE_NAME } = createRequire(import.meta.url)(
 const DEFAULT_URL = 'http://localhost:4200';
 const PLUGIN_VAR = 'lhPlugin';
 
+const CATEGORY_TO_GROUP: Record<string, LighthouseGroupSlug> = {
+  performance: 'performance',
+  a11y: 'accessibility',
+  'best-practices': 'best-practices',
+  seo: 'seo',
+};
+
 const CATEGORIES: CategoryCodegenConfig[] = [
   {
     slug: 'performance',
@@ -86,25 +95,26 @@ export const lighthouseSetupBinding = {
   generateConfig: (answers: Record<string, PluginAnswer>) => {
     const options = parseAnswers(answers);
     const hasCategories = options.categories.length > 0;
-    const formattedUrls = formatUrls(options.urls);
     const imports = [
       {
         moduleSpecifier: PACKAGE_NAME,
         defaultImport: 'lighthousePlugin',
         ...(hasCategories ? { namedImports: ['lighthouseGroupRefs'] } : {}),
       },
     ];
+    const pluginCall = formatPluginCall(options);
+
     if (!hasCategories) {
       return {
         imports,
-        pluginInit: [`lighthousePlugin(${formattedUrls}),`],
+        pluginInit: [`${pluginCall},`],
       };
     }
     return {
       imports,
       pluginDeclaration: {
         identifier: PLUGIN_VAR,
-        expression: `lighthousePlugin(${formattedUrls})`,
+        expression: pluginCall,
       },
       pluginInit: [`${PLUGIN_VAR},`],
       categories: createCategories(options),
@@ -121,6 +131,19 @@ function parseAnswers(
   };
 }
 
+function formatPluginCall({ urls, categories }: LighthouseOptions): string {
+  const formattedUrls = formatUrls(urls);
+  const groups = categories.flatMap(slug => {
+    const group = CATEGORY_TO_GROUP[slug];
+    return group ? [group] : [];
+  });
+  if (groups.length === 0 || groups.length === LIGHTHOUSE_GROUP_SLUGS.length) {
+    return `lighthousePlugin(${formattedUrls})`;
+  }
+  const onlyGroups = groups.map(singleQuote).join(', ');
+  return `lighthousePlugin(${formattedUrls}, { onlyGroups: [${onlyGroups}] })`;
+}
+
 function formatUrls([first, ...rest]: [string, ...string[]]): string {
   if (rest.length === 0) {
     return singleQuote(first);

packages/plugin-lighthouse/src/lib/binding.unit.test.ts

@@ -73,6 +73,21 @@ describe('lighthouseSetupBinding', () => {
       ]);
     });
 
+    it('should pass onlyGroups when not all categories are selected', () => {
+      const { pluginDeclaration } = binding.generateConfig({
+        ...defaultAnswers,
+        'lighthouse.categories': ['performance', 'seo'],
+      });
+      expect(pluginDeclaration!.expression).toContain(
+        "onlyGroups: ['performance', 'seo']",
+      );
+    });
+
+    it('should omit onlyGroups when all categories are selected', () => {
+      const { pluginDeclaration } = binding.generateConfig(defaultAnswers);
+      expect(pluginDeclaration!.expression).not.toContain('onlyGroups');
+    });
+
     it('should use custom URL in plugin declaration', () => {
       expect(
         binding.generateConfig({