adding gmbh company type

1 files changed, 32 insertions, 0 deletions

diff --git a/id/server/idserverlib/src/main/java/at/gv/egovernment/moa/id/protocols/stork2/MandateRetrievalRequest.java b/id/server/idserverlib/src/main/java/at/gv/egovernment/moa/id/protocols/stork2/MandateRetrievalRequest.java
index 95b59314d..1d081fb1c 100644
--- a/id/server/idserverlib/src/main/java/at/gv/egovernment/moa/id/protocols/stork2/MandateRetrievalRequest.java
+++ b/id/server/idserverlib/src/main/java/at/gv/egovernment/moa/id/protocols/stork2/MandateRetrievalRequest.java
@@ -31,6 +31,7 @@ import java.io.StringWriter;
 import java.math.BigInteger;
 import java.util.ArrayList;
 import java.util.Arrays;
+import java.util.regex.Pattern;
 
 /**
  *
@@ -131,6 +132,18 @@ public class MandateRetrievalRequest implements IAction {
                     String eLPIdentifier = geteLPIdentifier(mandateContainer, currentAttribute);
                     PersonalAttribute personalAttribute = new PersonalAttribute(currentAttribute.getName(), currentAttribute.isRequired(), Arrays.asList(eLPIdentifier), "Available");
                     attributeList.add(personalAttribute);
+                } else if (currentAttribute.getName().equals("type")) {
+                    String type = getCompanyType(mandateContainer, currentAttribute);
+                    PersonalAttribute personalAttribute;
+                    if (type.length() > 0) {
+                        attributeList.add(new PersonalAttribute(currentAttribute.getName(), currentAttribute.isRequired(), Arrays.asList(type), "Available"));
+                    } else {
+                        attributeList.add(new PersonalAttribute(currentAttribute.getName(), currentAttribute.isRequired(), Arrays.asList(type), "NotAvailable"));
+                    }
+                } else if (currentAttribute.getName().equals("translatableType")) {
+                    String translatableType = "";
+                    PersonalAttribute personalAttribute = new PersonalAttribute(currentAttribute.getName(), currentAttribute.isRequired(), Arrays.asList(translatableType), "Available");
+                    attributeList.add(personalAttribute);
                 }
 
             }
@@ -241,6 +254,25 @@ public class MandateRetrievalRequest implements IAction {
     }
 
 
+    private String getCompanyType(MandateContainer mandateContainer, PersonalAttribute sourceAttribute) throws MOAIDException {
+        String legalName = getLegalName(mandateContainer, sourceAttribute);
+
+        ArrayList<Pattern> gmbhPatterns = new ArrayList<Pattern>();
+        gmbhPatterns.add(Pattern.compile(".+ gmbh$", Pattern.CASE_INSENSITIVE));
+        gmbhPatterns.add(Pattern.compile(".+ Handelsges\\.m\\.b\\.H\\.$", Pattern.CASE_INSENSITIVE));
+        gmbhPatterns.add(Pattern.compile(".+ Gesellschaft m\\.b\\.H\\.$", Pattern.CASE_INSENSITIVE));
+        gmbhPatterns.add(Pattern.compile(".+ gesmbh$", Pattern.CASE_INSENSITIVE));
+
+        for (Pattern pattern : gmbhPatterns) {
+            if (pattern.matcher(legalName).matches()) {
+                return "GmbH";
+            }
+        }
+
+        return "";
+    }
+
+
     private RepresentationPersonType getRepresentative(MandateContainer mandateContainer, PersonalAttribute sourceAttribute) {
         RepresentationPersonType representative = new RepresentationPersonType();