obsws-cli/obsws_cli/input.py

103 lines
2.4 KiB
Python

"""module containing commands for manipulating inputs."""
from typing import Annotated
import typer
from . import validate
from .alias import AliasGroup
app = typer.Typer(cls=AliasGroup)
@app.callback()
def main():
"""Control inputs in OBS."""
@app.command('list | ls')
def list(
ctx: typer.Context,
input: Annotated[bool, typer.Option(help='Filter by input type.')] = False,
output: Annotated[bool, typer.Option(help='Filter by output type.')] = False,
colour: Annotated[bool, typer.Option(help='Filter by colour source type.')] = False,
):
"""List all inputs."""
resp = ctx.obj.get_input_list()
kinds = []
if input:
kinds.append('input')
if output:
kinds.append('output')
if colour:
kinds.append('colour')
if not any([input, output, colour]):
kinds = ['input', 'output', 'colour']
inputs = filter(
lambda input_: any(kind in input_.get('inputKind') for kind in kinds),
resp.inputs,
)
typer.echo('\n'.join(input_.get('inputName') for input_ in inputs))
@app.command()
def mute(ctx: typer.Context, input_name: str):
"""Mute an input."""
if not validate.input_in_inputs(ctx, input_name):
typer.echo(
f"Input '{input_name}' not found.",
err=True,
)
raise typer.Exit(code=1)
ctx.obj.set_input_mute(
name=input_name,
muted=True,
)
typer.echo(f"Input '{input_name}' muted.")
@app.command()
def unmute(ctx: typer.Context, input_name: str):
"""Unmute an input."""
if not validate.input_in_inputs(ctx, input_name):
typer.echo(
f"Input '{input_name}' not found.",
err=True,
)
raise typer.Exit(code=1)
ctx.obj.set_input_mute(
name=input_name,
muted=False,
)
typer.echo(f"Input '{input_name}' unmuted.")
@app.command('toggle | tg')
def toggle(ctx: typer.Context, input_name: str):
"""Toggle an input."""
if not validate.input_in_inputs(ctx, input_name):
typer.echo(
f"Input '{input_name}' not found.",
err=True,
)
raise typer.Exit(code=1)
# Get the current mute state
resp = ctx.obj.get_input_mute(name=input_name)
new_state = not resp.input_muted
ctx.obj.set_input_mute(
name=input_name,
muted=new_state,
)
typer.echo(
f"Input '{input_name}' {'muted' if new_state else 'unmuted'}.",
)