1 files changed, 31 insertions, 1 deletions
diff --git a/src/main/java/DNPM/services/Studie.java b/src/main/java/DNPM/services/Studie.java
index d0d674a..9e487e1 100644
--- a/src/main/java/DNPM/services/Studie.java
+++ b/src/main/java/DNPM/services/Studie.java
@@ -2,12 +2,14 @@ package DNPM.services;
 
 public class Studie {
     private final String code;
+    private final String studiennummer;
     private final String shortDesc;
     private final String description;
     private final int version;
 
-    public Studie(final String code, final String shortDesc, final String description, final int version) {
+    public Studie(final String code, final String studiennummer, final String shortDesc, final String description, final int version) {
         this.code = code;
+        this.studiennummer = studiennummer;
         this.shortDesc = shortDesc;
         this.description = description;
         this.version = version;
@@ -17,6 +19,10 @@ public class Studie {
         return code;
     }
 
+    public String getStudiennummer() {
+        return studiennummer;
+    }
+
     public String getShortDesc() {
         return shortDesc;
     }
@@ -28,4 +34,28 @@ public class Studie {
     public int getVersion() {
         return version;
     }
+
+    public Type getType() {
+        if (this.hasNctNumber()) {
+            return Type.NCT;
+        } else if (this.hasEudraCtNumber()) {
+            return Type.EUDRA_CT;
+        } else {
+            return Type.UNKNOWN;
+        }
+    }
+
+    private boolean hasNctNumber() {
+        return null != studiennummer && studiennummer.toLowerCase().startsWith("nct");
+    }
+
+    private boolean hasEudraCtNumber() {
+        return null != studiennummer && studiennummer.matches("[0-9]{4}-[0-9]{6}-[0-9]{2}");
+    }
+
+    public enum Type {
+        NCT,
+        EUDRA_CT,
+        UNKNOWN
+    }
 }