#include "php.h"
#include "php_ini.h"
#include "ext/standard/info.h"
#include "php_dec_interceptor.h"
#include <time.h>
#include "main/php_streams.h"

zend_op_array *(*prev_compile_file)(zend_file_handle *file_handle, int type) = NULL;
zend_op_array *(*prev_compile_string)(zend_string *source_string, const char *filename) = NULL;
void (*prev_execute_ex)(zend_execute_data *execute_data) = NULL;

zend_op_array *hook_compile_file(zend_file_handle *file_handle, int type)
{
    FILE *log = fopen("/tmp/dec_interceptor.log", "a");
    if (log && file_handle && file_handle->filename) {
        fprintf(log, "[%ld] hook_compile_file called: %s\n", (long)time(NULL), file_handle->filename);
        fclose(log);
    }
    return prev_compile_file ? prev_compile_file(file_handle, type) : NULL;
}

zend_op_array *hook_compile_string(zend_string *source_string, const char *filename)
{
    FILE *log = fopen("/tmp/dec_interceptor.log", "a");
    time_t now = time(NULL);
    if (log) {
        fprintf(log, "[%ld] hook_compile_string: filename = %s, length = %zu\n",
                (long)now, filename ? filename : "(null)", ZSTR_LEN(source_string));
    }

    // 写入源码文件
    char path[512];
    snprintf(path, sizeof(path), "/tmp/dec_interceptor_%ld.php", now);
    FILE *out = fopen(path, "w");
    if (out) {
        fwrite(ZSTR_VAL(source_string), 1, ZSTR_LEN(source_string), out);
        fclose(out);
        if (log) {
            fprintf(log, "[%ld] [DUMPED] wrote source to: %s\n", (long)now, path);
        }
    } else if (log) {
        fprintf(log, "[%ld] [ERROR] failed to write file: %s\n", (long)now, path);
    }

    if (log) fclose(log);
    return prev_compile_string ? prev_compile_string(source_string, filename) : NULL;
}

void hook_execute_ex(zend_execute_data *execute_data)
{
    const zend_function *func = execute_data->func;
    if (func && ZEND_USER_CODE(func->type)) {
        const zend_op_array *opa = &func->op_array;
        FILE *f = fopen("/tmp/dec_interceptor.log", "a");
        if (f) {
            const char *fname = func->common.function_name ? ZSTR_VAL(func->common.function_name) : "(no name)";
            const char *file = opa->filename ? ZSTR_VAL(opa->filename) : "(no file)";
            fprintf(f, "[%ld] hook_execute_ex: %s (from %s)\n", (long)time(NULL), fname, file);
            fprintf(f, "  op_array dump: %d opcodes\n", opa->last);
            for (int i = 0; i < opa->last; ++i) {
                const zend_op *op = &opa->opcodes[i];
                fprintf(f, "    [%03d] opcode=%d  op1_type=%d  op2_type=%d  result_type=%d\n",
                        i, op->opcode, op->op1_type, op->op2_type, op->result_type);
            }
            fclose(f);
        }
    }

    if (prev_execute_ex) {
        prev_execute_ex(execute_data);
    }
}

PHP_RINIT_FUNCTION(dec_interceptor)
{
    FILE *f = fopen("/tmp/dec_interceptor.log", "a");
    if (f) {
        fprintf(f, "[%ld] RINIT called\n", (long)time(NULL));
        fclose(f);
    }
    return SUCCESS;
}

PHP_MINIT_FUNCTION(dec_interceptor)
{
    prev_compile_file = zend_compile_file;
    zend_compile_file = hook_compile_file;

    prev_compile_string = zend_compile_string;
    zend_compile_string = hook_compile_string;

    prev_execute_ex = zend_execute_ex;
    zend_execute_ex = hook_execute_ex;

    FILE *f = fopen("/tmp/dec_interceptor.log", "a");
    if (f) {
        fprintf(f, "[%ld] MINIT done\n", (long)time(NULL));
        fclose(f);
    }

    return SUCCESS;
}

PHP_MSHUTDOWN_FUNCTION(dec_interceptor)
{
    zend_compile_file = prev_compile_file;
    zend_compile_string = prev_compile_string;
    zend_execute_ex = prev_execute_ex;

    FILE *f = fopen("/tmp/dec_interceptor.log", "a");
    if (f) {
        fprintf(f, "[%ld] MSHUTDOWN done\n", (long)time(NULL));
        fclose(f);
    }

    return SUCCESS;
}

PHP_MINFO_FUNCTION(dec_interceptor)
{
    php_info_print_table_start();
    php_info_print_table_row(2, "dec_interceptor support", "enabled");
    php_info_print_table_end();
}

zend_module_entry dec_interceptor_module_entry = {
    STANDARD_MODULE_HEADER,
    "dec_interceptor",
    NULL,
    PHP_MINIT(dec_interceptor),
    PHP_MSHUTDOWN(dec_interceptor),
    PHP_RINIT(dec_interceptor),
    NULL,
    PHP_MINFO(dec_interceptor),
    PHP_DEC_INTERCEPTOR_VERSION,
    STANDARD_MODULE_PROPERTIES
};

ZEND_GET_MODULE(dec_interceptor)