#include <ctype.h>
#include "shared_context.h"
#include "common_utils.h"
#include "feature_signTx.h"
#include "uint256.h"
#include "eth_plugin_handler.h"
#include "network.h"
#include "common_ui.h"
#include "ui_callbacks.h"
#include "apdu_constants.h"
#include "crypto_helpers.h"
#include "format.h"
#include "manage_asset_info.h"

#define ERR_SILENT_MODE_CHECK_FAILED 0x6001

static bool g_use_standard_ui;

static uint32_t splitBinaryParameterPart(char *result, size_t result_size, uint8_t *parameter) {
    uint32_t i;
    for (i = 0; i < 8; i++) {
        if (parameter[i] != 0x00) {
            break;
        }
    }
    if (i == 8) {
        result[0] = '0';
        result[1] = '0';
        result[2] = '\0';
        return 2;
    } else {
        format_hex(parameter + i, 8 - i, result, result_size);
        return ((8 - i) * 2);
    }
}

customStatus_e customProcessor(txContext_t *context) {
    if (((context->txType == LEGACY && context->currentField == LEGACY_RLP_DATA) ||
         (context->txType == EIP2930 && context->currentField == EIP2930_RLP_DATA) ||
         (context->txType == EIP1559 && context->currentField == EIP1559_RLP_DATA)) &&
        (context->currentFieldLength != 0)) {
        context->content->dataPresent = true;
        // If handling a new contract rather than a function call, abort immediately
        if (tmpContent.txContent.destinationLength == 0) {
            return CUSTOM_NOT_HANDLED;
        }
        // If data field is less than 4 bytes long, do not try to use a plugin.
        if (context->currentFieldLength < 4) {
            return CUSTOM_NOT_HANDLED;
        }
        if (context->currentFieldPos == 0) {
            ethPluginInitContract_t pluginInit;
            // If handling the beginning of the data field, assume that the function selector is
            // present
            if (context->commandLength < 4) {
                PRINTF("Missing function selector\n");
                return CUSTOM_FAULT;
            }
            dataContext.tokenContext.pluginStatus = ETH_PLUGIN_RESULT_UNAVAILABLE;
            // If contract debugging mode is activated, do not go through the plugin activation
            // as they wouldn't be displayed if the plugin consumes all data but fallbacks
            if (!N_storage.contractDetails) {
                eth_plugin_prepare_init(&pluginInit,
                                        context->workBuffer,
                                        context->currentFieldLength);
                dataContext.tokenContext.pluginStatus =
                    eth_plugin_perform_init(tmpContent.txContent.destination, &pluginInit);
            }
            PRINTF("pluginstatus %d\n", dataContext.tokenContext.pluginStatus);
            eth_plugin_result_t status = dataContext.tokenContext.pluginStatus;
            if (status == ETH_PLUGIN_RESULT_ERROR) {
                PRINTF("Plugin error\n");
                return CUSTOM_FAULT;
            } else if (status >= ETH_PLUGIN_RESULT_SUCCESSFUL) {
                dataContext.tokenContext.fieldIndex = 0;
                dataContext.tokenContext.fieldOffset = 0;
                copyTxData(context, NULL, 4);
                if (context->currentFieldLength == 4) {
                    return CUSTOM_NOT_HANDLED;
                }
            }
        }
        uint32_t blockSize;
        uint32_t copySize;
        uint32_t fieldPos = context->currentFieldPos;
        if (fieldPos == 0) {  // not reached if a plugin is available
            if (!N_storage.contractDetails) {
                return CUSTOM_NOT_HANDLED;
            }
            dataContext.tokenContext.fieldIndex = 0;
            dataContext.tokenContext.fieldOffset = 0;
            blockSize = 4;
        } else {
            if (!N_storage.contractDetails &&
                dataContext.tokenContext.pluginStatus <= ETH_PLUGIN_RESULT_UNSUCCESSFUL) {
                return CUSTOM_NOT_HANDLED;
            }
            blockSize = 32 - (dataContext.tokenContext.fieldOffset % 32);
        }

        // If the last parameter is of type `bytes` then we might have an
        // edge case where the data is not a multiple of 32. Set `blockSize` accordingly
        if ((context->currentFieldLength - fieldPos) < blockSize) {
            blockSize = context->currentFieldLength - fieldPos;
        }

        copySize = (context->commandLength < blockSize ? context->commandLength : blockSize);

        PRINTF("currentFieldPos %d copySize %d\n", context->currentFieldPos, copySize);

        copyTxData(context,
                   dataContext.tokenContext.data + dataContext.tokenContext.fieldOffset,
                   copySize);

        if (context->currentFieldPos == context->currentFieldLength) {
            PRINTF("\n\nIncrementing one\n");
            context->currentField++;
            context->processingField = false;
        }

        dataContext.tokenContext.fieldOffset += copySize;

        if (copySize == blockSize) {
            // Can process or display
            if (dataContext.tokenContext.pluginStatus >= ETH_PLUGIN_RESULT_SUCCESSFUL) {
                ethPluginProvideParameter_t pluginProvideParameter;
                eth_plugin_prepare_provide_parameter(&pluginProvideParameter,
                                                     dataContext.tokenContext.data,
                                                     dataContext.tokenContext.fieldIndex * 32 + 4);
                if (!eth_plugin_call(ETH_PLUGIN_PROVIDE_PARAMETER,
                                     (void *) &pluginProvideParameter)) {
                    PRINTF("Plugin parameter call failed\n");
                    return CUSTOM_FAULT;
                }
                dataContext.tokenContext.fieldIndex++;
                dataContext.tokenContext.fieldOffset = 0;
                memset(dataContext.tokenContext.data, 0, sizeof(dataContext.tokenContext.data));
                return CUSTOM_HANDLED;
            }

            if (fieldPos != 0) {
                dataContext.tokenContext.fieldIndex++;
            }
            dataContext.tokenContext.fieldOffset = 0;
            if (fieldPos == 0) {
                format_hex(dataContext.tokenContext.data,
                           4,
                           strings.tmp.tmp,
                           sizeof(strings.tmp.tmp));
                ui_confirm_selector();
            } else {
                uint32_t offset = 0;
                uint32_t i;
                snprintf(strings.tmp.tmp2,
                         sizeof(strings.tmp.tmp2),
                         "Field %d",
                         dataContext.tokenContext.fieldIndex);
                for (i = 0; i < 4; i++) {
                    offset += splitBinaryParameterPart(strings.tmp.tmp + offset,
                                                       sizeof(strings.tmp.tmp) - offset,
                                                       dataContext.tokenContext.data + 8 * i);
                    if (i != 3) {
                        strings.tmp.tmp[offset++] = ':';
                    }
                }
                ui_confirm_parameter();
            }
        } else {
            return CUSTOM_HANDLED;
        }

        return CUSTOM_SUSPENDED;
    }
    return CUSTOM_NOT_HANDLED;
}

void report_finalize_error(void) {
    reset_app_context();
    io_seproxyhal_send_status(APDU_RESPONSE_INVALID_DATA);
    ui_idle();
}

static void address_to_string(uint8_t *in,
                              size_t in_len,
                              char *out,
                              size_t out_len,
                              uint64_t chainId) {
    if (in_len != 0) {
        if (!getEthDisplayableAddress(in, out, out_len, chainId)) {
            THROW(APDU_RESPONSE_ERROR_NO_INFO);
        }
    } else {
        strlcpy(out, "Contract", out_len);
    }
}

static void raw_fee_to_string(uint256_t *rawFee, char *displayBuffer, uint32_t displayBufferSize) {
    uint64_t chain_id = get_tx_chain_id();
    const char *feeTicker = get_displayable_ticker(&chain_id, chainConfig);
    uint8_t tickerOffset = 0;
    uint32_t i;

    tostring256(rawFee, 10, (char *) (G_io_apdu_buffer + 100), 100);
    i = 0;
    while (G_io_apdu_buffer[100 + i]) {
        i++;
    }
    adjustDecimals((char *) (G_io_apdu_buffer + 100),
                   i,
                   (char *) G_io_apdu_buffer,
                   100,
                   WEI_TO_ETHER);
    i = 0;
    tickerOffset = 0;
    memset(displayBuffer, 0, displayBufferSize);

    while (feeTicker[tickerOffset]) {
        if ((uint32_t) tickerOffset >= displayBufferSize) {
            break;
        }

        displayBuffer[tickerOffset] = feeTicker[tickerOffset];
        tickerOffset++;
    }
    if ((uint32_t) tickerOffset < displayBufferSize) displayBuffer[tickerOffset++] = ' ';
    while (G_io_apdu_buffer[i]) {
        if ((uint32_t) (tickerOffset) + i >= displayBufferSize) {
            break;
        }
        displayBuffer[tickerOffset + i] = G_io_apdu_buffer[i];
        i++;
    }

    if ((uint32_t) (tickerOffset) + i < displayBufferSize) {
        displayBuffer[tickerOffset + i] = '\0';
    }
}

// Compute the fees, transform it to a string, prepend a ticker to it and copy everything to
// `displayBuffer` output
static void max_transaction_fee_to_string(const txInt256_t *BEGasPrice,
                                          const txInt256_t *BEGasLimit,
                                          char *displayBuffer,
                                          uint32_t displayBufferSize) {
    // Use temporary variables to convert values to uint256_t
    uint256_t gasPrice = {0};
    uint256_t gasLimit = {0};
    // Use temporary variable to store the result of the operation in uint256_t
    uint256_t rawFee = {0};

    PRINTF("Gas price %.*H\n", BEGasPrice->length, BEGasPrice->value);
    PRINTF("Gas limit %.*H\n", BEGasLimit->length, BEGasLimit->value);
    convertUint256BE(BEGasPrice->value, BEGasPrice->length, &gasPrice);
    convertUint256BE(BEGasLimit->value, BEGasLimit->length, &gasLimit);
    mul256(&gasPrice, &gasLimit, &rawFee);
    raw_fee_to_string(&rawFee, displayBuffer, displayBufferSize);
}

static void nonce_to_string(const txInt256_t *nonce, char *out, size_t out_size) {
    uint256_t nonce_uint256;
    convertUint256BE(nonce->value, nonce->length, &nonce_uint256);
    tostring256(&nonce_uint256, 10, out, out_size);
}

static void get_network_as_string(char *out, size_t out_size) {
    uint64_t chain_id = get_tx_chain_id();
    const char *name = get_network_name_from_chain_id(&chain_id);

    if (name == NULL) {
        // No network name found so simply copy the chain ID as the network name.
        if (!u64_to_string(chain_id, out, out_size)) {
            THROW(0x6502);
        }
    } else {
        // Network name found, simply copy it.
        strlcpy(out, name, out_size);
    }
}

static void get_public_key(uint8_t *out, uint8_t outLength) {
    uint8_t raw_pubkey[65];

    if (outLength < ADDRESS_LENGTH) {
        return;
    }
    if (bip32_derive_get_pubkey_256(CX_CURVE_256K1,
                                    tmpCtx.transactionContext.bip32.path,
                                    tmpCtx.transactionContext.bip32.length,
                                    raw_pubkey,
                                    NULL,
                                    CX_SHA512) != CX_OK) {
        THROW(APDU_RESPONSE_UNKNOWN);
    }

    getEthAddressFromRawKey(raw_pubkey, out);
}

/* Local implementation of strncasecmp, workaround of the segfaulting base implem
 * Remove once strncasecmp is fixed
 */
static int strcasecmp_workaround(const char *str1, const char *str2) {
    unsigned char c1, c2;
    do {
        c1 = *str1++;
        c2 = *str2++;
        if (toupper(c1) != toupper(c2)) {
            return toupper(c1) - toupper(c2);
        }
    } while (c1 != '\0');
    return 0;
}

__attribute__((noinline)) static bool finalize_parsing_helper(void) {
    char displayBuffer[50];
    uint8_t decimals = WEI_TO_ETHER;
    uint64_t chain_id = get_tx_chain_id();
    const char *ticker = get_displayable_ticker(&chain_id, chainConfig);
    ethPluginFinalize_t pluginFinalize;
    cx_err_t error = CX_INTERNAL_ERROR;

    // Verify the chain
    if (chainConfig->chainId != ETHEREUM_MAINNET_CHAINID) {
        uint64_t id = get_tx_chain_id();

        if (chainConfig->chainId != id) {
            PRINTF("Invalid chainID %u expected %u\n", id, chainConfig->chainId);
            reset_app_context();
            report_finalize_error();
            return false;
        }
    }
    // Store the hash
    CX_CHECK(cx_hash_no_throw((cx_hash_t *) &global_sha3,
                              CX_LAST,
                              tmpCtx.transactionContext.hash,
                              0,
                              tmpCtx.transactionContext.hash,
                              32));

    uint8_t msg_sender[ADDRESS_LENGTH] = {0};
    get_public_key(msg_sender, sizeof(msg_sender));

    address_to_string(msg_sender,
                      ADDRESS_LENGTH,
                      strings.common.fromAddress,
                      sizeof(strings.common.fromAddress),
                      chainConfig->chainId);
    PRINTF("FROM address displayed: %s\n", strings.common.fromAddress);
    // Finalize the plugin handling
    if (dataContext.tokenContext.pluginStatus >= ETH_PLUGIN_RESULT_SUCCESSFUL) {
        eth_plugin_prepare_finalize(&pluginFinalize);

        pluginFinalize.address = msg_sender;

        if (!eth_plugin_call(ETH_PLUGIN_FINALIZE, (void *) &pluginFinalize)) {
            PRINTF("Plugin finalize call failed\n");
            report_finalize_error();
            return false;
        }
        // Lookup tokens if requested
        ethPluginProvideInfo_t pluginProvideInfo;
        eth_plugin_prepare_provide_info(&pluginProvideInfo);
        if ((pluginFinalize.tokenLookup1 != NULL) || (pluginFinalize.tokenLookup2 != NULL)) {
            if (pluginFinalize.tokenLookup1 != NULL) {
                PRINTF("Lookup1: %.*H\n", ADDRESS_LENGTH, pluginFinalize.tokenLookup1);
                pluginProvideInfo.item1 = get_asset_info_by_addr(pluginFinalize.tokenLookup1);
                if (pluginProvideInfo.item1 != NULL) {
                    PRINTF("Token1 ticker: %s\n", pluginProvideInfo.item1->token.ticker);
                }
            }
            if (pluginFinalize.tokenLookup2 != NULL) {
                PRINTF("Lookup2: %.*H\n", ADDRESS_LENGTH, pluginFinalize.tokenLookup2);
                pluginProvideInfo.item2 = get_asset_info_by_addr(pluginFinalize.tokenLookup2);
                if (pluginProvideInfo.item2 != NULL) {
                    PRINTF("Token2 ticker: %s\n", pluginProvideInfo.item2->token.ticker);
                }
            }
            if (eth_plugin_call(ETH_PLUGIN_PROVIDE_INFO, (void *) &pluginProvideInfo) <=
                ETH_PLUGIN_RESULT_UNSUCCESSFUL) {
                PRINTF("Plugin provide token call failed\n");
                report_finalize_error();
                return false;
            }
            pluginFinalize.result = pluginProvideInfo.result;
        }
        if (pluginFinalize.result != ETH_PLUGIN_RESULT_FALLBACK) {
            // Handle the right interface
            switch (pluginFinalize.uiType) {
                case ETH_UI_TYPE_GENERIC:
                    // Use the dedicated ETH plugin UI
                    g_use_standard_ui = false;
                    tmpContent.txContent.dataPresent = false;
                    // Add the number of screens + the number of additional screens to get the total
                    // number of screens needed.
                    dataContext.tokenContext.pluginUiMaxItems =
                        pluginFinalize.numScreens + pluginProvideInfo.additionalScreens;
                    break;
                case ETH_UI_TYPE_AMOUNT_ADDRESS:
                    // Use the standard ETH UI as this plugin uses the amount/address UI
                    g_use_standard_ui = true;
                    tmpContent.txContent.dataPresent = false;
                    if ((pluginFinalize.amount == NULL) || (pluginFinalize.address == NULL)) {
                        PRINTF("Incorrect amount/address set by plugin\n");
                        report_finalize_error();
                        return false;
                    }
                    memmove(tmpContent.txContent.value.value, pluginFinalize.amount, 32);
                    tmpContent.txContent.value.length = 32;
                    memmove(tmpContent.txContent.destination, pluginFinalize.address, 20);
                    tmpContent.txContent.destinationLength = 20;
                    if (pluginProvideInfo.item1 != NULL) {
                        decimals = pluginProvideInfo.item1->token.decimals;
                        ticker = pluginProvideInfo.item1->token.ticker;
                    }
                    break;
                default:
                    PRINTF("ui type %d not supported\n", pluginFinalize.uiType);
                    report_finalize_error();
                    return false;
            }
        }
    }

    if (G_called_from_swap) {
        if (G_swap_response_ready) {
            // Unreachable given current return to exchange mechanism. Safeguard against regression
            PRINTF("FATAL: safety against double sign triggered\n");
            os_sched_exit(-1);
        }
        G_swap_response_ready = true;
    }

    // User has just validated a swap but ETH received apdus about a non standard plugin / contract
    if (G_called_from_swap && !g_use_standard_ui) {
        PRINTF("ERR_SILENT_MODE_CHECK_FAILED, G_called_from_swap\n");
        THROW(ERR_SILENT_MODE_CHECK_FAILED);
    }

    // Prepare destination address and amount to display
    if (g_use_standard_ui) {
        // Format the address in a temporary buffer, if in swap case compare it with validated
        // address, else commit it
        address_to_string(tmpContent.txContent.destination,
                          tmpContent.txContent.destinationLength,
                          displayBuffer,
                          sizeof(displayBuffer),
                          chainConfig->chainId);
        if (G_called_from_swap) {
            // Ensure the values are the same that the ones that have been previously validated
            if (strcasecmp_workaround(strings.common.toAddress, displayBuffer) != 0) {
                PRINTF("ERR_SILENT_MODE_CHECK_FAILED, address check failed\n");
                THROW(ERR_SILENT_MODE_CHECK_FAILED);
            }
        } else {
            strlcpy(strings.common.toAddress, displayBuffer, sizeof(strings.common.toAddress));
        }
        PRINTF("TO address displayed: %s\n", strings.common.toAddress);

        // Format the amount in a temporary buffer, if in swap case compare it with validated
        // amount, else commit it
        if (!amountToString(tmpContent.txContent.value.value,
                            tmpContent.txContent.value.length,
                            decimals,
                            ticker,
                            displayBuffer,
                            sizeof(displayBuffer))) {
            PRINTF("OVERFLOW, amount to string failed\n");
            THROW(EXCEPTION_OVERFLOW);
        }

        if (G_called_from_swap) {
            // Ensure the values are the same that the ones that have been previously validated
            if (strcmp(strings.common.fullAmount, displayBuffer) != 0) {
                PRINTF("ERR_SILENT_MODE_CHECK_FAILED, amount check failed\n");
                PRINTF("Expected %s\n", strings.common.fullAmount);
                PRINTF("Received %s\n", displayBuffer);
                THROW(ERR_SILENT_MODE_CHECK_FAILED);
            }
        } else {
            strlcpy(strings.common.fullAmount, displayBuffer, sizeof(strings.common.fullAmount));
        }
        PRINTF("Amount displayed: %s\n", strings.common.fullAmount);
    }

    // Compute the max fee in a temporary buffer, if in swap case compare it with validated max fee,
    // else commit it
    max_transaction_fee_to_string(&tmpContent.txContent.gasprice,
                                  &tmpContent.txContent.startgas,
                                  displayBuffer,
                                  sizeof(displayBuffer));
    if (G_called_from_swap) {
        // Ensure the values are the same that the ones that have been previously validated
        if (strcmp(strings.common.maxFee, displayBuffer) != 0) {
            PRINTF("ERR_SILENT_MODE_CHECK_FAILED, fees check failed\n");
            PRINTF("Expected %s\n", strings.common.maxFee);
            PRINTF("Received %s\n", displayBuffer);
            THROW(ERR_SILENT_MODE_CHECK_FAILED);
        }
    } else {
        strlcpy(strings.common.maxFee, displayBuffer, sizeof(strings.common.maxFee));
    }

    PRINTF("Fees displayed: %s\n", strings.common.maxFee);

    // Prepare nonce to display
    nonce_to_string(&tmpContent.txContent.nonce,
                    strings.common.nonce,
                    sizeof(strings.common.nonce));
    PRINTF("Nonce: %s\n", strings.common.nonce);

    // Prepare network field
    get_network_as_string(strings.common.network_name, sizeof(strings.common.network_name));
    PRINTF("Network: %s\n", strings.common.network_name);
    return true;
end:
    return false;
}

void start_signature_flow(void) {
    if (g_use_standard_ui) {
        ux_approve_tx(false);
    } else {
        dataContext.tokenContext.pluginUiState = PLUGIN_UI_OUTSIDE;
        dataContext.tokenContext.pluginUiCurrentItem = 0;
        ux_approve_tx(true);
    }
}

void finalizeParsing(void) {
    g_use_standard_ui = true;

    if (!finalize_parsing_helper()) {
        return;
    }
    // If called from swap, the user has already validated a standard transaction
    // And we have already checked the fields of this transaction above
    if (G_called_from_swap && g_use_standard_ui) {
        io_seproxyhal_touch_tx_ok(NULL);
    } else {
        // If blind-signing detected, start the warning flow beforehand
        if (tmpContent.txContent.dataPresent) {
            ui_warning_contract_data();
        } else {
            start_signature_flow();
        }
    }
}