ONAP-PAP-REST/src/main/java/org/onap/policy/pap/xacml/rest/controller/BRMSDictionaryController.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP-PAP-REST
   4  * ================================================================================
   5  * Copyright (C) 2017-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
  21 package org.onap.policy.pap.xacml.rest.controller;
  22
  23 import java.io.IOException;
  24 import java.io.StringWriter;
  25 import java.nio.charset.StandardCharsets;
  26 import java.util.Date;
  27 import java.util.List;
  28
  29 import javax.servlet.http.HttpServletRequest;
  30 import javax.servlet.http.HttpServletResponse;
  31
  32 import org.apache.commons.io.IOUtils;
  33 import org.onap.policy.api.PEDependency;
  34 import org.onap.policy.common.logging.flexlogger.FlexLogger;
  35 import org.onap.policy.common.logging.flexlogger.Logger;
  36 import org.onap.policy.pap.xacml.rest.components.CreateBRMSRuleTemplate;
  37 import org.onap.policy.pap.xacml.rest.util.DictionaryUtils;
  38 import org.onap.policy.rest.dao.CommonClassDao;
  39 import org.onap.policy.rest.jpa.BRMSController;
  40 import org.onap.policy.rest.jpa.BRMSDependency;
  41 import org.onap.policy.rest.jpa.BRMSParamTemplate;
  42 import org.onap.policy.rest.jpa.OnapName;
  43 import org.onap.policy.rest.jpa.UserInfo;
  44 import org.onap.policy.utils.PolicyUtils;
  45 import org.onap.policy.xacml.api.XACMLErrorConstants;
  46 import org.springframework.beans.factory.annotation.Autowired;
  47 import org.springframework.http.MediaType;
  48 import org.springframework.stereotype.Controller;
  49 import org.springframework.web.bind.annotation.RequestMapping;
  50 import org.springframework.web.bind.annotation.RequestMethod;
  51 import org.springframework.web.servlet.ModelAndView;
  52
  53 import com.fasterxml.jackson.databind.DeserializationFeature;
  54 import com.fasterxml.jackson.databind.JsonNode;
  55 import com.fasterxml.jackson.databind.ObjectMapper;
  56
  57
  58 @Controller
  59 public class BRMSDictionaryController{
  60
  61     private static final Logger LOGGER  = FlexLogger.getLogger(BRMSDictionaryController.class);
  62
  63     private static final String VALIDATIONRESPONSE = "Validation";
  64
  65     private static CommonClassDao commonClassDao;
  66     private static String rule;
  67     private static String successMsg = "Success";
  68     private static String duplicateResponseString = "Duplicate";
  69     private static String ruleName = "ruleName";
  70     private static String errorMessage  = "Error";
  71     private static String operation = "operation";
  72     private static String dictionaryFields ="dictionaryFields";
  73     private static String userid = "userid";
  74     private static String dependencyName = "dependencyName";
  75     private static String controllerName = "controllerName";
  76     private static String brmsParamDatas = "brmsParamDictionaryDatas";
  77     private static String brmsDependencyDatas = "brmsDependencyDictionaryDatas";
  78     private static String brmsControllerDatas = "brmsControllerDictionaryDatas";
  79
  80     @Autowired
  81     public BRMSDictionaryController(CommonClassDao commonClassDao){
  82         BRMSDictionaryController.commonClassDao = commonClassDao;
  83     }
  84
  85     public static void setCommonClassDao(CommonClassDao commonClassDao2) {
  86         BRMSDictionaryController.commonClassDao = commonClassDao2;
  87     }
  88
  89     public BRMSDictionaryController() {
  90         super();
  91     }
  92
  93     private DictionaryUtils getDictionaryUtilsInstance(){
  94         return DictionaryUtils.getDictionaryUtils();
  95     }
  96
  97     @RequestMapping(value={"/get_BRMSParamDataByName"}, method={RequestMethod.GET} , produces=MediaType.APPLICATION_JSON_VALUE)
  98     public void getBRMSParamDictionaryByNameEntityData(HttpServletResponse response){
  99         DictionaryUtils utils = getDictionaryUtilsInstance();
 100         utils.getDataByEntity(response, brmsParamDatas, ruleName, BRMSParamTemplate.class);
 101     }
 102
 103     @RequestMapping(value={"/get_BRMSParamData"}, method={RequestMethod.GET} , produces=MediaType.APPLICATION_JSON_VALUE)
 104     public void getBRMSParamDictionaryEntityData(HttpServletResponse response){
 105         DictionaryUtils utils = getDictionaryUtilsInstance();
 106         utils.getData(response, brmsParamDatas, BRMSParamTemplate.class);
 107     }
 108
 109     @RequestMapping(value={"/brms_dictionary/set_BRMSParamData"}, method={RequestMethod.POST})
 110     public static void setRuleData(HttpServletRequest request) throws IOException{
 111         StringWriter writer = new StringWriter();
 112         IOUtils.copy(request.getInputStream() , writer, StandardCharsets.UTF_8);
 113         String cleanStreamBoundary =  writer.toString().replaceFirst("------(.*)(?s).*octet-stream", "");
 114         rule = cleanStreamBoundary.substring(0, cleanStreamBoundary.lastIndexOf("end")+4);
 115     }
 116
 117     @RequestMapping(value={"/brms_dictionary/save_BRMSParam"}, method={RequestMethod.POST})
 118     public ModelAndView saveBRMSParamDictionary(HttpServletRequest request, HttpServletResponse response) throws IOException {
 119         DictionaryUtils utils = getDictionaryUtilsInstance();
 120         try {
 121             boolean fromAPI = utils.isRequestFromAPI(request);
 122             ObjectMapper mapper = new ObjectMapper();
 123             mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
 124             JsonNode root = mapper.readTree(request.getReader());
 125
 126             BRMSParamTemplate bRMSParamTemplateData;
 127             String userId = null;
 128             if(fromAPI){
 129                 bRMSParamTemplateData = mapper.readValue(root.get(dictionaryFields).toString(), BRMSParamTemplate.class);
 130                 userId = "API";
 131             }else{
 132                 bRMSParamTemplateData = mapper.readValue(root.get("brmsParamDictionaryData").toString(), BRMSParamTemplate.class);
 133                 userId = root.get(userid).textValue();
 134             }
 135             UserInfo userInfo = utils.getUserInfo(userId);
 136
 137             List<Object> duplicateData =  commonClassDao.checkDuplicateEntry(bRMSParamTemplateData.getRuleName(), ruleName, BRMSParamTemplate.class);
 138             boolean duplicateflag = false;
 139             if(!duplicateData.isEmpty()){
 140                 BRMSParamTemplate data = (BRMSParamTemplate) duplicateData.get(0);
 141                 if(request.getParameter(operation) != null && "update".equals(request.getParameter(operation))){
 142                     bRMSParamTemplateData.setId(data.getId());
 143                 }else if((request.getParameter(operation) != null && !"update".equals(request.getParameter(operation))) ||
 144                         (request.getParameter(operation) == null && (data.getId() != bRMSParamTemplateData.getId()))){
 145                     duplicateflag = true;
 146                 }
 147             }
 148             String responseString = null;
 149             boolean validation = false;
 150             if(rule != null && CreateBRMSRuleTemplate.validateRuleParams(rule)){
 151                 bRMSParamTemplateData.setRule(rule);
 152                 validation = true;
 153                 if(!duplicateflag){
 154                     if(bRMSParamTemplateData.getId() == 0){
 155                         bRMSParamTemplateData.setUserCreatedBy(userInfo);
 156                         commonClassDao.save(bRMSParamTemplateData);
 157                     }else{
 158                         commonClassDao.update(bRMSParamTemplateData);
 159                     }
 160                     responseString = mapper.writeValueAsString(commonClassDao.getData(BRMSParamTemplate.class));
 161                 }else{
 162                     responseString = duplicateResponseString;
 163                 }
 164             }
 165
 166             if(!validation){
 167                 responseString = VALIDATIONRESPONSE;
 168             }
 169             if(fromAPI){
 170                 if (responseString!=null && !(duplicateResponseString).equals(responseString) && !VALIDATIONRESPONSE.equals(responseString)) {
 171                     responseString = successMsg;
 172                 }
 173                 ModelAndView result = new ModelAndView();
 174                 result.setViewName(responseString);
 175                 return result;
 176             }else{
 177                 utils.setResponseData(response, brmsParamDatas, responseString);
 178             }
 179         }catch(Exception e){
 180             utils.setErrorResponseData(response, e);
 181         }
 182         return null;
 183     }
 184
 185     @RequestMapping(value={"/brms_dictionary/remove_brmsParam"}, method={RequestMethod.POST})
 186     public void removeBRMSParamDictionary(HttpServletRequest request, HttpServletResponse response)throws IOException {
 187         DictionaryUtils utils = getDictionaryUtilsInstance();
 188         utils.removeData(request, response, brmsParamDatas, BRMSParamTemplate.class);
 189     }
 190
 191     @RequestMapping(value={"/get_BRMSDependencyDataByName"}, method={RequestMethod.GET} , produces=MediaType.APPLICATION_JSON_VALUE)
 192     public void getBRMSDependencyDictionaryByNameEntityData(HttpServletResponse response){
 193         DictionaryUtils utils = getDictionaryUtilsInstance();
 194         utils.getDataByEntity(response, brmsDependencyDatas, dependencyName, BRMSDependency.class);
 195     }
 196
 197     @RequestMapping(value={"/get_BRMSDependencyData"}, method={RequestMethod.GET} , produces=MediaType.APPLICATION_JSON_VALUE)
 198     public void getBRMSDependencyDictionaryEntityData(HttpServletResponse response){
 199         DictionaryUtils utils = getDictionaryUtilsInstance();
 200         utils.getData(response, brmsDependencyDatas, BRMSDependency.class);
 201     }
 202
 203     @RequestMapping(value={"/brms_dictionary/save_BRMSDependencyData"}, method={RequestMethod.POST})
 204     public ModelAndView saveBRMSDependencyDictionary(HttpServletRequest request, HttpServletResponse response) throws IOException {
 205         DictionaryUtils utils = getDictionaryUtilsInstance();
 206         try {
 207             LOGGER.debug("DictionaryController:  saveBRMSDependencyDictionary() is called");
 208             boolean fromAPI = utils.isRequestFromAPI(request);
 209             ObjectMapper mapper = new ObjectMapper();
 210             mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
 211             JsonNode root = mapper.readTree(request.getReader());
 212
 213             BRMSDependency brmsDependency;
 214             String userId = null;
 215             if(fromAPI){
 216                 brmsDependency = mapper.readValue(root.get(dictionaryFields).toString(), BRMSDependency.class);
 217                 userId = "API";
 218             }else{
 219                 brmsDependency = mapper.readValue(root.get("brmsDependencyDictionaryData").toString(), BRMSDependency.class);
 220                 userId = root.get(userid).textValue();
 221             }
 222             UserInfo userInfo = utils.getUserInfo(userId);
 223
 224             List<Object> duplicateData =  commonClassDao.checkDuplicateEntry(brmsDependency.getDependencyName(), dependencyName, BRMSDependency.class);
 225             boolean duplicateflag = false;
 226             if(!duplicateData.isEmpty()){
 227                 BRMSDependency data = (BRMSDependency) duplicateData.get(0);
 228                 if(request.getParameter(operation) != null && "update".equals(request.getParameter(operation))){
 229                     brmsDependency.setId(data.getId());
 230                 }else if((request.getParameter(operation) != null && !"update".equals(request.getParameter(operation))) ||
 231                         (request.getParameter(operation) == null && (data.getId() != brmsDependency.getId()))){
 232                     duplicateflag = true;
 233                 }
 234             }
 235             LOGGER.audit("the userId from the onap portal is: " + userId);
 236             String responseString = null;
 237             if(brmsDependency.getDependency()!=null && !("").equals(brmsDependency.getDependency().trim())){
 238                 PEDependency dependency = null;
 239                 try{
 240                     dependency = PolicyUtils.jsonStringToObject(brmsDependency.getDependency(), PEDependency.class);
 241                 }catch(Exception e){
 242                     LOGGER.error(XACMLErrorConstants.ERROR_SCHEMA_INVALID + "wrong data given for BRMS PEDependency Dictionary : " + brmsDependency.getDependency(),e);
 243                 }
 244                 if(dependency==null){
 245                     responseString = errorMessage;
 246                 }else{
 247                     if(!duplicateflag){
 248                         brmsDependency.setUserModifiedBy(userInfo);
 249                         if(brmsDependency.getId() == 0){
 250                             brmsDependency.setUserCreatedBy(userInfo);
 251                             commonClassDao.save(brmsDependency);
 252                         }else{
 253                             brmsDependency.setModifiedDate(new Date());
 254                             commonClassDao.update(brmsDependency);
 255                         }
 256                         responseString = mapper.writeValueAsString(commonClassDao.getData(BRMSDependency.class));
 257                     }else{
 258                         responseString = duplicateResponseString;
 259                     }
 260                 }
 261             }
 262
 263             if(fromAPI){
 264                 if(responseString!=null && !duplicateResponseString.equals(responseString) && !errorMessage.equals(responseString)){
 265                     responseString = successMsg;
 266                 }
 267                 ModelAndView result = new ModelAndView();
 268                 result.setViewName(responseString);
 269                 return result;
 270             }else{
 271                 utils.setResponseData(response, brmsDependencyDatas, responseString);
 272             }
 273         } catch (Exception e){
 274             utils.setErrorResponseData(response, e);
 275         }
 276         return null;
 277     }
 278
 279     @RequestMapping(value={"/brms_dictionary/remove_brmsDependency"}, method={RequestMethod.POST})
 280     public void removeBRMSDependencyDictionary(HttpServletRequest request, HttpServletResponse response) throws IOException {
 281         DictionaryUtils utils = getDictionaryUtilsInstance();
 282         utils.removeData(request, response, brmsDependencyDatas, BRMSDependency.class);
 283     }
 284
 285
 286     @RequestMapping(value={"/get_BRMSControllerDataByName"}, method={RequestMethod.GET} , produces=MediaType.APPLICATION_JSON_VALUE)
 287     public void getBRMSControllerDictionaryByNameEntityData(HttpServletResponse response){
 288         DictionaryUtils utils = getDictionaryUtilsInstance();
 289         utils.getDataByEntity(response, brmsControllerDatas, controllerName, BRMSController.class);
 290     }
 291
 292     @RequestMapping(value={"/get_BRMSControllerData"}, method={RequestMethod.GET} , produces=MediaType.APPLICATION_JSON_VALUE)
 293     public void getBRMSControllerDictionaryEntityData(HttpServletResponse response){
 294         DictionaryUtils utils = getDictionaryUtilsInstance();
 295         utils.getData(response, brmsControllerDatas, BRMSController.class);
 296     }
 297
 298     @RequestMapping(value={"/brms_dictionary/save_BRMSControllerData"}, method={RequestMethod.POST})
 299     public ModelAndView saveBRMSControllerDictionary(HttpServletRequest request, HttpServletResponse response) throws IOException{
 300         DictionaryUtils utils = getDictionaryUtilsInstance();
 301         try {
 302             LOGGER.debug("DictionaryController:  saveBRMSControllerDictionary() is called");
 303             boolean fromAPI = utils.isRequestFromAPI(request);
 304             ObjectMapper mapper = new ObjectMapper();
 305             mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
 306             JsonNode root = mapper.readTree(request.getReader());
 307             BRMSController brmsController;
 308             String userId = null;
 309             if(fromAPI){
 310                 brmsController = mapper.readValue(root.get(dictionaryFields).toString(), BRMSController.class);
 311                 userId = "API";
 312             }else{
 313                 brmsController = mapper.readValue(root.get("brmsControllerDictionaryData").toString(), BRMSController.class);
 314                 userId = root.get(userid).textValue();
 315             }
 316             UserInfo userInfo = utils.getUserInfo(userId);
 317
 318             List<Object> duplicateData =  commonClassDao.checkDuplicateEntry(brmsController.getControllerName(), controllerName, BRMSController.class);
 319             boolean duplicateflag = false;
 320             if(!duplicateData.isEmpty()){
 321                 BRMSController data = (BRMSController) duplicateData.get(0);
 322                 if(request.getParameter(operation) != null && "update".equals(request.getParameter(operation))){
 323                     brmsController.setId(data.getId());
 324                 }else if((request.getParameter(operation) != null && !"update".equals(request.getParameter(operation))) ||
 325                         (request.getParameter(operation) == null && (data.getId() != brmsController.getId()))){
 326                     duplicateflag = true;
 327                 }
 328             }
 329             String responseString = null;
 330             if(brmsController.getController()!=null && !("").equals(brmsController.getController().trim())){
 331                 PEDependency dependency = null;
 332                 try{
 333                     dependency = PolicyUtils.jsonStringToObject(brmsController.getController(), PEDependency.class);
 334                 }catch(Exception e){
 335                     LOGGER.error(XACMLErrorConstants.ERROR_SCHEMA_INVALID + "wrong data given for BRMS Controller Dictionary : " + brmsController.getController(),e);
 336                 }
 337                 if(dependency==null){
 338                     responseString = errorMessage;
 339                 }else{
 340                     if(!duplicateflag){
 341                         brmsController.setUserModifiedBy(userInfo);
 342                         if(brmsController.getId() == 0){
 343                             brmsController.setUserCreatedBy(userInfo);
 344                             commonClassDao.save(brmsController);
 345                         }else{
 346                             brmsController.setModifiedDate(new Date());
 347                             commonClassDao.update(brmsController);
 348                         }
 349                         responseString = mapper.writeValueAsString(commonClassDao.getData(OnapName.class));
 350                     }else{
 351                         responseString = duplicateResponseString;
 352                     }
 353                 }
 354             }
 355             if(fromAPI){
 356                 if (responseString!=null && !(duplicateResponseString).equals(responseString) && !(errorMessage).equals(responseString)) {
 357                     responseString = successMsg;
 358                 }
 359                 ModelAndView result = new ModelAndView();
 360                 result.setViewName(responseString);
 361                 return result;
 362             }else{
 363                 utils.setResponseData(response, brmsControllerDatas, responseString);
 364             }
 365         } catch (Exception e){
 366             utils.setErrorResponseData(response, e);
 367         }
 368         return null;
 369     }
 370
 371     @RequestMapping(value={"/brms_dictionary/remove_brmsController"}, method={RequestMethod.POST})
 372     public void removeBRMSControllerDictionary(HttpServletRequest request, HttpServletResponse response) throws IOException{
 373         DictionaryUtils utils = getDictionaryUtilsInstance();
 374         utils.removeData(request, response, brmsControllerDatas, BRMSController.class);
 375     }
 376
 377     public BRMSDependency getDependencyDataByID(String dependencyName){
 378         return (BRMSDependency) commonClassDao.getEntityItem(BRMSDependency.class, BRMSDictionaryController.dependencyName, dependencyName);
 379     }
 380
 381     public BRMSController getControllerDataByID(String controllerName){
 382         return (BRMSController) commonClassDao.getEntityItem(BRMSController.class, BRMSDictionaryController.controllerName, controllerName);
 383     }
 384 }