package audio

import (
	"bytes"
	"context"
	"fmt"
	"io"
	"time"

	"github.com/youpy/go-wav"
	"github.com/hajimehoshi/go-mp3"
	"go.uber.org/zap"
)

// PlayWav 播放 WAV 文件(阻塞),直到完成或 context 取消
func PlayWav(ctx context.Context, r io.ReadCloser) error {
	otoCtx, err := initContext()
	if err != nil {
		return fmt.Errorf("音频上下文初始化失败: %w", err)
	}

	// Read the entire file into memory since wav.NewReader needs ReadAt
	data, err := io.ReadAll(r)
	if err != nil {
		r.Close()
		return fmt.Errorf("读取 WAV 文件失败: %w", err)
	}
	r.Close()

	// Create a reader from the buffered data
	dec := wav.NewReader(bytes.NewReader(data))

	// 获取音频格式信息
	format, err := dec.Format()
	if err != nil {
		return fmt.Errorf("获取 WAV 格式失败: %w", err)
	}

	duration, _ := dec.Duration()
	sourceRate := int(format.SampleRate)
	channels := int(format.NumChannels)

	zap.S().Infof("WAV 音频: %d ch, %d Hz, 时长: %v",
		channels, sourceRate, duration)

	// 需要重采样
	var reader io.Reader = dec
	if needsResampling(sourceRate) {
		zap.S().Infof("重采样: %d Hz → %d Hz", sourceRate, UniversalSampleRate)
		resampleReader, err := newResamplingReader(dec, sourceRate, UniversalSampleRate, channels)
		if err != nil {
			return fmt.Errorf("创建重采样器失败: %w", err)
		}
		reader = resampleReader
	}

	player := otoCtx.NewPlayer(reader)
	defer player.Close()

	player.Play()

	// 等待播放完成
	done := make(chan struct{})
	go func() {
		for !player.IsPlaying() {
			time.Sleep(10 * time.Millisecond)
		}
		for player.IsPlaying() {
			time.Sleep(10 * time.Millisecond)
		}
		time.Sleep(200 * time.Millisecond)
		close(done)
	}()

	select {
	case <-done:
		return nil
	case <-ctx.Done():
		return ctx.Err()
	}
}

// PlayMP3 播放 MP3 文件(阻塞),直到完成或 context 取消
func PlayMP3(ctx context.Context, r io.ReadCloser) error {
	otoCtx, err := initContext()
	if err != nil {
		return fmt.Errorf("音频上下文初始化失败: %w", err)
	}

	dec, err := mp3.NewDecoder(r)
	if err != nil {
		r.Close()
		return fmt.Errorf("MP3 解码失败: %w", err)
	}
	defer r.Close()

	// MP3 解码器信息
	sampleRate := int(dec.SampleRate())
	sampleCount := dec.Length()
	channels := 2 // MP3 通常是立体声
	duration := time.Duration(float64(sampleCount)/float64(sampleRate)*1000) * time.Millisecond

	zap.S().Infof("MP3 音频: %d Hz, 时长约: %v", sampleRate, duration)

	// 需要重采样
	var reader io.Reader = dec
	if needsResampling(sampleRate) {
		zap.S().Infof("重采样: %d Hz → %d Hz", sampleRate, UniversalSampleRate)
		resampleReader, err := newResamplingReader(dec, sampleRate, UniversalSampleRate, channels)
		if err != nil {
			return fmt.Errorf("创建重采样器失败: %w", err)
		}
		reader = resampleReader
	}

	player := otoCtx.NewPlayer(reader)
	defer player.Close()

	player.Play()

	// 等待播放完成
	done := make(chan struct{})
	go func() {
		for !player.IsPlaying() {
			time.Sleep(10 * time.Millisecond)
		}
		for player.IsPlaying() {
			time.Sleep(10 * time.Millisecond)
		}
		time.Sleep(200 * time.Millisecond)
		close(done)
	}()

	select {
	case <-done:
		return nil
	case <-ctx.Done():
		return ctx.Err()
	}
}