ProxyC/Builder/TrojanGo.py

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

import json


def sslConfig(proxyInfo: dict) -> dict:
    return {**{
        'verify': proxyInfo['verify']
    }, **({} if proxyInfo['sni'] == '' else {
        'sni': proxyInfo['sni']
    }), **({} if proxyInfo['alpn'] is None else {
        'alpn': proxyInfo['alpn'].split(',')
    })}


def wsConfig(proxyInfo: dict) -> dict:
    if proxyInfo['ws'] is None:
        return {'enabled': False}
    wsObject = {
        'enabled': True,
        'path': proxyInfo['ws']['path']
    }
    if proxyInfo['ws']['host'] != '':
        wsObject['host'] = proxyInfo['ws']['host']
    return wsObject


def ssConfig(proxyInfo: dict) -> dict:
    return {**{
        'enabled': False if proxyInfo['ss'] is None else True
    }, **({} if proxyInfo['ss'] is None else {
        'method': proxyInfo['ss']['method'],
        'password': proxyInfo['ss']['passwd'],
    })}


def pluginConfig(proxyInfo: dict) -> dict:
    return {**{
        'enabled': False if proxyInfo['plugin'] is None else True
    }, **({} if proxyInfo['plugin'] is None else {
        'type': 'shadowsocks',
        'command': proxyInfo['plugin']['type'],
        'option': proxyInfo['plugin']['param'],
    })}


def load(proxyInfo: dict, socksInfo: dict, configFile: str) -> tuple[list, str, dict]:
    trojanGoConfig = {
        'run_type': 'client',
        'local_addr': socksInfo['addr'],
        'local_port': socksInfo['port'],
        'remote_addr': proxyInfo['server'],
        'remote_port': proxyInfo['port'],
        'password': [
            proxyInfo['passwd']
        ],
        'log_level': 0,  # 0 -> debug level
        'ssl': sslConfig(proxyInfo),
        'websocket': wsConfig(proxyInfo),
        'shadowsocks': ssConfig(proxyInfo),
        'transport_plugin': pluginConfig(proxyInfo),
    }
    return ['trojan-go', '-config', configFile], json.dumps(trojanGoConfig), {}