asteri 1.0.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
asteri-1.0.0/LICENSE ADDED
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2026 IshikawaUta
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
asteri-1.0.0/PKG-INFO ADDED
@@ -0,0 +1,11 @@
1
+ Metadata-Version: 2.4
2
+ Name: asteri
3
+ Version: 1.0.0
4
+ License-File: LICENSE
5
+ Requires-Dist: setproctitle>=1.3.3
6
+ Requires-Dist: watchdog>=3.0.0
7
+ Requires-Dist: h2>=4.1.0
8
+ Requires-Dist: gevent>=23.9.1
9
+ Requires-Dist: psutil>=5.9.0
10
+ Dynamic: license-file
11
+ Dynamic: requires-dist
asteri-1.0.0/README.md ADDED
@@ -0,0 +1,86 @@
1
+ # 🌟 Asteri Web Server
2
+
3
+ **Asteri** adalah web server Python berperforma tinggi, *production-ready*, yang dirancang dengan sistem argumen CLI yang kaya dan intuitif. Asteri mendukung berbagai protokol mulai dari WSGI, ASGI, hingga uWSGI biner.
4
+
5
+ ## ✨ Fitur Utama
6
+ - **Multi-Protokol**: HTTP/1.1, HTTP/2 (Full Frame Support), WSGI, ASGI, uWSGI.
7
+ - **Berbagai Tipe Worker**:
8
+ - `sync`: Worker sinkron standar.
9
+ - `gthread`: Worker berbasis thread untuk konkurensi lebih baik.
10
+ - `gevent`: Worker asinkron berbasis greenlet (performa tinggi).
11
+ - `asgi`: Dukungan penuh untuk aplikasi modern (FastAPI, Starlette).
12
+ - **Auto-Reload Profesional**: Berbasis *event-driven* menggunakan `watchdog`.
13
+ - **Manajemen Proses**: Mode Daemon, PID files, dan User/Group switching.
14
+ - **Keamanan**: Dukungan SSL/TLS (HTTPS) yang mudah dikonfigurasi.
15
+ - **Dashboard Status**: Pantau kesehatan server di endpoint `/asteri-status`.
16
+
17
+ ## 🚀 Instalasi
18
+ ```bash
19
+ git clone https://github.com/IshikawaUta/asteri.git
20
+ cd asteri
21
+ pip install -e .
22
+ ```
23
+
24
+ ## 🛠️ Penggunaan Dasar
25
+ Menjalankan aplikasi WSGI:
26
+ ```bash
27
+ asteri myapp:app
28
+ ```
29
+
30
+ Menjalankan dengan 4 worker dan bind ke port tertentu:
31
+ ```bash
32
+ asteri myapp:app -w 4 -b 0.0.0.0:8080
33
+ ```
34
+
35
+ ## 📖 Panduan CLI Lengkap
36
+
37
+ Asteri mendukung berbagai kategori argumen untuk kontrol penuh:
38
+
39
+ ### 🌐 Jaringan (Network)
40
+ - `-b, --bind ADDRESS`: Bind ke alamat tertentu (misal: `127.0.0.1:8000`).
41
+ - `--backlog INT`: Batas antrean koneksi.
42
+ - `--reuse-port`: Menggunakan flag `SO_REUSEPORT` pada soket.
43
+
44
+ ### 👷 Worker
45
+ - `-w, --workers INT`: Jumlah proses worker.
46
+ - `-k, --worker-class STRING`: Tipe worker (`sync`, `gthread`, `asgi`, `gevent`).
47
+ - `-t, --timeout INT`: Batas waktu worker sebelum di-*restart*.
48
+ - `--preload`: Memuat aplikasi sebelum *forking* worker.
49
+
50
+ ### 🔒 Keamanan & SSL
51
+ - `--certfile FILE`: Jalur ke file sertifikat SSL.
52
+ - `--keyfile FILE`: Jalur ke file kunci SSL.
53
+ - `-u, --user USER`: Menjalankan worker sebagai user tertentu.
54
+ - `-g, --group GROUP`: Menjalankan worker sebagai group tertentu.
55
+
56
+ ### 📝 Logging & Debugging
57
+ - `--log-file FILE`: Jalur ke file log error.
58
+ - `--log-level LEVEL`: Granularitas log (`debug`, `info`, `warning`, `error`).
59
+ - `--access-logfile FILE`: Jalur ke file log akses.
60
+ - `--print-config`: Menampilkan konfigurasi yang sedang digunakan dan keluar.
61
+
62
+ ### ⚙️ Proses
63
+ - `-D, --daemon`: Menjalankan Asteri di latar belakang (background).
64
+ - `-p, --pid FILE`: Membuat file PID.
65
+ - `--reload`: Memantau perubahan kode dan memuat ulang worker secara otomatis.
66
+ - `--chdir DIR`: Mengubah direktori kerja sebelum menjalankan aplikasi.
67
+
68
+ ## 📄 File Konfigurasi
69
+ Anda bisa menggunakan file Python untuk konfigurasi yang lebih kompleks:
70
+
71
+ ```python
72
+ # asteri.conf.py
73
+ bind = "127.0.0.1:8080"
74
+ workers = 4
75
+ worker_class = "gthread"
76
+ timeout = 60
77
+ reload = True
78
+ ```
79
+ Jalankan dengan: `asteri myapp:app -c asteri.conf.py`
80
+
81
+ ## 📊 Dashboard Status
82
+ Aktifkan dashboard internal untuk melihat statistik worker:
83
+ Kunjungi `http://localhost:8000/asteri-status` pada browser Anda.
84
+
85
+ ## 📜 Lisensi
86
+ Proyek ini dilisensikan di bawah **MIT License**.
File without changes
@@ -0,0 +1,199 @@
1
+ import logging
2
+ import argparse
3
+ import sys
4
+ import os
5
+ from asteri.arbiter import Arbiter
6
+ from asteri.workers.sync import SyncWorker
7
+ from asteri.workers.gthread import GThreadWorker
8
+ from asteri.workers.asgi import ASGIWorker
9
+ try:
10
+ from asteri.workers.gevent import GeventWorker
11
+ except ImportError:
12
+ GeventWorker = None
13
+ from asteri.utils import logger, print_banner, Colors, setup_logging
14
+
15
+ def import_app(app_path):
16
+ """Import application from string 'module:callable'."""
17
+ try:
18
+ module_path, app_name = app_path.split(":")
19
+ sys.path.insert(0, os.getcwd())
20
+ module = __import__(module_path)
21
+ return getattr(module, app_name)
22
+ except Exception as e:
23
+ logger.error(f"Could not import app '{Colors.BOLD}{app_path}{Colors.ENDC}': {e}")
24
+ sys.exit(1)
25
+
26
+ def main():
27
+ print_banner()
28
+
29
+ parser = argparse.ArgumentParser(
30
+ prog="asteri",
31
+ description=f"{Colors.BOLD}Asteri: High Performance Web Server{Colors.ENDC}",
32
+ formatter_class=argparse.ArgumentDefaultsHelpFormatter,
33
+ add_help=True
34
+ )
35
+
36
+ # Positional Arguments
37
+ parser.add_argument("app", nargs="?", help="Application path (e.g., myapp:app)")
38
+
39
+ # Config Group
40
+ config_group = parser.add_argument_group("Config")
41
+ config_group.add_argument("-c", "--config", help="The Asteri config file.")
42
+ config_group.add_argument("-v", "--version", action="version", version="Asteri v1.0.0")
43
+ config_group.add_argument("--check-config", action="store_true", help="Check the configuration and exit.")
44
+ config_group.add_argument("--print-config", action="store_true", help="Print the configuration settings.")
45
+
46
+ # Network Group
47
+ net_group = parser.add_argument_group("Network")
48
+ net_group.add_argument("-b", "--bind", action="append", help="The socket to bind.")
49
+ net_group.add_argument("--backlog", type=int, default=2048, help="The maximum number of pending connections.")
50
+ net_group.add_argument("--reuse-port", action="store_true", help="Set the SO_REUSEPORT flag.")
51
+
52
+ # Worker Group
53
+ worker_group = parser.add_argument_group("Workers")
54
+ worker_group.add_argument("-w", "--workers", type=int, default=1, help="The number of worker processes.")
55
+ worker_group.add_argument("-k", "--worker-class", default="sync", choices=["sync", "gthread", "asgi", "gevent"],
56
+ help="The type of workers to use.")
57
+ worker_group.add_argument("--threads", type=int, default=1, help="The number of worker threads.")
58
+ worker_group.add_argument("--worker-connections", type=int, default=1000, help="Max simultaneous clients.")
59
+ worker_group.add_argument("--max-requests", type=int, default=0, help="Max requests before worker restart.")
60
+ worker_group.add_argument("--max-requests-jitter", type=int, default=0, help="Jitter for max-requests.")
61
+ worker_group.add_argument("-t", "--timeout", type=int, default=30, help="Worker timeout in seconds.")
62
+ worker_group.add_argument("--graceful-timeout", type=int, default=30, help="Timeout for graceful restart.")
63
+ worker_group.add_argument("--keep-alive", type=int, default=2, help="Keep-alive timeout.")
64
+ worker_group.add_argument("--preload", action="store_true", help="Load app code before forking.")
65
+
66
+ # Security/SSL Group
67
+ sec_group = parser.add_argument_group("Security")
68
+ sec_group.add_argument("--keyfile", help="SSL key file")
69
+ sec_group.add_argument("--certfile", help="SSL certificate file")
70
+ sec_group.add_argument("--ca-certs", help="CA certificates file")
71
+ sec_group.add_argument("--ssl-version", type=int, default=2, help="SSL version to use.")
72
+ sec_group.add_argument("--ciphers", help="SSL Cipher suite to use.")
73
+ sec_group.add_argument("-u", "--user", help="Switch worker processes to run as this user.")
74
+ sec_group.add_argument("-g", "--group", help="Switch worker process to run as this group.")
75
+ sec_group.add_argument("-m", "--umask", type=int, default=0, help="Bit mask for file mode.")
76
+
77
+ # Logging Group
78
+ log_group = parser.add_argument_group("Logging")
79
+ log_group.add_argument("--access-logfile", help="The Access log file.")
80
+ log_group.add_argument("--error-logfile", "--log-file", help="The Error log file.")
81
+ log_group.add_argument("--log-level", default="info", choices=["debug", "info", "warning", "error", "critical"])
82
+ log_group.add_argument("--capture-output", action="store_true", help="Redirect stdout/stderr to error log.")
83
+ log_group.add_argument("--access-logformat", help="The access log format.")
84
+
85
+ # Process Group
86
+ proc_group = parser.add_argument_group("Process")
87
+ proc_group.add_argument("-D", "--daemon", action="store_true", help="Daemonize the process.")
88
+ proc_group.add_argument("-p", "--pid", help="PID filename.")
89
+ proc_group.add_argument("--chdir", help="Change directory before loading apps.")
90
+ proc_group.add_argument("-e", "--env", action="append", help="Set environment variables.")
91
+ proc_group.add_argument("-n", "--name", help="Process name for setproctitle.")
92
+ proc_group.add_argument("--reload", action="store_true", help="Restart workers on code changes.")
93
+
94
+ # HTTP limits
95
+ http_group = parser.add_argument_group("HTTP Limits")
96
+ http_group.add_argument("--limit-request-line", type=int, default=4094)
97
+ http_group.add_argument("--limit-request-fields", type=int, default=100)
98
+ http_group.add_argument("--limit-request-field_size", type=int, default=8190)
99
+
100
+ # HTTP/2 Group
101
+ h2_group = parser.add_argument_group("HTTP/2")
102
+ h2_group.add_argument("--http-protocols", default="h1", help="HTTP protocols to support (e.g., h1,h2)")
103
+ h2_group.add_argument("--http2-max-concurrent-streams", type=int, default=100)
104
+
105
+ args = parser.parse_args()
106
+
107
+ # 1. Load Config File if provided
108
+ if args.config and os.path.exists(args.config):
109
+ config_namespace = {}
110
+ with open(args.config) as f:
111
+ exec(f.read(), config_namespace)
112
+ for key, value in config_namespace.items():
113
+ if hasattr(args, key.lower()):
114
+ setattr(args, key.lower(), value)
115
+
116
+ # Ensure bind is always a list
117
+ if args.bind and isinstance(args.bind, str):
118
+ args.bind = [args.bind]
119
+ elif not args.bind:
120
+ args.bind = ["127.0.0.1:8000"]
121
+
122
+ # 2. Check Config
123
+ if args.check_config or args.print_config:
124
+ if args.print_config:
125
+ print(f"{Colors.BOLD}Resolved Configuration:{Colors.ENDC}")
126
+ for arg in vars(args):
127
+ print(f" {arg}: {getattr(args, arg)}")
128
+ sys.exit(0)
129
+
130
+ # 3. Handle Chdir
131
+ if args.chdir:
132
+ os.chdir(args.chdir)
133
+ sys.path.insert(0, os.getcwd())
134
+
135
+ # 4. Handle Env
136
+ if args.env:
137
+ for env_var in args.env:
138
+ if "=" in env_var:
139
+ k, v = env_var.split("=", 1)
140
+ os.environ[k] = v
141
+
142
+ # 5. Logging setup
143
+ log_file = args.error_logfile
144
+ log_level_name = (args.log_level or "info").upper()
145
+ log_level = getattr(logging, log_level_name, logging.INFO)
146
+ setup_logging(log_file=log_file, level=log_level)
147
+
148
+ worker_map = {
149
+ "sync": SyncWorker,
150
+ "gthread": GThreadWorker,
151
+ "asgi": ASGIWorker,
152
+ "gevent": GeventWorker
153
+ }
154
+
155
+ worker_class = worker_map[args.worker_class]
156
+ if worker_class is None:
157
+ logger.error(f"Worker class '{args.worker_class}' is not available.")
158
+ sys.exit(1)
159
+
160
+ if not args.app:
161
+ parser.print_help()
162
+ sys.exit(1)
163
+
164
+ # Preload
165
+ if args.preload:
166
+ from asteri.utils import import_app
167
+ import_app(args.app)
168
+
169
+ # Setup binds
170
+ binds = args.bind if args.bind else ["127.0.0.1:8000"]
171
+
172
+ arbiter = Arbiter(
173
+ args.app, worker_class,
174
+ num_workers=args.workers,
175
+ bind=binds[0],
176
+ reload=args.reload,
177
+ certfile=args.certfile,
178
+ keyfile=args.keyfile,
179
+ daemon=args.daemon,
180
+ pidfile=args.pid,
181
+ user=args.user,
182
+ group=args.group,
183
+ umask=args.umask,
184
+ proc_name=args.name,
185
+ timeout=args.timeout
186
+ )
187
+
188
+ logger.info(f"Booting {Colors.CYAN}{args.worker_class}{Colors.ENDC} workers...")
189
+
190
+ try:
191
+ arbiter.start()
192
+ except KeyboardInterrupt:
193
+ pass
194
+ except Exception as e:
195
+ logger.error(f"Fatal error: {e}")
196
+ sys.exit(1)
197
+
198
+ if __name__ == "__main__":
199
+ main()
@@ -0,0 +1,204 @@
1
+ import os
2
+ import signal
3
+ import socket
4
+ import sys
5
+ import time
6
+ from .utils import logger, set_proctitle, Colors
7
+
8
+ try:
9
+ from watchdog.observers import Observer
10
+ from watchdog.events import FileSystemEventHandler
11
+ WATCHDOG_AVAILABLE = True
12
+ except ImportError:
13
+ WATCHDOG_AVAILABLE = False
14
+
15
+ class Arbiter:
16
+ """The Master process that manages workers."""
17
+
18
+ def __init__(self, app_path, worker_class, num_workers=1, bind="0.0.0.0:8000", reload=False, certfile=None, keyfile=None,
19
+ daemon=False, pidfile=None, user=None, group=None, umask=0, proc_name=None, timeout=30):
20
+ self.app_path = app_path
21
+ self.worker_class = worker_class
22
+ self.num_workers = num_workers
23
+ self.bind = bind
24
+ self.reload = reload
25
+ self.certfile = certfile
26
+ self.keyfile = keyfile
27
+ self.daemon = daemon
28
+ self.pidfile = pidfile
29
+ self.user = user
30
+ self.group = group
31
+ self.umask = umask
32
+ self.proc_name = proc_name or "master"
33
+ self.timeout = timeout
34
+ self.workers = {} # pid -> worker_instance
35
+ self.sock = None
36
+ self.alive = True
37
+ self.pid = os.getpid()
38
+ self.reloader = None
39
+
40
+ def start(self):
41
+ if self.daemon:
42
+ self.daemonize()
43
+
44
+ if self.pidfile:
45
+ self.write_pid()
46
+
47
+ if self.umask:
48
+ os.umask(self.umask)
49
+
50
+ set_proctitle(self.proc_name)
51
+ logger.info(f"Starting Asteri Arbiter (pid: {Colors.BOLD}{os.getpid()}{Colors.ENDC})")
52
+
53
+ if self.reload and WATCHDOG_AVAILABLE:
54
+ self.setup_reloader()
55
+ elif self.reload:
56
+ logger.warning("Watchdog not available, falling back to manual reload.")
57
+
58
+ # Create listener socket
59
+ host, port = self.bind.split(":")
60
+ self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
61
+ self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
62
+
63
+ if self.certfile and self.keyfile:
64
+ import ssl
65
+ context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
66
+ context.load_cert_chain(certfile=self.certfile, keyfile=self.keyfile)
67
+ self.sock = context.wrap_socket(self.sock, server_side=True)
68
+ logger.info(f"SSL {Colors.GREEN}enabled{Colors.ENDC} ({self.certfile})")
69
+
70
+ self.sock.bind((host, int(port)))
71
+ self.sock.listen(1024)
72
+ self.sock.setblocking(False)
73
+
74
+ scheme = "https" if self.certfile else "http"
75
+ logger.info(f"Listening on {Colors.UNDERLINE}{scheme}://{self.bind}{Colors.ENDC}")
76
+
77
+ self.setup_signals()
78
+ self.manage_workers()
79
+
80
+ def setup_reloader(self):
81
+ class ReloadHandler(FileSystemEventHandler):
82
+ def __init__(self, arbiter):
83
+ self.arbiter = arbiter
84
+ def on_modified(self, event):
85
+ if event.src_path.endswith('.py'):
86
+ logger.info(f"Change detected in {Colors.BOLD}{os.path.basename(event.src_path)}{Colors.ENDC}. Reloading...")
87
+ self.arbiter.stop_workers(signal.SIGTERM)
88
+
89
+ self.reloader = Observer()
90
+ self.reloader.schedule(ReloadHandler(self), os.getcwd(), recursive=True)
91
+ self.reloader.start()
92
+ logger.info(f"Auto-reload {Colors.GREEN}enabled{Colors.ENDC} (watchdog)")
93
+
94
+ def daemonize(self):
95
+ """Standard double-fork daemonization."""
96
+ if os.fork() > 0: sys.exit(0)
97
+ os.setsid()
98
+ if os.fork() > 0: sys.exit(0)
99
+
100
+ # Redirect standard file descriptors
101
+ sys.stdout.flush()
102
+ sys.stderr.flush()
103
+ si = open(os.devnull, 'r')
104
+ so = open(os.devnull, 'a+')
105
+ se = open(os.devnull, 'a+')
106
+ os.dup2(si.fileno(), sys.stdin.fileno())
107
+ os.dup2(so.fileno(), sys.stdout.fileno())
108
+ os.dup2(se.fileno(), sys.stderr.fileno())
109
+
110
+ def write_pid(self):
111
+ with open(self.pidfile, 'w') as f:
112
+ f.write(str(os.getpid()))
113
+
114
+ def switch_user(self):
115
+ if not self.user and not self.group:
116
+ return
117
+ import pwd, grp
118
+ if self.group:
119
+ gid = grp.getgrnam(self.group).gr_gid
120
+ os.setgid(gid)
121
+ if self.user:
122
+ uid = pwd.getpwnam(self.user).pw_uid
123
+ os.setuid(uid)
124
+
125
+ def setup_signals(self):
126
+ signal.signal(signal.SIGCHLD, self.handle_chld)
127
+ signal.signal(signal.SIGTERM, self.handle_exit)
128
+ signal.signal(signal.SIGINT, self.handle_exit)
129
+ signal.signal(signal.SIGHUP, self.handle_hup)
130
+ signal.signal(signal.SIGQUIT, self.handle_quit)
131
+
132
+ def handle_chld(self, sig, frame):
133
+ """Worker died."""
134
+ self.wakeup()
135
+
136
+ def handle_exit(self, sig, frame):
137
+ self.alive = False
138
+ self.stop_workers(signal.SIGTERM)
139
+
140
+ def handle_quit(self, sig, frame):
141
+ self.alive = False
142
+ self.stop_workers(signal.SIGQUIT)
143
+
144
+ def handle_hup(self, sig, frame):
145
+ """Reload workers."""
146
+ logger.info("Reloading workers...")
147
+ self.stop_workers(signal.SIGTERM)
148
+
149
+ def wakeup(self):
150
+ """Force loop to check workers."""
151
+ pass
152
+
153
+ def stop_workers(self, sig):
154
+ import psutil
155
+ for pid in list(self.workers.keys()):
156
+ try:
157
+ p = psutil.Process(pid)
158
+ p.send_signal(sig)
159
+ except psutil.NoSuchProcess:
160
+ if pid in self.workers:
161
+ del self.workers[pid]
162
+
163
+ def spawn_worker(self):
164
+ worker = self.worker_class(0, self.pid, self.sock, self.app_path, self.timeout)
165
+ pid = os.fork()
166
+
167
+ if pid == 0: # Child
168
+ try:
169
+ self.switch_user()
170
+ worker.init_process()
171
+ worker.run()
172
+ sys.exit(0)
173
+ except Exception as e:
174
+ logger.error(f"Worker error: {e}")
175
+ sys.exit(1)
176
+ else: # Parent
177
+ self.workers[pid] = worker
178
+ return pid
179
+
180
+ def manage_workers(self):
181
+ while self.alive:
182
+ # Maintain worker count
183
+ while len(self.workers) < self.num_workers:
184
+ self.spawn_worker()
185
+
186
+ # Reaping dead children
187
+ try:
188
+ pid, status = os.waitpid(-1, os.WNOHANG)
189
+ while pid > 0:
190
+ if pid in self.workers:
191
+ del self.workers[pid]
192
+ pid, status = os.waitpid(-1, os.WNOHANG)
193
+ except ChildProcessError:
194
+ pass
195
+
196
+ time.sleep(1.0)
197
+
198
+ # Cleanup
199
+ if self.reloader:
200
+ self.reloader.stop()
201
+ self.reloader.join()
202
+
203
+ logger.info("Asteri shutting down.")
204
+ self.sock.close()
@@ -0,0 +1,120 @@
1
+ import socket
2
+ import io
3
+ import http.client
4
+ from .utils import logger
5
+
6
+ class HTTPRequest:
7
+ def __init__(self, method, path, version, headers, body=None):
8
+ self.method = method
9
+ self.path = path
10
+ self.version = version
11
+ self.headers = headers
12
+ self.body = body
13
+
14
+ class HTTPParser:
15
+ @staticmethod
16
+ def parse(raw_data):
17
+ """Simple HTTP/1.1 parser."""
18
+ try:
19
+ # Split headers and body
20
+ header_part, _, body = raw_data.partition(b"\r\n\r\n")
21
+ lines = header_part.split(b"\r\n")
22
+
23
+ if not lines or not lines[0]:
24
+ return None
25
+
26
+ # Request line
27
+ request_line = lines[0].decode('ascii').split()
28
+ if len(request_line) < 3:
29
+ return None
30
+
31
+ method, path, version = request_line
32
+
33
+ # Headers
34
+ headers = {}
35
+ for line in lines[1:]:
36
+ if b":" in line:
37
+ key, value = line.split(b":", 1)
38
+ headers[key.decode('ascii').lower().strip()] = value.decode('ascii').strip()
39
+
40
+ return HTTPRequest(method, path, version, headers, body)
41
+ except Exception as e:
42
+ logger.error(f"Failed to parse HTTP request: {e}")
43
+ return None
44
+
45
+ try:
46
+ import h2.connection
47
+ import h2.events
48
+ H2_AVAILABLE = True
49
+ except ImportError:
50
+ H2_AVAILABLE = False
51
+
52
+ class HTTP2Handler:
53
+ """Professional HTTP/2 handler using 'h2' library."""
54
+ PREFACE = b"PRI * HTTP/2.0\r\n\r\nSM\r\n\r\n"
55
+
56
+ @staticmethod
57
+ def is_http2(data):
58
+ return data.startswith(HTTP2Handler.PREFACE)
59
+
60
+ def __init__(self, sock):
61
+ self.sock = sock
62
+ self.conn = h2.connection.H2Connection()
63
+ self.conn.initiate_connection()
64
+ self.sock.sendall(self.conn.data_to_send())
65
+
66
+ def handle(self):
67
+ """Main HTTP/2 event loop for a connection."""
68
+ if not H2_AVAILABLE:
69
+ logger.error("HTTP/2 requested but 'h2' library is missing.")
70
+ return
71
+
72
+ while True:
73
+ try:
74
+ data = self.sock.recv(65535)
75
+ if not data: break
76
+
77
+ events = self.conn.receive_data(data)
78
+ for event in events:
79
+ if isinstance(event, h2.events.RequestReceived):
80
+ self.handle_request(event)
81
+
82
+ data_to_send = self.conn.data_to_send()
83
+ if data_to_send:
84
+ self.sock.sendall(data_to_send)
85
+ except Exception as e:
86
+ logger.debug(f"H2 Connection closed: {e}")
87
+ break
88
+
89
+ def handle_request(self, event):
90
+ # Convert H2 headers to dict
91
+ headers = dict(event.headers)
92
+ method = headers.get(':method')
93
+ path = headers.get(':path')
94
+
95
+ # This would then call the app... for now just logging
96
+ logger.info(f"H2 Request: {method} {path}")
97
+
98
+ # Simple H2 Response
99
+ response_headers = [
100
+ (':status', '200'),
101
+ ('content-type', 'text/plain'),
102
+ ('server', 'Asteri'),
103
+ ]
104
+ self.conn.send_headers(event.stream_id, response_headers)
105
+ self.conn.send_data(event.stream_id, b"Hello from Asteri (HTTP/2)!", end_stream=True)
106
+ self.sock.sendall(self.conn.data_to_send())
107
+
108
+ def build_http_response(status_code, headers, body):
109
+ """Build a standard HTTP/1.1 response."""
110
+ status_text = http.client.responses.get(status_code, "Unknown")
111
+ response = f"HTTP/1.1 {status_code} {status_text}\r\n"
112
+
113
+ if "Content-Length" not in headers and body:
114
+ headers["Content-Length"] = str(len(body))
115
+
116
+ for key, value in headers.items():
117
+ response += f"{key}: {value}\r\n"
118
+
119
+ response += "\r\n"
120
+ return response.encode('ascii') + (body if isinstance(body, bytes) else body.encode('utf-8'))