/*******************************************************************************
 * Copyright 2013 SAP AG
 * 
 * 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.
 ******************************************************************************/
package com.sap.core.odata.processor.core.jpa.model;

import java.util.ArrayList;
import java.util.List;

import com.sap.core.odata.api.edm.provider.ComplexProperty;
import com.sap.core.odata.api.edm.provider.ComplexType;
import com.sap.core.odata.api.edm.provider.Facets;
import com.sap.core.odata.api.edm.provider.Key;
import com.sap.core.odata.api.edm.provider.Property;
import com.sap.core.odata.api.edm.provider.PropertyRef;
import com.sap.core.odata.api.edm.provider.SimpleProperty;
import com.sap.core.odata.processor.api.jpa.access.JPAEdmBuilder;
import com.sap.core.odata.processor.api.jpa.exception.ODataJPAModelException;
import com.sap.core.odata.processor.api.jpa.model.JPAEdmComplexTypeView;
import com.sap.core.odata.processor.api.jpa.model.JPAEdmKeyView;
import com.sap.core.odata.processor.api.jpa.model.JPAEdmPropertyView;

public class JPAEdmKey extends JPAEdmBaseViewImpl implements JPAEdmKeyView {

  private JPAEdmPropertyView propertyView;
  private JPAEdmComplexTypeView complexTypeView = null;
  private boolean isBuildModeComplexType = false;
  private Key key;

  public JPAEdmKey(final JPAEdmProperty view) {
    super(view);
    propertyView = view;
  }

  public JPAEdmKey(final JPAEdmComplexTypeView complexTypeView,
      final JPAEdmPropertyView propertyView) {
    super(complexTypeView);
    this.propertyView = propertyView;
    this.complexTypeView = complexTypeView;
    isBuildModeComplexType = true;
  }

  @Override
  public JPAEdmBuilder getBuilder() {
    if (builder == null) {
      builder = new JPAEdmKeyBuider();
    }

    return builder;
  }

  @Override
  public Key getEdmKey() {
    return key;
  }

  private class JPAEdmKeyBuider implements JPAEdmBuilder {

    @Override
    public void build() throws ODataJPAModelException {

      List<PropertyRef> propertyRefList = null;
      if (key == null) {
        key = new Key();
      }

      if (key.getKeys() == null) {
        propertyRefList = new ArrayList<PropertyRef>();
        key.setKeys(propertyRefList);
      } else {
        propertyRefList = key.getKeys();
      }

      if (isBuildModeComplexType) {
        ComplexType complexType = complexTypeView
            .searchEdmComplexType(propertyView
                .getJPAAttribute().getJavaType().getName());
        normalizeComplexKey(complexType, propertyRefList);
      } else {
        PropertyRef propertyRef = new PropertyRef();
        propertyRef.setName(propertyView.getEdmSimpleProperty().getName());
        Facets facets = (Facets) propertyView.getEdmSimpleProperty().getFacets();
        if (facets == null) {
          propertyView.getEdmSimpleProperty().setFacets(new Facets().setNullable(false));
        } else {
          facets.setNullable(false);
        }
        propertyRefList.add(propertyRef);
      }

    }

    //TODO think how to stop the recursion if A includes B and B includes A!!!!!!
    public void normalizeComplexKey(final ComplexType complexType, final List<PropertyRef> propertyRefList) {
      for (Property property : complexType.getProperties()) {
        try {

          SimpleProperty simpleProperty = (SimpleProperty) property;
          Facets facets = (Facets) simpleProperty.getFacets();
          if (facets == null) {
            simpleProperty.setFacets(new Facets().setNullable(false));
          } else {
            facets.setNullable(false);
          }
          PropertyRef propertyRef = new PropertyRef();
          propertyRef.setName(simpleProperty.getName());
          propertyRefList.add(propertyRef);

        } catch (ClassCastException e) {
          ComplexProperty complexProperty = (ComplexProperty) property;
          normalizeComplexKey(complexTypeView.searchEdmComplexType(complexProperty.getType()), propertyRefList);
        }

      }
    }
  }
}