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():
|
def main():
|
||||||
print("Hello from stapler!")
|
params = parse_parameters()
|
||||||
|
server = StaplerServer(params)
|
||||||
|
server.start()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|||||||
+4
-1
@@ -4,7 +4,10 @@ version = "0.1.0"
|
|||||||
description = "Static pages as simple as a gzip file"
|
description = "Static pages as simple as a gzip file"
|
||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
requires-python = ">=3.14"
|
requires-python = ">=3.14"
|
||||||
dependencies = []
|
dependencies = [
|
||||||
|
"importlib>=1.0.4",
|
||||||
|
"toml>=0.10.2",
|
||||||
|
]
|
||||||
|
|
||||||
[dependency-groups]
|
[dependency-groups]
|
||||||
dev = [
|
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
|
revision = 3
|
||||||
requires-python = ">=3.14"
|
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]]
|
[[package]]
|
||||||
name = "ruff"
|
name = "ruff"
|
||||||
version = "0.15.10"
|
version = "0.15.10"
|
||||||
@@ -28,9 +34,13 @@ wheels = [
|
|||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "staple"
|
name = "stapler"
|
||||||
version = "0.1.0"
|
version = "0.1.0"
|
||||||
source = { virtual = "." }
|
source = { virtual = "." }
|
||||||
|
dependencies = [
|
||||||
|
{ name = "importlib" },
|
||||||
|
{ name = "toml" },
|
||||||
|
]
|
||||||
|
|
||||||
[package.dev-dependencies]
|
[package.dev-dependencies]
|
||||||
dev = [
|
dev = [
|
||||||
@@ -39,6 +49,10 @@ dev = [
|
|||||||
]
|
]
|
||||||
|
|
||||||
[package.metadata]
|
[package.metadata]
|
||||||
|
requires-dist = [
|
||||||
|
{ name = "importlib", specifier = ">=1.0.4" },
|
||||||
|
{ name = "toml", specifier = ">=0.10.2" },
|
||||||
|
]
|
||||||
|
|
||||||
[package.metadata.requires-dev]
|
[package.metadata.requires-dev]
|
||||||
dev = [
|
dev = [
|
||||||
@@ -46,6 +60,15 @@ dev = [
|
|||||||
{ name = "ty", specifier = ">=0.0.29" },
|
{ 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]]
|
[[package]]
|
||||||
name = "ty"
|
name = "ty"
|
||||||
version = "0.0.29"
|
version = "0.0.29"
|
||||||
|
|||||||
Reference in New Issue
Block a user