package com.mattrixwv.cipherstream.controller.combination;


import org.slf4j.MDC;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import com.fasterxml.jackson.databind.node.ObjectNode;
import com.mattrixwv.cipherstream.aspect.CipherStreamLoggingAspect;
import com.mattrixwv.cipherstream.combination.ADFGVX;
import com.mattrixwv.cipherstream.utils.CipherInfoUtil;
import com.mattrixwv.cipherstream.utils.CipherParameterUtil;

import lombok.extern.slf4j.Slf4j;


@Slf4j
@RestController
@RequestMapping("/cipherStream/adfgvx")
public class AdfgvxCipherController{
	@GetMapping
	public ObjectNode getCipherInfo(){
		MDC.put(CipherStreamLoggingAspect.CIPHER_NAME_LOGGING, CipherInfoUtil.ADFGVX_CIPHER_NAME);
		log.info("Getting info for {}", CipherInfoUtil.ADFGVX_CIPHER_NAME);


		return CipherInfoUtil.buildInfoNode(CipherInfoUtil.ADFGVX_CIPHER_NAME, CipherInfoUtil.ADFGVX_CIPHER_DESCRIPTION);
	}

	@PostMapping("/encode")
	public ObjectNode encodeAdfgvx(@RequestBody ObjectNode cipherParams){
		MDC.put(CipherStreamLoggingAspect.CIPHER_NAME_LOGGING, CipherInfoUtil.ADFGVX_CIPHER_NAME);
		log.info("Encoding {}", CipherInfoUtil.ADFGVX_CIPHER_NAME);


		CipherParameterUtil.verifySquareKeyword(cipherParams);
		boolean preserveCapitals = cipherParams.get(CipherParameterUtil.PRESERVE_CAPITALS).asBoolean();
		boolean preserveWhitespace = cipherParams.get(CipherParameterUtil.PRESERVE_WHITESPACE).asBoolean();
		boolean preserveSymbols = cipherParams.get(CipherParameterUtil.PRESERVE_SYMBOLS).asBoolean();
		String keyword = cipherParams.get(CipherParameterUtil.KEYWORD).asText();
		String squareKeyword = cipherParams.get(CipherParameterUtil.SQUARE_KEYWORD).asText();
		String inputString = cipherParams.get(CipherParameterUtil.INPUT_STRING).asText();


		ADFGVX adfgvx = new ADFGVX(preserveCapitals, preserveWhitespace, preserveSymbols);
		String outputString = adfgvx.encode(squareKeyword, keyword, inputString);

		cipherParams.put(CipherParameterUtil.OUTPUT_STRING, outputString);


		return cipherParams;
	}

	@PostMapping("/decode")
	public ObjectNode decodeAdfgvx(@RequestBody ObjectNode cipherParams){
		MDC.put(CipherStreamLoggingAspect.CIPHER_NAME_LOGGING, CipherInfoUtil.ADFGVX_CIPHER_NAME);
		log.info("Decoding {}", CipherInfoUtil.ADFGVX_CIPHER_NAME);


		CipherParameterUtil.verifySquareKeyword(cipherParams);
		boolean preserveCapitals = cipherParams.get(CipherParameterUtil.PRESERVE_CAPITALS).asBoolean();
		boolean preserveWhitespace = cipherParams.get(CipherParameterUtil.PRESERVE_WHITESPACE).asBoolean();
		boolean preserveSymbols = cipherParams.get(CipherParameterUtil.PRESERVE_SYMBOLS).asBoolean();
		String keyword = cipherParams.get(CipherParameterUtil.KEYWORD).asText();
		String squareKeyword = cipherParams.get(CipherParameterUtil.SQUARE_KEYWORD).asText();
		String inputString = cipherParams.get(CipherParameterUtil.INPUT_STRING).asText();


		ADFGVX adfgvx = new ADFGVX(preserveCapitals, preserveWhitespace, preserveSymbols);
		String outputString = adfgvx.decode(squareKeyword, keyword, inputString);

		cipherParams.put(CipherParameterUtil.OUTPUT_STRING, outputString);


		return cipherParams;
	}
}