catalog-ui/app/scripts/services/relation-icons-service.ts

/*-
 * ============LICENSE_START=======================================================
 * SDC
 * ================================================================================
 * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ============LICENSE_END=========================================================
 */
/// <reference path="../references"/>
module Sdc.Services {
  'use strict';

  export interface IRelationIconsService {
      getIconPath(iconName:string):string;
  }

  export class RelationIconsService implements IRelationIconsService {
      constructor() {}
      private icons: Array<string> = [
          'AttachesTo',
          'BindsTo',
          'DependsOn',
          'HostedOn',
          'LinksTo',
          'RoutesTo'
      ];

      public getIconPath = (relationshipType:string): string => {
        let result:string = 'ConnectedTo';
        let baseUrl:string =  '/styles/images/relationship-icons/';

          if (relationshipType) {
              let arr = relationshipType.split('.'); // looks like tosca.relationships.AttachesTo
              relationshipType = arr[arr.length - 1];
              if (this.icons.indexOf(relationshipType) > -1) {
                  result = relationshipType;
              }
              if('LinksTo'==result){
                  return '';
              }
          }

        return baseUrl + result + '.svg';
      }

  }
}