package com.aurea.maven.plugins.sonic.sdm.util;
   
   import java.util.ArrayList;
   import java.util.List;
   
   import com.sonicsw.deploy.artifact.ESBArtifact;
   import com.sonicsw.sonicxq.ServiceTypeDocument.ServiceType;
   
   public class ESBServiceTypesContribution extends AbstractModelContribution {
  
    public ESBServiceTypesContribution(String name, ESBDeploymentModelBuilder builder) {
      super(name, builder);
    }
    
    @SuppressWarnings("unchecked")
  public Object getContribution() throws Exception {
      
      ESBDeploymentModel model = getBuilder().getModel();
      
      List<ServiceType> result = (List<ServiceType>)model.getModelObject(getContributionName());
      
      if (result == null) {
        result = new ArrayList<ServiceType>();
        for(Object obj: getArtifactList(ESBArtifact.SERVICE_TYPE, ServiceType.class)) result.add((ServiceType)obj);
      }
      return result;
    }
  
  }