基于Go语言编写一个能在Windows上录音的小程序

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() {
	// 列出所有输入设备
	inputDevices, _ := winmm.EnumInDevices()
	for _, dev := range inputDevices {
		fmt.Println("输入设备:", dev.Id(), dev.Name())
	}

	// 列出所有输出设备
	outputDevices, _ := winmm.EnumOutDevices()
	for _, dev := range outputDevices {
		fmt.Println("输出设备:", dev.Id(), dev.Name())
	}

	// 等待用户按回车继续
	fmt.Println("按回车键开始录音...")
	_, _ = fmt.Scanln()

	// 1. 创建录音所需的音频格式：立体声（2通道）、8kHz 采样率、16位深度
	waveFormat := winmm.NewWaveFormat(winmm.ChStereo, winmm.SpS08kHz, winmm.BpS16)

	// 创建用于录音的缓冲区，时长为 2 秒
	recordBuffer := winmm.NewWaveHeader(waveFormat, 2)

	// 创建录音对象并打开默认录音设备（WaveMapper）
	recorder := winmm.NewWaveIn()
	_ = recorder.Open(winmm.WaveMapper, waveFormat)

	// 准备缓冲区，并获取其唯一标识 key（用于后续操作）
	bufferKey, _ := recorder.PrepareBuffer(recordBuffer)

	fmt.Println("正在录音...（按回车键停止）")

	// 将缓冲区添加到录音设备，并开始录音
	_ = recorder.AddBuffer(bufferKey)
	_ = recorder.Start()

	// 稍微延迟，确保录音已开始
	time.Sleep(200 * time.Millisecond)

	// 启动一个 goroutine，监听用户是否按下回车键以停止录音
	shouldStop := false
	go func() {
		_, _ = fmt.Scanln()
		shouldStop = true
	}()

	// 创建输出 WAV 文件
	outFile, err := os.Create("output.wav")
	if err != nil {
		log.Fatal("无法创建输出文件:", err)
	}
	defer outFile.Close()

	// 创建 WAV 编码器：8kHz、16位、2通道（立体声）、1 块（通常为 1）
	encoder := wav.NewEncoder(outFile, 8000, 16, 2, 1)

	// 初始化用于写入的音频缓冲区
	writeBuffer := audio.IntBuffer{
		Format: &audio.Format{
			NumChannels: 2,   // 立体声
			SampleRate:  8000, // 8kHz 采样率
		},
	}

	// 持续监听录音数据，直到用户停止
	for !shouldStop {
		select {
		case data := <-recorder.Channel:
			// 将接收到的原始字节数据转换为 int16 样本，并追加到 writeBuffer
			copyDataToBuffer(data.GetBufferSlice(), &writeBuffer)
			// 重新将缓冲区加入录音队列（循环使用）
			_ = recorder.AddBuffer(bufferKey)
			_ = recorder.Start()
		default:
			// 短暂休眠，避免忙等待
			time.Sleep(10 * time.Millisecond)
		}
	}

	// 停止录音后，释放缓冲区
	_ = recorder.UnPrepareBuffer(bufferKey)
	// 关闭录音设备
	_ = recorder.Close()

	// 将录制的音频数据写入 WAV 文件（包含 RIFF 头和 PCM 数据）
	if err := encoder.Write(&writeBuffer); err != nil {
		log.Fatal("写入 WAV 文件失败:", err)
	}
	if err := encoder.Close(); err != nil {
		log.Fatal("关闭 WAV 编码器失败:", err)
	}

	fmt.Println("录音已保存为 output.wav")
}

// copyDataToBuffer 将原始字节数据（小端序 int16）转换为 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 转为 int 并追加到缓冲区
		buffer.Data = append(buffer.Data, int(sample))
	}
}