Create dedicated server entry point to prevent crashes

codeflash/lsp/server.py

@@ -3,7 +3,7 @@
 from pathlib import Path
 from typing import TYPE_CHECKING, Any
 
-from lsprotocol.types import INITIALIZE
+from lsprotocol.types import INITIALIZE, MessageType, LogMessageParams
 from pygls import uris
 from pygls.protocol import LanguageServerProtocol, lsp_method
 from pygls.server import LanguageServer
@@ -55,3 +55,24 @@ def initialize_optimizer(self, config_file: Path) -> None:
         args.no_pr = True  # LSP server should not create PRs
         args = process_pyproject_config(args)
         self.optimizer = Optimizer(args)
+
+    def show_message_log(self, message: str, message_type: str) -> None:
+        """Send a log message to the client's output channel.
+
+        Args:
+            message: The message to log
+            message_type: String type - "Info", "Warning", "Error", or "Log"
+        """
+        # Convert string message type to LSP MessageType enum
+        type_mapping = {
+            "Info": MessageType.Info,
+            "Warning": MessageType.Warning, 
+            "Error": MessageType.Error,
+            "Log": MessageType.Log
+        }
+
+        lsp_message_type = type_mapping.get(message_type, MessageType.Info)
+
+        # Send log message to client (appears in output channel)
+        log_params = LogMessageParams(type=lsp_message_type, message=message)
+        self.lsp.notify("window/logMessage", log_params)

codeflash/lsp/server_entry.py

@@ -0,0 +1,42 @@
+"""
+This script is the dedicated entry point for the Codeflash Language Server.
+It initializes the server and redirects its logs to stderr so that the
+VS Code client can display them in the output channel.
+
+This script is run by the VS Code extension and is not intended to be
+executed directly by users.
+"""
+import logging
+import sys
+import os
+from codeflash.lsp.beta import server
+
+# Configure logging to stderr for VS Code output channel
+def setup_logging():
+    # Clear any existing handlers to prevent conflicts
+    root_logger = logging.getLogger()
+    root_logger.handlers.clear()
+
+    # Set up stderr handler for VS Code output channel with [LSP-Server] prefix
+    handler = logging.StreamHandler(sys.stderr)
+    handler.setFormatter(logging.Formatter("[LSP-Server] %(asctime)s [%(levelname)s]: %(message)s"))
+
+    # Configure root logger
+    root_logger.addHandler(handler)
+    root_logger.setLevel(logging.INFO)
+
+    # Also configure the pygls logger specifically
+    pygls_logger = logging.getLogger('pygls')
+    pygls_logger.setLevel(logging.INFO)
+
+    return root_logger
+
+# Set up logging
+log = setup_logging()
+
+# Silence the console module to prevent stdout pollution
+from codeflash.cli_cmds.console import console
+console.quiet = True
+
+log.info("Starting Codeflash Language Server...")
+server.start_io()