ngx-scrolltop/schematics/ng-add/add-declaration-to-module.rule.ts

Lightweight, Material Design inspired button for scroll-to-top of the page. No dependencies. Pure Angular!

import { normalize } from '@angular-devkit/core';
import { Rule, SchematicsException, Tree } from '@angular-devkit/schematics';
import * as ts from 'typescript';
import { addSymbolToNgModuleMetadata } from '../utility/ast-utils';
import { InsertChange } from '../utility/change';

export function addDeclarationToAppModule(appModule: string): Rule {
  return (host: Tree) => {
    if (!appModule) {
      return host;
    }

    // Part I: Construct path and read file
    const modulePath = normalize('/' + appModule);

    const text = host.read(modulePath);
    if (text === null) {
      throw new SchematicsException(`File ${modulePath} does not exist.`);
    }
    const sourceText = text.toString('utf-8');
    const source = ts.createSourceFile(
      modulePath,
      sourceText,
      ts.ScriptTarget.Latest,
      true
    );

    // Part II: Find out, what to change
    const changes = addSymbolToNgModuleMetadata(
      source,
      modulePath,
      'imports',
      'NgxScrollTopModule',
      'ngx-scrolltop'
    );

    // Part III: Apply changes
    const recorder = host.beginUpdate(modulePath);
    for (const change of changes) {
      if (change instanceof InsertChange) {
        recorder.insertLeft(change.pos, change.toAdd);
      }
    }
    host.commitUpdate(recorder);

    return host;
  };
}