dotfiles/.config/scripts/pipewire/link-virtual-mic
Sheldon Lee 2e93b1a285 Fuck bash scripting sometimes
- Rewrote the script in python
2023-04-12 08:19:03 +01:00

79 lines
2.1 KiB
Python
Executable File

#!/usr/bin/python
import argparse
import subprocess
import time
mappingMic = {
"output\\.rnnoise_source:capture_1": "mixed-mic:input_FL",
"output\\.rnnoise_source:capture_2": "mixed-mic:input_FR"
}
mappingLoopback = {
"loopback:monitor_FL": "mixed-mic:input_FL",
"loopback:monitor_FR": "mixed-mic:input_FR"
}
parser = argparse.ArgumentParser(
prog="link-virtual-mic",
description="link pipewire virtual mics.")
parser.add_argument("-a", "--all", action="store_true")
parser.add_argument("-d", "--disconnect", action="store_true")
def main():
args = parser.parse_args()
connectAll = args.all and not args.disconnect
disconnectAll = args.all and args.disconnect
disconnectMic = args.all and args.disconnect
disconnectLoopback = args.disconnect
tries = -1
while tries < 20:
tries += 1
micSuccess = True
loopbackSuccess = True
if disconnectAll or not args.disconnect:
micSuccess = manageLinks(mappingMic, disconnectMic)
if connectAll or disconnectLoopback:
loopbackSuccess = manageLinks(mappingLoopback, disconnectLoopback)
if not micSuccess or not loopbackSuccess:
time.sleep(1)
continue
break
def manageLinks(mapping, disconnect=False):
for key, value in mapping.items():
cmdOpts = {"shell": True, "capture_output": True, "text": True}
cmd = "pw-link -o | grep -e '{}'".format(key)
completedProcess = subprocess.run(cmd, **cmdOpts)
if completedProcess.returncode == 1:
return False
output = completedProcess.stdout.split('\n')[0]
cmd = "pw-link -i | grep -e '{}'".format(value)
completedProcess = subprocess.run(cmd, **cmdOpts)
if completedProcess.returncode == 1:
return False
input = completedProcess.stdout.split('\n')[0]
if disconnect:
cmd = ["pw-link", "-d", output, input]
else:
cmd = ["pw-link", output, input]
print(' '.join(cmd))
subprocess.run(cmd, capture_output=True)
return True
if __name__ == "__main__":
main()