cleanup folders

Author: bordumb

Makefile

@@ -0,0 +1,25 @@
+.PHONY: proto run build install test clean
+
+proto:
+	poetry run python -m grpc_tools.protoc -I./proto \
+	  --python_out=./src/boot_python/generated \
+	  --grpc_python_out=./src/boot_python/generated \
+	  ./proto/plugin.proto
+	# Patch absolute import to relative so package imports work
+	sed -i '' 's/^import plugin_pb2 as plugin__pb2/from . import plugin_pb2 as plugin__pb2/' \
+	  src/boot_python/generated/plugin_pb2_grpc.py
+
+run:
+	poetry run boot-python 2>/dev/null | head -1
+
+build: proto
+	poetry build
+
+install: build
+	pipx install dist/boot_python-*.whl
+
+test:
+	poetry run pytest -q
+
+clean:
+	rm -rf dist build .pytest_cache .ruff_cache

poetry.lock

@@ -1,37 +1,37 @@
-# boot-python/pyproject.toml
 [tool.poetry]
 name = "boot-python"
-version = "0.1.0-alpha"
-description = "A prompt-provider plugin for generating Python projects with boot-code."
+version = "0.1.1-alpha"
+description = "Prompt-provider plugin for boot-code (Python)."
 authors = ["Bordumb <bordumbb@gmail.com>"]
 readme = "README.md"
-# Add this line to tell Poetry where to find the package source code.
-packages = [{include = "src"}]
+packages = [
+  { include = "boot_python", from = "src" }
+]
+
+# Include runtime assets (prompts) and optionally proto (dev)
+include = [
+  { path = "prompts/**/*", format = "sdist" },
+  { path = "prompts/**/*", format = "wheel" },
+  # proto is not needed at runtime; include only if you want it in sdists:
+  { path = "proto/plugin.proto", format = "sdist" }
+]
 
 [tool.poetry.dependencies]
-# This section lists the libraries your plugin needs to run.
-python = ">=3.11"
-# Core libraries for the gRPC server functionality.
+python = "^3.9"
 grpcio = "^1.74.0"
 protobuf = "==6.31.1"
-# Library for parsing the spec.toml file content.
-toml = "^0.10.2"
+tomli = { version = "^2.0.1", python = "<3.11" }  # for TOML parsing on Py<=3.10
 
 [tool.poetry.group.dev.dependencies]
-# This section lists tools for development, like compiling protobuf files.
 grpcio-tools = "^1.74.0"
-pytest = ">=7.0.0"
-pytest-asyncio = ">=0.21.0"
-black = ">=24.0.0"
-mypy = ">=1.0.0"
-ruff = ">=0.1.0"
+pytest = "^8.0.0"
 
 [tool.poetry.scripts]
-# This defines the entry point for the executable. When a user runs `boot-python`,
-# it will execute the `main` function inside the `src/main.py` file.
-# This is how `boot-code` will start your plugin.
-boot-python = "src.main:main"
+boot-python = "boot_python.main:main"
 
 [build-system]
-requires = ["poetry-core"]
+requires = ["poetry-core>=1.9.0"]
 build-backend = "poetry.core.masonry.api"
+
+[tool.pytest.ini_options]
+pythonpath = ["src"]

pyproject.toml

@@ -1,8 +1,28 @@
 # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
+import warnings
 
-import plugin_pb2 as plugin__pb2
+from . import plugin_pb2 as plugin__pb2
+
+GRPC_GENERATED_VERSION = '1.74.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in plugin_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
 
 
 class BootCodePluginStub(object):
@@ -18,7 +38,7 @@ def __init__(self, channel):
                 '/plugin.BootCodePlugin/GetPromptComponents',
                 request_serializer=plugin__pb2.GetPromptComponentsRequest.SerializeToString,
                 response_deserializer=plugin__pb2.GetPromptComponentsResponse.FromString,
-                )
+                _registered_method=True)
 
 
 class BootCodePluginServicer(object):
@@ -42,6 +62,7 @@ def add_BootCodePluginServicer_to_server(servicer, server):
     generic_handler = grpc.method_handlers_generic_handler(
             'plugin.BootCodePlugin', rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('plugin.BootCodePlugin', rpc_method_handlers)
 
 
  # This class is part of an EXPERIMENTAL API.
@@ -59,8 +80,18 @@ def GetPromptComponents(request,
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/plugin.BootCodePlugin/GetPromptComponents',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/plugin.BootCodePlugin/GetPromptComponents',
             plugin__pb2.GetPromptComponentsRequest.SerializeToString,
             plugin__pb2.GetPromptComponentsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

src/boot_python/__init__.py

@@ -0,0 +1,66 @@
+import logging
+import os
+import signal
+import socket
+import sys
+import time
+from concurrent import futures
+
+import grpc
+
+from boot_python.generated import plugin_pb2_grpc
+from boot_python.server import BootPluginServicer
+
+# All logs -> STDERR; keep STDOUT pristine for handshake
+logging.basicConfig(stream=sys.stderr, level=logging.INFO, format="%(levelname)s %(message)s")
+
+
+def _pick_loopback_port() -> tuple[str, int]:
+    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    s.bind(("127.0.0.1", 0))
+    host, port = s.getsockname()
+    s.close()
+    return host, port
+
+
+def main() -> None:
+    # Start gRPC server
+    host, port = _pick_loopback_port()
+    server = grpc.server(futures.ThreadPoolExecutor(max_workers=4))
+    plugin_pb2_grpc.add_BootCodePluginServicer_to_server(BootPluginServicer(), server)
+    server.add_insecure_port(f"{host}:{port}")
+
+    # Print handshake to STDOUT only (no extra whitespace/newlines!)
+    # Format: 1|1|tcp|HOST:PORT|grpc
+    print(f"1|1|tcp|{host}:{port}|grpc", end="", flush=True)
+
+    server.start()
+    logging.info("boot-python started on %s:%d", host, port)
+    
+   # Test fast-exit: if set, skip the blocking loop so imports/patches won't hang.
+    if os.environ.get("BOOT_PYTHON_TEST_EXIT_AFTER_HANDSHAKE") == "1":
+        logging.info("Exiting immediately due to BOOT_PYTHON_TEST_EXIT_AFTER_HANDSHAKE=1")
+        server.stop(grace=None)
+        logging.info("boot-python stopped")
+        return
+
+    # Graceful shutdown
+    stop_event = {"stop": False}
+
+    def _sigterm(_sig, _frm):
+        logging.info("Received shutdown signal")
+        stop_event["stop"] = True
+        server.stop(grace=None)
+
+    signal.signal(signal.SIGINT, _sigterm)
+    signal.signal(signal.SIGTERM, _sigterm)
+
+    try:
+        while not stop_event["stop"]:
+            time.sleep(0.2)
+    finally:
+        logging.info("boot-python stopped")
+
+
+if __name__ == "__main__":
+    main()

src/boot_python/generated/__init__.py

@@ -0,0 +1,66 @@
+from __future__ import annotations
+
+import io
+import logging
+import os
+from pathlib import Path
+from typing import Dict, Tuple
+
+import grpc
+
+try:
+    import tomllib  # Py>=3.11
+except ModuleNotFoundError:  # pragma: no cover
+    import tomli as tomllib  # Py<=3.10
+
+from boot_python.generated import plugin_pb2, plugin_pb2_grpc
+
+PROMPTS_DIR = Path(__file__).resolve().parent.parent.parent / "prompts"
+
+
+def _read_text(p: Path) -> str:
+    with p.open("r", encoding="utf-8") as f:
+        return f.read()
+
+
+def _load_prompts() -> Dict[str, str]:
+    files = ["base_instructions.txt", "language_rules.txt", "review_instructions.txt", "README.md.template"]
+    components: Dict[str, str] = {}
+    for name in files:
+        fp = PROMPTS_DIR / name
+        if fp.exists():
+            components[name] = _read_text(fp)
+        else:
+            logging.warning("Prompt file missing: %s", fp)
+    return components
+
+
+def _derive_user_spec_prompt(spec_toml: str) -> str:
+    """
+    Minimal, safe TOML decode to craft a user-specific prompt.
+    Keep it defensive: if TOML parse fails, fall back gracefully.
+    """
+    try:
+        data = tomllib.loads(spec_toml)
+        project = data.get("project", {})
+        name = project.get("name", "unknown-project")
+        descr = project.get("description", "")
+        lang  = project.get("language", "python")
+        return f"User requests a {lang} project named '{name}'. Description: {descr}"
+    except Exception as e:  # don’t fail plugin on TOML issues
+        logging.warning("Failed to parse spec TOML: %s", e)
+        return "User requests a python project. Description: (unavailable)"
+
+
+class BootPluginServicer(plugin_pb2_grpc.BootCodePluginServicer):
+    def GetPromptComponents(
+        self,
+        request: plugin_pb2.GetPromptComponentsRequest,
+        context: grpc.ServicerContext,
+    ) -> plugin_pb2.GetPromptComponentsResponse:
+        components = _load_prompts()
+        user_spec_prompt = _derive_user_spec_prompt(request.spec_toml_content or "")
+        return plugin_pb2.GetPromptComponentsResponse(
+            components=components,
+            user_spec_prompt=user_spec_prompt,
+        )