import pyaudio
import subprocess
import sys
import noisereduce as nr
import numpy as np
from datetime import datetime, timedelta
import hashlib



# 设置参数
FORMAT = pyaudio.paInt16
CHANNELS = 1
RATE = 44100
CHUNK = 1024
RTMP_URL = 'rtmp://127.0.0.1/live/stream'  # 替换为你的 RTMP 服务器地址
DEVICE_INDEX = 0

'''
rtmp服务器搭建指南:
1.编译安装nginx和rtmp module,下载源码.
```
https://nginx.org/download/nginx-1.27.2.tar.gz
https://github.com/arut/nginx-rtmp-module.git
```

生成makefile
```
./configure --prefix=/Users/linus/nginx \
        --with-cc-opt=-Wno-unused-but-set-variable \
        --add-module=/Users/linus/nginx-rtmp-module;
```

2.配置nginx,在nginx.conf末尾追加
```
rtmp {
    server {
        listen 1935;  # RTMP 端口
        chunk_size 4096;
        allow publish 127.0.0.1;
        deny publish all;
        application live {
            live on;  # 开启直播
            record off;  # 关闭录制（如果不需要）
        }
    }
}
```

3.启动nginx

4.运行本代码
python audio-rtmp.py

5.打开vlc，选择网络播放,地址
rtmp://127.0.0.1/live/stream



'''

def start_rtmp_server():
    # 初始化 PyAudio
    audio = pyaudio.PyAudio()
    # 打开音频流
    stream = audio.open(format=FORMAT,
                        channels=CHANNELS,
                        rate=RATE,
                        input=True,
                        frames_per_buffer=CHUNK)
    # 启动 FFmpeg 进程
    ffmpeg_cmd = [
        'ffmpeg',
        '-f', 's16le',           # 输入格式
        '-ar', str(RATE),        # 采样率
        '-ac', str(CHANNELS),    # 声道数
        '-i', '-',                # 输入来自标准输入
        '-c:a', 'aac',           # 音频编码器
        '-f', 'flv',             # 输出格式
        RTMP_URL
    ]
    process = subprocess.Popen(ffmpeg_cmd, stdin=subprocess.PIPE)
    print(f"开始录音并推送到 RTMP 服务器{RTMP_URL}")
    try:
        while True:
            try:
                # 从音频流中读取数据
                data = stream.read(CHUNK)
                # 将字节数据转换为 NumPy 数组
                audio_data = np.frombuffer(data, dtype=np.int16)
                # 进行降噪处理
                reduced_noise = nr.reduce_noise(y=audio_data, sr=RATE)
                # 将处理后的数据转换回字节格式
                output_data = reduced_noise.astype(np.int16).tobytes()
                # 将处理后的数据写入 FFmpeg 进程
                process.stdin.write(output_data)
            except BrokenPipeError:
                print('[warn] pipe write.')
            #except Exception:
            #    print('[warn] opps.')
    except KeyboardInterrupt:
        print("停止录音.")
    # 清理资源
    stream.stop_stream()
    stream.close()
    audio.terminate()
    process.stdin.close()
    process.wait()

def choose_device():
    # 初始化 PyAudio
    audio = pyaudio.PyAudio()
    # 列出所有音频输入设备
    for i in range(audio.get_device_count()):
        device_info = audio.get_device_info_by_index(i)
        if device_info['maxInputChannels'] > 0:
            print(f"Device {i}: {device_info['name']}")
    audio.terminate()
    print(40 * '*')
    print('>>请输入麦克风编号:', end='', flush=True)
    DEVICE_INDEX = sys.stdin.readline().strip()

def create_rtmp_url():
    '''如果使用腾讯云直播服务,创建推流地址'''
    domain = 'push.tlivecloud.com'
    app_name = 'live'
    stream_name = 'stream'
    key = 'key123'
    current_date = datetime.now()
    next_month = current_date + timedelta(days=30)
    _time = int(next_month.timestamp())
    hex_time = str(hex(_time))[2:].upper()
    _txSecret = key + stream_name + hex_time
    txSecret = hashlib.md5(_txSecret.encode('utf-8')).hexdigest()
    txTime = hex_time
    rtmp_url = f'rtmp://{domain}/{app_name}/{stream_name}?txSecret={txSecret}&txTime={txTime}'
    print('rtmp_url:', rtmp_url)
    return rtmp_url

if __name__ == '__main__':
    #choose_device()
    start_rtmp_server()