mirror of
https://github.com/onyx-and-iris/obsws-cli.git
synced 2025-05-16 22:40:23 +01:00
101 lines
2.6 KiB
Python
101 lines
2.6 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['obsws'].get_input_list()
|
|
|
|
# For each kind flag, if it is set to True, add it to the kinds list.
|
|
# If no flags are set, default to all kinds.
|
|
# Define a mapping of kind names to their corresponding flags
|
|
kind_flags = {
|
|
'input': input,
|
|
'output': output,
|
|
'color': colour,
|
|
}
|
|
|
|
# Collect all kinds where the corresponding flag is set to True
|
|
kinds = [kind for kind, flag in kind_flags.items() if flag]
|
|
|
|
# If no flags are set, default to all kinds
|
|
if not kinds:
|
|
kinds = list(kind_flags.keys())
|
|
|
|
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['obsws'].set_input_mute(
|
|
name=input_name,
|
|
muted=True,
|
|
)
|
|
|
|
|
|
@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['obsws'].set_input_mute(
|
|
name=input_name,
|
|
muted=False,
|
|
)
|
|
|
|
|
|
@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['obsws'].get_input_mute(name=input_name)
|
|
new_state = not resp.input_muted
|
|
|
|
ctx.obj['obsws'].set_input_mute(
|
|
name=input_name,
|
|
muted=new_state,
|
|
)
|