els-addon-typed-templates/src/lib/hbs-extractor.ts

Ember Language Server Typed Templates

import { keyForItem } from "./hbs-transform";
import { isSimpleBlockComponentElement } from "./ast-helpers";
import { serverForProject } from "./ts-service";
export function extractRelationships(items, projectRoot) {

  let registry = {
    component: {},
    helper: {}
  };

  const server = serverForProject(projectRoot);
  if (server) {
    registry = server.getRegistry(projectRoot);
  }

  let isComponent = (name) => {
    return name in registry.component;
  }

  // projectRoot
  const componentsForImport: string[] = [];
  const parents = {};
  const scopes = {};
  const klass = {};
  const blockPaths: any = [];
  function addChilds(items, key) {
    items.forEach(item => {
      if (item.type === "MustacheStatement" || item.type === "BlockStatement") {
        parents[key].push(keyForItem(item));
      } else if (item.type === "ElementNode") {
        if (isSimpleBlockComponentElement(item)) {
          parents[key].push(keyForItem({
            type: 'BlockStatement',
            loc: item.loc
          }));
        }
        item.modifiers.forEach(mod => {
          parents[key].push(keyForItem(mod));
        });
        item.attributes.forEach((attr)=>{
          if (attr.value.type === 'ConcatStatement') {
            attr.value.parts.forEach((part)=>{
              if (part.type === 'MustacheStatement') {
                addChilds([part], key);
                parents[key].push(keyForItem(part));
              }
            })
          } else if (attr.value.type === 'MustacheStatement') {
            addChilds([attr.value], key);
            parents[key].push(keyForItem(attr.value));
          }
        });
      }
      addChilds(item.program ? item.program.body : item.children || [], key);
      if (item.inverse) {
        addChilds(item.inverse.body || [], key);
      }
      if (item.hash) {
        item.hash.pairs.forEach(attr => {
          if (attr.value.type === 'ConcatStatement') {
            attr.value.parts.forEach((part)=>{
              if (part.type === 'MustacheStatement') {
                addChilds([part], key);
                parents[key].push(keyForItem(part));
              }
            })
          } else if (attr.value.type === 'MustacheStatement') {
            addChilds([attr.value], key);
            parents[key].push(keyForItem(attr.value));
          }
        });
      }
    });
  }

  items.slice(0).forEach(b => {
    let n = b.slice(0);
    let pointer: any = null;
    while (n.length) {
      let exp = n.shift();
      const key = keyForItem(exp);

      if (!pointer) {
        pointer = key;
        parents[pointer] = [];
        if (exp.blockParams) {
          scopes[pointer] = exp.blockParams;
        } else {
          scopes[pointer] = exp.program ? (exp.program.blockParams || []) : [];
        }
      }
      if (!(key in parents)) {
        parents[key] = [];
      }

      addChilds(exp.program ? exp.program.body : exp.children || [], pointer);
      addChilds(exp.inverse ? exp.inverse.body : [], pointer);

      parents[key].push(keyForItem(exp.path));
      klass[key] = exp;
      klass[keyForItem(exp.path)] = exp.path;
      if (exp.type === "BlockStatement") {
        blockPaths.push(keyForItem(exp.path));
        if (exp.isComponent) {
          componentsForImport.push(exp.path.original);
        }
      } else if (exp.type === "MustacheStatement") {
        if (exp.path.type === 'PathExpression') {
          if (isComponent(exp.path.original)) {
            componentsForImport.push(exp.path.original);
          }
        }
      }

      exp.params && exp.params.forEach(p => {
        klass[keyForItem(p)] = p;
        parents[pointer].push(keyForItem(p));
      });
      exp.hash && exp.hash.pairs.forEach(p => {
        klass[keyForItem(p.value)] = p.value;
        parents[pointer].push(keyForItem(p.value));
      });
    }
  });

  return {
    componentsForImport,
    parents,
    scopes,
    klass,
    blockPaths
  };
}