From 32e149fcb5d30f3e74709d1dffdb98d84a66116d Mon Sep 17 00:00:00 2001
From: Erik Pitko <pitko.erik@gmail.com>
Date: Wed, 7 Oct 2020 15:59:46 +0200
Subject: [PATCH] Added eduperson_entitlement scope

---
 .../cz/metacentrum/fake_oidc/OidcController.java   |  3 +++
 src/main/java/cz/metacentrum/fake_oidc/User.java   | 14 ++++++++++++++
 src/main/resources/application.yml                 |  3 +++
 3 files changed, 20 insertions(+)

diff --git a/src/main/java/cz/metacentrum/fake_oidc/OidcController.java b/src/main/java/cz/metacentrum/fake_oidc/OidcController.java
index 76fa0c0..6d00967 100644
--- a/src/main/java/cz/metacentrum/fake_oidc/OidcController.java
+++ b/src/main/java/cz/metacentrum/fake_oidc/OidcController.java
@@ -160,6 +160,9 @@ public ResponseEntity<?> userinfo(@RequestHeader("Authorization") String auth,
         if (scopes.contains("email")) {
             m.put("email", user.getEmail());
         }
+        if (scopes.contains("eduperson_entitlement")) {
+        	m.put("eduPersonEntitlement", user.getEntitlement());
+        }
         return ResponseEntity.ok().body(m);
     }
 
diff --git a/src/main/java/cz/metacentrum/fake_oidc/User.java b/src/main/java/cz/metacentrum/fake_oidc/User.java
index b9a0e2e..5c6b8d2 100644
--- a/src/main/java/cz/metacentrum/fake_oidc/User.java
+++ b/src/main/java/cz/metacentrum/fake_oidc/User.java
@@ -1,5 +1,7 @@
 package cz.metacentrum.fake_oidc;
 
+import java.util.Arrays;
+
 public class User {
     private String logname;
     private String password;
@@ -9,6 +11,7 @@ public class User {
     private String family_name;
     private String email;
     private String preferred_username;
+    private String[] entitlement;
 
     public String getLogname() {
         return logname;
@@ -66,6 +69,14 @@ public void setEmail(String email) {
         this.email = email;
     }
 
+    public String[] getEntitlement() {
+        return entitlement;
+    }
+
+    public void setEntitlement(String[] entitlement) {
+        this.entitlement = entitlement;
+    }
+
     public String getPreferred_username() {
         return preferred_username;
     }
@@ -73,6 +84,8 @@ public String getPreferred_username() {
     public void setPreferred_username(String preferred_username) {
         this.preferred_username = preferred_username;
     }
+    
+    
 
     @Override
     public String toString() {
@@ -85,6 +98,7 @@ public String toString() {
                 ", family_name='" + family_name + '\'' +
                 ", email='" + email + '\'' +
                 ", preferred_username='" + preferred_username + '\'' +
+                ", eduPersonEntitlement='" + Arrays.toString(entitlement) + '\'' +
                 '}';
     }
 }
diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml
index f252502..505ada3 100644
--- a/src/main/resources/application.yml
+++ b/src/main/resources/application.yml
@@ -22,3 +22,6 @@ oidc:
     family_name: "Perun"
     preferred_username: "perun"
     email: "perun@cesnet.cz"
+    entitlement: 
+      - urn:geant:cesnet.cz:group:einfra#perun.cesnet.cz
+      - urn:geant:cesnet.cz:group:einfra:members#perun.cesnet.cz