#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from Basis.Functions import hostFormat


def loadOrigin(proxyInfo: dict) -> list:  # origin stream
    return ['client'] + [
        '--server', '%s:%i' % (hostFormat(proxyInfo['server'], v6Bracket = True), proxyInfo['port']),
        '--password', proxyInfo['passwd'],
    ] + (['--udpovertcp'] if proxyInfo['stream']['uot'] else [])


def loadWebsocket(proxyInfo: dict) -> list:
    isTls = proxyInfo['stream']['secure'] is not None
    wsAddress = (('wss' if isTls else 'ws') + '://%s:%i%s') % (
        hostFormat(proxyInfo['stream']['host'], v6Bracket = True), proxyInfo['port'], proxyInfo['stream']['path']
    )
    brookCommand = [
        'wssclient' if isTls else 'wsclient',
        '--address', '%s:%i' % (hostFormat(proxyInfo['server'], v6Bracket = True), proxyInfo['port']),
        '--password', proxyInfo['passwd'],
    ] + (['--withoutBrookProtocol'] if proxyInfo['stream']['raw'] else [])
    if not isTls:
        return brookCommand + ['--wsserver', wsAddress]
    return brookCommand + ['--wssserver', wsAddress] + (
        [] if proxyInfo['stream']['secure']['verify'] else ['--insecure']
    )


def load(proxyInfo: dict, socksInfo: dict, configFile: str) -> tuple[list, str, dict]:
    brookCommand = ['brook', '--debug', '--listen', ':'] + {  # debug module listen on random port
        'origin': loadOrigin,
        'ws': loadWebsocket,
    }[proxyInfo['stream']['type']](proxyInfo) + [
        '--socks5', '%s:%i' % (hostFormat(socksInfo['addr'], v6Bracket = True), socksInfo['port'])
    ]
    return brookCommand, 'Config file %s no need' % configFile, {}  # command, fileContent, envVar