archived-sslocal-manager-alpine/app.py

import os
import subprocess
import re
import platform
from flask import Flask, render_template, request, redirect

app = Flask(__name__)

kernel_ver = platform.release()
match = re.search(r'(\d+)\.(\d+)', kernel_ver)
cmv, csv = (int(match.group(1)), int(match.group(2))) if match else (0, 0)
TFO = ""
if cmv >= 3 and csv > 7:
    TFO = "--tcp-fast-open"

ARGS=os.getenv("ARGS","")

def get_status():
    try:
        with open('/var/run/ssserver.pid', 'r') as f:
            pid = f.read().strip()
        
        # 检查进程是否存在
        os.kill(int(pid), 0)
        return (True, pid)
    except (FileNotFoundError, ValueError, ProcessLookupError):
        return (False, None)

@app.route('/')
def index():
    status = get_status()
    return render_template('index.html', 
                          running=status[0],
                          pid=status[1])

@app.route('/status')
def status():
    running, pid = get_status()
    if running:
        return f"Running With PID {pid}", 200
    return "Not Running", 502

@app.route('/action', methods=['POST'])
def action():
    running, pid = get_status()
    
    if running:
        subprocess.run(["kill", pid])
        os.remove('/var/run/ssserver.pid')
    else:
        cmd = [
            "ssserver",
            "--log-without-time",
            "-c", "/.ssconfig.json",
            "-a", "nobody",
            "--dns", "114.114.114.114,8.8.8.8",
            "--daemonize-pid", "/var/run/ssserver.pid"
        ]
        if TFO:
            cmd.append(TFO)
        if ARGS:
            cmd.extend(ARGS.split())
        subprocess.Popen(cmd)
    
    return redirect('/')

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=6000)