package com.mattrixwv.cipherstream.controller.polysubstitution;


import java.util.List;

import org.slf4j.MDC;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.PropertySource;
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.mattrixwv.cipherstream.aspect.CipherStreamLoggingAspect;
import com.mattrixwv.cipherstream.polysubstitution.RailFence;
import com.mattrixwv.cipherstream.utils.CipherInfoUtil;
import com.mattrixwv.cipherstream.utils.CipherParameterUtil;

import lombok.extern.slf4j.Slf4j;
import tools.jackson.databind.node.ObjectNode;


@Slf4j
@RestController
@RequestMapping("/railFence")
@PropertySource("classpath:ciphers.properties")
public class RailFenceController{
	@Value("${cipher.poly.railfence.name}")
	private String railFenceName;
	@Value("${cipher.poly.railfence.description}")
	private String railFenceDescription;
	@Value("#{${cipher.poly.railfence.explanation}}")
	private List<String> railFenceExplanation;
	@Value("#{${cipher.poly.railfence.facts}}")
	private List<String> railFenceFacts;


	@GetMapping
	public ObjectNode getCipherInfo(){
		MDC.put(CipherStreamLoggingAspect.CIPHER_NAME_LOGGING, railFenceName);
		log.info("Getting info for {}", railFenceName);


		return CipherInfoUtil.buildInfoNode(railFenceName, railFenceDescription, railFenceExplanation, railFenceFacts);
	}

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


		CipherParameterUtil.verifyRailFenceParams(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();
		int rails = cipherParams.get(CipherParameterUtil.RAIL_FENCE_RAILS).asInt();
		String inputString = cipherParams.get(CipherParameterUtil.INPUT_STRING).asString();


		RailFence railFence = new RailFence(preserveCapitals, preserveWhitespace, preserveSymbols);
		String outputString = railFence.encode(rails, inputString);

		cipherParams.put(CipherParameterUtil.OUTPUT_STRING, outputString);


		return cipherParams;
	}

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


		CipherParameterUtil.verifyRailFenceParams(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();
		int rails = cipherParams.get(CipherParameterUtil.RAIL_FENCE_RAILS).asInt();
		String inputString = cipherParams.get(CipherParameterUtil.INPUT_STRING).asString();


		RailFence railFence = new RailFence(preserveCapitals, preserveWhitespace, preserveSymbols);
		String outputString = railFence.decode(rails, inputString);

		cipherParams.put(CipherParameterUtil.OUTPUT_STRING, outputString);


		return cipherParams;
	}
}