from __future__ import annotations

import csv
from pathlib import Path

REPO_ROOT = Path(__file__).resolve().parents[2]
EXAMPLE_DATA_DIR = REPO_ROOT / "example_data"
SEARCH_CSV = EXAMPLE_DATA_DIR / "search_addresses_london.csv"
REFERENCE_CSV = EXAMPLE_DATA_DIR / "reference_addresses_london.csv"


def _build_search_rows() -> list[dict[str, str]]:
    return [
        {
            "address_line_1": "Flat 1 12 Brixton Road",
            "address_line_2": "London",
            "postcode": "SW9 6BU",
        },
        {
            "address_line_1": "221B Baker Street",
            "address_line_2": "London",
            "postcode": "NW1 6XE",
        },
        {
            "address_line_1": "10 Downing Street",
            "address_line_2": "London",
            "postcode": "SW1A 2AA",
        },
        {
            "address_line_1": "1 Waterloo Road",
            "address_line_2": "London",
            "postcode": "SE1 8TY",
        },
        {
            "address_line_1": "9 Clapham Common North Side",
            "address_line_2": "London",
            "postcode": "SW4 0QW",
        },
        {
            "address_line_1": "3 Coldharbour Lane",
            "address_line_2": "London",
            "postcode": "SW9 8LH",
        },
        {
            "address_line_1": "17 Acre Lane",
            "address_line_2": "London",
            "postcode": "SW2 5SP",
        },
        {
            "address_line_1": "65 Camberwell Church Street",
            "address_line_2": "London",
            "postcode": "SE5 8TR",
        },
        {
            "address_line_1": "88 Fictional Road",
            "address_line_2": "London",
            "postcode": "SE1 1AA",
        },
        {
            "address_line_1": "19 Imaginary Parade",
            "address_line_2": "London",
            "postcode": "SW8 9ZZ",
        },
    ]


def _build_reference_rows() -> list[dict[str, str]]:
    return [
        {
            "addr1": "Flat 1",
            "addr2": "12 Brixton Road",
            "addr3": "London",
            "addr4": "",
            "postcode": "SW9 6BU",
        },
        {
            "addr1": "",
            "addr2": "221B Baker Street",
            "addr3": "London",
            "addr4": "",
            "postcode": "NW1 6XE",
        },
        {
            "addr1": "",
            "addr2": "10 Downing Street",
            "addr3": "London",
            "addr4": "City of Westminster",
            "postcode": "SW1A 2AA",
        },
        {
            "addr1": "",
            "addr2": "1 Waterloo Road",
            "addr3": "London",
            "addr4": "Lambeth",
            "postcode": "SE1 8TY",
        },
        {
            "addr1": "",
            "addr2": "9 Clapham Common North Side",
            "addr3": "London",
            "addr4": "Clapham",
            "postcode": "SW4 0QW",
        },
        {
            "addr1": "",
            "addr2": "3 Coldharbour Lane",
            "addr3": "London",
            "addr4": "Brixton",
            "postcode": "SW9 8LH",
        },
        {
            "addr1": "",
            "addr2": "17 Acre Lane",
            "addr3": "London",
            "addr4": "Brixton",
            "postcode": "SW2 5SP",
        },
        {
            "addr1": "",
            "addr2": "65 Camberwell Church Street",
            "addr3": "London",
            "addr4": "Camberwell",
            "postcode": "SE5 8TR",
        },
        {
            "addr1": "",
            "addr2": "5 Vauxhall Bridge Road",
            "addr3": "London",
            "addr4": "",
            "postcode": "SW1V 2RE",
        },
        {
            "addr1": "",
            "addr2": "24 Kennington Road",
            "addr3": "London",
            "addr4": "",
            "postcode": "SE1 7BL",
        },
        {
            "addr1": "",
            "addr2": "41 Atlantic Road",
            "addr3": "London",
            "addr4": "Brixton",
            "postcode": "SW9 8JL",
        },
        {
            "addr1": "",
            "addr2": "3 Electric Avenue",
            "addr3": "London",
            "addr4": "Brixton",
            "postcode": "SW9 8JX",
        },
        {
            "addr1": "",
            "addr2": "48 Denmark Hill",
            "addr3": "London",
            "addr4": "Camberwell",
            "postcode": "SE5 8RS",
        },
        {
            "addr1": "",
            "addr2": "30 Streatham High Road",
            "addr3": "London",
            "addr4": "Streatham",
            "postcode": "SW16 1DB",
        },
        {
            "addr1": "",
            "addr2": "52 Tulse Hill",
            "addr3": "London",
            "addr4": "",
            "postcode": "SW2 2QA",
        },
        {
            "addr1": "",
            "addr2": "120 Norwood Road",
            "addr3": "London",
            "addr4": "Herne Hill",
            "postcode": "SE24 9AF",
        },
        {
            "addr1": "",
            "addr2": "44 Landor Road",
            "addr3": "London",
            "addr4": "Stockwell",
            "postcode": "SW9 9PH",
        },
        {
            "addr1": "",
            "addr2": "18 Brixton Hill",
            "addr3": "London",
            "addr4": "",
            "postcode": "SW2 1QA",
        },
        {
            "addr1": "",
            "addr2": "101 Clapham Park Road",
            "addr3": "London",
            "addr4": "Clapham",
            "postcode": "SW4 7EW",
        },
        {
            "addr1": "",
            "addr2": "9 Herne Hill",
            "addr3": "London",
            "addr4": "",
            "postcode": "SE24 9NE",
        },
    ]


def _write_csv(path: Path, rows: list[dict[str, str]]) -> None:
    with path.open("w", newline="", encoding="utf-8") as file_obj:
        writer = csv.DictWriter(file_obj, fieldnames=list(rows[0].keys()))
        writer.writeheader()
        writer.writerows(rows)


def main() -> None:
    EXAMPLE_DATA_DIR.mkdir(parents=True, exist_ok=True)

    search_rows = _build_search_rows()
    reference_rows = _build_reference_rows()

    _write_csv(SEARCH_CSV, search_rows)
    _write_csv(REFERENCE_CSV, reference_rows)

    print(f"Wrote search fixture: {SEARCH_CSV}")
    print(f"Wrote reference fixture: {REFERENCE_CSV}")
    print(f"search rows={len(search_rows)}, reference rows={len(reference_rows)}")


if __name__ == "__main__":
    main()