75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
import argparse, platform, os
|
|
from discord import Intents, Client, Member, opus
|
|
from discord.ext import commands
|
|
from NoiseGatev2 import NoiseGate
|
|
|
|
# Load the proper OPUS library for the device being used
|
|
async def load_opus():
|
|
# Check the system type and load the correct library
|
|
# Linux ARM AARCH64 running 32bit OS
|
|
processor = platform.machine()
|
|
print("Processor: ", processor)
|
|
if os.name == 'nt':
|
|
if processor == "AMD64":
|
|
print(f"Loaded OPUS library for AMD64")
|
|
opus.load_opus('./opus/libopus_amd64.dll')
|
|
return "AMD64"
|
|
else:
|
|
if processor == "aarch64":
|
|
print(f"Loaded OPUS library for aarch64")
|
|
opus.load_opus('./opus/libopus_aarcch64.so')
|
|
return "aarch64"
|
|
elif processor == "armv7l":
|
|
print(f"Loaded OPUS library for armv7l")
|
|
opus.load_opus('./opus/libopus_armv7l.so')
|
|
return "armv7l"
|
|
|
|
|
|
def main(clientId='OTQzNzQyMDQwMjU1MTE1MzA0.Yg3eRA.ZxEbRr55xahjfaUmPY8pmS-RHTY', channelId=367396189529833476, NGThreshold=50, deviceId=1):
|
|
intents = Intents.default()
|
|
|
|
client = commands.Bot(command_prefix='!', intents=intents)
|
|
|
|
@client.event
|
|
async def on_ready():
|
|
print(f'We have logged in as {client.user}')
|
|
|
|
channelIdToJoin = client.get_channel(channelId)
|
|
print("Channel", channelIdToJoin)
|
|
|
|
print("Loading opus")
|
|
await load_opus()
|
|
|
|
if opus.is_loaded():
|
|
print("Joining voice")
|
|
channelConnection = await channelIdToJoin.connect(timeout=60.0, reconnect=True)
|
|
print("Voice Connected")
|
|
streamHandler = NoiseGate(
|
|
_input_device_index=deviceId,
|
|
_voice_connection=channelConnection,
|
|
_noise_gate_threshold=NGThreshold)
|
|
# Start the audio stream
|
|
streamHandler.run()
|
|
print("stream running")
|
|
|
|
|
|
client.run(clientId)
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("deviceId", type=int, help="The ID of the audio device to use")
|
|
parser.add_argument("channelId", type=int, help="The ID of the voice channel to use")
|
|
parser.add_argument("clientId", type=str, help="The discord client ID")
|
|
parser.add_argument("-n", "--NGThreshold", type=int, help="Change the noisegate threshold. This defaults to 50")
|
|
args = parser.parse_args()
|
|
|
|
if (not args.NGThreshold):
|
|
args.NGThreshold = 50
|
|
|
|
print("Arguments:", args)
|
|
|
|
main(
|
|
clientId=args.clientId,
|
|
channelId=args.channelId,
|
|
NGThreshold=args.NGThreshold,
|
|
deviceId=args.deviceId
|
|
) |