/**
 * Copyright 2005-2018 The Kuali Foundation
 *
 * Licensed under the Educational Community 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.opensource.org/licenses/ecl2.php
 *
 * 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.
 */
package org.kuali.rice.krad.labs.encryption;

import org.kuali.rice.core.api.CoreApiServiceLocator;
import org.kuali.rice.core.api.encryption.EncryptionService;
import org.kuali.rice.krad.util.GlobalVariables;
import org.kuali.rice.krad.web.controller.UifControllerBase;
import org.kuali.rice.krad.web.form.UifFormBase;
import org.springframework.stereotype.Controller;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.security.GeneralSecurityException;

/**
 * Controller for the encrypt/decrypt utility
 *
 * @author Kuali Rice Team (rice.collab@kuali.org)
 */
@Controller
@RequestMapping(value = "/encryption")
public class EncryptionController extends UifControllerBase {

    public static final String INPUT_FIELD = "input";
    public static final String ENCRYPTION_ERROR = "labs.encryption.error";
    private  EncryptionService encryptionService;

    /**
     * Polulate the encryptionServiceName on the form.
     */
    @Override
    protected EncryptionForm createInitialForm() {
        EncryptionForm encryptionForm = new EncryptionForm();
        encryptionForm.setEncryptionServiceName(getEncryptionService().getClass().getSimpleName());
        return encryptionForm;
    }

    /**
     * Encrypt the text of the input field.
     *
     * <p>
     * The encryptedText and decryptedText filds are populated when encryption is successful.  An error message
     * is displayed otherwise.
     * </p>
     */
    @RequestMapping(params = "methodToCall=encrypt")
    public ModelAndView encrypt(@ModelAttribute("KualiForm") UifFormBase form, BindingResult result,
            HttpServletRequest request, HttpServletResponse response) {
        EncryptionForm encryptionForm = (EncryptionForm) form;

        try {
            encryptionForm.setEncryptedText(getEncryptionService().encrypt(
                    encryptionForm.getInput()));
            encryptionForm.setDecryptedText(encryptionForm.getInput());
        }
        catch (GeneralSecurityException gse) {
            GlobalVariables.getMessageMap().putError(INPUT_FIELD, ENCRYPTION_ERROR, gse.toString());
        }

        return getModelAndView(form);
    }

    /**
     * Decrypt the text of the input field.
     *
     * <p>
     * The encryptedText and decryptedText filds are populated when decryption is successful.  An error message
     * is displayed otherwise.
     * </p>
     */
    @RequestMapping(params = "methodToCall=decrypt")
    public ModelAndView decrypt(@ModelAttribute("KualiForm") UifFormBase form, BindingResult result,
            HttpServletRequest request, HttpServletResponse response) {
        EncryptionForm encryptionForm = (EncryptionForm) form;

        try {
            encryptionForm.setEncryptedText(encryptionForm.getInput());
            encryptionForm.setDecryptedText(getEncryptionService().decrypt(
                    encryptionForm.getInput()));
        }
        catch (GeneralSecurityException gse) {
            GlobalVariables.getMessageMap().putError(INPUT_FIELD, ENCRYPTION_ERROR, gse.toString());
        }

        return getModelAndView(form);
    }

    /**
     * Get the encryption service
     *
     * @return EncryptionService
     */
    private EncryptionService getEncryptionService() {
        if (encryptionService == null) {
            encryptionService = CoreApiServiceLocator.getEncryptionService();
        }

        return encryptionService;
    }
}
