feat: activate profile for client

Author: JoJoJoJoJoJoJo

src/mcpm/clients/base.py

@@ -13,6 +13,8 @@
 from ruamel.yaml import YAML
 
 from mcpm.schemas.server_config import ServerConfig, STDIOServerConfig
+from mcpm.utils.config import ROUTER_SERVER_NAME
+from mcpm.utils.router_server import format_server_url
 
 logger = logging.getLogger(__name__)
 
@@ -132,6 +134,30 @@ def is_client_installed(self) -> bool:
         """
         pass
 
+    @abc.abstractmethod
+    def activate_profile(self, profile_name: str, router_config: Dict[str, Any]) -> bool:
+        """
+        Activate a profile in the client config
+
+        Args:
+            profile_name: Name of the profile
+            router_config: Router configuration
+
+        Returns:
+            bool: Success or failure
+        """
+        pass
+
+    @abc.abstractmethod
+    def deactivate_profile(self) -> bool:
+        """
+        Deactivate a profile in the client config
+
+        Returns:
+            bool: Success or failure
+        """
+        pass
+
 
 class JSONClientManager(BaseClientManager):
     """
@@ -350,6 +376,33 @@ def is_client_installed(self) -> bool:
         # Can be overridden by subclasses
         return os.path.isdir(os.path.dirname(self.config_path))
 
+    def activate_profile(self, profile_name: str, router_config: Dict[str, Any]) -> bool:
+        """Activate a profile in the client config
+
+        Args:
+            profile_name: Name of the profile
+
+        Returns:
+            bool: Success or failure
+        """
+        host = router_config["host"]
+        port = router_config["port"]
+        default_base_url = f"http://{host}:{port}/sse"
+
+        server_config = self._format_router_server(profile_name, default_base_url)
+        return self.add_server(server_config)
+
+    def _format_router_server(self, profile_name, base_url) -> ServerConfig:
+        return format_server_url(self.client_key, profile_name, base_url)
+
+    def deactivate_profile(self) -> bool:
+        """Deactivate a profile in the client config
+
+        Returns:
+            bool: Success or failure
+        """
+        return self.remove_server(ROUTER_SERVER_NAME)
+
 
 class YAMLClientManager(BaseClientManager):
     """
@@ -589,3 +642,30 @@ def is_client_installed(self) -> bool:
         """
         # Check if the config directory exists
         return os.path.isdir(os.path.dirname(self.config_path))
+
+    def activate_profile(self, profile_name: str, router_config: Dict[str, Any]) -> bool:
+        """Activate a profile in the client config
+
+        Args:
+            profile_name: Name of the profile
+
+        Returns:
+            bool: Success or failure
+        """
+        host = router_config["host"]
+        port = router_config["port"]
+        default_base_url = f"http://{host}:{port}/sse"
+
+        server_config = self._format_router_server(profile_name, default_base_url)
+        return self.add_server(server_config)
+
+    def _format_router_server(self, profile_name, base_url) -> ServerConfig:
+        return format_server_url(self.client_key, profile_name, base_url)
+
+    def deactivate_profile(self) -> bool:
+        """Deactivate a profile in the client config
+
+        Returns:
+            bool: Success or failure
+        """
+        return self.remove_server(ROUTER_SERVER_NAME)

src/mcpm/clients/client_registry.py

@@ -17,6 +17,7 @@
 from mcpm.clients.managers.fiveire import FiveireManager
 from mcpm.clients.managers.goose import GooseClientManager
 from mcpm.clients.managers.windsurf import WindsurfManager
+from mcpm.utils.config import ConfigManager
 from mcpm.utils.scope import CLIENT_PREFIX, PROFILE_PREFIX
 
 logger = logging.getLogger(__name__)
@@ -205,3 +206,37 @@ def determine_active_scope(cls) -> str | None:
         if client:
             return f"{CLIENT_PREFIX}{client}"
         return None
+
+    @classmethod
+    def activate_profile(cls, client_name: str, profile_name: str) -> bool:
+        """
+        Activate a profile in the client config
+
+        Args:
+            client_name: Name of the client
+            profile_name: Name of the profile
+
+        Returns:
+            bool: Success or failure
+        """
+        router_config = ConfigManager().get_router_config()
+        client = cls.get_client_manager(client_name)
+        if client is None:
+            return False
+        return client.activate_profile(profile_name, router_config)
+
+    @classmethod
+    def deactivate_profile(cls, client_name: str) -> bool:
+        """
+        Deactivate a profile in the client config
+
+        Args:
+            profile_name: Name of the profile
+
+        Returns:
+            bool: Success or failure
+        """
+        client = cls.get_client_manager(client_name)
+        if client is None:
+            return False
+        return client.deactivate_profile()

src/mcpm/clients/managers/claude_desktop.py

@@ -7,6 +7,8 @@
 from typing import Any, Dict
 
 from mcpm.clients.base import JSONClientManager
+from mcpm.schemas.server_config import ServerConfig
+from mcpm.utils.router_server import format_server_url_with_proxy_param
 
 logger = logging.getLogger(__name__)
 
@@ -111,6 +113,9 @@ def is_server_disabled(self, server_name: str) -> bool:
         config = self._load_config()
         return "disabledServers" in config and server_name in config["disabledServers"]
 
+    def _format_router_server(self, profile_name, base_url) -> ServerConfig:
+        return format_server_url_with_proxy_param(self.client_key, profile_name, base_url)
+
     # Uses base class implementation of remove_server
 
     # Uses base class implementation of get_server

src/mcpm/clients/managers/continue_extension.py

@@ -10,6 +10,7 @@
 
 from mcpm.clients.base import YAMLClientManager
 from mcpm.schemas.server_config import ServerConfig, STDIOServerConfig
+from mcpm.utils.router_server import format_server_url_with_proxy_headers
 
 logger = logging.getLogger(__name__)
 
@@ -203,3 +204,6 @@ def from_client_format(self, server_name: str, client_config: Dict[str, Any]) ->
         }
         server_data.update(client_config)
         return TypeAdapter(ServerConfig).validate_python(server_data)
+
+    def _format_router_server(self, profile_name, base_url) -> ServerConfig:
+        return format_server_url_with_proxy_headers(self.client_key, profile_name, base_url)

src/mcpm/commands/profile.py

@@ -5,6 +5,7 @@
 from mcpm.clients.client_registry import ClientRegistry
 from mcpm.profile.profile_config import ProfileConfigManager
 from mcpm.schemas.server_config import STDIOServerConfig
+from mcpm.utils.config import ConfigManager
 
 profile_config_manager = ProfileConfigManager()
 console = Console()
@@ -19,9 +20,9 @@ def profile():
 
 @click.command()
 @click.argument("profile_name")
-@click.option("--client", "-c", default="client", help="Client of the profile")
+@click.option("--client", "-c", help="Client of the profile")
 @click.help_option("-h", "--help")
-def activate(profile_name, client):
+def activate(profile_name, client=None):
     """Activate a profile.
 
     Sets the specified profile as the active profile.
@@ -33,18 +34,37 @@ def activate(profile_name, client):
 
     # Set the active profile
     client_registry = ClientRegistry()
-    if client_registry.set_active_profile(profile_name):
+    config_manager = ConfigManager()
+
+    if client:
+        console.print(f"[bold cyan]Activating profile '{profile_name}' in client '{client}'...[/]")
+        client_manager = ClientRegistry.get_client_manager(client)
+        if client_manager is None:
+            console.print(f"[bold red]Error:[/] Client '{client}' not found.")
+            return
+        success = client_manager.activate_profile(profile_name, config_manager.get_router_config())
+    else:
+        client = ClientRegistry.get_active_client()
+        if client is None:
+            console.print("[bold yellow]No active client found.[/]\n")
+            return
+        console.print(f"[bold cyan]Activating profile '{profile_name}' in active client '{client}'...[/]")
+        client_manager = ClientRegistry.get_client_manager(client)
+        if client_manager is None:
+            console.print(f"[bold red]Error:[/] Client '{client}' not found.")
+            return
+        success = client_manager.activate_profile(profile_name, config_manager.get_router_config())
+    if success:
+        client_registry.set_active_profile(profile_name)
         console.print(f"\n[green]Profile '{profile_name}' activated successfully.[/]\n")
     else:
         console.print(f"[bold red]Error:[/] Failed to activate profile '{profile_name}'.")
 
-    # TODO: add url to the client config
-
 
 @click.command()
-@click.option("--client", "-c", default="client", help="Client of the profile")
+@click.option("--client", "-c", help="Client of the profile")
 @click.help_option("-h", "--help")
-def deactivate(client):
+def deactivate(client=None):
     """Deactivate a profile.
 
     Unsets the active profile.
@@ -56,12 +76,30 @@ def deactivate(client):
         return
     console.print(f"\n[green]Deactivating profile '{active_profile}'...[/]")
     client_registry = ClientRegistry()
-    if client_registry.set_active_profile(None):
-        console.print(f"\n[green]Profile '{active_profile}' deactivated successfully.[/]\n")
-    else:
-        console.print(f"[bold red]Error:[/] Failed to deactivate profile '{active_profile}'.")
 
-    # TODO: remove url from the client config
+    if client:
+        console.print(f"[bold cyan]Deactivating profile '{active_profile}' in client '{client}'...[/]")
+        client_manager = ClientRegistry.get_client_manager(client)
+        if client_manager is None:
+            console.print(f"[bold red]Error:[/] Client '{client}' not found.")
+            return
+        success = client_manager.deactivate_profile()
+    else:
+        client = ClientRegistry.get_active_client()
+        if client is None:
+            console.print("[bold yellow]No active client found.[/]\n")
+            return
+        console.print(f"[bold cyan]Deactivating profile '{active_profile}' in active client '{client}'...[/]")
+        client_manager = ClientRegistry.get_client_manager(client)
+        if client_manager is None:
+            console.print(f"[bold red]Error:[/] Client '{client}' not found.")
+            return
+        success = client_manager.deactivate_profile()
+    if success:
+        client_registry.set_active_profile(None)
+        console.print(f"\n[yellow]Profile '{active_profile}' deactivated successfully.[/]\n")
+    else:
+        console.print(f"[bold red]Error:[/] Failed to deactivate profile '{active_profile}' in client '{client}'.")
 
 
 @profile.command(name="ls")