import json
import os
from typing import Dict, List

from pydantic import parse_file_as

from mev_inspect.schemas.blocks import Block
from mev_inspect.schemas.sandwiches import Sandwich

THIS_FILE_DIRECTORY = os.path.dirname(__file__)
TEST_BLOCKS_DIRECTORY = os.path.join(THIS_FILE_DIRECTORY, "blocks")
TEST_SANDWICHES_DIRECTORY = os.path.join(THIS_FILE_DIRECTORY, "sandwiches")


def load_test_sandwiches(block_number: int) -> List[Sandwich]:
    sandwiches_path = f"{TEST_SANDWICHES_DIRECTORY}/{block_number}.json"
    return parse_file_as(List[Sandwich], sandwiches_path)


def load_test_block(block_number: int) -> Block:
    block_path = f"{TEST_BLOCKS_DIRECTORY}/{block_number}.json"
    defaults = {"block_timestamp": 0}

    with open(block_path, "r") as block_file:
        block_json = json.load(block_file)
        for item in block_json["receipts"]:
            if "status" not in item:
                item["status"] = "0x1"
        return Block(
            **{
                **defaults,
                **block_json,
            }
        )


def load_comp_markets() -> Dict[str, str]:
    comp_markets_path = f"{THIS_FILE_DIRECTORY}/comp_markets.json"
    with open(comp_markets_path, "r") as markets_file:
        markets = json.load(markets_file)
        return markets


def load_cream_markets() -> Dict[str, str]:
    cream_markets_path = f"{THIS_FILE_DIRECTORY}/cream_markets.json"
    with open(cream_markets_path, "r") as markets_file:
        markets = json.load(markets_file)
        return markets