contract;

// This contract receives messages sent from an instance of FuelERC20Gateway
// deployed at the L1. The messages are relayed by the message predicate.
// Upon processing a message, funds are minted to the receiver.

// Hexens FUEL1-21: This contract might be able to receive BASE_ASSET and
// without a rescue function, the BASE_ASSET funds might get stuck.
// Special care must be put when sending messages from the gateway to this
// contract so that they do not cointain BASE_ASSET funds

mod data_structures;
mod errors;
mod events;
mod utils;

use contract_message_receiver::MessageReceiver;
use errors::BridgeFungibleTokenError;
use data_structures::{
    constants::{
        CONTRACT_DEPOSIT,
        CONTRACT_DEPOSIT_WITH_DATA,
        DEPOSIT,
    },
    deposit_message::{
        DepositMessage,
        DepositType,
    },
    message_data::MessageData,
    metadata_message::MetadataMessage,
};
use events::{ClaimRefundEvent, DepositEvent, RefundRegisteredEvent, WithdrawalEvent};
use interface::bridge::Bridge;
use sway_libs::reentrancy::reentrancy_guard;
use std::{
    asset::{
        burn,
        mint,
        transfer,
    },
    call_frames::msg_asset_id,
    context::msg_amount,
    flags::{
        disable_panic_on_overflow,
        enable_panic_on_overflow,
    },
    hash::Hash,
    hash::sha256,
    inputs::input_message_sender,
    message::send_message,
    primitive_conversions::u64::*,
    storage::{
        storage_string::*,
    },
    string::String,
};
use utils::encode_data;
use standards::{
    src20::{
        SetDecimalsEvent,
        SetNameEvent,
        SetSymbolEvent,
        SRC20,
        TotalSupplyEvent,
    },
    src7::{
        Metadata,
        SetMetadataEvent,
        SRC7,
    },
};

const FUEL_ASSET_DECIMALS: u8 = 9u8;

configurable {
    BRIDGED_TOKEN_GATEWAY: b256 = 0x00000000000000000000000096c53cd98B7297564716a8f2E1de2C83928Af2fe,
}

storage {
    bridge {
        asset_to_sub_id: StorageMap<AssetId, SubId> = StorageMap {},
        asset_to_token_id: StorageMap<AssetId, b256> = StorageMap {},
        refund_amounts: StorageMap<b256, StorageMap<b256, u256>> = StorageMap {},
        tokens_minted: StorageMap<AssetId, u64> = StorageMap {},
        l1_addresses: StorageMap<AssetId, b256> = StorageMap {},
        l1_symbols: StorageMap<b256, StorageString> = StorageMap {},
        l1_names: StorageMap<b256, StorageString> = StorageMap {},
        decimals: StorageMap<b256, u8> = StorageMap {},
        total_assets: u64 = 0,
    },
}

// Implement the process_message function required to be a message receiver
impl MessageReceiver for Contract {
    #[payable]
    #[storage(read, write)]
    fn process_message(msg_idx: u64) {
        // Protect against reentrancy attacks that could allow replaying messages
        reentrancy_guard();

        let input_sender: b256 = input_message_sender(msg_idx).unwrap().into();
        require(
            input_sender == BRIDGED_TOKEN_GATEWAY,
            BridgeFungibleTokenError::UnauthorizedSender,
        );

        match MessageData::parse(msg_idx) {
            MessageData::Deposit(deposit) => _process_deposit(deposit, msg_idx),
            MessageData::Metadata(metadata) => _process_metadata(metadata),
        };
    }
}

impl Bridge for Contract {
    #[storage(read, write)]
    fn claim_refund(from: b256, token_address: b256, token_id: b256) {
        let asset = _generate_sub_id_from_metadata(token_address, token_id);
        let amount = storage::bridge.refund_amounts.get(from).get(asset).try_read().unwrap_or(u256::zero());
        require(
            amount != u256::zero(),
            BridgeFungibleTokenError::NoRefundAvailable,
        );

        // reset the refund amount to 0
        storage::bridge
            .refund_amounts
            .get(from)
            .insert(asset, u256::zero());

        // send a message to unlock this amount on the base layer gateway contract
        send_message(
            BRIDGED_TOKEN_GATEWAY,
            encode_data(from, amount.as_b256(), token_address, token_id),
            0,
        );

        log(ClaimRefundEvent {
            amount,
            from,
            token_address,
            token_id,
        });
    }

    #[payable]
    #[storage(read, write)]
    fn withdraw(to: b256) {
        let amount: u64 = msg_amount();
        require(amount != 0, BridgeFungibleTokenError::NoCoinsSent);
        require(
            to != b256::zero(),
            BridgeFungibleTokenError::WithdrawalToZeroAddress,
        );

        let asset_id = msg_asset_id();
        let sub_id = _asset_to_sub_id(asset_id);
        let token_id = _asset_to_token_id(asset_id);
        let l1_address = _asset_to_l1_address(asset_id);

        // Hexens Fuel1-4: Might benefit from a custom error message
        let new_total_supply = storage::bridge.tokens_minted.get(asset_id).read() - amount;
        storage::bridge
            .tokens_minted
            .insert(asset_id, new_total_supply);
        burn(sub_id, amount);

        let sender = msg_sender().unwrap();
        log(TotalSupplyEvent {
            asset: asset_id,
            supply: new_total_supply,
            sender,
        });

        // send a message to unlock this amount on the base layer gateway contract
        send_message(
            BRIDGED_TOKEN_GATEWAY,
            encode_data(to, amount.as_u256().as_b256(), l1_address, token_id),
            0,
        );
        log(WithdrawalEvent {
            to: to,
            from: sender,
            amount: amount,
        });
    }

    fn bridged_token_gateway() -> b256 {
        BRIDGED_TOKEN_GATEWAY
    }

    #[storage(read)]
    fn asset_to_sub_id(asset_id: AssetId) -> SubId {
        _asset_to_sub_id(asset_id)
    }

    #[storage(read)]
    fn asset_to_l1_address(asset_id: AssetId) -> b256 {
        _asset_to_l1_address(asset_id)
    }
}

impl SRC20 for Contract {
    #[storage(read)]
    fn total_assets() -> u64 {
        storage::bridge.total_assets.try_read().unwrap_or(0)
    }

    #[storage(read)]
    fn total_supply(asset: AssetId) -> Option<u64> {
        storage::bridge.tokens_minted.get(asset).try_read()
    }

    #[storage(read)]
    fn name(asset: AssetId) -> Option<String> {
        let l1_address = _asset_to_l1_address(asset);
        storage::bridge.l1_names.get(l1_address).read_slice()
    }

    #[storage(read)]
    fn symbol(asset: AssetId) -> Option<String> {
        let l1_address = _asset_to_l1_address(asset);
        storage::bridge.l1_symbols.get(l1_address).read_slice()
    }

    #[storage(read)]
    fn decimals(asset: AssetId) -> Option<u8> {
        match storage::bridge.l1_addresses.get(asset).try_read() {
            Some(l1_address) => storage::bridge.decimals.get(l1_address).try_read(),
            None => None,
        }
    }
}

impl SRC7 for Contract {
    #[storage(read)]
    fn metadata(asset: AssetId, key: String) -> Option<Metadata> {
        let sub_id = storage::bridge.asset_to_sub_id.get(asset).try_read();

        if (sub_id.is_none()) {
            return None;
        };

        let result = if key == String::from_ascii_str("bridged:chain") {
            Some(Metadata::String(String::from_ascii_str("1")))
        } else if key == String::from_ascii_str("bridged:address") {
            Some(Metadata::B256(storage::bridge.l1_addresses.get(asset).read()))
        } else if key == String::from_ascii_str("bridged:decimals") {
            let l1_address = storage::bridge.l1_addresses.get(asset).read();
            Some(Metadata::Int(storage::bridge.decimals.get(l1_address).read().into()))
        } else if key == String::from_ascii_str("bridged:id") {
            Some(Metadata::B256(storage::bridge.asset_to_token_id.get(asset).read()))
        } else {
            None
        };

        result
    }
}

// Storage-dependant private functions
#[storage(write)]
fn register_refund(
    from: b256,
    token_address: b256,
    token_id: b256,
    amount: b256,
) {
    let asset = _generate_sub_id_from_metadata(token_address, token_id);

    let previous_amount = storage::bridge.refund_amounts.get(from).get(asset).try_read().unwrap_or(u256::zero());
    let new_amount = amount.as_u256() + previous_amount;

    storage::bridge
        .refund_amounts
        .get(from)
        .insert(asset, new_amount);
    log(RefundRegisteredEvent {
        from,
        token_address,
        token_id,
        amount,
    });
}

#[storage(read)]
fn _asset_to_sub_id(asset_id: AssetId) -> SubId {
    let sub_id = storage::bridge.asset_to_sub_id.get(asset_id).try_read();
    require(sub_id.is_some(), BridgeFungibleTokenError::AssetNotFound);
    sub_id.unwrap()
}

#[storage(read)]
fn _asset_to_token_id(asset_id: AssetId) -> b256 {
    let token_id = storage::bridge.asset_to_token_id.get(asset_id).try_read();
    require(token_id.is_some(), BridgeFungibleTokenError::AssetNotFound);
    token_id.unwrap()
}

#[storage(read)]
fn _asset_to_l1_address(asset_id: AssetId) -> b256 {
    let l1_address = storage::bridge.l1_addresses.get(asset_id).try_read();
    require(
        l1_address
            .is_some(),
        BridgeFungibleTokenError::AssetNotFound,
    );
    l1_address.unwrap()
}

#[storage(read, write)]
fn _process_deposit(message_data: DepositMessage, msg_idx: u64) {
    require(
        message_data
            .amount != b256::zero(),
        BridgeFungibleTokenError::NoCoinsSent,
    );

    let amount: u64 = match <u64 as TryFrom<u256>>::try_from(message_data.amount.as_u256()) {
        Some(value) => value,
        None => {
            register_refund(
                message_data
                    .from,
                message_data
                    .token_address,
                message_data
                    .token_id,
                message_data
                    .amount,
            );
            return;
        }
    };
    let sub_id = _generate_sub_id_from_metadata(message_data.token_address, message_data.token_id);
    let asset_id = AssetId::new(ContractId::this(), sub_id);

    let _ = disable_panic_on_overflow();

    let current_total_supply = storage::bridge.tokens_minted.get(asset_id).try_read().unwrap_or(0);
    let new_total_supply = current_total_supply + amount;

    if new_total_supply < current_total_supply {
        register_refund(
            message_data
                .from,
            message_data
                .token_address,
            message_data
                .token_id,
            message_data
                .amount,
        );
        return;
    }

    let _ = enable_panic_on_overflow();

    storage::bridge
        .tokens_minted
        .insert(asset_id, new_total_supply);

    // Store asset metadata if it is the first time that funds are bridged
    let sender = Identity::Address(Address::from(BRIDGED_TOKEN_GATEWAY));
    if storage::bridge.asset_to_sub_id.get(asset_id).try_read().is_none()
    {
        log(SetMetadataEvent {
            asset: asset_id,
            metadata: Some(Metadata::String(String::from_ascii_str("1"))),
            key: String::from_ascii_str("bridged:chain"),
            sender,
        });

        storage::bridge.asset_to_sub_id.insert(asset_id, sub_id);
        storage::bridge
            .asset_to_token_id
            .insert(asset_id, message_data.token_id);
        log(SetMetadataEvent {
            asset: asset_id,
            metadata: Some(Metadata::B256(message_data.token_id)),
            key: String::from_ascii_str("bridged:id"),
            sender,
        });

        storage::bridge
            .total_assets
            .write(storage::bridge.total_assets.try_read().unwrap_or(0) + 1);
        storage::bridge
            .l1_addresses
            .insert(asset_id, message_data.token_address);
        log(SetMetadataEvent {
            asset: asset_id,
            metadata: Some(Metadata::B256(message_data.token_address)),
            key: String::from_ascii_str("bridged:address"),
            sender,
        });

        if message_data.decimals < FUEL_ASSET_DECIMALS {
            storage::bridge
                .decimals
                .insert(message_data.token_address, message_data.decimals);
            log(SetDecimalsEvent {
                asset: asset_id,
                decimals: message_data.decimals,
                sender,
            });
            log(SetMetadataEvent {
                asset: asset_id,
                metadata: Some(Metadata::Int(message_data.decimals.into())),
                key: String::from_ascii_str("bridged:decimals"),
                sender,
            });
        } else {
            storage::bridge
                .decimals
                .insert(message_data.token_address, FUEL_ASSET_DECIMALS);
            log(SetDecimalsEvent {
                asset: asset_id,
                decimals: FUEL_ASSET_DECIMALS,
                sender,
            });
            log(SetMetadataEvent {
                asset: asset_id,
                metadata: Some(Metadata::Int(FUEL_ASSET_DECIMALS.into())),
                key: String::from_ascii_str("bridged:decimals"),
                sender,
            });
        }
    };
    // mint tokens & update storage
    mint(sub_id, amount);

    log(TotalSupplyEvent {
        asset: asset_id,
        supply: new_total_supply,
        sender,
    });

    match message_data.deposit_type {
        DepositType::Address | DepositType::Contract => {
            transfer(message_data.to, asset_id, amount)
        },
        DepositType::ContractWithData => {
            let contract_id: b256 = message_data.to.as_contract_id().unwrap().into();
            let dest_contract = abi(MessageReceiver, contract_id);
            // TODO: Hexens Fuel1-2, if this call fails, funds may get stuck
            dest_contract
                .process_message {
                    coins: amount,
                    asset_id: asset_id.into(),
                }(msg_idx);
        }
    };

    log(DepositEvent {
        to: message_data.to,
        from: message_data.from,
        amount: amount,
    });
}

#[storage(read, write)]
fn _process_metadata(metadata: MetadataMessage) {
    let sub_id = _generate_sub_id_from_metadata(metadata.token_address, metadata.token_id);
    let asset_id = AssetId::new(ContractId::this(), sub_id);

    // Important to note: in order to register metadata for an asset, 
    // it must have been deposited first
    let l1_address = _asset_to_l1_address(asset_id);

    storage::bridge
        .l1_names
        .get(l1_address)
        .write_slice(metadata.name);
    storage::bridge
        .l1_symbols
        .get(l1_address)
        .write_slice(metadata.symbol);

    let sender = Identity::Address(Address::from(BRIDGED_TOKEN_GATEWAY));

    log(SetNameEvent {
        asset: asset_id,
        name: Some(metadata.name),
        sender,
    });

    log(SetSymbolEvent {
        asset: asset_id,
        symbol: Some(metadata.symbol),
        sender,
    });
}

fn _generate_sub_id_from_metadata(token_address: b256, token_id: b256) -> b256 {
    // SRC8: sub_id = sha256(chain_id ++ token_address + token_id)
    sha256((String::from_ascii_str("1"), token_address, token_id))
}