#!/usr/bin/env python3

import sys
import os
import logging
import tempfile
from urllib.parse import quote_plus as urllib_quote_plus
import subprocess

ENABLE_SSHD = '/usr/bin/steamos-polkit-helpers/steamos-enable-sshd'

logger = logging.getLogger(os.path.realpath(__file__))

import devkit_utils

if __name__ == '__main__':
    logger.setLevel(logging.INFO)
    ch = logging.StreamHandler()
    ch.setLevel(logging.INFO)
    ch.setFormatter(logging.Formatter('%(name)s:%(message)s'))
    logger.addHandler(ch)

    request = open(sys.argv[1], 'rt').read()
    requestIP = 'Unknown'
    try:
        requestIP = sys.argv[2]
    except NameError:
        logger.debug('request IP not given as argument to hook')
        pass

    if os.environ.get('DEVKIT_BYPASS_STEAM_PROMPT', None) == '1':
        # Still using for the standalone dev-kit service application
        logger.warning('Bypassing the Steam client approval - this is potentially insecure!')
        MAGIC_PHRASE = '900b919520e4cf601998a71eec318fec'
        tokens = request.split(' ')
        if tokens[-1].strip('\n') != MAGIC_PHRASE:
            raise Exception('Invalid request')
        sys.exit(0)

    key_identifier = request.split(' ')[2]
    request = f'Development host at IP {requestIP} key: {key_identifier!r}'

    devkit_utils.validate_steam_client()

    with tempfile.TemporaryDirectory(prefix='approve-ssh-key') as tempdir:
        logger.debug('Issue approve-ssh-key request to the running Steam client')
        response = os.path.join(tempdir, 'response')
        cmd = 'approve-ssh-key?response={}&request={}'.format(
            urllib_quote_plus(response),
            urllib_quote_plus(request),
        )
        devkit_utils.execute_steam_client_command(cmd)
        with devkit_utils.wait_on_file_response(response, timeout=30) as f:
            logger.debug('Got response from Steam client')
            # Make sure sshd is enabled to support development features
            if os.path.exists(ENABLE_SSHD):
                subprocess.check_call(ENABLE_SSHD)
            else:
                subprocess.check_call('sudo systemctl enable --now sshd', shell=True)