create-forgeon 0.3.15 → 0.3.16

package/src/modules/rate-limit.mjs

@@ -12,6 +12,7 @@ import {
   ensureValidatorSchema,
   upsertEnvLines,
 } from './shared/patch-utils.mjs';
+import { ensureWebProbeDefinition, resolveProbeTargets } from './shared/probes.mjs';
 
 function copyFromPreset(packageRoot, targetRoot, relativePath) {
   const source = path.join(packageRoot, 'templates', 'module-presets', 'rate-limit', relativePath);
@@ -113,7 +114,11 @@ function patchAppModule(targetRoot) {
   fs.writeFileSync(filePath, `${content.trimEnd()}\n`, 'utf8');
 }
 
-function patchHealthController(targetRoot) {
+function patchHealthController(targetRoot, probeTargets) {
+  if (!probeTargets.allowApi) {
+    return;
+  }
+
   const filePath = path.join(targetRoot, 'apps', 'api', 'src', 'health', 'health.controller.ts');
   if (!fs.existsSync(filePath)) {
     return;
@@ -139,69 +144,18 @@ function patchHealthController(targetRoot) {
   fs.writeFileSync(filePath, `${content.trimEnd()}\n`, 'utf8');
 }
 
-function patchWebApp(targetRoot) {
-  const filePath = path.join(targetRoot, 'apps', 'web', 'src', 'App.tsx');
-  if (!fs.existsSync(filePath)) {
-    return;
-  }
-
-  let content = fs.readFileSync(filePath, 'utf8').replace(/\r\n/g, '\n');
-  content = content
-    .replace(/^\s*\{\/\* forgeon:probes:actions:start \*\/\}\r?\n?/gm, '')
-    .replace(/^\s*\{\/\* forgeon:probes:actions:end \*\/\}\r?\n?/gm, '')
-    .replace(/^\s*\{\/\* forgeon:probes:results:start \*\/\}\r?\n?/gm, '')
-    .replace(/^\s*\{\/\* forgeon:probes:results:end \*\/\}\r?\n?/gm, '');
-
-  if (!content.includes('rateLimitProbeResult')) {
-    const stateAnchors = [
-      '  const [dbProbeResult, setDbProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [authProbeResult, setAuthProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [validationProbeResult, setValidationProbeResult] = useState<ProbeResult | null>(null);',
-    ];
-    const stateAnchor = stateAnchors.find((line) => content.includes(line));
-    if (stateAnchor) {
-      content = ensureLineAfter(
-        content,
-        stateAnchor,
-        '  const [rateLimitProbeResult, setRateLimitProbeResult] = useState<ProbeResult | null>(null);',
-      );
-    }
-  }
-
-  if (!content.includes('Check rate limit (click repeatedly)')) {
-    const probePath = content.includes("runProbe(setHealthResult, '/health')")
-      ? '/health/rate-limit'
-      : '/api/health/rate-limit';
-    const button = `        <button onClick={() => runProbe(setRateLimitProbeResult, '${probePath}')}>\n          Check rate limit (click repeatedly)\n        </button>`;
-
-    const actionsStart = content.indexOf('<div className="actions">');
-    if (actionsStart >= 0) {
-      const actionsEnd = content.indexOf('\n      </div>', actionsStart);
-      if (actionsEnd >= 0) {
-        content = `${content.slice(0, actionsEnd)}\n${button}${content.slice(actionsEnd)}`;
-      }
-    }
-  }
-
-  if (!content.includes("{renderResult('Rate limit probe response', rateLimitProbeResult)}")) {
-    const resultLine = "      {renderResult('Rate limit probe response', rateLimitProbeResult)}";
-    const networkLine = '      {networkError ? <p className="error">{networkError}</p> : null}';
-    if (content.includes(networkLine)) {
-      content = content.replace(networkLine, `${resultLine}\n${networkLine}`);
-    } else {
-      const anchors = [
-        "{renderResult('Auth probe response', authProbeResult)}",
-        "{renderResult('DB probe response', dbProbeResult)}",
-        "{renderResult('Validation probe response', validationProbeResult)}",
-      ];
-      const anchor = anchors.find((line) => content.includes(line));
-      if (anchor) {
-        content = ensureLineAfter(content, anchor, resultLine);
-      }
-    }
-  }
-
-  fs.writeFileSync(filePath, `${content.trimEnd()}\n`, 'utf8');
+function registerWebProbe(targetRoot, probeTargets) {
+  ensureWebProbeDefinition({
+    targetRoot,
+    probeTargets,
+    definition: {
+      id: 'rate-limit',
+      title: 'Rate Limit',
+      buttonLabel: 'Check rate limit (click repeatedly)',
+      resultTitle: 'Rate limit probe response',
+      path: '/health/rate-limit',
+    },
+  });
 }
 
 function patchApiDockerfile(targetRoot) {
@@ -326,10 +280,12 @@ Operational notes:
 
 export function applyRateLimitModule({ packageRoot, targetRoot }) {
   copyFromPreset(packageRoot, targetRoot, path.join('packages', 'rate-limit'));
+  const probeTargets = resolveProbeTargets({ targetRoot, moduleId: 'rate-limit' });
+
   patchApiPackage(targetRoot);
   patchAppModule(targetRoot);
-  patchHealthController(targetRoot);
-  patchWebApp(targetRoot);
+  patchHealthController(targetRoot, probeTargets);
+  registerWebProbe(targetRoot, probeTargets);
   patchApiDockerfile(targetRoot);
   patchCompose(targetRoot);
   patchReadme(targetRoot);

package/src/modules/rbac.mjs

@@ -10,6 +10,7 @@ import {
   ensureLineBefore,
   ensureNestCommonImport,
 } from './shared/patch-utils.mjs';
+import { ensureWebProbeDefinition, resolveProbeTargets } from './shared/probes.mjs';
 
 function copyFromPreset(packageRoot, targetRoot, relativePath) {
   const source = path.join(packageRoot, 'templates', 'module-presets', 'rbac', relativePath);
@@ -118,7 +119,11 @@ function patchAppModule(targetRoot) {
   fs.writeFileSync(filePath, `${content.trimEnd()}\n`, 'utf8');
 }
 
-function patchHealthController(targetRoot) {
+function patchHealthController(targetRoot, probeTargets) {
+  if (!probeTargets.allowApi) {
+    return;
+  }
+
   const filePath = path.join(targetRoot, 'apps', 'api', 'src', 'health', 'health.controller.ts');
   if (!fs.existsSync(filePath)) {
     return;
@@ -160,70 +165,23 @@ function patchHealthController(targetRoot) {
   fs.writeFileSync(filePath, `${content.trimEnd()}\n`, 'utf8');
 }
 
-function patchWebApp(targetRoot) {
-  const filePath = path.join(targetRoot, 'apps', 'web', 'src', 'App.tsx');
-  if (!fs.existsSync(filePath)) {
-    return;
-  }
-
-  let content = fs.readFileSync(filePath, 'utf8').replace(/\r\n/g, '\n');
-  content = content
-    .replace(/^\s*\{\/\* forgeon:probes:actions:start \*\/\}\r?\n?/gm, '')
-    .replace(/^\s*\{\/\* forgeon:probes:actions:end \*\/\}\r?\n?/gm, '')
-    .replace(/^\s*\{\/\* forgeon:probes:results:start \*\/\}\r?\n?/gm, '')
-    .replace(/^\s*\{\/\* forgeon:probes:results:end \*\/\}\r?\n?/gm, '');
-
-  if (!content.includes('rbacProbeResult')) {
-    const stateAnchors = [
-      '  const [rateLimitProbeResult, setRateLimitProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [dbProbeResult, setDbProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [authProbeResult, setAuthProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [validationProbeResult, setValidationProbeResult] = useState<ProbeResult | null>(null);',
-    ];
-    const stateAnchor = stateAnchors.find((line) => content.includes(line));
-    if (stateAnchor) {
-      content = ensureLineAfter(
-        content,
-        stateAnchor,
-        '  const [rbacProbeResult, setRbacProbeResult] = useState<ProbeResult | null>(null);',
-      );
-    }
-  }
-
-  if (!content.includes('Check RBAC access')) {
-    const useProxyPath = content.includes("runProbe(setHealthResult, '/health')");
-    const probePath = useProxyPath ? '/health/rbac' : '/api/health/rbac';
-    const button = `        <button\n          onClick={() =>\n            runProbe(setRbacProbeResult, '${probePath}', {\n              headers: { 'x-forgeon-permissions': 'health.rbac' },\n            })\n          }\n        >\n          Check RBAC access\n        </button>`;
-
-    const actionsStart = content.indexOf('<div className="actions">');
-    if (actionsStart >= 0) {
-      const actionsEnd = content.indexOf('\n      </div>', actionsStart);
-      if (actionsEnd >= 0) {
-        content = `${content.slice(0, actionsEnd)}\n${button}${content.slice(actionsEnd)}`;
-      }
-    }
-  }
-
-  if (!content.includes("{renderResult('RBAC probe response', rbacProbeResult)}")) {
-    const resultLine = "      {renderResult('RBAC probe response', rbacProbeResult)}";
-    const networkLine = '      {networkError ? <p className="error">{networkError}</p> : null}';
-    if (content.includes(networkLine)) {
-      content = content.replace(networkLine, `${resultLine}\n${networkLine}`);
-    } else {
-      const anchors = [
-        "      {renderResult('Rate limit probe response', rateLimitProbeResult)}",
-        "      {renderResult('Auth probe response', authProbeResult)}",
-        "      {renderResult('DB probe response', dbProbeResult)}",
-        "      {renderResult('Validation probe response', validationProbeResult)}",
-      ];
-      const anchor = anchors.find((line) => content.includes(line));
-      if (anchor) {
-        content = ensureLineAfter(content, anchor, resultLine);
-      }
-    }
-  }
-
-  fs.writeFileSync(filePath, `${content.trimEnd()}\n`, 'utf8');
+function registerWebProbe(targetRoot, probeTargets) {
+  ensureWebProbeDefinition({
+    targetRoot,
+    probeTargets,
+    definition: {
+      id: 'rbac',
+      title: 'RBAC',
+      buttonLabel: 'Check RBAC access',
+      resultTitle: 'RBAC probe response',
+      path: '/health/rbac',
+      request: {
+        headers: {
+          'x-forgeon-permissions': 'health.rbac',
+        },
+      },
+    },
+  });
 }
 
 function patchApiDockerfile(targetRoot) {
@@ -321,10 +279,12 @@ Current scope:
 
 export function applyRbacModule({ packageRoot, targetRoot }) {
   copyFromPreset(packageRoot, targetRoot, path.join('packages', 'rbac'));
+  const probeTargets = resolveProbeTargets({ targetRoot, moduleId: 'rbac' });
+
   patchApiPackage(targetRoot);
   patchAppModule(targetRoot);
-  patchHealthController(targetRoot);
-  patchWebApp(targetRoot);
+  patchHealthController(targetRoot, probeTargets);
+  registerWebProbe(targetRoot, probeTargets);
   patchApiDockerfile(targetRoot);
   patchReadme(targetRoot);
 }

package/src/modules/scheduler.mjs

@@ -3,15 +3,13 @@ import path from 'node:path';
 import { copyRecursive, writeJson } from '../utils/fs.mjs';
 import {
   ensureBuildSteps,
-  ensureClassMember,
   ensureDependency,
-  ensureImportLine,
   ensureLineAfter,
   ensureLineBefore,
-  ensureLoadItem,
-  ensureValidatorSchema,
   upsertEnvLines,
 } from './shared/patch-utils.mjs';
+import { patchAppModuleRegistration, patchHealthControllerServiceProbe } from './shared/nest-runtime-wiring.mjs';
+import { ensureWebProbeDefinition, resolveProbeTargets } from './shared/probes.mjs';
 
 function copyFromPreset(packageRoot, targetRoot, relativePath) {
   const source = path.join(packageRoot, 'templates', 'module-presets', 'scheduler', relativePath);
@@ -35,169 +33,47 @@ function patchApiPackage(targetRoot) {
 }
 
 function patchAppModule(targetRoot) {
-  const filePath = path.join(targetRoot, 'apps', 'api', 'src', 'app.module.ts');
-  if (!fs.existsSync(filePath)) {
-    return;
-  }
-
-  let content = fs.readFileSync(filePath, 'utf8').replace(/\r\n/g, '\n');
-  content = ensureImportLine(
-    content,
-    "import { ForgeonSchedulerModule, schedulerConfig, schedulerEnvSchema } from '@forgeon/scheduler';",
-  );
-  content = ensureLoadItem(content, 'schedulerConfig');
-  content = ensureValidatorSchema(content, 'schedulerEnvSchema');
-
-  if (!content.includes('    ForgeonSchedulerModule,')) {
-    if (content.includes('    ForgeonQueueModule,')) {
-      content = ensureLineAfter(content, '    ForgeonQueueModule,', '    ForgeonSchedulerModule,');
-    } else if (content.includes('    ForgeonI18nModule.register({')) {
-      content = ensureLineBefore(content, '    ForgeonI18nModule.register({', '    ForgeonSchedulerModule,');
-    } else if (content.includes('    ForgeonAuthModule.register({')) {
-      content = ensureLineBefore(content, '    ForgeonAuthModule.register({', '    ForgeonSchedulerModule,');
-    } else if (content.includes('    ForgeonAuthModule.register(),')) {
-      content = ensureLineBefore(content, '    ForgeonAuthModule.register(),', '    ForgeonSchedulerModule,');
-    } else if (content.includes('    DbPrismaModule,')) {
-      content = ensureLineAfter(content, '    DbPrismaModule,', '    ForgeonSchedulerModule,');
-    } else if (content.includes('    ForgeonLoggerModule,')) {
-      content = ensureLineAfter(content, '    ForgeonLoggerModule,', '    ForgeonSchedulerModule,');
-    } else if (content.includes('    ForgeonSwaggerModule,')) {
-      content = ensureLineAfter(content, '    ForgeonSwaggerModule,', '    ForgeonSchedulerModule,');
-    } else {
-      content = ensureLineAfter(content, '    CoreErrorsModule,', '    ForgeonSchedulerModule,');
-    }
-  }
-
-  fs.writeFileSync(filePath, `${content.trimEnd()}\n`, 'utf8');
+  patchAppModuleRegistration(targetRoot, {
+    importLine: "import { ForgeonSchedulerModule, schedulerConfig, schedulerEnvSchema } from '@forgeon/scheduler';",
+    loadItem: 'schedulerConfig',
+    envSchema: 'schedulerEnvSchema',
+    moduleLine: '    ForgeonSchedulerModule,',
+    beforeAnchors: [
+      '    ForgeonI18nModule.register({',
+      '    ForgeonAuthModule.register({',
+      '    ForgeonAuthModule.register(),',
+    ],
+    afterAnchors: [
+      '    ForgeonQueueModule,',
+      '    DbPrismaModule,',
+      '    ForgeonLoggerModule,',
+      '    ForgeonSwaggerModule,',
+    ],
+  });
 }
 
-function patchHealthController(targetRoot) {
-  const filePath = path.join(targetRoot, 'apps', 'api', 'src', 'health', 'health.controller.ts');
-  if (!fs.existsSync(filePath)) {
-    return;
-  }
-
-  let content = fs.readFileSync(filePath, 'utf8').replace(/\r\n/g, '\n');
-  content = ensureImportLine(content, "import { ForgeonSchedulerService } from '@forgeon/scheduler';");
-
-  if (!content.includes('private readonly schedulerService: ForgeonSchedulerService')) {
-    const constructorMatch = content.match(/constructor\(([\s\S]*?)\)\s*\{/m);
-    if (constructorMatch) {
-      const original = constructorMatch[0];
-      const inner = constructorMatch[1].trimEnd();
-      const normalizedInner = inner.replace(/,\s*$/, '');
-      const separator = normalizedInner.length > 0 ? ',' : '';
-      const next = `constructor(${normalizedInner}${separator}
-    private readonly schedulerService: ForgeonSchedulerService,
-  ) {`;
-      content = content.replace(original, next);
-    } else {
-      const classAnchor = 'export class HealthController {';
-      if (content.includes(classAnchor)) {
-        content = content.replace(
-          classAnchor,
-          `${classAnchor}
-  constructor(private readonly schedulerService: ForgeonSchedulerService) {}
-`,
-        );
-      }
-    }
-  }
-
-  if (!content.includes("@Get('scheduler')")) {
-    const method = `
-  @Get('scheduler')
-  async getSchedulerProbe() {
-    return this.schedulerService.getProbeStatus();
-  }
-`;
-    content = ensureClassMember(content, 'HealthController', method, { beforeNeedle: 'private translate(' });
-  }
-
-  fs.writeFileSync(filePath, `${content.trimEnd()}\n`, 'utf8');
+function patchHealthController(targetRoot, probeTargets) {
+  patchHealthControllerServiceProbe(targetRoot, probeTargets, {
+    importLine: "import { ForgeonSchedulerService } from '@forgeon/scheduler';",
+    constructorMember: 'private readonly schedulerService: ForgeonSchedulerService',
+    routePath: 'scheduler',
+    methodName: 'getSchedulerProbe',
+    serviceCall: 'this.schedulerService.getProbeStatus()',
+  });
 }
 
-function patchWebApp(targetRoot) {
-  const filePath = path.join(targetRoot, 'apps', 'web', 'src', 'App.tsx');
-  if (!fs.existsSync(filePath)) {
-    return;
-  }
-
-  let content = fs.readFileSync(filePath, 'utf8').replace(/\r\n/g, '\n');
-  content = content
-    .replace(/^\s*\{\/\* forgeon:probes:actions:start \*\/\}\r?\n?/gm, '')
-    .replace(/^\s*\{\/\* forgeon:probes:actions:end \*\/\}\r?\n?/gm, '')
-    .replace(/^\s*\{\/\* forgeon:probes:results:start \*\/\}\r?\n?/gm, '')
-    .replace(/^\s*\{\/\* forgeon:probes:results:end \*\/\}\r?\n?/gm, '');
-
-  if (!content.includes('schedulerProbeResult')) {
-    const stateAnchors = [
-      '  const [queueProbeResult, setQueueProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [filesImageProbeResult, setFilesImageProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [filesQuotasProbeResult, setFilesQuotasProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [filesAccessProbeResult, setFilesAccessProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [filesVariantsProbeResult, setFilesVariantsProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [filesProbeResult, setFilesProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [rbacProbeResult, setRbacProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [rateLimitProbeResult, setRateLimitProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [authProbeResult, setAuthProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [dbProbeResult, setDbProbeResult] = useState<ProbeResult | null>(null);',
-      '  const [validationProbeResult, setValidationProbeResult] = useState<ProbeResult | null>(null);',
-    ];
-    const stateAnchor = stateAnchors.find((line) => content.includes(line));
-    if (stateAnchor) {
-      content = ensureLineAfter(
-        content,
-        stateAnchor,
-        '  const [schedulerProbeResult, setSchedulerProbeResult] = useState<ProbeResult | null>(null);',
-      );
-    }
-  }
-
-  if (!content.includes('Check scheduler health')) {
-    const probePath = content.includes("runProbe(setHealthResult, '/health')")
-      ? '/health/scheduler'
-      : '/api/health/scheduler';
-    const button = `        <button onClick={() => runProbe(setSchedulerProbeResult, '${probePath}')}>
-          Check scheduler health
-        </button>`;
-
-    const actionsStart = content.indexOf('<div className="actions">');
-    if (actionsStart >= 0) {
-      const actionsEnd = content.indexOf('\n      </div>', actionsStart);
-      if (actionsEnd >= 0) {
-        content = `${content.slice(0, actionsEnd)}\n${button}${content.slice(actionsEnd)}`;
-      }
-    }
-  }
-
-  if (!content.includes("{renderResult('Scheduler probe response', schedulerProbeResult)}")) {
-    const resultLine = "      {renderResult('Scheduler probe response', schedulerProbeResult)}";
-    const networkLine = '      {networkError ? <p className="error">{networkError}</p> : null}';
-    if (content.includes(networkLine)) {
-      content = content.replace(networkLine, `${resultLine}\n${networkLine}`);
-    } else {
-      const anchors = [
-        "{renderResult('Queue probe response', queueProbeResult)}",
-        "{renderResult('Files image probe response', filesImageProbeResult)}",
-        "{renderResult('Files quotas probe response', filesQuotasProbeResult)}",
-        "{renderResult('Files access probe response', filesAccessProbeResult)}",
-        "{renderResult('Files variants probe response', filesVariantsProbeResult)}",
-        "{renderResult('Files probe response', filesProbeResult)}",
-        "{renderResult('RBAC probe response', rbacProbeResult)}",
-        "{renderResult('Rate limit probe response', rateLimitProbeResult)}",
-        "{renderResult('Auth probe response', authProbeResult)}",
-        "{renderResult('DB probe response', dbProbeResult)}",
-        "{renderResult('Validation probe response', validationProbeResult)}",
-      ];
-      const anchor = anchors.find((line) => content.includes(line));
-      if (anchor) {
-        content = ensureLineAfter(content, anchor, resultLine);
-      }
-    }
-  }
-
-  fs.writeFileSync(filePath, `${content.trimEnd()}\n`, 'utf8');
+function registerWebProbe(targetRoot, probeTargets) {
+  ensureWebProbeDefinition({
+    targetRoot,
+    probeTargets,
+    definition: {
+      id: 'scheduler',
+      title: 'Scheduler',
+      buttonLabel: 'Check scheduler health',
+      resultTitle: 'Scheduler probe response',
+      path: '/health/scheduler',
+    },
+  });
 }
 
 function patchApiDockerfile(targetRoot) {
@@ -346,11 +222,12 @@ Operational notes:
 
 export function applySchedulerModule({ packageRoot, targetRoot }) {
   copyFromPreset(packageRoot, targetRoot, path.join('packages', 'scheduler'));
-
+  const probeTargets = resolveProbeTargets({ targetRoot, moduleId: 'scheduler' });
+
   patchApiPackage(targetRoot);
   patchAppModule(targetRoot);
-  patchHealthController(targetRoot);
-  patchWebApp(targetRoot);
+  patchHealthController(targetRoot, probeTargets);
+  registerWebProbe(targetRoot, probeTargets);
   patchApiDockerfile(targetRoot);
   patchCompose(targetRoot);
   patchReadme(targetRoot);
@@ -365,4 +242,4 @@ export function applySchedulerModule({ packageRoot, targetRoot }) {
     'SCHEDULER_TIMEZONE=UTC',
     'SCHEDULER_HEARTBEAT_CRON=*/5 * * * *',
   ]);
-}
+}

package/src/modules/shared/nest-runtime-wiring.mjs

@@ -0,0 +1,110 @@
+import fs from 'node:fs';
+import path from 'node:path';
+import {
+  ensureClassMember,
+  ensureImportLine,
+  ensureLineAfter,
+  ensureLineBefore,
+  ensureLoadItem,
+  ensureValidatorSchema,
+} from './patch-utils.mjs';
+
+function normalize(content) {
+  return content.replace(/\r\n/g, '\n');
+}
+
+export function patchAppModuleRegistration(targetRoot, options) {
+  const {
+    importLine,
+    loadItem,
+    envSchema,
+    moduleLine,
+    afterAnchors = [],
+    beforeAnchors = [],
+    fallbackAnchor = '    CoreErrorsModule,',
+  } = options;
+
+  const filePath = path.join(targetRoot, 'apps', 'api', 'src', 'app.module.ts');
+  if (!fs.existsSync(filePath)) {
+    return;
+  }
+
+  let content = normalize(fs.readFileSync(filePath, 'utf8'));
+  content = ensureImportLine(content, importLine);
+  content = ensureLoadItem(content, loadItem);
+  content = ensureValidatorSchema(content, envSchema);
+
+  if (!content.includes(moduleLine)) {
+    const beforeAnchor = beforeAnchors.find((anchor) => content.includes(anchor));
+    if (beforeAnchor) {
+      content = ensureLineBefore(content, beforeAnchor, moduleLine);
+    } else {
+      const afterAnchor = afterAnchors.find((anchor) => content.includes(anchor)) ?? fallbackAnchor;
+      content = ensureLineAfter(content, afterAnchor, moduleLine);
+    }
+  }
+
+  fs.writeFileSync(filePath, `${content.trimEnd()}\n`, 'utf8');
+}
+
+export function patchHealthControllerServiceProbe(targetRoot, probeTargets, options) {
+  if (!probeTargets.allowApi) {
+    return;
+  }
+
+  const {
+    importLine,
+    constructorMember,
+    routePath,
+    methodName,
+    serviceCall,
+    className = 'HealthController',
+    classAnchor = 'export class HealthController {',
+    beforeNeedles = [],
+    beforeNeedle = 'private translate(',
+  } = options;
+
+  const filePath = path.join(targetRoot, 'apps', 'api', 'src', 'health', 'health.controller.ts');
+  if (!fs.existsSync(filePath)) {
+    return;
+  }
+
+  let content = normalize(fs.readFileSync(filePath, 'utf8'));
+  content = ensureImportLine(content, importLine);
+
+  if (!content.includes(constructorMember)) {
+    const constructorMatch = content.match(/constructor\(([\s\S]*?)\)\s*\{/m);
+    if (constructorMatch) {
+      const original = constructorMatch[0];
+      const inner = constructorMatch[1].trimEnd();
+      const normalizedInner = inner.replace(/,\s*$/, '');
+      const separator = normalizedInner.length > 0 ? ',' : '';
+      const next = `constructor(${normalizedInner}${separator}
+    ${constructorMember},
+  ) {`;
+      content = content.replace(original, next);
+    } else if (content.includes(classAnchor)) {
+      content = content.replace(
+        classAnchor,
+        `${classAnchor}
+  constructor(${constructorMember}) {}
+`,
+      );
+    }
+  }
+
+  const routeDecorator = `@Get('${routePath}')`;
+  if (!content.includes(routeDecorator)) {
+    const method = `
+  ${routeDecorator}
+  async ${methodName}() {
+    return ${serviceCall};
+  }
+`;
+    const resolvedBeforeNeedle =
+      beforeNeedles.find((needle) => content.includes(needle)) ?? beforeNeedle;
+    content = ensureClassMember(content, className, method, { beforeNeedle: resolvedBeforeNeedle });
+  }
+
+  fs.writeFileSync(filePath, `${content.trimEnd()}\n`, 'utf8');
+}