# SPDX-License-Identifier: AGPL-3.0-only
# Copyright (c) 2026 sol pbc

import importlib

from solstone.think.utils import day_path


def test_get_raw_file(tmp_path, monkeypatch):
    utils = importlib.import_module("solstone.think.utils")
    monkeypatch.setenv("SOLSTONE_JOURNAL", str(tmp_path))
    day_dir = day_path("20240101")

    # Create segments under default stream
    segment_123000 = day_dir / "default" / "123000_300"
    segment_123000.mkdir(parents=True)
    segment_090000 = day_dir / "default" / "090000_300"
    segment_090000.mkdir(parents=True)

    (segment_123000 / "monitor_1_diff.png").write_bytes(b"data")
    (segment_123000 / "monitor_1_diff.json").write_text(
        '{"visual_description": "screen", "raw": "monitor_1_diff.png"}'
    )

    (segment_090000 / "raw.flac").write_bytes(b"data")
    # Write JSONL format: metadata first, then entry
    (segment_090000 / "audio.jsonl").write_text(
        '{"raw": "raw.flac"}\n{"text": "hello"}\n'
    )

    path, mime, meta = utils.get_raw_file(
        "20240101", "default/123000_300/monitor_1_diff.json"
    )
    assert path == "monitor_1_diff.png"
    assert mime == "image/png"
    assert meta["visual_description"] == "screen"

    path, mime, meta = utils.get_raw_file("20240101", "default/090000_300/audio.jsonl")
    assert path == "raw.flac"
    assert mime == "audio/flac"
    # JSONL format returns a list: [entry1_dict, ...]
    assert isinstance(meta, list) and len(meta) == 1
    assert meta[0]["text"] == "hello"