#!/usr/bin/env python3
# Set Huion H430P parameters from saved profiles using xsetwacom
# Driver: https://github.com/DIGImend/digimend-kernel-drivers
#
# Write your config into ~/.config/huion-tablet-yml
# Example:
#
# presets:
#   my-preset:
#     screen: DP2-2
#     area: 0 0 24384 15240
#     buttons:
#       pad:
#         1: key Ctrl Z
#         2: key Ctrl Shift Z
#         3: key +
#         8: key -
# then run huion-tablet.py my-preset

from subprocess import run
import argparse
import os.path
import yaml

def set_parameter(device, *args):
    if device == 'stylus':
        device = 'HUION Huion Tablet Pen stylus'
    elif device == 'pad':
        device = 'HUION Huion Tablet Pad pad'
    args = map(str, args)
    run(['xsetwacom', 'set', device, *args], check=True)

parser = argparse.ArgumentParser(description='setup huion h430p tablet')
parser.add_argument('--area', type=str, default='')
parser.add_argument('--screen', type=str, default='')
parser.add_argument('preset', metavar='PRESET', type=str, nargs='?', help='a preset')
args = parser.parse_args()

area = '0 0 24384 15240'
screen = ''

if args.preset is not None:
    with open(os.path.expanduser('~/.config/huion-tablet.yml'), 'r') as f:
        config = yaml.load(f, Loader=yaml.SafeLoader)
        preset = config['presets'][args.preset]
        area = preset['area']
        screen = preset['screen']
        if 'buttons' in preset:
            for device in preset['buttons']:
                for button, mapping in preset['buttons'][device].items():
                    set_parameter(device, 'button', button, *str(mapping).split(' '))

if args.area != '':
    area = args.area
if args.screen != '':
    screen = args.screen

area = tuple(map(int, area.split(' ')))

set_parameter('stylus', 'Area', ' '.join(map(str, area)))
set_parameter('stylus', 'MapToOutput', screen)