41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
from pathlib import Path
|
|
from typing import Callable
|
|
|
|
from discord.ext import voice_recv
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class MeetingRecorder:
|
|
"""Wrapper around discord-ext-voice-recv's listen/stop_listening API."""
|
|
|
|
def __init__(self, voice_client: voice_recv.VoiceRecvClient, output_path: str):
|
|
self.vc = voice_client
|
|
self.output_path = output_path
|
|
self.recording = False
|
|
self.sink: voice_recv.AudioSink | None = None
|
|
|
|
async def start(self, after_callback: Callable[[Exception | None], None]) -> None:
|
|
if self.vc.is_listening():
|
|
raise RuntimeError("Voice client is already listening")
|
|
|
|
Path(self.output_path).parent.mkdir(parents=True, exist_ok=True)
|
|
wave_sink = voice_recv.WaveSink(self.output_path)
|
|
self.sink = voice_recv.SilenceGeneratorSink(wave_sink)
|
|
self.vc.listen(self.sink, after=after_callback)
|
|
self.recording = True
|
|
logger.info("Voice receive started: output_path=%s", self.output_path)
|
|
|
|
async def stop(self) -> None:
|
|
if not self.recording:
|
|
logger.info("Stop requested, but recorder was already inactive: output_path=%s", self.output_path)
|
|
return
|
|
|
|
if self.vc.is_listening():
|
|
self.vc.stop_listening()
|
|
logger.info("Voice receive stop requested: output_path=%s", self.output_path)
|
|
self.recording = False
|