mod common;

use common::{ensure_plugin_built, fixture_path, CompileBuilder, ObfuscationConfig};

#[test]
fn test_your_feature() {
    // 确保插件已构建
    ensure_plugin_built();

    // 配置混淆选项
    let config = ObfuscationConfig {
        your_option: Some(true),
        ..ObfuscationConfig::disabled()  // 禁用其他所有选项
    };

    // 编译测试文件
    let result = CompileBuilder::new(
        fixture_path("your_module", "your_test.c"),
        "output_binary_name",
    )
    .config(config)
    .optimization("O2")  // 可选
    .compile();

    // 验证编译成功
    result.assert_success();

    // 运行并验证输出
    let run = result.run();
    run.assert_success();

    let lines = run.stdout_lines();
    assert_eq!(lines[0], "Expected output");
}

ObfuscationConfig {
    // 字符串加密
    string_encryption: Option<bool>,
    string_algorithm: Option<String>,      // "xor" | "simd_xor"
    string_decrypt_timing: Option<String>, // "lazy" | "global"
    string_stack_alloc: Option<bool>,
    string_inline_decrypt_fn: Option<bool>,
    string_max_encryption_count: Option<u32>,

    // 间接分支
    indirect_branch: Option<bool>,
    indirect_branch_flags: Option<String>, // "chained_dummy_block"

    // 间接调用
    indirect_call: Option<bool>,

    // 控制流
    flatten: Option<bool>,
    bogus_control_flow: Option<bool>,
    vm_flatten: Option<bool>,

    // 基本块
    shuffle_blocks: Option<bool>,
    shuffle_blocks_flags: Option<String>,  // "random" | "reverse" | "rotate"
    split_basic_block: Option<bool>,

    // 其他
    mba: Option<bool>,
    function_wrapper: Option<bool>,
}

CompileBuilder::new(source_path, output_name)
    .config(obfuscation_config)  // 设置混淆配置
    .optimization("O2")          // 设置优化级别
    .std("c++17")                // 设置 C/C++ 标准
    .arg("-Wall")                // 添加额外编译参数
    .without_plugin()            // 不使用混淆插件（用于基准对比）
    .compile()                   // 执行编译