use defuse_near_utils::{CURRENT_ACCOUNT_ID, method_name};
use near_plugins::{AccessControllable, Pausable, access_control_any, pause};
use near_sdk::{Allowance, Promise, PublicKey, assert_one_yocto, near, require};

use crate::{
    contract::{Contract, ContractExt, Role},
    intents::{Intents, RelayerKeys},
};

const EXECUTE_INTENTS_FUNC: &str = method_name!(Contract::execute_intents);

#[near]
impl RelayerKeys for Contract {
    #[pause(name = "intents")]
    #[payable]
    #[access_control_any(roles(Role::DAO, Role::RelayerKeysManager))]
    fn add_relayer_key(&mut self, public_key: PublicKey) -> Promise {
        assert_one_yocto();
        Self::ext(CURRENT_ACCOUNT_ID.clone())
            .do_add_relayer_key(public_key.clone())
            .add_access_key_allowance(
                public_key,
                Allowance::Unlimited,
                CURRENT_ACCOUNT_ID.clone(),
                EXECUTE_INTENTS_FUNC,
            )
    }

    #[private]
    fn do_add_relayer_key(&mut self, public_key: PublicKey) {
        require!(self.relayer_keys.insert(public_key), "key already exists",);
    }

    #[pause(name = "intents")]
    #[access_control_any(roles(Role::DAO, Role::RelayerKeysManager))]
    #[payable]
    fn delete_relayer_key(&mut self, public_key: PublicKey) -> Promise {
        assert_one_yocto();
        require!(self.relayer_keys.remove(&public_key), "key not found");

        Promise::new(CURRENT_ACCOUNT_ID.clone()).delete_key(public_key)
    }
}