# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========

from unittest.mock import AsyncMock, patch

import pytest

from mirage.accessor.slack import SlackAccessor
from mirage.cache.index.ram import RAMIndexCacheStore
from mirage.commands.builtin.slack import COMMANDS
from mirage.commands.config import CommandOpts
from mirage.core.slack.config import SlackConfig
from mirage.types import PathSpec

CHANNELS = [
    {
        "id": "C1",
        "name": "general",
        "created": 1
    },
    {
        "id": "C2",
        "name": "random",
        "created": 2
    },
]

GENERAL = "/channels/general__C1"
RANDOM = "/channels/random__C2"


def _find_command():
    for fn in COMMANDS:
        for rc in getattr(fn, "_registered_commands", []):
            if rc.name == "find" and rc.filetype is None:
                return fn
    raise AssertionError("factory find not registered for slack")


def _spec(virtual: str) -> PathSpec:
    return PathSpec(virtual=virtual,
                    directory=virtual,
                    resource_path=virtual.strip("/"))


async def _run(paths, *texts: str, **flags) -> list[str]:
    accessor = SlackAccessor(SlackConfig(token="xoxb-test"))
    find = _find_command()
    with patch("mirage.core.slack.readdir.list_channels",
               new_callable=AsyncMock,
               return_value=CHANNELS):
        stdout, _io = await find(
            accessor, paths, list(texts),
            CommandOpts(index=RAMIndexCacheStore(), flags={**flags}))
    data = stdout if isinstance(stdout, bytes) else b""
    return data.decode().splitlines()


@pytest.mark.asyncio
async def test_walk_lists_channel_dirs():
    lines = await _run([_spec("/channels")], maxdepth="1")
    assert "/channels" in lines
    assert GENERAL in lines
    assert RANDOM in lines


@pytest.mark.asyncio
async def test_path_pattern_is_honored():
    lines = await _run([_spec("/channels")], maxdepth="1", path="*general*")
    assert lines == [GENERAL]


@pytest.mark.asyncio
async def test_size_is_honored_dirs_count_as_zero():
    lines = await _run([_spec("/channels")], maxdepth="1", size="+0c")
    assert lines == []
    lines = await _run([_spec("/channels")], maxdepth="1", size="-1k")
    assert GENERAL in lines


FILE_CHANNELS = [{"id": "C1", "name": "general", "created": 1700000000}]

FILE_MESSAGES = [{
    "ts":
    "1700000000.000100",
    "text":
    "here you go",
    "files": [{
        "id": "F1",
        "name": "report.pdf",
        "url_private_download": "https://files.slack.com/x",
        "size": 10,
        "timestamp": 1700000000,
        "mimetype": "application/pdf",
    }],
}]

DAY = "2023-11-14"


async def _run_with_files(paths, *texts: str, **flags) -> list[str]:
    accessor = SlackAccessor(SlackConfig(token="xoxb-test"))
    find = _find_command()
    with patch("mirage.core.slack.readdir.list_channels",
               new_callable=AsyncMock,
               return_value=FILE_CHANNELS), \
         patch("mirage.core.slack.readdir._latest_message_ts",
               new_callable=AsyncMock,
               return_value=1700000000.0001), \
         patch("mirage.core.slack.readdir.fetch_messages_for_day",
               new_callable=AsyncMock,
               return_value=FILE_MESSAGES):
        stdout, _io = await find(
            accessor, paths, list(texts),
            CommandOpts(index=RAMIndexCacheStore(), flags={**flags}))
    data = stdout if isinstance(stdout, bytes) else b""
    return data.decode().splitlines()


@pytest.mark.asyncio
async def test_type_f_lists_uploaded_files():
    lines = await _run_with_files([_spec("/channels")], type="f")
    assert f"{GENERAL}/{DAY}/chat.jsonl" in lines
    assert f"{GENERAL}/{DAY}/files/report__F1.pdf" in lines
    assert f"{GENERAL}/{DAY}/files" not in lines


@pytest.mark.asyncio
async def test_type_d_lists_files_dir_not_blobs():
    lines = await _run_with_files([_spec("/channels")], type="d")
    assert f"{GENERAL}/{DAY}/files" in lines
    assert f"{GENERAL}/{DAY}/files/report__F1.pdf" not in lines
