@nx/angular/src/generators/setup-ssr/lib/add-hydration.js

The Nx Plugin for Angular contains executors, generators, and utilities for managing Angular applications and libraries within an Nx workspace. It provides: - Integration with libraries such as Storybook, Jest, ESLint, Tailwind CSS, Playwright and Cypre

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.addHydration = addHydration;
const devkit_1 = require("@nx/devkit");
const js_1 = require("@nx/js");
const ensure_typescript_1 = require("@nx/js/src/utils/typescript/ensure-typescript");
const ts_solution_setup_1 = require("@nx/js/src/utils/typescript/ts-solution-setup");
const ast_utils_1 = require("../../../utils/nx-devkit/ast-utils");
const version_utils_1 = require("../../utils/version-utils");
let tsModule;
let tsquery;
function addHydration(tree, options) {
    const projectConfig = (0, devkit_1.readProjectConfiguration)(tree, options.project);
    const sourceRoot = (0, ts_solution_setup_1.getProjectSourceRoot)(projectConfig, tree);
    if (!tsModule) {
        tsModule = (0, ensure_typescript_1.ensureTypescript)();
        tsquery = require('@phenomnomnominal/tsquery').tsquery;
    }
    let pathToClientConfigFile;
    if (options.standalone) {
        pathToClientConfigFile = (0, devkit_1.joinPathFragments)(sourceRoot, 'app/app.config.ts');
    }
    else {
        pathToClientConfigFile = (0, devkit_1.joinPathFragments)(sourceRoot, 'app/app.module.ts');
        if (!tree.exists(pathToClientConfigFile)) {
            pathToClientConfigFile = (0, devkit_1.joinPathFragments)(sourceRoot, 'app/app-module.ts');
        }
    }
    const sourceText = tree.read(pathToClientConfigFile, 'utf-8');
    let sourceFile = tsModule.createSourceFile(pathToClientConfigFile, sourceText, tsModule.ScriptTarget.Latest, true);
    const provideClientHydrationCallExpression = tsquery(sourceFile, 'ObjectLiteralExpression PropertyAssignment:has(Identifier[name=providers]) ArrayLiteralExpression CallExpression:has(Identifier[name=provideClientHydration])')[0];
    if (provideClientHydrationCallExpression) {
        return;
    }
    const addImport = (source, symbolName, packageName, filePath, isDefault = false) => {
        return (0, js_1.insertImport)(tree, source, filePath, symbolName, packageName, isDefault);
    };
    const { major: angularMajorVersion } = (0, version_utils_1.getInstalledAngularVersionInfo)(tree);
    sourceFile = addImport(sourceFile, 'provideClientHydration', '@angular/platform-browser', pathToClientConfigFile);
    if (angularMajorVersion >= 19) {
        sourceFile = addImport(sourceFile, 'withEventReplay', '@angular/platform-browser', pathToClientConfigFile);
    }
    const provider = angularMajorVersion >= 19
        ? 'provideClientHydration(withEventReplay())'
        : 'provideClientHydration()';
    if (options.standalone) {
        (0, ast_utils_1.addProviderToAppConfig)(tree, pathToClientConfigFile, provider);
    }
    else {
        (0, ast_utils_1.addProviderToModule)(tree, sourceFile, pathToClientConfigFile, provider);
    }
}