LittleServerPython/LittleServerMain.py at main · BalloonUpdate/LittleServerPython · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
import ctypes
import json
import sys
import time
from http import HTTPStatus
from http.server import HTTPServer, SimpleHTTPRequestHandler
import yaml
from file import File

inDev = not getattr(sys, 'frozen', False)


def getMetadata():
    temp = File(getattr(sys, '_MEIPASS', ''))
    return json.loads(temp('meta.json').content)


version = getMetadata()['version'] if not inDev else '0.0.0'
commit = getMetadata()['commit'] if not inDev else ''
compile_time = getMetadata()['compile_time'] if not inDev else ''


class UpdaterHttpRequestHandler(SimpleHTTPRequestHandler):
    def do_GET(self):
        if self.path == '/index.yml':
            self.respondWithContent(yaml.safe_dump({
                'update': 'res',
                **indexYaml
            }, sort_keys=False))
        elif self.path == '/' + resDir.name + '.yml':
            self.respondWithContent(yaml.safe_dump(self.hashDir(resDir), sort_keys=False))
        else:
            super(UpdaterHttpRequestHandler, self).do_GET()

    def respondWithContent(self, data: str):
        self.send_response(HTTPStatus.OK)
        self.send_header("Content-type", 'text/plain')
        self.send_header("Content-Length", str(len(data)))
        self.send_header("Last-Modified", self.date_time_string())
        self.end_headers()

        self.wfile.write(data.encode('utf-8'))

    def hashDir(self, directory: File):
        structure = []
        for f in directory:
            if f.isFile:
                structure.append({'name': f.name, 'length': f.length, 'hash': f.sha1})
            if f.isDirectory:
                structure.append({'name': f.name, 'children': self.hashDir(f)})
        return structure


def check():
    if not workDir.exists:
        print('找不到工作目录' + workDir.path)
        sys.exit(1)

    if not configFile.exists:
        print('找不到配置文件' + configFile.name)
        sys.exit(1)

    if not resDir.exists:
        print('找不到资源目录' + resDir.path)
        sys.exit(1)


if __name__ == '__main__':
    try:
        print('正在启动文件更新助手服务端单文件版v'+version+' (开源地址: https://github.com/updater-for-minecraft)')

        try:
            ctypes.windll.kernel32.SetConsoleTitleW("文件更新助手服务端 "+version)
        except:
            pass

        workDir = File('.')
        resDir = workDir('res')
        configFile = workDir('config.yml')

        check()

        configObject = yaml.load(configFile.content, yaml.SafeLoader)
        address = configObject['address']
        port = configObject['port']

        indexYaml = {**configObject}

        print('文件更新助手服务端已经启动!')
        print('API地址: http://'+('127.0.0.1' if address == '0.0.0.0' else address)+':'+str(port)+'/index.yml')

        server = HTTPServer((address, port), UpdaterHttpRequestHandler)
        server.serve_forever()
    except BaseException as e:
        raise e