diff --git a/data/.gitignore b/data/.gitignore new file mode 100644 index 0000000..c96a04f --- /dev/null +++ b/data/.gitignore @@ -0,0 +1,2 @@ +* +!.gitignore \ No newline at end of file diff --git a/main.py b/main.py index 084d7cd..7619a41 100644 --- a/main.py +++ b/main.py @@ -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__": diff --git a/pyproject.toml b/pyproject.toml index 58a6ca5..aad558a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -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 = [ diff --git a/src/__init__.py b/src/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/src/params.py b/src/params.py new file mode 100644 index 0000000..792c3b5 --- /dev/null +++ b/src/params.py @@ -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) diff --git a/src/project.py b/src/project.py new file mode 100644 index 0000000..0bd088b --- /dev/null +++ b/src/project.py @@ -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") diff --git a/src/server.py b/src/server.py new file mode 100644 index 0000000..860d048 --- /dev/null +++ b/src/server.py @@ -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 diff --git a/uv.lock b/uv.lock index bbb8eba..a9d7edd 100644 --- a/uv.lock +++ b/uv.lock @@ -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"