package com.liwei.blowfish;



import javax.crypto.Cipher;

import javax.crypto.spec.IvParameterSpec;

import javax.crypto.spec.SecretKeySpec;



import org.apache.commons.codec.binary.Base64;

import org.apache.commons.codec.digest.DigestUtils;

import org.apache.commons.lang.StringUtils;

import org.apache.log4j.Logger;



/**

 * ??Blowfish?????????

 * 

 */

public class BlowfishEncrypter {

    /** logger */

    private static final Logger      logger         = Logger.getLogger(BlowfishEncrypter.class);



    // ----- ?? -----

    private static String            CIPHER_NAME    = "Blowfish/CFB8/NoPadding";

    private static String            KEY_SPEC_NAME  = "Blowfish";

    private static String            CHARSET        = "GBK";



    private String                   CIPHER_KEY     = "Ay16quwz5ic-=TAOBAO=-4z5kHgYuy1X";

    private SecretKeySpec            secretKeySpec;

    private IvParameterSpec          ivParameterSpec;



    private static final ThreadLocal encrypter_pool = new ThreadLocal();



    // ----- ???? -----

    private Cipher                   cipher;



    /**

     * ????????

     */

    private BlowfishEncrypter() {

        this("Ay16quwz5ic-=TAOBAO=-4z5kHgYuy1X");

    }



    /**

     * ?????

     */

    public BlowfishEncrypter(String CIPHER_KEY) {

        this.CIPHER_KEY = CIPHER_KEY;

        secretKeySpec = new SecretKeySpec(CIPHER_KEY.getBytes(), KEY_SPEC_NAME);

        ivParameterSpec = new IvParameterSpec((DigestUtils.md5Hex(CIPHER_KEY).substring(0, 8))

            .getBytes());



        try {

            cipher = Cipher.getInstance(CIPHER_NAME);

        } catch (Exception e) {

            logger.error("[BlowfishEncrypter]", e);

        }

    }



    /**

     * ??????????????

     * 

     * <p>

     * ???????????????ThreadLocal????????????

     * </p>

     * 

     * @return

     */

    public static BlowfishEncrypter getEncrypter() {

        BlowfishEncrypter encrypter = (BlowfishEncrypter) encrypter_pool.get();



        if (encrypter == null) {

            encrypter = new BlowfishEncrypter();

            encrypter_pool.set(encrypter);

        }



        return encrypter;

    }



    /**

     * ????????

     * 

     * @param str

     * 

     * @return ??????????????????null

     */

    public String encrypt(String str) {

        if (StringUtils.isNotBlank(str)) {

            try {

                cipher.init(Cipher.ENCRYPT_MODE, secretKeySpec, ivParameterSpec);



                byte[] clearBytes = str.getBytes(CHARSET);

                byte[] cryptoBytes = cipher.doFinal(clearBytes);



                return new String(Base64.encodeBase64(cryptoBytes, false), CHARSET);

            } catch (Exception e) {

                logger.error("???????: ", e);

            }

        }



        return null;

    }



    /**

     * ????????

     * 

     * @param str

     * 

     * @return ??????????????????null

     */

    public String decrypt(String str) {

        if (StringUtils.isNotBlank(str)) {

            try {

                byte[] cryptoBase64Bytes = str.getBytes(CHARSET);

                byte[] cryptoBytes = Base64.decodeBase64(cryptoBase64Bytes);



                cipher.init(Cipher.DECRYPT_MODE, secretKeySpec, ivParameterSpec);



                byte[] clearBytes = cipher.doFinal(cryptoBytes);



                return new String(clearBytes, CHARSET);

            } catch (Exception e) {

                logger.warn("???????: ", e);

            }

        }



        return null;

    }



}