PKU VPN 2 - 真正实现 PKU VPN 和 Clash 兼容使用

# -*- encoding: utf-8 -*-
#@Author  :   Arthals
#@File    :   updater.py
#@Time    :   2023/04/28 20:42:24
#@Contact :   zhuozhiyongde@126.com
#@Software:   Visual Studio Code

import requests
import os
import yaml


def update():
    url = 'https://yoursubscribe.url'

    r = requests.get(url, timeout=10)
    data = yaml.safe_load(r.text)

    # 添加外部访问密码
    data['secret'] = 'xxxxxxx'

    # 添加 PKU VPN
    pku_proxy = {
        'name': 'PKU',
        'server': '127.0.0.1',
        'port': 11080,
        'type': 'socks5'
    }

    data['proxies'].insert(0, pku_proxy)

    pku_group = {'name': '🎓 北京大学', 'type': 'select', 'proxies': ['PKU']}

    data['proxy-groups'].insert(0, pku_group)
    # 一定要让 vpn 在前面，否则不能兼容 Clash 的 TUN 模式
    data['rules'].insert(0, 'DOMAIN-SUFFIX,vpn.pku.edu.cn,DIRECT')
    data['rules'].insert(1, 'DOMAIN-SUFFIX,pku.edu.cn,🎓 北京大学')

    # 添加自定义规则
    with open('custom_rules.txt', 'r', encoding='utf-8') as f:
        rules = f.readlines()
        for rule in rules:
            rule = '%s,🔰 节点选择' % rule.rstrip("\n")
            data['rules'].insert(0, rule)

    # 添加 fallback 策略组
    fallback_group = {
        'name': '🍂 Fallback',
        'type': 'fallback',
        'proxies': ['♻️ 自动选择'],
        'url': 'http://www.gstatic.com/generate_204',
        'interval': 300
    }
    data['proxy-groups'].insert(0, fallback_group)

    # 导出 yaml 文件
    yaml.Dumper.ignore_aliases = lambda *args: True
    with open('Arthals.yaml', 'w') as file:
        yaml.dump(data, file, allow_unicode=True, default_flow_style=False)
        # print(t.encode('UTF-8'))


if __name__ == '__main__':
    try:
        update()
        os.system('pm2 restart Clash')
        os.system(
            'export https_proxy=http://127.0.0.1:7890 http_proxy=http://127.0.0.1:7890 all_proxy=socks5://127.0.0.1:7890'
        )

    except Exception as e:
        print(repr(e))

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
#@Author  :   Arthals
#@File    :   CustomAPI.py
#@Time    :   2023/09/20 00:46:28
#@Contact :   zhuozhiyongde@126.com
#@Software:   Visual Studio Code

from fastapi import FastAPI, Response

app = FastAPI()

def get_clash_config():
    # 打开/home/ubuntu/.config/clash/Arthals.yaml，并以UTF-8编码返回
    try:
        with open('/home/ubuntu/.config/clash/Arthals.yaml',
                  'r',
                  encoding='utf-8') as f:
            resp = f.read()
            return resp
    except Exception:
        return None

@app.get('/clash')
async def get_config(token: str):
    # 检查params中是否有token，且token是否正确
    if token != 'xxxxxxx':
        return Response(status_code=403)

    return Response(content=get_clash_config(), media_type='text/plain')

tree -P "Custom-API|\.config/clash" --matchdirs --prune -a -L 4
.
├── .config
│   └── clash
│       ├── Arthals-without-pku.yaml
│       ├── Arthals.yaml
│       ├── cache.db
│       ├── clash
│       ├── clash-old
│       ├── clash_rule_providers.yaml
│       ├── clash_rules.yaml
│       ├── config.yaml
│       ├── Country.mmdb
│       ├── ecosystem.config.js
│       ├── raw-update.sh
│       ├── update.log
│       ├── updater.py
│       └── update.sh
└── Custom-API
    ├── CustomAPI.py
    └── ecosystem.config.js

allow-lan: true
external-controller: :9090
log-level: info
mode: Rule
port: 7890
proxies:
-   name: PKU
    port: 11080
    server: 127.0.0.1
    type: socks5
...
proxy-groups:
-   name: 🎓 北京大学
    proxies:
    - PKU
    - DIRECT
    type: select
...
rules:
- DOMAIN-SUFFIX,pku.edu.cn,🎓 北京大学
- RULE-SET,private,DIRECT
- RULE-SET,reject,REJECT
...
- MATCH,DIRECT
secret: xxxxxxx
socks-port: 7891

parsers:
  - url: https://you-subscribe-url
    yaml:
      prepend-proxy-groups:
        - name: 🎓 北京大学
          type: select
          proxies:
            - PKU
            - DIRECT
      prepend-proxies:
        - name: PKU
          port: 11080
          server: localhost
          type: socks5
      prepend-rules:
        - DOMAIN-SUFFIX,pku.edu.cn,🎓 北京大学