/*-
 * ============LICENSE_START=======================================================
 * ONAP-PAP-REST
 * ================================================================================
 * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ============LICENSE_END=========================================================
 */

package org.onap.policy.pap.xacml.rest.controller;

import java.io.IOException;
import java.io.PrintWriter;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.json.JSONObject;
import org.onap.policy.common.logging.flexlogger.FlexLogger;
import org.onap.policy.common.logging.flexlogger.Logger;
import org.onap.policy.pap.xacml.rest.util.JsonMessage;
import org.onap.policy.rest.dao.CommonClassDao;
import org.onap.policy.rest.jpa.RiskType;
import org.onap.policy.rest.jpa.SafePolicyWarning;
import org.onap.policy.rest.jpa.UserInfo;
import org.onap.policy.utils.PolicyUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.servlet.ModelAndView;

import com.fasterxml.jackson.databind.DeserializationFeature;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;

@Controller
public class SafePolicyController {

	private static final Logger LOGGER  = FlexLogger.getLogger(SafePolicyController.class);
	
	private static CommonClassDao commonClassDao;
	private static String duplicateResponseString = "Duplicate";
	private static String operation = "operation";
	private static String apiflag = "apiflag";
	private static String utf8 = "UTF-8";
	private static String applicationJsonContentType = "application / json";

	@Autowired
	public SafePolicyController(CommonClassDao commonClassDao){
		SafePolicyController.commonClassDao = commonClassDao;
	}
	
	public SafePolicyController(){}	
	
	public UserInfo getUserInfo(String loginId){
		return (UserInfo) commonClassDao.getEntityItem(UserInfo.class, "userLoginId", loginId);
	}
	
	@RequestMapping(value = { "/get_RiskTypeDataByName" }, method = {
			org.springframework.web.bind.annotation.RequestMethod.GET }, produces = MediaType.APPLICATION_JSON_VALUE)
	public void getRiskTypeDictionaryByNameEntityData(HttpServletResponse response) {
		try {
			Map<String, Object> model = new HashMap<>();
			ObjectMapper mapper = new ObjectMapper();
			model.put("riskTypeDictionaryDatas", mapper.writeValueAsString(commonClassDao.getDataByColumn(RiskType.class, "name")));
			JsonMessage msg = new JsonMessage(mapper.writeValueAsString(model));
			JSONObject j = new JSONObject(msg);
			response.getWriter().write(j.toString());
		} catch (Exception e) {
			LOGGER.error("Exception Occured"+e);
		}
	}

	@RequestMapping(value = { "/get_RiskTypeData" }, method = {
			org.springframework.web.bind.annotation.RequestMethod.GET }, produces = MediaType.APPLICATION_JSON_VALUE)
	public void getOnapNameDictionaryEntityData(HttpServletResponse response) {
		try {
			Map<String, Object> model = new HashMap<>();
			ObjectMapper mapper = new ObjectMapper();
			model.put("riskTypeDictionaryDatas", mapper.writeValueAsString(commonClassDao.getData(RiskType.class)));
			JsonMessage msg = new JsonMessage(mapper.writeValueAsString(model));
			JSONObject j = new JSONObject(msg);
            response.addHeader("successMapKey", "success"); 
            response.addHeader(operation, "getDictionary");
			response.getWriter().write(j.toString());
		} catch (Exception e) {
            LOGGER.error(e);
            response.setStatus(HttpServletResponse.SC_BAD_REQUEST);                             
            response.addHeader("error", "dictionaryDBQuery");
		}
	}

	@RequestMapping(value = { "/sp_dictionary/save_riskType" }, method = {
			org.springframework.web.bind.annotation.RequestMethod.POST })
	public ModelAndView saveRiskTypeDictionary(HttpServletRequest request, HttpServletResponse response)
			throws IOException {
		try {
			boolean duplicateflag = false;
            boolean isFakeUpdate = false;
            boolean fromAPI = false;
            if (request.getParameter(apiflag)!=null && ("api").equalsIgnoreCase(request.getParameter(apiflag))) {
                fromAPI = true;
            }
			ObjectMapper mapper = new ObjectMapper();
			mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
			JsonNode root = mapper.readTree(request.getReader());
            RiskType riskTypeData;
            String userId = null;
            if (fromAPI) {
                riskTypeData = (RiskType) mapper.readValue(root.get("dictionaryFields").toString(),
                        RiskType.class);
                userId = "API";
                
                //check if update operation or create, get id for data to be updated and update attributeData
                if ("update".equalsIgnoreCase(request.getParameter(operation))){
                    List<Object> duplicateData =  commonClassDao.checkDuplicateEntry(riskTypeData.getRiskName(), "name", RiskType.class);
                    RiskType data = (RiskType) duplicateData.get(0);
                    int id = data.getId();
                    
                    if(id==0){
                        isFakeUpdate=true;
                        riskTypeData.setId(1);
                    } else {
                        riskTypeData.setId(id);
                    }
                    
                    riskTypeData.setUserCreatedBy(this.getUserInfo(userId));
                }
            } else {
            	riskTypeData = (RiskType) mapper.readValue(root.get("riskTypeDictionaryData").toString(), RiskType.class);
            	userId = root.get("userid").textValue();
            }
			 
			if (riskTypeData.getId() == 0) {
				List<Object> duplicateData =  commonClassDao.checkDuplicateEntry(riskTypeData.getRiskName(), "name", RiskType.class);
				if(!duplicateData.isEmpty()){
					duplicateflag = true;
				}else{
					riskTypeData.setUserCreatedBy(getUserInfo(userId));
					riskTypeData.setUserModifiedBy(getUserInfo(userId));
					commonClassDao.save(riskTypeData);
				}
			} else {
				if (!isFakeUpdate) {
					riskTypeData.setUserModifiedBy(this.getUserInfo(userId));
					riskTypeData.setModifiedDate(new Date());
					commonClassDao.update(riskTypeData);
				}
			}
            String responseString = "";
            if(duplicateflag){
                responseString = duplicateResponseString;
            }else{
                responseString = mapper.writeValueAsString(commonClassDao.getData(RiskType.class));
            }
            
            if (fromAPI) {
                if (responseString!=null && !responseString.equals(duplicateResponseString)) {
                    if(isFakeUpdate){
                        responseString = "Exists";
                    } else {
                        responseString = "Success";
                    }
                }
                ModelAndView result = new ModelAndView();
                result.setViewName(responseString);
                return result;
            } else {
                response.setCharacterEncoding(utf8);
                response.setContentType(applicationJsonContentType);
                request.setCharacterEncoding(utf8);
 
                PrintWriter out = response.getWriter();
                JSONObject j = new JSONObject("{riskTypeDictionaryDatas: " + responseString + "}");
                out.write(j.toString());
                return null;
            }
        }catch (Exception e) {
        	LOGGER.error(e);
			response.setCharacterEncoding(utf8);
			request.setCharacterEncoding(utf8);
			PrintWriter out = response.getWriter();
			out.write(PolicyUtils.CATCH_EXCEPTION);
		}
		return null;
	}

	@RequestMapping(value = { "/sp_dictionary/remove_riskType" }, method = {
			org.springframework.web.bind.annotation.RequestMethod.POST })
	public ModelAndView removeOnapDictionary(HttpServletRequest request, HttpServletResponse response)
			throws IOException{
		try {
			ObjectMapper mapper = new ObjectMapper();
			mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
			JsonNode root = mapper.readTree(request.getReader());
			RiskType onapData = (RiskType) mapper.readValue(root.get("data").toString(), RiskType.class);
			commonClassDao.delete(onapData);
			response.setCharacterEncoding(utf8);
			response.setContentType(applicationJsonContentType);
			request.setCharacterEncoding(utf8);

			PrintWriter out = response.getWriter();

			String responseString = mapper.writeValueAsString(commonClassDao.getData(RiskType.class));
			JSONObject j = new JSONObject("{riskTypeDictionaryDatas: " + responseString + "}");
			out.write(j.toString());

			return null;
		} catch (Exception e) {
			LOGGER.error(e);
			response.setCharacterEncoding(utf8);
			request.setCharacterEncoding(utf8);
			PrintWriter out = response.getWriter();
			out.write(PolicyUtils.CATCH_EXCEPTION);
		}
		return null;
	}

	@RequestMapping(value = { "/get_SafePolicyWarningDataByName" }, method = {
			org.springframework.web.bind.annotation.RequestMethod.GET }, produces = MediaType.APPLICATION_JSON_VALUE)
	public void getSafePolicyWarningEntityDataByName(HttpServletResponse response) {
		try {
			Map<String, Object> model = new HashMap<>();
			ObjectMapper mapper = new ObjectMapper();
			model.put("safePolicyWarningDatas",
					mapper.writeValueAsString(commonClassDao.getDataByColumn(SafePolicyWarning.class, "name")));
			JsonMessage msg = new JsonMessage(mapper.writeValueAsString(model));
			JSONObject j = new JSONObject(msg);
			response.getWriter().write(j.toString());
		} catch (Exception e) {
			LOGGER.error("Exception Occured"+e);
		}
	}

	@RequestMapping(value = { "/get_SafePolicyWarningData" }, method = {
			org.springframework.web.bind.annotation.RequestMethod.GET }, produces = MediaType.APPLICATION_JSON_VALUE)
	public void getSafePolicyWarningeEntityData(HttpServletResponse response) {
		try {
			Map<String, Object> model = new HashMap<>();
			ObjectMapper mapper = new ObjectMapper();
			model.put("safePolicyWarningDatas",
					mapper.writeValueAsString(commonClassDao.getData(SafePolicyWarning.class)));
			JsonMessage msg = new JsonMessage(mapper.writeValueAsString(model));
			JSONObject j = new JSONObject(msg);
            response.addHeader("successMapKey", "success"); 
            response.addHeader(operation, "getDictionary");
			response.getWriter().write(j.toString());
		} catch (Exception e) {
			LOGGER.error(e);
            response.setStatus(HttpServletResponse.SC_BAD_REQUEST);                             
            response.addHeader("error", "dictionaryDBQuery");
		}
	}

	@RequestMapping(value = { "/sp_dictionary/save_safePolicyWarning" }, method = {
			org.springframework.web.bind.annotation.RequestMethod.POST })
	public ModelAndView saveSafePolicyWarningDictionary(HttpServletRequest request, HttpServletResponse response)
			throws IOException {
		try {
			boolean duplicateflag = false;
            boolean isFakeUpdate = false;
            boolean fromAPI = false;
            if (request.getParameter(apiflag)!=null && ("api").equalsIgnoreCase(request.getParameter(apiflag))) {
                fromAPI = true;
            }
			ObjectMapper mapper = new ObjectMapper();
			mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
			JsonNode root = mapper.readTree(request.getReader());
			SafePolicyWarning safePolicyWarning;
            if (fromAPI) {
                safePolicyWarning = (SafePolicyWarning) mapper
                        .readValue(root.get("dictionaryFields").toString(), SafePolicyWarning.class);
                
                //check if update operation or create, get id for data to be updated and update attributeData
                if (("update").equals(request.getParameter(operation))) {
                    List<Object> duplicateData =  commonClassDao.checkDuplicateEntry(safePolicyWarning.getName(), "name", SafePolicyWarning.class);
                    SafePolicyWarning data = (SafePolicyWarning) duplicateData.get(0);
                    int id = data.getId();
                    
                    if(id==0){
                        isFakeUpdate=true;
                        safePolicyWarning.setId(1);
                    } else {
                        safePolicyWarning.setId(id);
                    } 
                }
            } else {
            	safePolicyWarning = (SafePolicyWarning) mapper.readValue(root.get("safePolicyWarningData").toString(), SafePolicyWarning.class);
            }

			if (safePolicyWarning.getId() == 0) {
				List<Object> duplicateData =  commonClassDao.checkDuplicateEntry(safePolicyWarning.getName(), "name", SafePolicyWarning.class);
				if(!duplicateData.isEmpty()){
					duplicateflag = true;
				}else{
					commonClassDao.save(safePolicyWarning);
				}
			} else {
				if(!isFakeUpdate) {
					commonClassDao.update(safePolicyWarning);
				}
			}
            String responseString = "";
            if(duplicateflag){
                responseString = duplicateResponseString;
            }else{
                responseString = mapper.writeValueAsString(commonClassDao.getData(SafePolicyWarning.class));
            }
            
            if (fromAPI) {
                if (responseString!=null && !responseString.equals(duplicateResponseString)) {
                    if(isFakeUpdate){
                        responseString = "Exists";
                    } else {
                        responseString = "Success";
                    }
                }
                ModelAndView result = new ModelAndView();
                result.setViewName(responseString);
                return result;
            } else {
                response.setCharacterEncoding(utf8);
                response.setContentType(applicationJsonContentType);
                request.setCharacterEncoding(utf8);
 
                PrintWriter out = response.getWriter();
                JSONObject j = new JSONObject("{safePolicyWarningDatas: " + responseString + "}");
                out.write(j.toString());
                return null;
            }
 
        }catch (Exception e) {
        	LOGGER.error(e);
			response.setCharacterEncoding(utf8);
			request.setCharacterEncoding(utf8);
			PrintWriter out = response.getWriter();
			out.write(PolicyUtils.CATCH_EXCEPTION);
		}
		return null;
	}

	@RequestMapping(value = { "/sp_dictionary/remove_SafePolicyWarning" }, method = {
			org.springframework.web.bind.annotation.RequestMethod.POST })
	public ModelAndView removeSafePolicyWarningDictionary(HttpServletRequest request, HttpServletResponse response)
			throws IOException {
		try {
			ObjectMapper mapper = new ObjectMapper();
			mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
			JsonNode root = mapper.readTree(request.getReader());
			SafePolicyWarning safePolicyWarningData = (SafePolicyWarning) mapper.readValue(root.get("data").toString(),
					SafePolicyWarning.class);
			commonClassDao.delete(safePolicyWarningData);
			response.setCharacterEncoding(utf8);
			response.setContentType(applicationJsonContentType);
			request.setCharacterEncoding(utf8);

			PrintWriter out = response.getWriter();

			String responseString = mapper.writeValueAsString(commonClassDao.getData(SafePolicyWarning.class));
			JSONObject j = new JSONObject("{groupPolicyScopeListDatas: " + responseString + "}");
			out.write(j.toString());

			return null;
		} catch (Exception e) {
			LOGGER.error(e);
			response.setCharacterEncoding(utf8);
			request.setCharacterEncoding(utf8);
			PrintWriter out = response.getWriter();
			out.write(PolicyUtils.CATCH_EXCEPTION);
		}
		return null;
	}

}