import tempfile
from pathlib import Path

from pyiceberg.schema import Schema
from pyiceberg.types import LongType, NestedField, StringType
from pyiceberg.io.pyarrow import schema_to_pyarrow
from pyiceberg.catalog import load_catalog

_iceberg_schema = Schema(
    NestedField(1, "id", LongType()),
    NestedField(2, "name", StringType()),
)

_arrow_schema = schema_to_pyarrow(_iceberg_schema)
print(f"Iceberg arrow schema: {_arrow_schema}")

_df = pl.LazyFrame(
    {
        "id": [1, 2],
        "name": ["test1", "test2"],
    }
)

with tempfile.TemporaryDirectory() as tmpdir:
    _df.sink_parquet(
        f"{tmpdir}/test.parquet", arrow_schema=_arrow_schema
    )

with tempfile.TemporaryDirectory() as _tmpdir:
    _warehouse = Path(_tmpdir)

    _catalog = load_catalog(
        "test",
        type="sql",
        uri=f"sqlite:///{_warehouse}/catalog.db",
        warehouse=str(_warehouse),
    )
    _catalog.create_namespace("ns")

    _table = _catalog.create_table(
        "ns.test",
        schema=Schema(
            NestedField(1, "id", LongType()),
            NestedField(2, "name", StringType()),
        ),
    )

    _data_dir = _warehouse / "data"
    _data_dir.mkdir()
    _file_path = _data_dir / "test.parquet"

    _df = pl.DataFrame(
        {
            "id": [1, 2, 3],
            "name": ["test1", "test2", "test3"],
        }
    )
    _df.write_parquet(_file_path)

    _table.add_files([str(_file_path)])

    _table.delete(delete_filter="id = 1")