jack2/tools/jack_control

#!/usr/bin/python3

from __future__ import print_function
name_base = 'org.jackaudio'
control_interface_name = name_base + '.JackControl'
configure_interface_name = name_base + '.Configure'
service_name = name_base + '.service'
control_iface = None
configure_iface = None

import sys
import os
from traceback import print_exc

import dbus

def bool_convert(str_value):
    if str_value == "0" or str_value.lower() in ["false","off","no","(null)"]:
        return False

    return bool(str_value)

def dbus_type_to_python_type(dbus_value):
    t = type(dbus_value)
    if t == dbus.Boolean:
        return bool(dbus_value)
    if t == dbus.Int32 or t == dbus.UInt32:
        return int(dbus_value)
    return dbus_value

def python_type_to_jackdbus_type(value, type_char):
    type_char = str(type_char)
    if type_char == "b":
        return bool_convert(value);
    elif type_char == "y":
        return dbus.Byte(ord(value));
    elif type_char == "i":
        return dbus.Int32(value)
    elif type_char == "u":
        return dbus.UInt32(value)

    return value

def dbus_typesig_to_type_string(type_char):
    type_char = str(type_char)
    if type_char == 'i':
        return "sint"
    if type_char == 'u':
        return "uint"
    if type_char == 'y':
        return "char"
    if type_char == 's':
        return "str"
    if type_char == 'b':
        return "bool"

    print('shit')
    return None                         # throw exception here?

def get_parameters(iface, path):
    params = iface.GetParametersInfo(path)

    #print params
    for param in params:
        typestr = dbus_typesig_to_type_string(param[0])
        name = param[1]
        #print name
        descr = param[2]
        #print descr
        isset, default, value = iface.GetParameterValue(path + [name])
        #print typestr
        if bool(isset):
            isset = "set"
        else:
            isset = "notset"
        value = dbus_type_to_python_type(value)
        default = dbus_type_to_python_type(default)

        print("%20s: %s (%s:%s:%s:%s)" %(name, descr, typestr, isset, default, value))

def print_help():
    print("Usage: %s [command] [command] ..." % os.path.basename(sys.argv[0]))
    print("Commands:")
    print("    shell                      - execute commands from stdin until End Of File (Ctrl+D in terminal)")
    print("    exit                       - exit jack dbus service (stops jack server if currently running)")
    print("    help                       - print this help text")
    print("    status                     - check whether jack server is started, return value is 0 if running and 1 otherwise")
    print("    start                      - start jack server if not currently started")
    print("    stop                       - stop jack server if currently started")
    print("    sm                         - switch master to currently selected driver")
    print("    dl                         - get list of available drivers")
    print("    dg                         - get currently selected driver")
    print("    ds <driver>                - select driver")
    print("    dp                         - get parameters of currently selected driver")
    print("    dpd <param>                - get long description for driver parameter")
    print("    dps <param> <value>        - set driver parameter")
    print("    dpr <param>                - reset driver parameter to its default value")
    print("    asd <driver>               - add slave driver")
    print("    rsd <driver>               - remove slave driver")
    print("    il                         - get list of available internals")
    print("    ip <name>                  - get parameters of given internal")
    print("    ipd <name> <param>         - get long description for internal parameter")
    print("    ips <name> <param> <value> - set internal parameter")
    print("    ipr <name> <param>         - reset internal parameter to its default value")
    print("    iload <name>               - load internal")
    print("    iunload <name>             - unload internal")
    print("    ep                         - get engine parameters")
    print("    epd <param>                - get long description for engine parameter")
    print("    eps <param> <value>        - set engine parameter")
    print("    epr <param>                - reset engine parameter to its default value")

def maybe_print_param_constraint(iface, param):
    is_range, is_strict, is_fake, values = iface.GetParameterConstraint(param)
    if is_range:
        print()
        print(("allowed range: %s to %s (inclusive)" % (values[0][0], values[1][0])))
    elif len(values):
        print()
        if is_strict:
            print("allowed values:")
        else:
            print("suggested values:")

        max_len = 0
        for value in values:
            if len(str(value[0])) > max_len:
                   max_len = len(str(value[0]))
        for value in values:
            print(("%*s'%s' - %s" % (1 + max_len - len(str(value[0])), "", str(value[0]), str(value[1]))))

def parse_argv(argv):
    global control_iface, configure_iface
    
    # check arguments
    index = 0
    while index < len(argv):
        arg = argv[index]
        index += 1
        try:
            if arg == 'exit':
                print("--- exit")
                control_iface.Exit()
            elif arg == 'status':
                print("--- status")
                if control_iface.IsStarted():
                    return (0, "started")
                else:
                    return (1, "stoped")
            elif arg == 'start':
                print("--- start")
                control_iface.StartServer()
            elif arg == 'stop':
                print("--- stop")
                control_iface.StopServer()
            elif arg == 'sm':
                print("--- switch master driver")
                control_iface.SwitchMaster()
            elif arg == 'ism':
                if control_iface.IsManuallyActivated():
                    print("Manually activated")
                else:
                    print("Automatically activated")
            elif arg == 'dl':
                print("--- drivers list")
                is_range, is_strict, is_fake_values, values = configure_iface.GetParameterConstraint(['engine', 'driver'])
                for value in values:
                    print(value[1])
            elif arg == 'dg':
                print("--- get selected driver")
                isset, default, value = configure_iface.GetParameterValue(['engine', 'driver'])
                print(value)
            elif arg == 'ds':
                if index >= len(argv):
                    return (1, "driver select command requires driver name argument")

                arg = argv[index]
                index += 1

                print("--- driver select \"%s\"" % arg)
                configure_iface.SetParameterValue(['engine', 'driver'], dbus.String(arg))
            elif arg == 'dp':
                print("--- get driver parameters (type:isset:default:value)")
                get_parameters(configure_iface, ['driver'])
            elif arg == 'dpd':
                if index >= len(argv):
                    return (1, "get driver parameter long description command requires parameter name argument")

                param = argv[index]
                index += 1

                print("--- get driver parameter description (%s)" % param)
                type_char, name, short_descr, long_descr = configure_iface.GetParameterInfo(['driver', param])
                print(long_descr)
                maybe_print_param_constraint(configure_iface, ['driver', param])
            elif arg == 'dps':
                if index + 1 >= len(argv):
                    return (1, "driver parameter set command requires parameter name and value arguments")

                param = argv[index]
                index += 1
                value = argv[index]
                index += 1

                print("--- driver param set \"%s\" -> \"%s\"" % (param, value))

                type_char, name, short_descr, long_descr = configure_iface.GetParameterInfo(['driver', param])
                configure_iface.SetParameterValue(['driver', param], python_type_to_jackdbus_type(value, type_char))
            elif arg == 'dpr':
                if index >= len(argv):
                    return (1, "driver parameter reset command requires parameter name argument")

                param = argv[index]
                index += 1

                print("--- driver param reset \"%s\"" % param)
                configure_iface.ResetParameterValue(['driver', param])
            elif arg == 'ep':
                print("--- get engine parameters (type:isset:default:value)")
                get_parameters(configure_iface, ['engine'])
            elif arg == 'epd':
                if index >= len(argv):
                    return (1, "get engine parameter long description command requires parameter name argument")

                param_name = argv[index]
                index += 1

                print("--- get engine parameter description (%s)" % param_name)

                type_char, name, short_descr, long_descr = configure_iface.GetParameterInfo(['engine', param_name])
                print(long_descr)
                maybe_print_param_constraint(configure_iface, ['engine', param_name])
            elif arg == 'eps':
                if index + 1 >= len(argv):
                    return (1, "engine parameter set command requires parameter name and value arguments")

                param = argv[index]
                index += 1
                value = argv[index]
                index += 1

                print("--- engine param set \"%s\" -> \"%s\"" % (param, value))

                type_char, name, short_descr, long_descr = configure_iface.GetParameterInfo(['engine', param])
                configure_iface.SetParameterValue(['engine', param], python_type_to_jackdbus_type(value, type_char))
            elif arg == 'epr':
                if index >= len(argv):
                    return (1, "engine parameter reset command requires parameter name")

                param = argv[index]
                index += 1

                print("--- engine param reset \"%s\"" % param)

                type_char, name, short_descr, long_descr = configure_iface.GetParameterInfo(['engine', param])
                configure_iface.ResetParameterValue(['engine', param])
            elif arg == 'il':
                print("--- internals list")
                is_leaf, internals = configure_iface.ReadContainer(['internals'])
                for internal in internals:
                    print(internal)
            elif arg == 'ip':
                print("--- get internal parameters (type:isset:default:value)")

                if index >= len(argv):
                    return (1, "internal parameters command requires internal name argument")

                internal_name = argv[index]
                index += 1

                get_parameters(configure_iface, ['internals', internal_name])
            elif arg == 'ipd':
                if index + 1 >= len(argv):
                    return (1, "get internal parameter long description command requires internal and parameter name arguments")

                name = argv[index]
                index += 1
                param = argv[index]
                index += 1

                print("--- get internal parameter description (%s)" % param)
                type_char, name, short_descr, long_descr = configure_iface.GetParameterInfo(['internals', name, param])
                print(long_descr)
            elif arg == 'ips':
                if index + 2 >= len(argv):
                    return (1, "internal parameter set command requires internal, parameter name and value arguments")

                internal_name = argv[index]
                index += 1
                param = argv[index]
                index += 1
                value = argv[index]
                index += 1

                print("--- internal param set \"%s\" -> \"%s\"" % (param, value))

                type_char, name, short_descr, long_descr = configure_iface.GetParameterInfo(['internals', internal_name, param])
                configure_iface.SetParameterValue(['internals', internal_name, param], python_type_to_jackdbus_type(value, type_char))
            elif arg == 'ipr':
                if index + 1 >= len(argv):
                    return (1, "reset internal parameter command requires internal and parameter name arguments")

                internal_name = argv[index]
                index += 1
                param = argv[index]
                index += 1

                print("--- internal param reset \"%s\"" % param)

                configure_iface.ResetParameterValue(['internals', internal_name, param])
            elif arg == 'iload':
                print("--- load internal")

                if index >= len(argv):
                    return (1, "load internal command requires internal name argument")

                name = argv[index]
                index += 1
                result = control_iface.LoadInternal(name)
            elif arg == 'iunload':
                print("--- unload internal")

                if index >= len(argv):
                    return (1, "unload internal command requires internal name argument")

                name = argv[index]
                index += 1
                result = control_iface.UnloadInternal(name)
            elif arg == 'asd':
                print("--- add slave driver")

                if index >= len(argv):
                    return (1, "add slave driver command requires driver name argument")

                name = argv[index]
                index += 1
                result = control_iface.AddSlaveDriver(name)
            elif arg == 'rsd':
                print("--- remove slave driver")

                if index >= len(argv):
                    return (1, "remove slave driver command requires driver name argument")

                name = argv[index]
                index += 1
                result = control_iface.RemoveSlaveDriver(name)
            elif arg == 'help':
                print_help()
            else:
                return (0, "Unknown command '%s'" % arg)
        except dbus.DBusException as e:
            return (1, "DBus exception: %s" % str(e))
    
    return (0, "")

def shell():
    from click.parser import split_arg_string
    while True:
        cmd = sys.stdin.readline()
        if (cmd == ''):
            break
        s,t = parse_argv(split_arg_string(cmd))
        print("%i: %s" % (s, t))

def main():
    global control_iface, configure_iface

    if len(sys.argv) == 1 or sys.argv[1] in ["-h", "--help"]:
        print_help()
        return 0

    bus = dbus.SessionBus()
    controller = bus.get_object(service_name, "/org/jackaudio/Controller")

    control_iface = dbus.Interface(controller, control_interface_name)
    configure_iface = dbus.Interface(controller, configure_interface_name)

    # check arguments
    if sys.argv[1] == "shell":
        shell()
    else:
        s, t = parse_argv(sys.argv[1:])
        if (t):
            print(t)
        return s

if __name__ == '__main__':
    sys.exit(main())