package weaver.authentication.handler.encoder;

import org.jasig.cas.authentication.handler.PasswordEncoder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.jdbc.core.simple.SimpleJdbcTemplate;
import weaver.loginsso.AES;

import javax.sql.DataSource;
import java.util.Map;

/**
 * Created by crazyDream on 2018/9/17.
 */
public class SM3PasswordEncoder implements PasswordEncoder {
    private final Logger log = LoggerFactory.getLogger(this.getClass());
    @Override
    public String encode(String password) {
        return null;
    }

    @Override
    public String encode(String username, String password) {
        /*String yjcust = null;
        try {
            yjcust = AES.decrypt(password, "yjcust");
        } catch (Exception e) {
            e.printStackTrace();
        }
        if (yjcust != null && yjcust.indexOf("-f-g-") > 0) {
            password=yjcust.substring(0,yjcust.indexOf("-f-g-"));
        }else if (password.indexOf("-h-i-")>0) {
            password = password.substring(0, password.indexOf("-h-i-"));
        }*/

        //取salt
        Map map = getJdbcTemplate().queryForMap(this.sql, username);
        String salt = (String) map.get("salt");
        if (salt != null && salt.indexOf("sm3_new#") >= 0) {
            salt = salt.substring(8, salt.length());
        }
        SM3 sm3 = new SM3();
        String encrypt = sm3.getEncrypt(password, salt);
        return encrypt;
    }

    @Override
    public String encode(String username, String password, Map<String, Object> otherParams) {
        return null;
    }

    private String sql;

    private SimpleJdbcTemplate jdbcTemplate;

    private DataSource dataSource;

    public void setSql(final String sql) {
        this.sql = sql;
    }


    public final void setDataSource(final DataSource dataSource) {
        this.jdbcTemplate = new SimpleJdbcTemplate(dataSource);
        this.dataSource = dataSource;
    }

    private final SimpleJdbcTemplate getJdbcTemplate() {
        return this.jdbcTemplate;
    }

    private final DataSource getDataSource() {
        return this.dataSource;
    }


}