CheeseAPI 0.0.1__py3-none-any.whl

CheeseAPI/request.py

@@ -0,0 +1,99 @@
+import json, traceback, re
+from typing import TypeVar, overload, Dict
+
+import CheeseType.network, xmltodict, CheeseLog
+
+from file import File
+
+T = TypeVar('T')
+
+class BaseItem:
+    def __init__(self):
+        self._values: Dict[str, str] = {}
+
+    def get(self, key: str, default: T = None) -> str | File | T:
+        if key not in self._values:
+            return default
+        return self._values[key]
+
+class Args(BaseItem):
+    def __init__(self, query: str):
+        super().__init__()
+
+        for q in query.split('&'):
+            q = q.split('=')
+            self._values[q[0]] = q[1] if len(q) > 1 else None
+
+class Form(BaseItem):
+    @overload
+    def __init__(self):
+        ...
+
+    @overload
+    def __init__(self, body: str):
+        ...
+
+    @overload
+    def __init__(self, body: bytes, spliter: bytes):
+        ...
+
+    def __init__(self, body: str | bytes | None = None, spliter: bytes | None = None):
+        super().__init__()
+
+        if body and not spliter:
+            for s in body.split('&'):
+                s = s.split('=')
+                self._values[s[0]] = s[1]
+
+        elif body and spliter:
+            body = body[2:-4].split(spliter)[1:-1]
+            for s in body:
+                key = re.findall(rb'\bname="(.*?)"', s)[0].decode()
+                value = s.split(b'\r\n\r\n')[1][:-4]
+                filename = re.findall(rb'\bfilename="(.*?)"', s)
+                if len(filename):
+                    self._values[key] = File(filename[0].decode(), value)
+                else:
+                    self._values[key] = value.decode()
+
+class Request:
+    def __init__(self, scope, body: bytes | None = None):
+        query_string = scope['query_string'].decode()
+
+        self.ip: CheeseType.network.IPv4 = scope['client'][0]
+        self.path: str = scope['path']
+        self.fullPath: str = self.path + '?' + query_string if query_string else self.path
+        self.scheme = scope['scheme']
+
+        self.headers: Dict[str, str] = {}
+
+        if scope['type'] in [ 'http', 'https' ]:
+            self.method: str = scope['method']
+            self.args = Args(query_string)
+            self.body = None
+            self.form: Form = Form()
+            for header in scope['headers']:
+                key = header[0].decode()
+                value = header[1].decode()
+                self.headers[key] = value
+                if key == 'content-type':
+                    try:
+                        if value in [ 'application/json', 'application/javascript' ]:
+                            self.body = json.loads(body)
+                        elif value == 'application/xml':
+                            self.body = xmltodict.parse(body)
+                        elif value in [ 'text/plain', 'text/html' ]:
+                            self.body = body.decode()
+                        elif value.startswith('multipart/form-data;'):
+                            self.form = Form(body, header[1].split(b'boundary=')[1].split(b';')[0])
+                        elif value == 'application/x-www-form-urlencoded':
+                            self.form = Form(body.decode())
+                    except:
+                        CheeseLog.danger(f'{self.method} {self.fullPath} cannot parse request.body correctly:\n{traceback.format_exc()}'[:-1], f'\033[36m{self.method} {self.fullPath}\033[0m cannot parse request.body correctly:\n{traceback.format_exc()}'[:-1])
+        else:
+            for header in scope['headers']:
+                key = header[0].decode()
+                value = header[1].decode()
+                self.headers[key] = value
+                if key == 'sec-websocket-key':
+                    self.sid: str = value

CheeseAPI/response.py

@@ -0,0 +1,384 @@
+import json
+from typing import Dict, Any
+
+from file import File
+
+CONTENT_TYPE = {
+    'tif': 'image/tiff',
+    '001': 'application/x-001',
+    '301': 'application/x-301',
+    '323': 'text/h323',
+    '906': 'application/x-906',
+    '907': 'drawing/907',
+    'a11': 'application/x-a11',
+    'acp': 'audio/x-mei-aac',
+    'ai': 'application/postscript',
+    'aif': 'audio/aiff',
+    'aifc': 'audio/aiff',
+    'aiff': 'audio/aiff',
+    'anv': 'application/x-anv',
+    'asa': 'text/asa',
+    'asf': 'video/x-ms-asf',
+    'asp': 'text/asp',
+    'asx': 'video/x-ms-asf',
+    'au': 'audio/basic',
+    'avi': 'video/avi',
+    'awf': 'application/vnd.adobe.workflow',
+    'biz': 'text/xml',
+    'bmp': 'application/x-bmp',
+    'bot': 'application/x-bot',
+    'c4t': 'application/x-c4t',
+    'c90': 'application/x-c90',
+    'cal': 'application/x-cals',
+    'cat': 'application/vnd.ms-pki.seccat',
+    'cdf': 'application/x-netcdf',
+    'cdr': 'application/x-cdr',
+    'cel': 'application/x-cel',
+    'cer': 'application/x-x509-ca-cert',
+    'cg4': 'application/x-g4',
+    'cgm': 'application/x-cgm',
+    'cit': 'application/x-cit',
+    'class': 'java/*',
+    'cml': 'text/xml',
+    'cmp': 'application/x-cmp',
+    'cmx': 'application/x-cmx',
+    'cot': 'application/x-cot',
+    'crl': 'application/pkix-crl',
+    'crt': 'application/x-x509-ca-cert',
+    'csi': 'application/x-csi',
+    'css': 'text/css',
+    'cut': 'application/x-cut',
+    'dbf': 'application/x-dbf',
+    'dbm': 'application/x-dbm',
+    'dbx': 'application/x-dbx',
+    'dcd': 'text/xml',
+    'dcx': 'application/x-dcx',
+    'der': 'application/x-x509-ca-cert',
+    'dgn': 'application/x-dgn',
+    'dib': 'application/x-dib',
+    'dll': 'application/x-msdownload',
+    'doc': 'application/msword',
+    'dot': 'application/msword',
+    'drw': 'application/x-drw',
+    'dtd': 'text/xml',
+    'dwf': 'Model/vnd.dwf',
+    'dwf': 'application/x-dwf',
+    'dwg': 'application/x-dwg',
+    'dxb': 'application/x-dxb',
+    'dxf': 'application/x-dxf',
+    'edn': 'application/vnd.adobe.edn',
+    'emf': 'application/x-emf',
+    'eml': 'message/rfc822',
+    'ent': 'text/xml',
+    'epi': 'application/x-epi',
+    'eps': 'application/x-ps',
+    'etd': 'application/x-ebx',
+    'exe': 'application/x-msdownload',
+    'fax': 'image/fax',
+    'fdf': 'application/vnd.fdf',
+    'fif': 'application/fractals',
+    'fo': 'text/xml',
+    'frm': 'application/x-frm',
+    'g4': 'application/x-g4',
+    'gbr': 'application/x-gbr',
+    '': 'application/x-',
+    'gif': 'image/gif',
+    'gl2': 'application/x-gl2',
+    'gp4': 'application/x-gp4',
+    'hgl': 'application/x-hgl',
+    'hmr': 'application/x-hmr',
+    'hpg': 'application/x-hpgl',
+    'hpl': 'application/x-hpl',
+    'hqx': 'application/mac-binhex40',
+    'hrf': 'application/x-hrf',
+    'hta': 'application/hta',
+    'htc': 'text/x-component',
+    'htm': 'text/html',
+    'html': 'text/html',
+    'htt': 'text/webviewhtml',
+    'htx': 'text/html',
+    'icb': 'application/x-icb',
+    'ico': 'image/x-icon',
+    'iff': 'application/x-iff',
+    'ig4': 'application/x-g4',
+    'igs': 'application/x-igs',
+    'iii': 'application/x-iphone',
+    'img': 'application/x-img',
+    'ins': 'application/x-internet-signup',
+    'isp': 'application/x-internet-signup',
+    'IVF': 'video/x-ivf',
+    'java': 'java/*',
+    'jfif': 'image/jpeg',
+    'jpe': 'image/jpeg',
+    'jpeg': 'image/jpeg',
+    'jpg': 'image/jpeg',
+    'js': 'application/x-javascript',
+    'jsp': 'text/html',
+    'la1': 'audio/x-liquid-file',
+    'lar': 'application/x-laplayer-reg',
+    'latex': 'application/x-latex',
+    'lavs': 'audio/x-liquid-secure',
+    'lbm': 'application/x-lbm',
+    'lmsff': 'audio/x-la-lms',
+    'ls': 'application/x-javascript',
+    'ltr': 'application/x-ltr',
+    'm1v': 'video/x-mpeg',
+    'm2v': 'video/x-mpeg',
+    'm3u': 'audio/mpegurl',
+    'm4e': 'video/mpeg4',
+    'mac': 'application/x-mac',
+    'man': 'application/x-troff-man',
+    'math': 'text/xml',
+    'mdb': 'application/msaccess',
+    'mfp': 'application/x-shockwave-flash',
+    'mht': 'message/rfc822',
+    'mhtml': 'message/rfc822',
+    'mi': 'application/x-mi',
+    'mid': 'audio/mid',
+    'midi': 'audio/mid',
+    'mil': 'application/x-mil',
+    'mml': 'text/xml',
+    'mnd': 'audio/x-musicnet-download',
+    'mns': 'audio/x-musicnet-stream',
+    'mocha': 'application/x-javascript',
+    'movie': 'video/x-sgi-movie',
+    'mp1': 'audio/mp1',
+    'mp2': 'audio/mp2',
+    'mp2v': 'video/mpeg',
+    'mp3': 'audio/mp3',
+    'mp4': 'video/mpeg4',
+    'mpa': 'video/x-mpg',
+    'mpd': 'application/vnd.ms-project',
+    'mpe': 'video/x-mpeg',
+    'mpeg': 'video/mpg',
+    'mpg': 'video/mpg',
+    'mpga': 'audio/rn-mpeg',
+    'mpp': 'application/vnd.ms-project',
+    'mps': 'video/x-mpeg',
+    'mpt': 'application/vnd.ms-project',
+    'mpv': 'video/mpg',
+    'mpv2': 'video/mpeg',
+    'mpw': 'application/vnd.ms-project',
+    'mpx': 'application/vnd.ms-project',
+    'mtx': 'text/xml',
+    'mxp': 'application/x-mmxp',
+    'net': 'image/pnetvue',
+    'nrf': 'application/x-nrf',
+    'nws': 'message/rfc822',
+    'odc': 'text/x-ms-odc',
+    'out': 'application/x-out',
+    'p10': 'application/pkcs10',
+    'p12': 'application/x-pkcs12',
+    'p7b': 'application/x-pkcs7-certificates',
+    'p7c': 'application/pkcs7-mime',
+    'p7m': 'application/pkcs7-mime',
+    'p7r': 'application/x-pkcs7-certreqresp',
+    'p7s': 'application/pkcs7-signature',
+    'pc5': 'application/x-pc5',
+    'pci': 'application/x-pci',
+    'pcl': 'application/x-pcl',
+    'pcx': 'application/x-pcx',
+    'pdf': 'application/pdf',
+    'pdx': 'application/vnd.adobe.pdx',
+    'pfx': 'application/x-pkcs12',
+    'pgl': 'application/x-pgl',
+    'pic': 'application/x-pic',
+    'pko': 'application/vnd.ms-pki.pko',
+    'pl': 'application/x-perl',
+    'plg': 'text/html',
+    'pls': 'audio/scpls',
+    'plt': 'application/x-plt',
+    'png': 'image/png',
+    'pot': 'application/vnd.ms-powerpoint',
+    'ppa': 'application/vnd.ms-powerpoint',
+    'ppm': 'application/x-ppm',
+    'pps': 'application/vnd.ms-powerpoint',
+    'ppt': 'application/vnd.ms-powerpoint',
+    'pr': 'application/x-pr',
+    'prf': 'application/pics-rules',
+    'prn': 'application/x-prn',
+    'prt': 'application/x-prt',
+    'ps': 'application/postscript',
+    'ptn': 'application/x-ptn',
+    'pwz': 'application/vnd.ms-powerpoint',
+    'r3t': 'text/vnd.rn-realtext3d',
+    'ra': 'audio/vnd.rn-realaudio',
+    'ram': 'audio/x-pn-realaudio',
+    'ras': 'application/x-ras',
+    'rat': 'application/rat-file',
+    'rdf': 'text/xml',
+    'rec': 'application/vnd.rn-recording',
+    'red': 'application/x-red',
+    'rgb': 'application/x-rgb',
+    'rjs': 'application/vnd.rn-realsystem-rjs',
+    'rjt': 'application/vnd.rn-realsystem-rjt',
+    'rlc': 'application/x-rlc',
+    'rle': 'application/x-rle',
+    'rm': 'application/vnd.rn-realmedia',
+    'rmf': 'application/vnd.adobe.rmf',
+    'rmi': 'audio/mid',
+    'rmj': 'application/vnd.rn-realsystem-rmj',
+    'rmm': 'audio/x-pn-realaudio',
+    'rmp': 'application/vnd.rn-rn_music_package',
+    'rms': 'application/vnd.rn-realmedia-secure',
+    'rmvb': 'application/vnd.rn-realmedia-vbr',
+    'rmx': 'application/vnd.rn-realsystem-rmx',
+    'rnx': 'application/vnd.rn-realplayer',
+    'rp': 'image/vnd.rn-realpix',
+    'rpm': 'audio/x-pn-realaudio-plugin',
+    'rsml': 'application/vnd.rn-rsml',
+    'rt': 'text/vnd.rn-realtext',
+    'rtf': 'application/msword',
+    'rv': 'video/vnd.rn-realvideo',
+    'sam': 'application/x-sam',
+    'sat': 'application/x-sat',
+    'sdp': 'application/sdp',
+    'sdw': 'application/x-sdw',
+    'sit': 'application/x-stuffit',
+    'slb': 'application/x-slb',
+    'sld': 'application/x-sld',
+    'slk': 'drawing/x-slk',
+    'smi': 'application/smil',
+    'smil': 'application/smil',
+    'smk': 'application/x-smk',
+    'snd': 'audio/basic',
+    'sol': 'text/plain',
+    'sor': 'text/plain',
+    'spc': 'application/x-pkcs7-certificates',
+    'spl': 'application/futuresplash',
+    'spp': 'text/xml',
+    'ssm': 'application/streamingmedia',
+    'sst': 'application/vnd.ms-pki.certstore',
+    'stl': 'application/vnd.ms-pki.stl',
+    'stm': 'text/html',
+    'sty': 'application/x-sty',
+    'svg': 'text/xml',
+    'swf': 'application/x-shockwave-flash',
+    'tdf': 'application/x-tdf',
+    'tg4': 'application/x-tg4',
+    'tga': 'application/x-tga',
+    'tif': 'image/tiff',
+    'tiff': 'image/tiff',
+    'tld': 'text/xml',
+    'top': 'drawing/x-top',
+    'torrent': 'application/x-bittorrent',
+    'tsd': 'text/xml',
+    'txt': 'text/plain',
+    'uin': 'application/x-icq',
+    'uls': 'text/iuls',
+    'vcf': 'text/x-vcard',
+    'vda': 'application/x-vda',
+    'vdx': 'application/vnd.visio',
+    'vml': 'text/xml',
+    'vpg': 'application/x-vpeg005',
+    'vsd': 'application/vnd.visio',
+    'vss': 'application/vnd.visio',
+    'vst': 'application/vnd.visio',
+    'vsw': 'application/vnd.visio',
+    'vsx': 'application/vnd.visio',
+    'vtx': 'application/vnd.visio',
+    'vxml': 'text/xml',
+    'wav': 'audio/wav',
+    'wax': 'audio/x-ms-wax',
+    'wb1': 'application/x-wb1',
+    'wb2': 'application/x-wb2',
+    'wb3': 'application/x-wb3',
+    'wbmp': 'image/vnd.wap.wbmp',
+    'wiz': 'application/msword',
+    'wk3': 'application/x-wk3',
+    'wk4': 'application/x-wk4',
+    'wkq': 'application/x-wkq',
+    'wks': 'application/x-wks',
+    'wm': 'video/x-ms-wm',
+    'wma': 'audio/x-ms-wma',
+    'wmd': 'application/x-ms-wmd',
+    'wmf': 'application/x-wmf',
+    'wml': 'text/vnd.wap.wml',
+    'wmv': 'video/x-ms-wmv',
+    'wmx': 'video/x-ms-wmx',
+    'wmz': 'application/x-ms-wmz',
+    'wp6': 'application/x-wp6',
+    'wpd': 'application/x-wpd',
+    'wpg': 'application/x-wpg',
+    'wpl': 'application/vnd.ms-wpl',
+    'wq1': 'application/x-wq1',
+    'wr1': 'application/x-wr1',
+    'wri': 'application/x-wri',
+    'wrk': 'application/x-wrk',
+    'ws': 'application/x-ws',
+    'ws2': 'application/x-ws',
+    'wsc': 'text/scriptlet',
+    'wsdl': 'text/xml',
+    'wvx': 'video/x-ms-wvx',
+    'xdp': 'application/vnd.adobe.xdp',
+    'xdr': 'text/xml',
+    'xfd': 'application/vnd.adobe.xfd',
+    'xfdf': 'application/vnd.adobe.xfdf',
+    'xhtml': 'text/html',
+    'xls': 'application/vnd.ms-excel',
+    'xlw': 'application/x-xlw',
+    'xml': 'text/xml',
+    'xpl': 'audio/scpls',
+    'xq': 'text/xml',
+    'xql': 'text/xml',
+    'xquery': 'text/xml',
+    'xsd': 'text/xml',
+    'xsl': 'text/xml',
+    'xslt': 'text/xml',
+    'xwd': 'application/x-xwd',
+    'x_b': 'application/x-x_b',
+    'sis': 'application/vnd.symbian.install',
+    'sisx': 'application/vnd.symbian.install',
+    'x_t': 'application/x-x_t',
+    'ipa': 'application/vnd.iphone',
+    'apk': 'application/vnd.android.package-archive',
+    'xap': 'application/x-silverlight-app',
+}
+
+class BaseResponse:
+    def __init__(self, body: Any = '', status: int = 200, headers: Dict[str, str] = {}) -> None:
+        self.status: int = status
+        self.headers: Dict[str, str] = headers
+        self.body: Any = body
+
+class Response(BaseResponse):
+    def __init__(self, body: Any = '', status: int = 200, headers: Dict[str, str] = {}):
+        _headers = {
+            'content-type': 'text/plain; charset=utf-8'
+        }
+        _headers.update(headers)
+        super().__init__(body, status, _headers)
+
+class JsonResponse(BaseResponse):
+    def __init__(self, body: Dict = {}, status: int = 200, headers: Dict[str, str] = {}):
+        _headers = {
+            'content-type': 'application/json; charset=utf-8'
+        }
+        _headers.update(headers)
+        super().__init__(json.dumps(body), status, _headers)
+
+class RedirectResponse(BaseResponse):
+    def __init__(self, url: str, status: int = 301, headers: Dict[str, str] = {}):
+        _headers = {
+            'Location': url
+        }
+        _headers.update(headers)
+        super().__init__(b'', status, _headers)
+
+class FileResponse(BaseResponse):
+    def __init__(self, file: File, isDownloaded: bool = False, headers: Dict[str, str] = {}):
+        fileSuffix = file.name.split('.')
+        if len(fileSuffix) == 1:
+            fileSuffix = ''
+        else:
+            fileSuffix = fileSuffix[-1]
+        if isDownloaded or fileSuffix not in CONTENT_TYPE:
+            _headers = {
+                'content-type': 'application/octet-stream'
+            }
+        else:
+            _headers = {
+                'content-type': CONTENT_TYPE[fileSuffix]
+            }
+        _headers.update(headers)
+        super().__init__(file.data, 200, _headers)

CheeseAPI/route.py

@@ -0,0 +1,140 @@
+import re, uuid
+from typing import Self, Callable, Dict, List, Tuple, Any
+
+PATTERNS: Dict[str, re.Pattern] = {
+    'str': {
+        'pattern': r'.+',
+        'type': str
+    },
+    'int': {
+        'pattern': r'[-+]?[0-9]+',
+        'type': int
+    },
+    'float': {
+        'pattern': r'[-+]?[0-9]+.[0-9]+',
+        'type': float
+    },
+    'uuid': {
+        'pattern': r'r"[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}"',
+        'type': uuid.UUID
+    }
+}
+
+_PATHS: dict = {}
+
+class _MatchPathNode:
+    def __init__(self):
+        self.children: dict = {}
+        self.key: str | None = None
+        self.isEnd: bool = False
+
+class _MatchPath:
+    def __init__(self):
+        self.root: _MatchPathNode = _MatchPathNode()
+
+    def insert(self, path: str, func: Callable, methods: List[str]):
+        node = self.root
+        for part in path.split('/')[1:]:
+            if re.match(r'<\w+:\w+>', part):
+                part = part[1:-1].split(':')
+                _part = f'<:{part[1]}>'
+                if _part not in node.children:
+                    node.children[_part] = _MatchPathNode()
+                node.children[_part].key = part[0]
+                part = _part
+            else:
+                if part not in node.children:
+                    node.children[part] = _MatchPathNode()
+            node = node.children[part]
+        node.isEnd = True
+
+        for method in methods:
+            node.children[method] = func
+
+    def match(self, path: str) -> Tuple[Dict[str, Callable], Dict[str, Any]]:
+        kwargs = {}
+        node = self.root
+        for part in path.split('/')[1:]:
+            if part not in node.children:
+                flag = False
+                for key, value in reversed(PATTERNS.items()):
+                    if re.fullmatch(value['pattern'], part) and f'<:{key}>' in node.children:
+                        kwargs[node.children[f'<:{key}>'].key] = value['type'](part)
+                        part = f'<:{key}>'
+                        flag = True
+                        break
+                if not flag:
+                    return None, {}
+            node = node.children[part]
+        if node.isEnd:
+            return node.children, kwargs
+        else:
+            return None, {}
+
+_MATCH_PATH: _MatchPath = _MatchPath()
+
+class Route:
+    def __init__(self, prefix: str = '', parent: Self | None = None):
+        self.parent: Route | None = parent
+        if self.parent is not None:
+            self.parent.routers.append(self)
+            prefix = self.parent.prefix + prefix
+        self.prefix: str = prefix
+        self.routers: List[Route] = []
+        self.paths: Dict[str, Dict[str, Callable]] = {}
+
+    def __call__(self, path: str, methods: List[str]):
+        def decorator(func):
+            self.default(path, methods, func)
+        return decorator
+
+    def default(self, path: str, methods: List[str], func: Callable):
+        _path = self.prefix + path
+        if _path not in _PATHS:
+            _PATHS[_path] = {}
+        if _path not in self.paths:
+            self.paths[_path] = {}
+
+        if not isinstance(_PATHS[_path], dict):
+            _PATHS[_path] = {}
+        for method in methods:
+            _PATHS[_path][method] = func
+            self.paths[_path][method] = func
+
+        _MATCH_PATH.insert(_path, func, methods)
+
+    def get(self, path: str):
+        def decorator(func):
+            self.default(path, [ 'GET' ], func)
+        return decorator
+
+    def post(self, path: str):
+        def decorator(func):
+            self.default(path, [ 'POST' ], func)
+        return decorator
+
+    def delete(self, path: str):
+        def decorator(func):
+            self.default(path, [ 'DELETE' ], func)
+        return decorator
+
+    def patch(self, path: str):
+        def decorator(func):
+            self.default(path, [ 'PATCH' ], func)
+        return decorator
+
+    def put(self, path: str):
+        def decorator(func):
+            self.default(path, [ 'PUT' ], func)
+        return decorator
+
+    def websocket(self, path: str):
+        def decorator(func):
+            self.default(path, [ 'WEBSOCKET' ], func)
+        return decorator
+
+def matchPath(path: str) -> Tuple[Dict[str, Callable], Dict[str, Any]]:
+    if path in _PATHS:
+        return _PATHS[path], {}
+
+    return _MATCH_PATH.match(path)