package main

import (
	"bytes"
	"encoding/binary"
	"fmt"
	"io"
	"log"
	"os"
	"time"

	"github.com/Onelio/winmm"
	"github.com/go-audio/audio"
	"github.com/go-audio/wav"
)

func main() {
	// 列出所有輸入設(shè)備
	inputDevices, _ := winmm.EnumInDevices()
	for _, dev := range inputDevices {
		fmt.Println("輸入設(shè)備:", dev.Id(), dev.Name())
	}

	// 列出所有輸出設(shè)備
	outputDevices, _ := winmm.EnumOutDevices()
	for _, dev := range outputDevices {
		fmt.Println("輸出設(shè)備:", dev.Id(), dev.Name())
	}

	// 等待用戶按回車?yán)^續(xù)
	fmt.Println("按回車鍵開始錄音...")
	_, _ = fmt.Scanln()

	// 1. 創(chuàng)建錄音所需的音頻格式：立體聲（2通道）、8kHz 采樣率、16位深度
	waveFormat := winmm.NewWaveFormat(winmm.ChStereo, winmm.SpS08kHz, winmm.BpS16)

	// 創(chuàng)建用于錄音的緩沖區(qū)，時長為 2 秒
	recordBuffer := winmm.NewWaveHeader(waveFormat, 2)

	// 創(chuàng)建錄音對象并打開默認(rèn)錄音設(shè)備（WaveMapper）
	recorder := winmm.NewWaveIn()
	_ = recorder.Open(winmm.WaveMapper, waveFormat)

	// 準(zhǔn)備緩沖區(qū)，并獲取其唯一標(biāo)識 key（用于后續(xù)操作）
	bufferKey, _ := recorder.PrepareBuffer(recordBuffer)

	fmt.Println("正在錄音...（按回車鍵停止）")

	// 將緩沖區(qū)添加到錄音設(shè)備，并開始錄音
	_ = recorder.AddBuffer(bufferKey)
	_ = recorder.Start()

	// 稍微延遲，確保錄音已開始
	time.Sleep(200 * time.Millisecond)

	// 啟動一個 goroutine，監(jiān)聽用戶是否按下回車鍵以停止錄音
	shouldStop := false
	go func() {
		_, _ = fmt.Scanln()
		shouldStop = true
	}()

	// 創(chuàng)建輸出 WAV 文件
	outFile, err := os.Create("output.wav")
	if err != nil {
		log.Fatal("無法創(chuàng)建輸出文件:", err)
	}
	defer outFile.Close()

	// 創(chuàng)建 WAV 編碼器：8kHz、16位、2通道（立體聲）、1 塊（通常為 1）
	encoder := wav.NewEncoder(outFile, 8000, 16, 2, 1)

	// 初始化用于寫入的音頻緩沖區(qū)
	writeBuffer := audio.IntBuffer{
		Format: &audio.Format{
			NumChannels: 2,   // 立體聲
			SampleRate:  8000, // 8kHz 采樣率
		},
	}

	// 持續(xù)監(jiān)聽錄音數(shù)據(jù)，直到用戶停止
	for !shouldStop {
		select {
		case data := <-recorder.Channel:
			// 將接收到的原始字節(jié)數(shù)據(jù)轉(zhuǎn)換為 int16 樣本，并追加到 writeBuffer
			copyDataToBuffer(data.GetBufferSlice(), &writeBuffer)
			// 重新將緩沖區(qū)加入錄音隊列（循環(huán)使用）
			_ = recorder.AddBuffer(bufferKey)
			_ = recorder.Start()
		default:
			// 短暫休眠，避免忙等待
			time.Sleep(10 * time.Millisecond)
		}
	}

	// 停止錄音后，釋放緩沖區(qū)
	_ = recorder.UnPrepareBuffer(bufferKey)
	// 關(guān)閉錄音設(shè)備
	_ = recorder.Close()

	// 將錄制的音頻數(shù)據(jù)寫入 WAV 文件（包含 RIFF 頭和 PCM 數(shù)據(jù)）
	if err := encoder.Write(&writeBuffer); err != nil {
		log.Fatal("寫入 WAV 文件失敗:", err)
	}
	if err := encoder.Close(); err != nil {
		log.Fatal("關(guān)閉 WAV 編碼器失敗:", err)
	}

	fmt.Println("錄音已保存為 output.wav")
}

// copyDataToBuffer 將原始字節(jié)數(shù)據(jù)（小端序 int16）轉(zhuǎn)換為 audio.IntBuffer 所需的 int 切片
func copyDataToBuffer(data []byte, buffer *audio.IntBuffer) {
	reader := bytes.NewReader(data)
	for {
		var sample int16
		err := binary.Read(reader, binary.LittleEndian, &sample)
		switch {
		case err == io.EOF:
			return // 讀取完畢
		case err != nil:
			log.Println("讀取音頻樣本時出錯:", err)
			return
		}
		// 將 int16 轉(zhuǎn)為 int 并追加到緩沖區(qū)
		buffer.Data = append(buffer.Data, int(sample))
	}
}