authz-core/src/main/java/com/att/authz/org/OrganizationFactory.java

   1 /*******************************************************************************\r
   2  * ============LICENSE_START====================================================\r
   3  * * org.onap.aaf\r
   4  * * ===========================================================================\r
   5  * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.\r
   6  * * ===========================================================================\r
   7  * * Licensed under the Apache License, Version 2.0 (the "License");\r
   8  * * you may not use this file except in compliance with the License.\r
   9  * * You may obtain a copy of the License at\r
  10  * * \r
  11  *  *      http://www.apache.org/licenses/LICENSE-2.0\r
  12  * * \r
  13  *  * Unless required by applicable law or agreed to in writing, software\r
  14  * * distributed under the License is distributed on an "AS IS" BASIS,\r
  15  * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  16  * * See the License for the specific language governing permissions and\r
  17  * * limitations under the License.\r
  18  * * ============LICENSE_END====================================================\r
  19  * *\r
  20  * * ECOMP is a trademark and service mark of AT&T Intellectual Property.\r
  21  * *\r
  22  ******************************************************************************/\r
  23 package com.att.authz.org;\r
  24 \r
  25 import java.lang.reflect.Constructor;\r
  26 import java.lang.reflect.InvocationTargetException;\r
  27 import java.util.Map;\r
  28 import java.util.concurrent.ConcurrentHashMap;\r
  29 \r
  30 import com.att.authz.env.AuthzEnv;\r
  31 import com.att.authz.env.AuthzTrans;\r
  32 import com.att.inno.env.APIException;\r
  33 import com.att.inno.env.Slot;\r
  34 \r
  35 /**\r
  36  * Organization Plugin Mechanism\r
  37  * \r
  38  * Define a NameSpace for the company (i.e. com.att), and put in Properties as \r
  39  * "Organization.[your NS" and assign the supporting Class.  \r
  40  * \r
  41  * Example:\r
  42  * Organization.com.att=com.att.authz.org.att.ATT\r
  43  *\r
  44  *\r
  45  */\r
  46 public class OrganizationFactory {\r
  47         public static final String ORG_SLOT = "ORG_SLOT";\r
  48         private static Organization defaultOrg = null;\r
  49         private static Map<String,Organization> orgs = new ConcurrentHashMap<String,Organization>();\r
  50         private static Slot orgSlot;\r
  51         \r
  52         public static void setDefaultOrg(AuthzEnv env, String orgClass) throws APIException {\r
  53                 orgSlot = env.slot(ORG_SLOT);\r
  54                 try {\r
  55                         @SuppressWarnings("unchecked")\r
  56                         Class<Organization> cls = (Class<Organization>) Class.forName(orgClass);\r
  57                         Constructor<Organization> cnst = cls.getConstructor(AuthzEnv.class);\r
  58                         defaultOrg = cnst.newInstance(env);\r
  59                 } catch (ClassNotFoundException | NoSuchMethodException | SecurityException | \r
  60                                 InstantiationException | IllegalAccessException | IllegalArgumentException | \r
  61                                 InvocationTargetException e) {\r
  62                         throw new APIException(e);\r
  63                 }\r
  64         }\r
  65         \r
  66         public static Organization obtain(AuthzEnv env,String orgNS) throws OrganizationException {\r
  67                 int at = orgNS.indexOf('@');\r
  68                 if(at<0) {\r
  69                         if(!orgNS.startsWith("com.")) {\r
  70                                 int dot1;\r
  71                                 if((dot1 = orgNS.lastIndexOf('.'))>-1) {\r
  72                                         int dot2;\r
  73                                         StringBuilder sb = new StringBuilder();\r
  74                                         if((dot2 = orgNS.lastIndexOf('.',dot1-1))>-1) {\r
  75                                                 sb.append(orgNS,dot1+1,orgNS.length());\r
  76                                                 sb.append('.');\r
  77                                                 sb.append(orgNS,dot2+1,dot1);\r
  78                                         } else {\r
  79                                                 sb.append(orgNS,dot1+1,orgNS.length());\r
  80                                                 sb.append('.');\r
  81                                                 sb.append(orgNS,at+1,dot1);\r
  82                                         }\r
  83                                         orgNS=sb.toString();\r
  84                                 }\r
  85                         }\r
  86                 } else {\r
  87                         // Only use two places (Enterprise) of domain\r
  88                         int dot;\r
  89                         if((dot= orgNS.lastIndexOf('.'))>-1) {\r
  90                                 StringBuilder sb = new StringBuilder();\r
  91                                 int dot2;\r
  92                                 if((dot2 = orgNS.lastIndexOf('.',dot-1))>-1) {\r
  93                                         sb.append(orgNS.substring(dot+1));\r
  94                                         sb.append(orgNS.subSequence(dot2, dot));\r
  95                                         orgNS = sb.toString();\r
  96                                 } else {\r
  97                                         sb.append(orgNS.substring(dot+1));\r
  98                                         sb.append('.');\r
  99                                         sb.append(orgNS.subSequence(at+1, dot));\r
 100                                         orgNS = sb.toString();\r
 101                                 }\r
 102                         }\r
 103                 }\r
 104                 Organization org = orgs.get(orgNS);\r
 105                 if(org == null) {\r
 106                         String orgClass = env.getProperty("Organization."+orgNS);\r
 107                         if(orgClass == null) {\r
 108                                 env.warn().log("There is no Organization." + orgNS + " property");\r
 109                         } else {\r
 110                                 for(Organization o : orgs.values()) {\r
 111                                         if(orgClass.equals(o.getClass().getName())) {\r
 112                                                 org = o;\r
 113                                         }\r
 114                                 }\r
 115                                 if(org==null) {\r
 116                                         try {\r
 117                                                 @SuppressWarnings("unchecked")\r
 118                                                 Class<Organization> cls = (Class<Organization>) Class.forName(orgClass);\r
 119                                                 Constructor<Organization> cnst = cls.getConstructor(AuthzEnv.class);\r
 120                                                 org = cnst.newInstance(env);\r
 121                                         } catch (ClassNotFoundException | NoSuchMethodException | SecurityException | \r
 122                                                         InstantiationException | IllegalAccessException | IllegalArgumentException | \r
 123                                                         InvocationTargetException e) {\r
 124                                                 throw new OrganizationException(e);\r
 125                                         }\r
 126                                 }\r
 127                                 orgs.put(orgNS, org);\r
 128                         }\r
 129                         if(org==null && defaultOrg!=null) {\r
 130                                 org=defaultOrg;\r
 131                                 orgs.put(orgNS, org);\r
 132                         }\r
 133                 }\r
 134                 \r
 135                 return org;\r
 136         }\r
 137 \r
 138         public static void set(AuthzTrans trans, String orgNS) throws OrganizationException {\r
 139                 Organization org = obtain(trans.env(),orgNS);\r
 140                 trans.put(orgSlot, org);\r
 141         }\r
 142         \r
 143         public static Organization get(AuthzTrans trans) {\r
 144                 return trans.get(orgSlot,defaultOrg);\r
 145         }\r
 146 \r
 147 }\r