feat: base server code
This commit is contained in:
@@ -0,0 +1,2 @@
|
||||
*
|
||||
!.gitignore
|
||||
@@ -1,5 +1,11 @@
|
||||
from src.params import parse_parameters
|
||||
from src.server import StaplerServer
|
||||
|
||||
|
||||
def main():
|
||||
print("Hello from stapler!")
|
||||
params = parse_parameters()
|
||||
server = StaplerServer(params)
|
||||
server.start()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
+4
-1
@@ -4,7 +4,10 @@ version = "0.1.0"
|
||||
description = "Static pages as simple as a gzip file"
|
||||
readme = "README.md"
|
||||
requires-python = ">=3.14"
|
||||
dependencies = []
|
||||
dependencies = [
|
||||
"importlib>=1.0.4",
|
||||
"toml>=0.10.2",
|
||||
]
|
||||
|
||||
[dependency-groups]
|
||||
dev = [
|
||||
|
||||
@@ -0,0 +1,46 @@
|
||||
import argparse
|
||||
import dataclasses
|
||||
import os
|
||||
import os.path
|
||||
|
||||
from . import project
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class Parameters:
|
||||
port: int
|
||||
host: str
|
||||
data_dir: str
|
||||
bind: str
|
||||
|
||||
@classmethod
|
||||
def from_namespace(cls, args: argparse.Namespace) -> "Parameters":
|
||||
return Parameters(**vars(args))
|
||||
|
||||
|
||||
def parse_parameters() -> Parameters:
|
||||
parser = argparse.ArgumentParser(
|
||||
project.get_name(), description=project.get_description()
|
||||
)
|
||||
parser.add_argument(
|
||||
"-p", "--port", type=int, default=8080, help="server port (default: 8080)"
|
||||
)
|
||||
parser.add_argument(
|
||||
"--host",
|
||||
default="localhost",
|
||||
help="server default host (default: localhost)",
|
||||
)
|
||||
parser.add_argument(
|
||||
"-d",
|
||||
"--data-dir",
|
||||
help="directory where files are/will be stored",
|
||||
default=os.path.join(os.getcwd(), "data"),
|
||||
)
|
||||
parser.add_argument(
|
||||
"-b",
|
||||
"--bind",
|
||||
default="0.0.0.0",
|
||||
help="server bind address (default: 0.0.0.0)",
|
||||
)
|
||||
args = parser.parse_args()
|
||||
return Parameters.from_namespace(args)
|
||||
@@ -0,0 +1,40 @@
|
||||
import os.path
|
||||
import toml
|
||||
import typing
|
||||
|
||||
__project_data = None
|
||||
|
||||
|
||||
def __get_project_data() -> None | dict[str, typing.Any]:
|
||||
global __project_data
|
||||
if __project_data is None:
|
||||
pyproject_toml_file = os.path.join(
|
||||
os.path.dirname(__file__), "..", "pyproject.toml"
|
||||
)
|
||||
if os.path.exists(pyproject_toml_file) and os.path.isfile(pyproject_toml_file):
|
||||
try:
|
||||
data = toml.load(pyproject_toml_file)
|
||||
if "project" in data:
|
||||
__project_data = data["project"]
|
||||
except TypeError, toml.TomlDecodeError, FileNotFoundError:
|
||||
pass
|
||||
return __project_data
|
||||
|
||||
|
||||
def __get_str_value(key: str) -> str:
|
||||
project_data = __get_project_data()
|
||||
if project_data is not None and key in project_data:
|
||||
return project_data[key]
|
||||
return "unknown"
|
||||
|
||||
|
||||
def get_version() -> str:
|
||||
return __get_str_value("version")
|
||||
|
||||
|
||||
def get_name() -> str:
|
||||
return __get_str_value("name")
|
||||
|
||||
|
||||
def get_description() -> str:
|
||||
return __get_str_value("description")
|
||||
@@ -0,0 +1,57 @@
|
||||
import http.server
|
||||
import http
|
||||
|
||||
from . import project, params
|
||||
|
||||
|
||||
class _StaplerRequestHandler(http.server.SimpleHTTPRequestHandler):
|
||||
protocol_version = "HTTP/2.0"
|
||||
server_version = "StaplerServer/" + project.get_version()
|
||||
|
||||
def __init__(self, *args, params: params.Parameters, **kwargs):
|
||||
self.default_host = params.host
|
||||
super().__init__(*args, directory=params.data_dir, **kwargs)
|
||||
|
||||
def list_directory(self, *_, **__):
|
||||
"""Disable default directory listing"""
|
||||
self.send_error(http.HTTPStatus.NOT_FOUND, "File not found")
|
||||
|
||||
def do_GET(self):
|
||||
if self.path == "/" and self.get_host() == self.default_host:
|
||||
self.server_index()
|
||||
return
|
||||
super().do_GET()
|
||||
|
||||
def get_host(self) -> str:
|
||||
return self.headers["Host"].split(":")[0]
|
||||
|
||||
def server_index(self):
|
||||
self.send_basic_body(self.server_version)
|
||||
|
||||
def send_basic_body(self, body: str, content_type: str = "text/plain"):
|
||||
encoded: bytes = body.encode()
|
||||
self.send_response(http.HTTPStatus.OK)
|
||||
self.send_header("Content-type", f"{content_type}; charset=UTF-8")
|
||||
self.send_header("Content-Length", str(len(encoded)))
|
||||
self.end_headers()
|
||||
self.wfile.write(encoded)
|
||||
|
||||
|
||||
class StaplerServer:
|
||||
def __init__(self, params: params.Parameters):
|
||||
self.default_host = params.host
|
||||
self.server = http.server.ThreadingHTTPServer(
|
||||
(params.bind, params.port),
|
||||
lambda req, client, server: _StaplerRequestHandler(
|
||||
req, client, server, params=params
|
||||
),
|
||||
)
|
||||
|
||||
def start(self):
|
||||
print(
|
||||
f"{_StaplerRequestHandler.server_version} serving on http://{self.default_host}:{self.server.server_port}..."
|
||||
)
|
||||
try:
|
||||
self.server.serve_forever()
|
||||
except KeyboardInterrupt:
|
||||
pass
|
||||
@@ -2,6 +2,12 @@ version = 1
|
||||
revision = 3
|
||||
requires-python = ">=3.14"
|
||||
|
||||
[[package]]
|
||||
name = "importlib"
|
||||
version = "1.0.4"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/31/77/3781f65cafe55480b56914def99022a5d2965a4bb269655c89ef2f1de3cd/importlib-1.0.4.zip", hash = "sha256:b6ee7066fea66e35f8d0acee24d98006de1a0a8a94a8ce6efe73a9a23c8d9826", size = 7074, upload-time = "2016-08-31T19:40:00.728Z" }
|
||||
|
||||
[[package]]
|
||||
name = "ruff"
|
||||
version = "0.15.10"
|
||||
@@ -28,9 +34,13 @@ wheels = [
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "staple"
|
||||
name = "stapler"
|
||||
version = "0.1.0"
|
||||
source = { virtual = "." }
|
||||
dependencies = [
|
||||
{ name = "importlib" },
|
||||
{ name = "toml" },
|
||||
]
|
||||
|
||||
[package.dev-dependencies]
|
||||
dev = [
|
||||
@@ -39,6 +49,10 @@ dev = [
|
||||
]
|
||||
|
||||
[package.metadata]
|
||||
requires-dist = [
|
||||
{ name = "importlib", specifier = ">=1.0.4" },
|
||||
{ name = "toml", specifier = ">=0.10.2" },
|
||||
]
|
||||
|
||||
[package.metadata.requires-dev]
|
||||
dev = [
|
||||
@@ -46,6 +60,15 @@ dev = [
|
||||
{ name = "ty", specifier = ">=0.0.29" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "toml"
|
||||
version = "0.10.2"
|
||||
source = { registry = "https://pypi.org/simple" }
|
||||
sdist = { url = "https://files.pythonhosted.org/packages/be/ba/1f744cdc819428fc6b5084ec34d9b30660f6f9daaf70eead706e3203ec3c/toml-0.10.2.tar.gz", hash = "sha256:b3bda1d108d5dd99f4a20d24d9c348e91c4db7ab1b749200bded2f839ccbe68f", size = 22253, upload-time = "2020-11-01T01:40:22.204Z" }
|
||||
wheels = [
|
||||
{ url = "https://files.pythonhosted.org/packages/44/6f/7120676b6d73228c96e17f1f794d8ab046fc910d781c8d151120c3f1569e/toml-0.10.2-py2.py3-none-any.whl", hash = "sha256:806143ae5bfb6a3c6e736a764057db0e6a0e05e338b5630894a5f779cabb4f9b", size = 16588, upload-time = "2020-11-01T01:40:20.672Z" },
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "ty"
|
||||
version = "0.0.29"
|
||||
|
||||
Reference in New Issue
Block a user