package com.profiler.modifier;

import com.profiler.config.ProfilerConfig;
import com.profiler.interceptor.bci.ByteCodeInstrumentor;
import com.profiler.modifier.arcus.ArcusClientModifier;
import com.profiler.modifier.connector.HTTPClientModifier;
import com.profiler.modifier.db.cubrid.CubridPreparedStatementModifier;
import com.profiler.modifier.db.cubrid.CubridResultSetModifier;
import com.profiler.modifier.db.cubrid.CubridStatementModifier;
import com.profiler.modifier.db.cubrid.CubridUStatementModifier;
import com.profiler.modifier.db.dbcp.DBCPBasicDataSourceModifier;
import com.profiler.modifier.db.dbcp.DBCPPoolModifier;
import com.profiler.modifier.db.mssql.MSSQLConnectionModifier;
import com.profiler.modifier.db.mssql.MSSQLPreparedStatementModifier;
import com.profiler.modifier.db.mssql.MSSQLResultSetModifier;
import com.profiler.modifier.db.mssql.MSSQLStatementModifier;
import com.profiler.modifier.db.mysql.*;
import com.profiler.modifier.db.oracle.OraclePreparedStatementModifier;
import com.profiler.modifier.db.oracle.OracleResultSetModifier;
import com.profiler.modifier.db.oracle.OracleStatementModifier;
import com.profiler.modifier.tomcat.CatalinaModifier;
import com.profiler.modifier.tomcat.StandardHostValveInvokeInterceptor;
import com.profiler.modifier.tomcat.TomcatConnectorModifier;
import com.profiler.modifier.tomcat.TomcatStandardServiceModifier;

import java.util.HashMap;
import java.util.Map;

public class DefaultModifierRegistry implements ModifierRegistry {
    // TODO 혹시 동시성을 고려 해야 되는지 검토.
    // 왠간해서는 동시성 상황이 안나올것으로 보임.
    private Map<String, Modifier> registry = new HashMap<String, Modifier>();

    private final ByteCodeInstrumentor byteCodeInstrumentor;
    private final ProfilerConfig profilerConfig;

    public DefaultModifierRegistry(ByteCodeInstrumentor byteCodeInstrumentor, ProfilerConfig profilerConfig) {
        this.byteCodeInstrumentor = byteCodeInstrumentor;
        this.profilerConfig = profilerConfig;
    }

    @Override
    public Modifier findModifier(String className) {
        return registry.get(className);
    }

    private void addModifier(Modifier modifier) {
        Modifier old = registry.put(modifier.getTargetClass(), modifier);
        if (old != null) {
            throw new IllegalStateException("Modifier already exist new:" + modifier.getClass() + " old:" + old.getTargetClass());
        }
    }

    public void addConnectorModifier() {
        HTTPClientModifier httpClientModifier = new HTTPClientModifier(byteCodeInstrumentor);
        addModifier(httpClientModifier);

        ArcusClientModifier arcusClientModifier = new ArcusClientModifier(byteCodeInstrumentor);
        addModifier(arcusClientModifier);
    }

    public void addTomcatModifier() {
        StandardHostValveInvokeInterceptor standardHostValveInvokeInterceptor = new StandardHostValveInvokeInterceptor(byteCodeInstrumentor);
        addModifier(standardHostValveInvokeInterceptor);

        Modifier tomcatStandardServiceModifier = new TomcatStandardServiceModifier(byteCodeInstrumentor);
        addModifier(tomcatStandardServiceModifier);

        Modifier tomcatConnectorModifier = new TomcatConnectorModifier(byteCodeInstrumentor);
        addModifier(tomcatConnectorModifier);

        Modifier tomcatCatalinaModifier = new CatalinaModifier(byteCodeInstrumentor);
        addModifier(tomcatCatalinaModifier);
    }

    public void addJdbcModifier() {
        // TODO 드라이버 존재 체크 로직을 앞단으로 이동 시킬수 없는지 검토
        if (!profilerConfig.isJdbcProfile()) {
            return;
        }

        if (profilerConfig.isJdbcProfileMySql()) {
            addMySqlDriver();
        }

        if (profilerConfig.isJdbcProfileMsSql()) {
            addMsSqlDriver();
        }

        if (profilerConfig.isJdbcProfileOracle()) {
            addOracleDriver();
        }
        if (profilerConfig.isJdbcProfileCubrid()) {
            addCubridDriver();
        }

        if (profilerConfig.isJdbcProfileDbcp()) {
            addDbcpDriver();
        }
    }

    private void addMySqlDriver() {
        // TODO MySqlDriver는 버전별로 Connection이 interface인지 class인지가 다름. 문제 없는지
        // 확인필요.

        Modifier mysqlNonRegisteringDriverModifier = new MySQLNonRegisteringDriverModifier(byteCodeInstrumentor);
        addModifier(mysqlNonRegisteringDriverModifier);

        Modifier mysqlConnectionImplModifier = new MySQLConnectionImplModifier(byteCodeInstrumentor);
        addModifier(mysqlConnectionImplModifier);

        Modifier mysqlStatementModifier = new MySQLStatementModifier(byteCodeInstrumentor);
        addModifier(mysqlStatementModifier);

        Modifier mysqlPreparedStatementModifier = new MySQLPreparedStatementModifier(byteCodeInstrumentor);
        addModifier(mysqlPreparedStatementModifier);

        MySQLPreparedStatementJDBC4Modifier myqlPreparedStatementJDBC4Modifier = new MySQLPreparedStatementJDBC4Modifier(byteCodeInstrumentor);
        addModifier(myqlPreparedStatementJDBC4Modifier);

        Modifier mysqlResultSetModifier = new MySQLResultSetModifier(byteCodeInstrumentor);
        addModifier(mysqlResultSetModifier);
    }

    private void addMsSqlDriver() {

        Modifier mssqlConnectionModifier = new MSSQLConnectionModifier(byteCodeInstrumentor);
        addModifier(mssqlConnectionModifier);

        Modifier mssqlStatementModifier = new MSSQLStatementModifier(byteCodeInstrumentor);
        addModifier(mssqlStatementModifier);

        Modifier mssqlPreparedStatementModifier = new MSSQLPreparedStatementModifier(byteCodeInstrumentor);
        addModifier(mssqlPreparedStatementModifier);

        Modifier mssqlResultSetModifier = new MSSQLResultSetModifier(byteCodeInstrumentor);
        addModifier(mssqlResultSetModifier);

    }

    private void addOracleDriver() {

        // TODO oracle의 경우 connection에 대한 impl이 없음. 확인필요.
        Modifier oraclePreparedStatementModifier = new OraclePreparedStatementModifier(byteCodeInstrumentor);
        addModifier(oraclePreparedStatementModifier);

        Modifier oracleStatement = new OracleStatementModifier(byteCodeInstrumentor);
        addModifier(oracleStatement);

        Modifier oracleResultSetModifier = new OracleResultSetModifier(byteCodeInstrumentor);
        addModifier(oracleResultSetModifier);
    }

    private void addCubridDriver() {

        // TODO cubrid의 경우도 connection에 대한 impl이 없음. 확인필요.
        Modifier cubridStatementModifier = new CubridStatementModifier(byteCodeInstrumentor);
        addModifier(cubridStatementModifier);

        Modifier cubridPreparedStatementModifier = new CubridPreparedStatementModifier(byteCodeInstrumentor);
        addModifier(cubridPreparedStatementModifier);

        Modifier cubridResultSetModifier = new CubridResultSetModifier(byteCodeInstrumentor);
        addModifier(cubridResultSetModifier);

        Modifier cubridUStatementModifier = new CubridUStatementModifier(byteCodeInstrumentor);
        addModifier(cubridUStatementModifier);
    }

    private void addDbcpDriver() {

        // TODO cubrid의 경우도 connection에 대한 impl이 없음. 확인필요.
        Modifier dbcpBasicDataSourceModifier = new DBCPBasicDataSourceModifier(byteCodeInstrumentor);
        addModifier(dbcpBasicDataSourceModifier);

        Modifier dbcpPoolModifier = new DBCPPoolModifier(byteCodeInstrumentor);
        addModifier(dbcpPoolModifier);
    }
}