input-remapper/bin/key-mapper-service
2020-12-06 18:42:56 +01:00

129 lines
3.5 KiB
Python
Executable File

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# key-mapper - GUI for device specific keyboard mappings
# Copyright (C) 2020 sezanzeb <proxima@hip70890b.de>
#
# This file is part of key-mapper.
#
# key-mapper is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# key-mapper is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with key-mapper. If not, see <https://www.gnu.org/licenses/>.
"""Starts injecting keycodes based on the configuration."""
import sys
import os
import atexit
import grp
from argparse import ArgumentParser
import gi
gi.require_version('GLib', '2.0')
from gi.repository import GLib
from pydbus import SessionBus
from keymapper.logger import update_verbosity, log_info, \
add_filehandler, logger
from keymapper.daemon import Daemon, BUS_NAME
from keymapper.state import system_mapping
from keymapper.paths import USER
from keymapper.dev.permissions import can_read_devices
def run(cmd):
"""Run and log a command."""
logger.info('Running `%s`...', cmd)
code = os.system(cmd)
if code != 0:
logger.error('Failed. exit code %d', code)
def group_exists(name):
"""Check if a group with that name exists."""
try:
grp.getgrnam(name)
return True
except KeyError:
return False
if __name__ == '__main__':
parser = ArgumentParser()
parser.add_argument(
'-d', '--debug', action='store_true', dest='debug',
help='Displays additional debug information',
default=False
)
parser.add_argument(
'-n', '--key-names', action='store_true', dest='key_names',
help='Print all available names for the mapping',
default=False
)
parser.add_argument(
'-s', '--setup-permissions', action='store_true', dest='setup',
help=(
'Requires sudo. Sets up all needed permissions to run '
'key-mapper as regular user'
),
default=False
)
options = parser.parse_args(sys.argv[1:])
update_verbosity(options.debug)
add_filehandler()
log_info()
if options.key_names:
print('\n'.join(system_mapping.list_names()))
sys.exit(0)
if options.setup:
if os.environ['USER'] != 'root' or USER == 'root':
# USER should contain the actual non-root user
logger.error('This only works using sudo')
sys.exit(1)
if group_exists('input'):
run(f'usermod -a -G input {USER}')
if group_exists('plugdev'):
run(f'usermod -a -G plugdev {USER}')
run(f'setfacl -m u:{USER}:rw- /dev/uinput')
logger.info(
'For these changes to take effect log out and in. '
'In some environments you may need to reboot.'
)
sys.exit(0)
can_read_devices()
bus = SessionBus()
loop = GLib.MainLoop()
daemon = Daemon(loop)
try:
bus.publish(BUS_NAME, daemon)
except RuntimeError:
logger.error('The service is already running')
sys.exit(1)
daemon.autoload()
atexit.register(lambda: daemon.stop(True))
loop.run()