package com.gt.agent.util;

import javax.crypto.Cipher;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;
import Decoder.BASE64Decoder;
import Decoder.BASE64Encoder;
// 加密
public class EncryptUtils {

    public static String Encrypt(String sSrc, String sKey) throws Exception {
        if (sKey == null) {
            System.out.print("Key為空null");
            return null;
        }
        // 判斷Key是否為16位
        if (sKey.length() != 16) {
            System.out.print("Key長度不是16位");
            return null;
        }
        byte[] raw = sKey.getBytes();
        SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");
        Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");//"演算法/模式/補碼方式"
        IvParameterSpec iv = new IvParameterSpec("0102030405060708".getBytes());//使用CBC模式，需要一個向量iv，可增加加密演算法的強度
        cipher.init(Cipher.ENCRYPT_MODE, skeySpec, iv);
        byte[] encrypted = cipher.doFinal(sSrc.getBytes());


        return new BASE64Encoder().encode(encrypted);//此處使用BASE64做轉碼功能，同時能起到2次加密的作用。
    }


    // 解密
    public static String Decrypt(String sSrc, String sKey) throws Exception {
        try {
            // 判斷Key是否正確
            if (sKey == null) {
                System.out.print("Key為空null");
                return null;
            }
            // 判斷Key是否為16位
            if (sKey.length() != 16) {
                System.out.print("Key長度不是16位");
                return null;
            }
            byte[] raw = sKey.getBytes("ASCII");
            SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");
            Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
            IvParameterSpec iv = new IvParameterSpec("0102030405060708"
                    .getBytes());
            cipher.init(Cipher.DECRYPT_MODE, skeySpec, iv);
            byte[] encrypted1 = new BASE64Decoder().decodeBuffer(sSrc);//先用base64解密
            try {
                byte[] original = cipher.doFinal(encrypted1);
                String originalString = new String(original);
                return originalString;
            } catch (Exception e) {
                System.out.println(e.toString());
                return null;
            }
        } catch (Exception ex) {
            System.out.println(ex.toString());
            return null;
        }
    }


    public static void main(String[] args) throws Exception {
        /*
         * 加密用的Key 可以用26個字母和數字組成，最好不要用保留字元，雖然不會錯，至於怎麼裁決，個人看情況而定
         * 此處使用AES-128-CBC加密模式，key需要為16位     +  base64加密。
         */
        String cKey = "123456789";
        StringBuffer markIds = new StringBuffer();
        markIds.append(1).append(",");
        String replace = markIds.substring(0, markIds.length() - 1).replace(",", "','");
        
        String time = "1490840793103";
        String sysName = "haha";

        String orgCode = "13456789";

//--自定義加密規則
        String keyString = sysName + time + orgCode;   
keyString = keyString.substring(keyString.length()-6, keyString.length()

)+keyString.substring(0, keyString.length() - 6);


        // 需要加密的字串
        String cSrc = keyString;
        System.out.println(cSrc);
        // 加密
        long lStart = System.currentTimeMillis();
        System.out.println(System.currentTimeMillis());
        String enString = EncryptUtils.Encrypt(cSrc, cKey);
        System.out.println("加密後的字串是：" + enString);


        long lUseTime = System.currentTimeMillis() - lStart;
        System.out.println("加密耗時：" + lUseTime + "毫秒");
        // 解密
        lStart = System.currentTimeMillis();
        String DeString = EncryptUtils.Decrypt(enString, cKey);
        System.out.println("解密後的字串是：" + DeString);
        lUseTime = System.currentTimeMillis() - lStart;
        System.out.println("解密耗時：" + lUseTime + "毫秒");
        
        System.out.println(System.currentTimeMillis());
    }
}