@RequiredArgsConstructor
public class AuditInterceptor extends JsqlParserSupport implements InnerInterceptor {

    @Override
    public void beforePrepare(StatementHandler sh, Connection connection, Integer transactionTimeout) {
        MPStatementHandler mpStatementHandler = PluginUtils.mpStatementHandler(sh);
        MappedStatement mappedStatement = mpStatementHandler.mappedStatement();
        SqlCommandType sqlCommandType = mappedStatement.getSqlCommandType();
        if (sqlCommandType == SqlCommandType.UPDATE || sqlCommandType == SqlCommandType.INSERT) {
            MPBoundSql mpBoundSql = mpStatementHandler.mPBoundSql();
            Class<?> entityType = mappedStatement.getParameterMap().getType();
            if (!mpBoundSql.sql().contains("REFRESH MATERIALIZED VIEW")) {
                mpBoundSql.sql(parserMulti(mpBoundSql.sql(), entityType));
            }
        }
    }

    @Override
    protected void processInsert(Insert insert, int index, String sql, Object obj) {
        Set<String> columnNames = getColumnNames(obj);
        if (CollectionUtils.isEmpty(columnNames)) {
            return;
        }
        // Set update time & tenant ID
        var now = new LongValue(System.currentTimeMillis());
        setInsertValue(insert, columnNames, AuditEntity.CREATED_COLUMN_NAME, now);
        setInsertValue(insert, columnNames, AuditEntity.MODIFIED_COLUMN_NAME, now);
        setInsertValue(insert, columnNames, AuditEntity.TENANT_ID_COLUMN_NAME, new StringValue(ContextUtils.getOrThrowTenantId()));
        setInsertValue(insert, columnNames, AuditEntity.ENVIRONMENT_COLUMN_NAME, new StringValue(ContextUtils.getEnvName()));
    }

    private Set<String> getColumnNames(Object obj) {
        TableInfo tableInfo = TableInfoHelper.getTableInfo((Class<?>) obj);
        if (tableInfo == null) {
            getLog().warn("Cannot find table info for class '{}'", obj);
            return Collections.emptySet();
        }
        return tableInfo.getFieldList().stream()
            .map(TableFieldInfo::getColumn)
            .collect(Collectors.toSet());
    }

    private LogTagHolder getLog() {
        return MosLog.wrap(log).tag1("AUDIT_INTERCEPTOR");
    }

    private void setInsertValue(Insert insert, Set<String> definedColumnNames, String columnName, Expression columnValue) {
        List<Column> columns = insert.getColumns();
        if (!isAccept(columnName, definedColumnNames, columns)) {
            return;
        }
        columns.add(new Column(columnName));
        ItemsList itemsList = insert.getItemsList();
        if (itemsList instanceof MultiExpressionList) {
            ((MultiExpressionList) itemsList).getExpressionLists().forEach(el -> el.getExpressions().add(columnValue));
        } else if (itemsList instanceof ExpressionList) {
            ((ExpressionList) insert.getItemsList()).getExpressions().add(columnValue);
        }
        // TODO remove after check
        MosLog.wrap(log).onPre().tag1("TAN_TEST").tag2("INSERT")
            .info("[{}][{}] Set insert value", insert.getTable().getName(), columnName);
    }

    /**
     * Check if column exist in table definition & not exist in INSERT/UPDATE command.
     */
    private boolean isAccept(String columnName, Set<String> definedColumnNames, List<Column> insertColumns) {
        if (!definedColumnNames.contains(columnName)) {
            return false;
        }
        Column insertColumn = insertColumns.stream()
            .filter(curInsertColumn -> StringUtils.equals(columnName, curInsertColumn.getColumnName())
                || StringUtils.equals("`" + columnName + "`", curInsertColumn.getColumnName()))
            .findFirst()
            .orElse(null);
        return (insertColumn == null);
    }

    @Override
    protected void processUpdate(Update update, int index, String sql, Object obj) {
        Set<String> columnNames = getColumnNames(obj);
        if (CollectionUtils.isEmpty(columnNames)) {
            return;
        }
        setUpdateValue(update, columnNames, AuditEntity.MODIFIED_COLUMN_NAME, new LongValue(System.currentTimeMillis()));
    }

    private void setUpdateValue(Update update, Set<String> definedColumnNames, String columnName, Expression columnValue) {
        List<Column> columns = update.getColumns();
        if (!isAccept(columnName, definedColumnNames, columns)) {
            return;
        }
        columns.add(new Column(columnName));
        List<Expression> expressions = update.getExpressions();
        expressions.add(columnValue);
        // TODO remove after check
        MosLog.wrap(log).onPre().tag1("TAN_TEST").tag2("UPDATE")
            .info("[{}][{}] Set update value", update.getTable().getName(), columnName);
    }

}