ONAP-PAP-REST/src/main/java/org/onap/policy/pap/xacml/rest/UpdateOthersPAPS.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;
  21
  22 import java.io.BufferedWriter;
  23 import java.io.File;
  24 import java.io.FileWriter;
  25 import java.io.IOException;
  26 import java.nio.charset.StandardCharsets;
  27 import java.nio.file.Files;
  28 import java.nio.file.Paths;
  29 import java.util.Base64;
  30 import java.util.HashMap;
  31 import java.util.List;
  32 import java.util.Map;
  33
  34 import javax.servlet.http.HttpServletRequest;
  35 import javax.servlet.http.HttpServletResponse;
  36
  37 import org.json.JSONObject;
  38 import org.onap.policy.common.logging.flexlogger.FlexLogger;
  39 import org.onap.policy.common.logging.flexlogger.Logger;
  40 import org.onap.policy.pap.xacml.rest.adapters.UpdateObjectData;
  41 import org.onap.policy.pap.xacml.rest.components.Policy;
  42 import org.onap.policy.pap.xacml.rest.util.JsonMessage;
  43 import org.onap.policy.rest.XACMLRestProperties;
  44 import org.onap.policy.rest.dao.CommonClassDao;
  45 import org.onap.policy.rest.jpa.ActionBodyEntity;
  46 import org.onap.policy.rest.jpa.ConfigurationDataEntity;
  47 import org.onap.policy.rest.jpa.PolicyDBDaoEntity;
  48 import org.onap.policy.utils.CryptoUtils;
  49 import org.onap.policy.xacml.api.XACMLErrorConstants;
  50 import org.springframework.beans.factory.annotation.Autowired;
  51 import org.springframework.http.HttpEntity;
  52 import org.springframework.http.HttpHeaders;
  53 import org.springframework.http.HttpMethod;
  54 import org.springframework.http.HttpStatus;
  55 import org.springframework.stereotype.Controller;
  56 import org.springframework.web.bind.annotation.RequestBody;
  57 import org.springframework.web.bind.annotation.RequestMapping;
  58 import org.springframework.web.bind.annotation.RequestMethod;
  59 import org.springframework.web.bind.annotation.ResponseBody;
  60 import org.springframework.web.client.HttpClientErrorException;
  61 import org.springframework.web.client.RestTemplate;
  62
  63 import com.fasterxml.jackson.databind.ObjectMapper;
  64
  65 @Controller
  66 public class UpdateOthersPAPS {
  67
  68     private static final Logger policyLogger    = FlexLogger.getLogger(UpdateOthersPAPS.class);
  69
  70     private static CommonClassDao commonClassDao;
  71
  72     private static final String contentType = "application/json";
  73     private static String configType =".Config_";
  74     private static String actionType =".Action_";
  75     private static String error ="error";
  76     public static CommonClassDao getCommonClassDao() {
  77         return commonClassDao;
  78     }
  79
  80     public static void setCommonClassDao(CommonClassDao commonClassDao) {
  81         UpdateOthersPAPS.commonClassDao = commonClassDao;
  82     }
  83
  84     @Autowired
  85     private UpdateOthersPAPS(CommonClassDao commonClassDao){
  86         UpdateOthersPAPS.commonClassDao = commonClassDao;
  87     }
  88
  89     public UpdateOthersPAPS() {
  90         //Empty Constructor
  91     }
  92
  93     @RequestMapping(value="/notifyOtherPAPs", method= RequestMethod.POST)
  94     public void  notifyOthersPAPsToUpdateConfigurations(HttpServletRequest request, HttpServletResponse response){
  95         Map<String, Object> model = new HashMap<>();
  96         ObjectMapper mapper = new ObjectMapper();
  97         UpdateObjectData body = new UpdateObjectData();
  98         body.setAction(request.getParameter("action"));
  99         body.setNewPolicyName(request.getParameter("newPolicyName"));
 100         body.setOldPolicyName(request.getParameter("oldPolicyName"));
 101
 102         String currentPap = XACMLRestProperties.getProperty("xacml.rest.pap.url");
 103         List<Object> getPAPUrls = commonClassDao.getData(PolicyDBDaoEntity.class);
 104         if(getPAPUrls != null && !getPAPUrls.isEmpty()){
 105             for(int i = 0; i < getPAPUrls.size(); i++){
 106                 PolicyDBDaoEntity papId = (PolicyDBDaoEntity) getPAPUrls.get(i);
 107                 String papUrl = papId.getPolicyDBDaoUrl();
 108                 if(!papUrl.equals(currentPap)){
 109                     String userName = papId.getUsername();
 110                     String password = papId.getPassword();
 111                     Base64.Encoder encoder = Base64.getEncoder();
 112                     String txt;
 113                     try{
 114                         txt = new String(CryptoUtils.decryptTxt(password), StandardCharsets.UTF_8);
 115                     } catch(Exception e){
 116                         policyLogger.debug(e);
 117                         //if we can't decrypt, might as well try it anyway
 118                         txt = password;
 119                     }
 120                     String encoding = encoder.encodeToString((userName+":"+txt).getBytes(StandardCharsets.UTF_8));
 121                     HttpHeaders headers = new HttpHeaders();
 122                     headers.set("Authorization", "Basic " + encoding);
 123                     headers.set("Content-Type", contentType);
 124
 125                     RestTemplate restTemplate = new RestTemplate();
 126                     HttpEntity<?> requestEntity = new HttpEntity<>(body, headers);
 127                     HttpClientErrorException exception = null;
 128
 129                     try{
 130                         restTemplate.exchange(papUrl + "onap/updateConfiguration", HttpMethod.POST, requestEntity, String.class);
 131                     }catch(Exception e){
 132                         policyLogger.error(XACMLErrorConstants.ERROR_PROCESS_FLOW + "Error while connecting to " + papUrl, e);
 133                         exception = new HttpClientErrorException(HttpStatus.INTERNAL_SERVER_ERROR, e.getMessage());
 134                         if("409 Conflict".equals(e.getMessage())){
 135                             policyLogger.error(e.getMessage());
 136                             response.addHeader(error, e.getMessage());
 137                         }
 138                     }
 139                     if(exception != null && exception.getStatusCode()!=null){
 140                         String message;
 141                         if(exception.getStatusCode().equals(HttpStatus.UNAUTHORIZED)){
 142                             message = XACMLErrorConstants.ERROR_PERMISSIONS +":"+exception.getStatusCode()+":" + "ERROR_AUTH_GET_PERM" ;
 143                             policyLogger.error(message);
 144                         }else if(exception.getStatusCode().equals(HttpStatus.BAD_REQUEST)){
 145                             message = XACMLErrorConstants.ERROR_DATA_ISSUE + ":"+exception.getStatusCode()+":" + exception.getResponseBodyAsString();
 146                             policyLogger.error(message);
 147                         }else if(exception.getStatusCode().equals(HttpStatus.NOT_FOUND)){
 148                             message = XACMLErrorConstants.ERROR_PROCESS_FLOW + "Error while connecting to " + papUrl + exception;
 149                             policyLogger.error(message);
 150                         }else{
 151                             message = XACMLErrorConstants.ERROR_PROCESS_FLOW + ":"+exception.getStatusCode()+":" + exception.getResponseBodyAsString();
 152                             policyLogger.error(message);
 153                         }
 154                         model.put(papUrl, message);
 155                     }else{
 156                         model.put(papUrl, "Success");
 157                     }
 158                 }
 159             }
 160             JsonMessage msg;
 161             try {
 162                 msg = new JsonMessage(mapper.writeValueAsString(model));
 163                 JSONObject j = new JSONObject(msg);
 164                 response.getWriter().write(j.toString());
 165             } catch (Exception e) {
 166                 policyLogger.error("Exception Occured"+e);
 167             }
 168         }
 169     }
 170
 171     @RequestMapping(value="/updateConfiguration", method= RequestMethod.POST)
 172     @ResponseBody
 173     public void updateConfiguration(@RequestBody UpdateObjectData data, HttpServletResponse response){
 174         String action = data.getAction();
 175         String newPolicyName = data.getNewPolicyName();
 176         String oldPolicyName = data.getOldPolicyName();
 177         try{
 178             if("rename".equals(action)){
 179                 if(oldPolicyName.contains(configType) || oldPolicyName.contains(actionType)){
 180                     File file;
 181                     if(oldPolicyName.contains(configType)){
 182                         file = new File(Policy.getConfigHome() + File.separator + oldPolicyName);
 183                     }else{
 184                         file = new File(Policy.getActionHome() + File.separator + oldPolicyName);
 185                     }
 186                     if(file.exists()){
 187                         File renamefile;
 188                         if(oldPolicyName.contains(configType)){
 189                             renamefile = new File(Policy.getConfigHome() + File.separator + newPolicyName);
 190                         }else{
 191                             renamefile = new File(Policy.getActionHome() + File.separator + newPolicyName);
 192                         }
 193                         if(file.renameTo(renamefile)){
 194                             policyLogger.info("Policy has been renamed Successfully"+newPolicyName);
 195                             response.addHeader("rename", "Success");
 196                         }else{
 197                             response.addHeader("rename", "Failure");
 198                         }
 199                     }
 200                 }
 201             }else if("delete".equals(action)){
 202                 if(oldPolicyName.contains(configType)){
 203                     Files.deleteIfExists(Paths.get(Policy.getConfigHome() + File.separator + oldPolicyName));
 204                 }else if(oldPolicyName.contains("Action_")){
 205                     Files.deleteIfExists(Paths.get(Policy.getActionHome() + File.separator + oldPolicyName));
 206                 }
 207             }else if("clonePolicy".equals(action) || "exportPolicy".equals(action)){
 208                 if(newPolicyName.contains(configType)){
 209                     ConfigurationDataEntity configEntiy = (ConfigurationDataEntity) commonClassDao.getEntityItem(ConfigurationDataEntity.class, "configurationName", newPolicyName);
 210                     saveConfigurationData(configEntiy, newPolicyName);
 211                 }else if(newPolicyName.contains(actionType)){
 212                     ActionBodyEntity actionEntiy = (ActionBodyEntity) commonClassDao.getEntityItem(ActionBodyEntity.class, "actionBodyName", newPolicyName);
 213                     saveActionBodyData(actionEntiy, newPolicyName);
 214                 }
 215             }
 216         } catch (IOException e) {
 217             policyLogger.error("Exception Occured While updating Configuration"+e);
 218         }
 219     }
 220
 221     private void saveConfigurationData(ConfigurationDataEntity configEntiy, String newPolicyName){
 222         try(FileWriter fw = new FileWriter(Policy.getConfigHome() + File.separator + newPolicyName)){
 223             BufferedWriter bw = new BufferedWriter(fw);
 224             bw.write(configEntiy.getConfigBody());
 225             bw.close();
 226         }catch (IOException e) {
 227             policyLogger.error("Exception Occured While closing the File input stream"+e);
 228         }
 229     }
 230
 231     private void saveActionBodyData(ActionBodyEntity actionEntiy , String newPolicyName){
 232         try(FileWriter fw  = new FileWriter(Policy.getActionHome() + File.separator + newPolicyName)){
 233             BufferedWriter bw = new BufferedWriter(fw);
 234             bw.write(actionEntiy.getActionBody());
 235             bw.close();
 236         }catch (IOException e) {
 237             policyLogger.error("Exception Occured While closing the File input stream"+e);
 238         }
 239     }
 240 }