assemble_context.ts

import { Context } from './Context';
import { Module } from './Module';
import { Package } from './Package';

export interface IAssembleContext {
  flush: () => void;
  setFTLModule: (module: Module) => void;
  getFTLModules: () => Array<Module>;
  setFTLGeneratedContent: (content: string) => void;
  getFTLGeneratedContent: () => string;
  getPackageCollection: () => Map<string, Map<string, Package>>;
  collection: {
    modules: Map<string, Module>;
    packages: {
      getAll: (fn: (pkg: Package) => void) => void;
      get: (name: string, version: string) => Package | undefined;
      add: (pkg: Package) => void;
    };
  };
}

export enum AssembleState {
  VENDOR_REQUIRED,
}
export function assembleContext(ctx: Context): IAssembleContext {
  let packages = new Map<string, Map<string, Package>>();
  let ftlModules: Array<Module> = [];
  let ftlContent: string;
  const obj = {
    setFTLModule: (module: Module) => {
      if (ftlModules.indexOf(module) === -1) {
        ftlModules.push(module);
      }
    },
    setFTLGeneratedContent: str => {
      ftlContent = str;
    },
    getFTLGeneratedContent: () => {
      return ftlContent;
    },
    getFTLModules: () => ftlModules,
    flush: () => {
      ftlModules = [];
      packages = new Map();
      ftlContent = '';
      obj.collection.modules = new Map();
    },
    getPackageCollection() {
      return packages;
    },
    collection: {
      modules: new Map<string, Module>(),

      packages: {
        getAll: (fn: (pkg: Package) => void) => {
          packages.forEach(items => {
            items.forEach(pkg => fn(pkg));
          });
        },
        get: (name: string, version: string) => {
          if (packages.has(name)) {
            return packages.get(name).get(version);
          }
        },
        add: (pkg: Package) => {
          const name = pkg.props.meta.name;
          const version = pkg.props.meta.version;
          if (packages.has(name)) {
            pkg.isFlat = false;
            packages.get(name).set(version, pkg);
          } else {
            const record = packages.set(name, new Map<string, Package>());
            pkg.isFlat = true;
            record.get(name).set(version, pkg);
          }
        },
      },
    },
  };
  return obj;
}