/*
 * Copyright (c) 2017, University of Oslo
 *
 * All rights reserved.
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 * Redistributions of source code must retain the above copyright notice, this
 * list of conditions and the following disclaimer.
 *
 * Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 * Neither the name of the HISP project nor the names of its contributors may
 * be used to endorse or promote products derived from this software without
 * specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package org.hisp.dhis.android.core.relationship;

import org.hisp.dhis.android.core.common.ModelBuilder;
import org.hisp.dhis.android.core.common.ObjectWithUid;

public class RelationshipConstraintModelBuilder
        extends ModelBuilder<RelationshipConstraint, RelationshipConstraintModel> {

    private RelationshipType relationshipType;
    private RelationshipConstraintType constraintType;

    RelationshipConstraintModelBuilder(RelationshipType relationshipType, RelationshipConstraintType type) {
        this.relationshipType = relationshipType;
        this.constraintType = type;
    }

    @Override
    public RelationshipConstraintModel buildModel(RelationshipConstraint relationshipConstraint) {
        return RelationshipConstraintModel.builder()
                .relationshipType(this.relationshipType.uid())
                .constraintType(this.constraintType)
                .relationshipEntity(relationshipConstraint.relationshipEntity())
                .trackedEntityType(getUidOrNull(relationshipConstraint.trackedEntityType()))
                .program(getUidOrNull(relationshipConstraint.program()))
                .programStage(getUidOrNull(relationshipConstraint.programStage()))
                .build();
    }

    private String getUidOrNull(ObjectWithUid object) {
        return object != null ? object.uid() : null;
    }
}

@@ -1857,54 +1857,13 @@
 ate 
-RelationshipType relationshipType;%0A    private
+final
  Rel
@@ -1885,27 +1885,29 @@
 aint
-Type constraintType
+Model.Builder builder
 ;%0A%0A 
@@ -2022,24 +2022,89 @@
         this
+.builder = RelationshipConstraintModel.builder()%0A                
 .relationshi
@@ -2108,19 +2108,17 @@
 shipType
- = 
+(
 relation
@@ -2125,17 +2125,23 @@
 shipType
-;
+.uid())
 %0A       
@@ -2141,20 +2141,24 @@
         
-this
+        
 .constra
@@ -2168,15 +2168,14 @@
 Type
- = 
+(
 type
+)
 ;%0A  
@@ -2311,161 +2311,20 @@
 urn 
-RelationshipConstraintModel.builder()%0A                .relationshipType(this.relationshipType.uid())%0A                .constraintType(this.constraintType)
+this.builder
 %0A   

refactor RelationshipConstraintModelBuilder