golf-mcp 0.2.9__py3-none-any.whl → 0.2.10__py3-none-any.whl

golf/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.2.9"
+__version__ = "0.2.10"
 
 # Import endpoints with fallback for dev mode
 try:

golf/core/builder.py

@@ -581,6 +581,70 @@ class CodeGenerator:
             # Default to older behavior for safety
             return False
 
+    def _generate_startup_section(self, project_path: Path) -> list[str]:
+        """Generate code section for startup.py execution during server runtime."""
+        startup_path = project_path / "startup.py"
+
+        if not startup_path.exists():
+            return []
+
+        return [
+            "",
+            "# Execute startup script for loading secrets and initialization",
+            "import importlib.util",
+            "import sys",
+            "import os",
+            "from pathlib import Path",
+            "",
+            "# Look for startup.py in the same directory as this server.py",
+            "startup_path = Path(__file__).parent / 'startup.py'",
+            "if startup_path.exists():",
+            "    try:",
+            "        # Save original environment for restoration",
+            "        try:",
+            "            original_dir = os.getcwd()",
+            "        except (FileNotFoundError, OSError):",
+            "            # Use server directory as fallback",
+            "            original_dir = str(Path(__file__).parent)",
+            "            os.chdir(original_dir)",
+            "        original_path = sys.path.copy()",
+            "        ",
+            "        # Set context for startup script execution",
+            "        script_dir = str(startup_path.parent)",
+            "        os.chdir(script_dir)",
+            "        sys.path.insert(0, script_dir)",
+            "        ",
+            "        # Debug output for startup script development",
+            "        if os.environ.get('GOLF_DEBUG'):",
+            "            print(f'Executing startup script: {startup_path}')",
+            "            print(f'Working directory: {os.getcwd()}')",
+            "            print(f'Python path: {sys.path[:3]}...')",  # Show first 3 entries
+            "        ",
+            "        # Load and execute startup script",
+            "        spec = importlib.util.spec_from_file_location('startup', startup_path)",
+            "        if spec and spec.loader:",
+            "            startup_module = importlib.util.module_from_spec(spec)",
+            "            spec.loader.exec_module(startup_module)",
+            "        else:",
+            "            print('Warning: Could not load startup.py', file=sys.stderr)",
+            "        ",
+            "    except Exception as e:",
+            "        import traceback",
+            "        print(f'Warning: Startup script execution failed: {e}', file=sys.stderr)",
+            "        print(traceback.format_exc(), file=sys.stderr)",
+            "        # Continue server startup despite script failure",
+            "        ",
+            "    finally:",
+            "        # Always restore original environment",
+            "        try:",
+            "            os.chdir(original_dir)",
+            "            sys.path[:] = original_path",
+            "        except Exception:",
+            "            # If directory restoration fails, at least fix the path",
+            "            sys.path[:] = original_path",
+            "",
+        ]
+
     def _generate_server(self) -> None:
         """Generate the main server entry point."""
         server_file = self.output_dir / "server.py"
@@ -935,6 +999,9 @@ class CodeGenerator:
             "",
         ]
 
+        # Generate startup section
+        startup_section = self._generate_startup_section(self.project_path)
+
         # OpenTelemetry setup code will be handled through imports and lifespan
 
         # Add auth setup code if auth is configured
@@ -1145,12 +1212,13 @@ class CodeGenerator:
             ]
 
         # Combine all sections
-        # Order: imports, env_section, auth_setup, server_code (mcp init),
+        # Order: imports, env_section, startup_section, auth_setup, server_code (mcp init),
         # early_telemetry_init, early_metrics_init, component_registrations,
         # metrics_route_code, health_check_code, main_code (run block)
         code = "\n".join(
             imports
             + env_section
+            + startup_section
             + auth_setup_code
             + server_code_lines
             + early_telemetry_init
@@ -1352,6 +1420,13 @@ def build_project(
     generator = CodeGenerator(project_path, settings, output_dir, build_env=build_env, copy_env=copy_env)
     generator.generate()
 
+    # Copy startup.py to output directory if it exists (after server generation)
+    startup_path = project_path / "startup.py"
+    if startup_path.exists():
+        dest_path = output_dir / "startup.py"
+        shutil.copy2(startup_path, dest_path)
+        console.print(get_status_text("success", "Startup script copied to build directory"))
+
     # Platform registration (only for prod builds)
     if build_env == "prod":
         console.print()

{golf_mcp-0.2.9.dist-info → golf_mcp-0.2.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: golf-mcp
-Version: 0.2.9
+Version: 0.2.10
 Summary: Framework for building MCP servers
 Author-email: Antoni Gmitruk <antoni@golf.dev>
 License-Expression: Apache-2.0

{golf_mcp-0.2.9.dist-info → golf_mcp-0.2.10.dist-info}/RECORD

@@ -1,4 +1,4 @@
-golf/__init__.py,sha256=CB9wu2mcYCSJopA-uO1-_87ebOSTDcXJ8uOktQ4va20,306
+golf/__init__.py,sha256=KrLUEDZhh_8Gu_JcwZgGUhFN9JR6Sjqt8v46gVPBsmE,307
 golf/_endpoints.py,sha256=-mvAmjx3YtqfAdajO13Kv7LKVBMdqsKqX0_3aCmCK4I,317
 golf/_endpoints.py.in,sha256=MeqcSF6xinnPknpBGF26xA9FYmSYKSWqCFXOSDlXiOA,216
 golf/_endpoints_fallback.py,sha256=CD6m45Ams1A9HVKowY8dCFUDMiFmJ8ZWSwHCENvU5u4,386
@@ -16,7 +16,7 @@ golf/commands/build.py,sha256=sLq9lSW4naq2vIlBreKI5SGnviQrhBWw13zfOZOKhuM,2293
 golf/commands/init.py,sha256=KkAg_3-KxBDFOcZqUHqcPAkDipykFVaLWpQ2tydnVPk,9617
 golf/commands/run.py,sha256=a8GSwLt6E2cUJespv-u3jbD-rNUMHqF3VArD1uXV-Vk,4299
 golf/core/__init__.py,sha256=4bKeskJ2fPaZqkz2xQScSa3phRLLrmrczwSL632jv-o,52
-golf/core/builder.py,sha256=Tx6AtWnnasY_Rfu1CEdhE7y3hZdN0RuhwLevJU9vZgo,66355
+golf/core/builder.py,sha256=-iUhj1G5SKSE9kG9Y2glxQPp3b45fqXVrYi3gQGmgbs,69963
 golf/core/builder_auth.py,sha256=SXPCpc5ipntoNqIAIA2ZCeGmEua6QVs7yC3MDtGKAro,8224
 golf/core/builder_metrics.py,sha256=j6Gtgd867o46JbDfSNGNsHt1QtV1XHKUJs1z8r4siQM,8830
 golf/core/builder_telemetry.py,sha256=86bp7UlMUN6JyQRrZ5EizovP6AJ_q65OANJTeJXDIKc,3421
@@ -48,9 +48,9 @@ golf/utilities/__init__.py,sha256=X9iY9yi3agz1GVcn8-qWeOCt8CSSsruHxqPNtiF63TY,53
 golf/utilities/context.py,sha256=DGGvhVe---QMhy0wtdWhNp-_WVk1NvAcOFn0uBKBpYo,1579
 golf/utilities/elicitation.py,sha256=MParZZZsY45s70-KXduHa6IvpWXnLW2FCPfrGijMaHs,5223
 golf/utilities/sampling.py,sha256=88nDv-trBE4gZQbcnMjXl3LW6TiIhv5zR_cuEIGjaIM,7233
-golf_mcp-0.2.9.dist-info/licenses/LICENSE,sha256=5_j2f6fTJmvfmUewzElhkpAaXg2grVoxKouOA8ihV6E,11348
-golf_mcp-0.2.9.dist-info/METADATA,sha256=67-svZP_no-mw7BBDGwoQJJ_RCcqCQH5uEAk7jfqUF0,9370
-golf_mcp-0.2.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-golf_mcp-0.2.9.dist-info/entry_points.txt,sha256=5y7rHYM8jGpU-nfwdknCm5XsApLulqsnA37MO6BUTYg,43
-golf_mcp-0.2.9.dist-info/top_level.txt,sha256=BQToHcBUufdyhp9ONGMIvPE40jMEtmI20lYaKb4hxOg,5
-golf_mcp-0.2.9.dist-info/RECORD,,
+golf_mcp-0.2.10.dist-info/licenses/LICENSE,sha256=5_j2f6fTJmvfmUewzElhkpAaXg2grVoxKouOA8ihV6E,11348
+golf_mcp-0.2.10.dist-info/METADATA,sha256=4YRKbcRjyZfBFPFcnzJ5ZV8hU3enJEVmQy2VumwVlpc,9371
+golf_mcp-0.2.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+golf_mcp-0.2.10.dist-info/entry_points.txt,sha256=5y7rHYM8jGpU-nfwdknCm5XsApLulqsnA37MO6BUTYg,43
+golf_mcp-0.2.10.dist-info/top_level.txt,sha256=BQToHcBUufdyhp9ONGMIvPE40jMEtmI20lYaKb4hxOg,5
+golf_mcp-0.2.10.dist-info/RECORD,,