120 lines
3.7 KiB
TypeScript
120 lines
3.7 KiB
TypeScript
import {Map, MapWrapper, ListWrapper} from 'angular2/src/facade/collection';
|
|
import {Type, isPresent, BaseException, stringify, isBlank} from 'angular2/src/facade/lang';
|
|
|
|
import {View} from 'angular2/src/core/annotations_impl/view';
|
|
import {ViewResolver} from 'angular2/src/core/compiler/view_resolver';
|
|
|
|
export class MockViewResolver extends ViewResolver {
|
|
_views: Map<Type, View> = new Map();
|
|
_inlineTemplates: Map<Type, string> = new Map();
|
|
_viewCache: Map<Type, View> = new Map();
|
|
_directiveOverrides: Map<Type, Map<Type, Type>> = new Map();
|
|
|
|
constructor() { super(); }
|
|
|
|
/**
|
|
* Overrides the {@link View} for a component.
|
|
*
|
|
* @param {Type} component
|
|
* @param {ViewDefinition} view
|
|
*/
|
|
setView(component: Type, view: View): void {
|
|
this._checkOverrideable(component);
|
|
this._views.set(component, view);
|
|
}
|
|
|
|
/**
|
|
* Overrides the inline template for a component - other configuration remains unchanged.
|
|
*
|
|
* @param {Type} component
|
|
* @param {string} template
|
|
*/
|
|
setInlineTemplate(component: Type, template: string): void {
|
|
this._checkOverrideable(component);
|
|
this._inlineTemplates.set(component, template);
|
|
}
|
|
|
|
/**
|
|
* Overrides a directive from the component {@link View}.
|
|
*
|
|
* @param {Type} component
|
|
* @param {Type} from
|
|
* @param {Type} to
|
|
*/
|
|
overrideViewDirective(component: Type, from: Type, to: Type): void {
|
|
this._checkOverrideable(component);
|
|
|
|
var overrides = this._directiveOverrides.get(component);
|
|
|
|
if (isBlank(overrides)) {
|
|
overrides = new Map();
|
|
this._directiveOverrides.set(component, overrides);
|
|
}
|
|
|
|
overrides.set(from, to);
|
|
}
|
|
|
|
/**
|
|
* Returns the {@link View} for a component:
|
|
* - Set the {@link View} to the overridden view when it exists or fallback to the default
|
|
* `ViewResolver`,
|
|
* see `setView`.
|
|
* - Override the directives, see `overrideViewDirective`.
|
|
* - Override the @View definition, see `setInlineTemplate`.
|
|
*
|
|
* @param component
|
|
* @returns {ViewDefinition}
|
|
*/
|
|
resolve(component: Type): View {
|
|
var view = this._viewCache.get(component);
|
|
if (isPresent(view)) return view;
|
|
|
|
view = this._views.get(component);
|
|
if (isBlank(view)) {
|
|
view = super.resolve(component);
|
|
}
|
|
|
|
var directives = view.directives;
|
|
var overrides = this._directiveOverrides.get(component);
|
|
|
|
if (isPresent(overrides) && isPresent(directives)) {
|
|
directives = ListWrapper.clone(view.directives);
|
|
MapWrapper.forEach(overrides, (to, from) => {
|
|
var srcIndex = directives.indexOf(from);
|
|
if (srcIndex == -1) {
|
|
throw new BaseException(
|
|
`Overriden directive ${stringify(from)} not found in the template of ${stringify(component)}`);
|
|
}
|
|
directives[srcIndex] = to;
|
|
});
|
|
view = new View(
|
|
{template: view.template, templateUrl: view.templateUrl, directives: directives});
|
|
}
|
|
|
|
var inlineTemplate = this._inlineTemplates.get(component);
|
|
if (isPresent(inlineTemplate)) {
|
|
view = new View({template: inlineTemplate, templateUrl: null, directives: view.directives});
|
|
}
|
|
|
|
this._viewCache.set(component, view);
|
|
return view;
|
|
}
|
|
|
|
/**
|
|
* Once a component has been compiled, the AppProtoView is stored in the compiler cache.
|
|
*
|
|
* Then it should not be possible to override the component configuration after the component
|
|
* has been compiled.
|
|
*
|
|
* @param {Type} component
|
|
*/
|
|
_checkOverrideable(component: Type): void {
|
|
var cached = this._viewCache.get(component);
|
|
|
|
if (isPresent(cached)) {
|
|
throw new BaseException(
|
|
`The component ${stringify(component)} has already been compiled, its configuration can not be changed`);
|
|
}
|
|
}
|
|
}
|