@print-one/grapesjs/src/trait_manager/model/TraitFactory.ts

Free and Open Source Web Builder Framework

import { TraitManagerConfig } from '../config/config';
import { isString } from 'underscore';
import Trait, { TraitProperties } from './Trait';
import EditorModel from '../../editor/model/Editor';

export default class TraitFactory {
  config: Partial<TraitManagerConfig>;

  constructor(config: Partial<TraitManagerConfig> = {}) {
    this.config = config;
  }

  /**
   * Build props object by their name
   */
  build(prop: string | TraitProperties, em: EditorModel): Trait {
    return isString(prop) ? this.buildFromString(prop, em) : new Trait(prop, em);
  }

  private buildFromString(name: string, em: EditorModel): Trait {
    const obj: TraitProperties = {
      name: name,
      type: 'text',
    };

    switch (name) {
      case 'target':
        obj.type = 'select';
        obj.default = false;
        obj.options = this.config.optionsTarget;
        break;
    }
    return new Trait(obj, em);
  }
}