use binaryninja::{
    binaryview::{BinaryView, BinaryViewExt},
    command::{register, Command},
    logger,
    string::BnString,
    symbol::Symbol,
};
use log::{info, LevelFilter};


struct TestCommand;

impl Command for TestCommand {
    fn action(&self, view: &BinaryView) {
        if let Some(func) = view.functions().iter().next() {
            info!("setting fn {:#x}", func.symbol().address());
            let bn_full_name = BnString::new("bn_full_name");
            let bn_short_name = BnString::new("bn_short_name");
            let symbol = Symbol::builder(
                func.symbol().sym_type(),
                func.symbol().raw_name(),
                func.symbol().address(),
            )
                .full_name(bn_full_name)
                .short_name(bn_short_name)
                .create();

            view.define_user_symbol(&symbol);
        }
    }

    fn valid(&self, _view: &BinaryView) -> bool {
        true
    }
}

#[no_mangle]
#[allow(non_snake_case)]
pub extern "C" fn CorePluginInit() -> bool {
    logger::init(LevelFilter::Info).expect("failed to initialize logging");

    register("Diff\\Test", "Test", TestCommand);

    true
}