From b446658fe266fa3fec7f87996046dc5744675661 Mon Sep 17 00:00:00 2001
From: Manuel Rego Casasnovas <mrego@igalia.com>
Date: Wed, 23 Dec 2009 17:23:15 +0100
Subject: [PATCH]  ItEr40S16CUImportacionOrganizacionsTraballoItEr39S17: Added
 some constraints on behaviour when you assign labels to an order element from
 the interface.

---
 .../business/labels/entities/Label.java       | 13 +++++
 .../orders/entities/OrderElement.java         | 54 +++++++++++++++----
 ...ssignedLabelsToOrderElementController.java | 12 ++++-
 .../AssignedLabelsToOrderElementModel.java    | 23 ++++++--
 4 files changed, 87 insertions(+), 15 deletions(-)

diff --git a/navalplanner-business/src/main/java/org/navalplanner/business/labels/entities/Label.java b/navalplanner-business/src/main/java/org/navalplanner/business/labels/entities/Label.java
index 2cc47464f..e08b54747 100644
--- a/navalplanner-business/src/main/java/org/navalplanner/business/labels/entities/Label.java
+++ b/navalplanner-business/src/main/java/org/navalplanner/business/labels/entities/Label.java
@@ -87,4 +87,17 @@ public class Label extends BaseEntity {
     public void removeOrderElement(OrderElement orderElement) {
         orderElements.add(orderElement);
     }
+
+    public boolean isEqualTo(Label label) {
+        if ((this.getName() != null) && (label.getName() != null)
+                && (this.getType() != null) && (label.getType() != null)
+                && (this.getType().getName() != null)
+                && (label.getType().getName() != null)
+                && this.getName().equals(label.getName())
+                && this.getType().getName().equals(label.getType().getName())) {
+            return true;
+        }
+        return false;
+    }
+
 }
diff --git a/navalplanner-business/src/main/java/org/navalplanner/business/orders/entities/OrderElement.java b/navalplanner-business/src/main/java/org/navalplanner/business/orders/entities/OrderElement.java
index 4420391c4..c784bc3c2 100644
--- a/navalplanner-business/src/main/java/org/navalplanner/business/orders/entities/OrderElement.java
+++ b/navalplanner-business/src/main/java/org/navalplanner/business/orders/entities/OrderElement.java
@@ -371,6 +371,15 @@ public abstract class OrderElement extends BaseEntity {
 
     public void addLabel(Label label) {
         Validate.notNull(label);
+
+        if (!checkAncestorsNoOtherLabelRepeated(label)) {
+            throw new IllegalArgumentException(
+                    _("Some ancestor has the same label assigned, "
+                            + "so this element is already inheriting this label"));
+        }
+
+        removeLabelOnChildren(label);
+
         labels.add(label);
     }
 
@@ -763,19 +772,46 @@ public abstract class OrderElement extends BaseEntity {
 
     private boolean containsLabel(HashSet<Label> labels, Label label) {
         for (Label each : labels) {
-            if ((each.getName() != null)
-                    && (label.getName() != null)
-                    && (each.getType() != null)
-                    && (label.getType() != null)
-                    && (each.getType().getName() != null)
-                    && (label.getType().getName() != null)
-                    && each.getName().equals(label.getName())
-                    && each.getType().getName().equals(
-                            label.getType().getName())) {
+            if (each.isEqualTo(label)) {
                 return true;
             }
         }
         return false;
     }
 
+    private boolean checkAncestorsNoOtherLabelRepeated(Label newLabel) {
+        for (Label label : labels) {
+            if (label.equals(newLabel)) {
+                return false;
+            }
+        }
+
+        if (parent != null) {
+            if (!((OrderElement) parent)
+                    .checkAncestorsNoOtherLabelRepeated(newLabel)) {
+                return false;
+            }
+        }
+        return true;
+    }
+
+    private void removeLabelOnChildren(Label newLabel) {
+        Label toRemove = null;
+
+        for (Label label : labels) {
+            if (label.equals(newLabel)) {
+                toRemove = label;
+                break;
+            }
+        }
+
+        if (toRemove != null) {
+            removeLabel(toRemove);
+        }
+
+        for (OrderElement child : getChildren()) {
+            child.removeLabelOnChildren(newLabel);
+        }
+    }
+
 }
\ No newline at end of file
diff --git a/navalplanner-webapp/src/main/java/org/navalplanner/web/orders/AssignedLabelsToOrderElementController.java b/navalplanner-webapp/src/main/java/org/navalplanner/web/orders/AssignedLabelsToOrderElementController.java
index 61edee581..49e86225d 100644
--- a/navalplanner-webapp/src/main/java/org/navalplanner/web/orders/AssignedLabelsToOrderElementController.java
+++ b/navalplanner-webapp/src/main/java/org/navalplanner/web/orders/AssignedLabelsToOrderElementController.java
@@ -116,7 +116,11 @@ public class AssignedLabelsToOrderElementController extends
         if (isAssigned(label)) {
             throw new WrongValueException(bdLabels, _("already assigned"));
         }
-        assignLabel(label);
+        try {
+            assignLabel(label);
+        } catch (IllegalArgumentException e) {
+            throw new WrongValueException(bdLabels, e.getMessage());
+        }
         bdLabels.clear();
     }
 
@@ -155,7 +159,11 @@ public class AssignedLabelsToOrderElementController extends
                         _("already assigned"));
             }
         }
-        assignLabel(label);
+        try {
+            assignLabel(label);
+        } catch (IllegalArgumentException e) {
+            throw new WrongValueException(txtLabelName, e.getMessage());
+        }
         clear(txtLabelName);
     }
 
diff --git a/navalplanner-webapp/src/main/java/org/navalplanner/web/orders/AssignedLabelsToOrderElementModel.java b/navalplanner-webapp/src/main/java/org/navalplanner/web/orders/AssignedLabelsToOrderElementModel.java
index 948305152..1583d0530 100644
--- a/navalplanner-webapp/src/main/java/org/navalplanner/web/orders/AssignedLabelsToOrderElementModel.java
+++ b/navalplanner-webapp/src/main/java/org/navalplanner/web/orders/AssignedLabelsToOrderElementModel.java
@@ -74,11 +74,27 @@ public class AssignedLabelsToOrderElementModel implements
 
     private void initializeOrderElement(OrderElement orderElement) {
         reattachLabels();
+        initialize(orderElement);
+
+        OrderElement parent = orderElement.getParent();
+        while (parent != null) {
+            initialize(parent);
+            parent = parent.getParent();
+        }
+
+        initializeChildren(orderElement);
+    }
+
+    private void initializeChildren(OrderElement orderElement) {
+        initialize(orderElement);
+        for (OrderElement child : orderElement.getChildren()) {
+            initializeChildren(child);
+        }
+    }
+
+    private void initialize(OrderElement orderElement) {
         orderDAO.reattach(orderElement);
         orderElement.getName();
-        if (orderElement.getParent() != null) {
-            orderElement.getParent().getName();
-        }
         initializeLabels(orderElement.getLabels());
     }
 
@@ -117,7 +133,6 @@ public class AssignedLabelsToOrderElementModel implements
             reattachLabels();
             OrderLineGroup parent = orderElement.getParent();
             while (parent != null) {
-                initializeOrderElement(parent);
                 result.addAll(parent.getLabels());
                 parent = parent.getParent();
             }