104 lines
3.5 KiB
Python
104 lines
3.5 KiB
Python
import pymumble_py3 as pymumble
|
|
from pymumble_py3.constants import *
|
|
import subprocess as sp
|
|
import time
|
|
import sys
|
|
import os
|
|
import fcntl
|
|
import audioop
|
|
import argparse
|
|
|
|
def message_received(message):
|
|
global is_streaming
|
|
global silence_time
|
|
command=message.message
|
|
if command == "/start":
|
|
is_streaming = True
|
|
silence_time = 0
|
|
mumble.users.myself.recording()
|
|
elif command == "/stop":
|
|
is_streaming = False
|
|
mumble.users.myself.unrecording()
|
|
|
|
parser = argparse.ArgumentParser(description='Regia pienamente automatizzata')
|
|
parser.add_argument('--channel', help='Set channel', default="")
|
|
parser.add_argument('--name', help='Set bot nickname', default="RadioRobbot")
|
|
parser.add_argument('--server', help='Set server', default="mumble.esiliati.org")
|
|
parser.add_argument('--port', help='Set port', type=int, default=64738)
|
|
parser.add_argument('--stream', action='store_true', help='Ignore commands in chat and stream everything')
|
|
parser.add_argument('--auto-suspend-stream', action='store_true', help='Ignore commands in chat and stream everything')
|
|
parser.add_argument('--max-silence-time', type=int, help='max silence time in seconds', default=30)
|
|
|
|
sys.argv.pop(0)
|
|
args = parser.parse_args(sys.argv)
|
|
|
|
|
|
pwd = "" # password
|
|
server = args.server
|
|
nick = args.name
|
|
channel = args.channel
|
|
port = args.port
|
|
is_streaming=False
|
|
stream_always= args.stream
|
|
auto_suspend_stream = args.auto_suspend_stream
|
|
silence_limit = 30
|
|
|
|
|
|
|
|
# Spin up a client and connect to mumble server
|
|
mumble = pymumble.Mumble(server, nick, password=pwd, port=port)
|
|
|
|
mumble.callbacks.set_callback(PYMUMBLE_CLBK_TEXTMESSAGERECEIVED, message_received)
|
|
mumble.set_receive_sound(1) # Enable receiving sound from mumble server
|
|
mumble.start()
|
|
mumble.is_ready() # Wait for client is ready
|
|
mumble.channels.find_by_name(channel).move_in()
|
|
mumble.users.myself.mute()
|
|
|
|
if is_streaming:
|
|
mumble.users.myself.recording()
|
|
|
|
BUFFER = 0.1
|
|
BITRATE = 48000
|
|
RESOLUTION = 10 # in ms
|
|
FLOAT_RESOLUTION = float(RESOLUTION) / 1000
|
|
MONO_CHUNK_SIZE = BITRATE * 2 * RESOLUTION / 1000
|
|
STEREO_CHUNK_SIZE = MONO_CHUNK_SIZE * 2
|
|
silent = b"\x00" * int(STEREO_CHUNK_SIZE)
|
|
cursor_time = None
|
|
cursor_time = time.time() - BUFFER
|
|
silence_time = 0
|
|
silence_limit_ms = silence_limit * 1000
|
|
|
|
while mumble.is_alive():
|
|
if cursor_time < time.time() - BUFFER:
|
|
base_sound = None
|
|
|
|
try:
|
|
for user in mumble.users.values(): # check the audio queue of each user
|
|
if user.sound.is_sound():
|
|
# available sound is to be treated now and not later
|
|
sound = user.sound.get_sound(FLOAT_RESOLUTION)
|
|
stereo_pcm = audioop.tostereo(sound.pcm, 2, 1, 1)
|
|
if base_sound == None:
|
|
base_sound = stereo_pcm
|
|
else:
|
|
base_sound = audioop.add(base_sound, stereo_pcm, 2)
|
|
except RuntimeError:
|
|
print("ignored exception in stderr...", file=sys.stderr)
|
|
|
|
if is_streaming or stream_always:
|
|
if base_sound:
|
|
silence_time = 0
|
|
sys.stdout.buffer.write(base_sound)
|
|
else:
|
|
silence_time += RESOLUTION
|
|
sys.stdout.buffer.write(silent)
|
|
|
|
if auto_suspend_stream and (silence_time >= silence_limit_ms):
|
|
is_streaming = False
|
|
mumble.users.myself.unrecording()
|
|
|
|
cursor_time += FLOAT_RESOLUTION
|
|
else:
|
|
time.sleep(FLOAT_RESOLUTION)
|