ONAP-PAP-REST/src/main/java/org/onap/policy/pap/xacml/rest/util/DictionaryUtils.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP-PAP-REST
   4  * ================================================================================
   5  * Copyright (C) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20 package org.onap.policy.pap.xacml.rest.util;
  21
  22 import java.io.IOException;
  23 import java.io.PrintWriter;
  24 import java.util.HashMap;
  25 import java.util.LinkedHashMap;
  26 import java.util.List;
  27 import java.util.Map;
  28
  29 import javax.servlet.http.HttpServletRequest;
  30 import javax.servlet.http.HttpServletResponse;
  31
  32 import org.apache.commons.logging.Log;
  33 import org.apache.commons.logging.LogFactory;
  34 import org.json.JSONObject;
  35 import org.onap.policy.rest.dao.CommonClassDao;
  36 import org.onap.policy.rest.jpa.Category;
  37 import org.onap.policy.rest.jpa.Datatype;
  38 import org.onap.policy.rest.jpa.UserInfo;
  39 import org.onap.policy.xacml.api.XACMLErrorConstants;
  40 import org.springframework.beans.factory.annotation.Autowired;
  41 import org.springframework.stereotype.Service;
  42 import org.springframework.web.servlet.ModelAndView;
  43
  44 import com.fasterxml.jackson.databind.DeserializationFeature;
  45 import com.fasterxml.jackson.databind.JsonNode;
  46 import com.fasterxml.jackson.databind.ObjectMapper;
  47
  48 @Service
  49 public class DictionaryUtils {
  50
  51         private static final Log LOGGER = LogFactory.getLog(DictionaryUtils.class);
  52
  53         private static String apiflag = "apiflag";
  54         private static String operation = "operation";
  55         private static String duplicateResponseString = "Duplicate";
  56         private static String utf8 = "UTF-8";
  57         private static String applicationJsonContentType = "application / json";
  58
  59         private static CommonClassDao commonClassDao;
  60
  61         private static DictionaryUtils dictionaryUtils;
  62
  63         public static synchronized DictionaryUtils getDictionaryUtils() {
  64             return dictionaryUtils != null ? dictionaryUtils : new DictionaryUtils();
  65         }
  66
  67         public static synchronized void setDictionaryUtils(DictionaryUtils dictionaryUtils) {
  68                 DictionaryUtils.dictionaryUtils = dictionaryUtils;
  69         }
  70
  71         @Autowired
  72         public DictionaryUtils(CommonClassDao commonClassDao){
  73                 DictionaryUtils.commonClassDao = commonClassDao;
  74         }
  75
  76         public DictionaryUtils(){
  77                 super();
  78         }
  79
  80         public UserInfo getUserInfo(String loginId){
  81                 return (UserInfo) commonClassDao.getEntityItem(UserInfo.class, "userLoginId", loginId);
  82         }
  83
  84         public boolean isRequestFromAPI(HttpServletRequest request){
  85                 return request.getParameter(apiflag)!=null && "api".equalsIgnoreCase(request.getParameter(apiflag));
  86         }
  87
  88         public String appendKey(List<Object> objects, String key1, String appendValue){
  89                 StringBuilder userValue = new StringBuilder();
  90                 int counter = 0;
  91                 for(Object attribute : objects){
  92                         if(attribute instanceof LinkedHashMap<?, ?>){
  93                                 String key = ((LinkedHashMap<?, ?>) attribute).get(key1).toString();
  94                                 if(counter>0){
  95                                         userValue.append(appendValue);
  96                                 }
  97                                 userValue.append(key);
  98                                 counter ++;
  99                         }
 100                 }
 101                 return userValue.toString();
 102         }
 103
 104         public String appendKeyValue(List<Object> objects, String append1, String append2){
 105                 StringBuilder header = new StringBuilder();
 106                 int counter = 0;
 107                 for(Object attribute : objects){
 108                         if(attribute instanceof LinkedHashMap<?, ?>){
 109                                 String key = ((LinkedHashMap<?, ?>) attribute).get("option").toString();
 110                                 String value = ((LinkedHashMap<?, ?>) attribute).get("number").toString();
 111                                 if(counter>0){
 112                                         header.append(append1);
 113                                 }
 114                                 header.append(key).append(append2).append(value);
 115                                 counter ++;
 116                         }
 117                 }
 118                 return header.toString();
 119         }
 120
 121         public Datatype getDataType(String datatype){
 122                 Datatype a = new Datatype();
 123                 if("string".equalsIgnoreCase(datatype)){
 124                         a.setId(26);
 125                 }else if("integer".equalsIgnoreCase(datatype)){
 126                         a.setId(12);
 127                 }else if("boolean".equalsIgnoreCase(datatype)){
 128                         a.setId(18);
 129                 }else if("double".equalsIgnoreCase(datatype)){
 130                         a.setId(25);
 131                 }else if("user".equalsIgnoreCase(datatype)){
 132                         a.setId(29);
 133                 }
 134                 return a;
 135         }
 136
 137         public Category getCategory(){
 138                 return (Category) commonClassDao.getDataById(Category.class, "shortName", "resource").get(0);
 139         }
 140
 141         public ModelAndView getResultForApi(String inResponseString){
 142                 String responseString = inResponseString;
 143                 if(responseString!=null && !duplicateResponseString.equals(responseString)){
 144                         responseString = "Success";
 145                 }
 146                 ModelAndView result = new ModelAndView();
 147                 result.setViewName(responseString);
 148                 return result;
 149         }
 150
 151         public void setResponseData(HttpServletResponse response, String key, String responseString) throws IOException{
 152                 response.setCharacterEncoding(utf8);
 153                 response.setContentType(applicationJsonContentType);
 154
 155                 PrintWriter out = response.getWriter();
 156                 JSONObject j = new JSONObject("{"+key+":" + responseString + "}");
 157                 out.write(j.toString());
 158         }
 159
 160         public void setErrorResponseData(HttpServletResponse response, Exception e) throws IOException{
 161                 LOGGER.error(XACMLErrorConstants.ERROR_PROCESS_FLOW + e);
 162                 response.setCharacterEncoding(utf8);
 163                 PrintWriter out = response.getWriter();
 164                 out.write(e.getMessage());
 165         }
 166
 167         @SuppressWarnings("rawtypes")
 168         public void getDataByEntity(HttpServletResponse response, String key, String value, Class className){
 169                 try{
 170                         Map<String, Object> model = new HashMap<>();
 171                         ObjectMapper mapper = new ObjectMapper();
 172                         model.put(key, mapper.writeValueAsString(commonClassDao.getDataByColumn(className, value)));
 173                         JsonMessage msg = new JsonMessage(mapper.writeValueAsString(model));
 174                         JSONObject j = new JSONObject(msg);
 175                         response.getWriter().write(j.toString());
 176                 }catch(Exception e){
 177                         LOGGER.error(XACMLErrorConstants.ERROR_PROCESS_FLOW + e);
 178                 }
 179         }
 180
 181         @SuppressWarnings("rawtypes")
 182         public void getData(HttpServletResponse response, String key, Class className){
 183                 try{
 184                         Map<String, Object> model = new HashMap<>();
 185                         ObjectMapper mapper = new ObjectMapper();
 186                         model.put(key, mapper.writeValueAsString(commonClassDao.getData(className)));
 187                         JsonMessage msg = new JsonMessage(mapper.writeValueAsString(model));
 188                         JSONObject j = new JSONObject(msg);
 189             response.addHeader("successMapKey", "success");
 190             response.addHeader(operation, "getDictionary");
 191                         response.getWriter().write(j.toString());
 192                 }catch(Exception e){
 193             LOGGER.error(XACMLErrorConstants.ERROR_PROCESS_FLOW + e);
 194             response.setStatus(HttpServletResponse.SC_BAD_REQUEST);
 195             response.addHeader("error", "dictionaryDBQuery");
 196                 }
 197         }
 198
 199         @SuppressWarnings("unchecked")
 200         public void removeData(HttpServletRequest request, HttpServletResponse response, String key, @SuppressWarnings("rawtypes") Class className) throws IOException{
 201                 try{
 202                         ObjectMapper mapper = new ObjectMapper();
 203                         mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
 204                         JsonNode root = mapper.readTree(request.getReader());
 205                         commonClassDao.delete((Object)mapper.readValue(root.get("data").toString(), className));
 206                         String responseString = mapper.writeValueAsString(commonClassDao.getData(className));
 207                         setResponseData(response, key, responseString);
 208                 }catch(Exception e){
 209                         setErrorResponseData(response, e);
 210                 }
 211         }
 212
 213 }