feat: support alias for profile

Author: JoJoJoJoJoJoJo

src/mcpm/clients/base.py

@@ -8,7 +8,7 @@
 import os
 import platform
 import re
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional, Union, override
 
 from pydantic import TypeAdapter
 from ruamel.yaml import YAML
@@ -136,13 +136,14 @@ def is_client_installed(self) -> bool:
         pass
 
     @abc.abstractmethod
-    def activate_profile(self, profile_name: str, router_config: Dict[str, Any]) -> bool:
+    def activate_profile(self, profile_name: str, router_config: Dict[str, Any], alias_name: str | None = None) -> bool:
         """
         Activate a profile in the client config
 
         Args:
             profile_name: Name of the profile
             router_config: Router configuration
+            alias_name: Alias name for the router in client config
 
         Returns:
             bool: Success or failure
@@ -406,11 +407,13 @@ 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:
+    def activate_profile(self, profile_name: str, router_config: Dict[str, Any], alias_name: str | None = None) -> bool:
         """Activate a profile in the client config
 
         Args:
             profile_name: Name of the profile
+            router_config: Router configuration
+            alias_name: Alias name for the router in client config
 
         Returns:
             bool: Success or failure
@@ -419,11 +422,11 @@ def activate_profile(self, profile_name: str, router_config: Dict[str, Any]) ->
         port = router_config["port"]
         default_base_url = f"http://{host}:{port}/sse"
 
-        server_config = self._format_router_server(profile_name, default_base_url)
+        server_config = self._format_router_server(profile_name, default_base_url, alias_name)
         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 _format_router_server(self, profile_name, base_url, alias_name: str | None = None) -> ServerConfig:
+        return format_server_url(self.client_key, profile_name, base_url, alias_name)
 
     def deactivate_profile(self) -> bool:
         """Deactivate a profile in the client config
@@ -673,7 +676,7 @@ 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:
+    def activate_profile(self, profile_name: str, router_config: Dict[str, Any], alias_name: str | None = None) -> bool:
         """Activate a profile in the client config
 
         Args:
@@ -686,12 +689,13 @@ def activate_profile(self, profile_name: str, router_config: Dict[str, Any]) ->
         port = router_config["port"]
         default_base_url = f"http://{host}:{port}/sse"
 
-        server_config = self._format_router_server(profile_name, default_base_url)
+        server_config = self._format_router_server(profile_name, default_base_url, alias_name)
         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 _format_router_server(self, profile_name, base_url, server_name: str | None = None) -> ServerConfig:
+        return format_server_url(self.client_key, profile_name, base_url, server_name)
 
+    @override
     def deactivate_profile(self) -> bool:
         """Deactivate a profile in the client config
 

src/mcpm/commands/target_operations/add.py

@@ -523,11 +523,11 @@ def _replace_argument_variables(value: str, prev_value: str, variables: dict) ->
 
 
 def add_profile_to_client(profile_name: str, client: str, alias: str | None = None, force: bool = False):
-    if not force and not Confirm.ask(f"Add this profile to {client}{' as ' + alias if alias else ''}?"):
+    if not force and not Confirm.ask(f"Add this profile {profile_name} to {client}{' as ' + alias if alias else ''}?"):
         console.print("[yellow]Operation cancelled.[/]")
         return
 
-    success = client_add_profile(profile_name, client)
+    success = client_add_profile(profile_name, client, alias)
     if success:
         console.print(f"[bold green]Successfully added profile {profile_name} to {client}![/]")
     else:

src/mcpm/commands/target_operations/common.py

@@ -94,7 +94,7 @@ def profile_get_server(profile: str, server: str) -> ServerConfig | None:
     return profile_manager.get_profile_server(profile, server)
 
 
-def client_add_profile(profile_name: str, client: str) -> bool:
+def client_add_profile(profile_name: str, client: str, alias_name: str | None = None) -> bool:
     client_manager = ClientRegistry.get_client_manager(client)
     if not client_manager:
         console.print(f"[bold red]Error:[/] Client '{client}' not found.")
@@ -104,5 +104,5 @@ def client_add_profile(profile_name: str, client: str) -> bool:
         console.print("[bold red]Error:[/] Router config not found.")
         return False
 
-    success = client_manager.activate_profile(profile_name, router_config)
+    success = client_manager.activate_profile(profile_name, router_config, alias_name)
     return success

src/mcpm/utils/router_server.py

@@ -2,25 +2,29 @@
 from mcpm.core.schema import ServerConfig, SSEServerConfig, STDIOServerConfig
 
 
-def format_server_url(client: str, profile: str, router_url: str) -> ServerConfig:
+def format_server_url(client: str, profile: str, router_url: str, server_name: str | None = None) -> ServerConfig:
     return SSEServerConfig(
-        name=ROUTER_SERVER_NAME,
+        name=server_name if server_name else ROUTER_SERVER_NAME,
         url=f"{router_url}?/client={client}&profile={profile}",
     )
 
 
-def format_server_url_with_proxy_param(client: str, profile: str, router_url: str) -> ServerConfig:
+def format_server_url_with_proxy_param(
+    client: str, profile: str, router_url: str, server_name: str | None = None
+) -> ServerConfig:
     result = STDIOServerConfig(
-        name=ROUTER_SERVER_NAME,
+        name=server_name if server_name else ROUTER_SERVER_NAME,
         command="uvx",
         args=["mcp-proxy", f"{router_url}?/client={client}&profile={profile}"],
     )
     return result
 
 
-def format_server_url_with_proxy_headers(client: str, profile: str, router_url: str) -> ServerConfig:
+def format_server_url_with_proxy_headers(
+    client: str, profile: str, router_url: str, server_name: str | None = None
+) -> ServerConfig:
     result = STDIOServerConfig(
-        name=ROUTER_SERVER_NAME,
+        name=server_name if server_name else ROUTER_SERVER_NAME,
         command="uvx",
         args=["mcp-proxy", router_url, "--headers", "profile", profile],
     )