import json
import wave
from pathlib import Path

import numpy as np

from app.audio import build_audio_profile, classify_audio
from app.config import settings
from app.profile_store import Profile


def _write_tone(path: Path, frequency: float, duration: float = 1.2) -> None:
    sample_rate = 16000
    time = np.arange(int(sample_rate * duration)) / sample_rate
    envelope = np.minimum(1.0, time * 8) * np.minimum(1.0, (duration - time) * 8)
    samples = 0.7 * np.sin(2 * np.pi * frequency * time) * envelope
    pcm = (samples * 32767).astype("<i2")
    path.parent.mkdir(parents=True, exist_ok=True)
    with wave.open(str(path), "wb") as output:
        output.setnchannels(1)
        output.setsampwidth(2)
        output.setframerate(sample_rate)
        output.writeframes(pcm.tobytes())


def test_audio_profile_classifies_and_rejects(tmp_path: Path) -> None:
    profiles = tmp_path / "profiles"
    artifacts = tmp_path / "artifacts"
    settings.profiles_dir = profiles
    settings.artifacts_dir = artifacts
    directory = profiles / "test" / "tones" / "v1"
    config = {
        "code": "test.tones.v1",
        "analysisType": "AUDIO_CLASSIFICATION",
        "decision": {
            "maxDistance": 0.45,
            "minDistanceMargin": 0.02,
            "unknownLabel": "UNKNOWN",
        },
    }
    directory.mkdir(parents=True)
    (directory / "profile.json").write_text(json.dumps(config), encoding="utf-8")
    reference_tones = {
        "POWER_ON": 440,
        "POWER_OFF": 660,
        "ARMED": 880,
        "DISARMED": 1100,
    }
    for label, frequency in reference_tones.items():
        _write_tone(directory / "references" / label / f"{label.lower()}.wav", frequency)
    profile = Profile("test.tones.v1", "AUDIO_CLASSIFICATION", directory, config)

    build_audio_profile(profile)
    classified_paths = {}
    for label, frequency in reference_tones.items():
        path = tmp_path / f"{label.lower()}-test.wav"
        _write_tone(path, frequency)
        classified_paths[label] = path
    other_path = tmp_path / "other-test.wav"
    _write_tone(other_path, 1600)

    for label, path in classified_paths.items():
        assert classify_audio(profile, path)["label"] == label
    assert classify_audio(profile, other_path)["label"] == "UNKNOWN"