feat: create server (#41)

Author: choieastsea

src/openstack_mcp_server/tools/compute_tools.py

@@ -1,3 +1,5 @@
+from typing import Any
+
 from fastmcp import FastMCP
 
 from openstack_mcp_server.tools.response.compute import Server
@@ -15,37 +17,73 @@ def register_tools(self, mcp: FastMCP):
         Register Compute-related tools with the FastMCP instance.
         """
 
-        mcp.tool()(self.get_compute_servers)
-        mcp.tool()(self.get_compute_server)
+        mcp.tool()(self.get_servers)
+        mcp.tool()(self.get_server)
+        mcp.tool()(self.create_server)
 
-    def get_compute_servers(self) -> list[Server]:
+    def get_servers(self) -> list[Server]:
         """
-        Get the list of Compute servers by invoking the registered tool.
+        Get the list of Compute servers.
 
-        :return: A list of Server objects representing the Compute servers.
+        :return: A list of Server objects.
         """
-        # Initialize connection
         conn = get_openstack_conn()
-
-        # List the servers
         server_list = []
         for server in conn.compute.servers():
-            server_list.append(
-                Server(name=server.name, id=server.id, status=server.status),
-            )
+            server_list.append(Server(**server))
 
         return server_list
 
-    def get_compute_server(self, id: str) -> Server:
+    def get_server(self, id: str) -> Server:
         """
-        Get a specific Compute server by invoking the registered tool.
+        Get a specific Compute server.
 
         :param id: The ID of the server to retrieve.
-        :return: A Server object representing the Compute server.
+        :return: A Server object.
         """
-        # Initialize connection
         conn = get_openstack_conn()
-
-        # Get a specific server (for example, the first one)
         server = conn.compute.get_server(id)
-        return Server(name=server.name, id=server.id, status=server.status)
+        return Server(**server)
+
+    def create_server(
+        self,
+        name: str,
+        image: str,
+        flavor: int,
+        network: str,
+        key_name: str | None = None,
+        security_groups: list[str] | None = None,
+        user_data: str | None = None,
+    ) -> Server:
+        """
+        Create a new Compute server.
+
+        :param name: The name of the server.
+        :param image: The ID of the image to use.
+        :param flavor: The (integer) ID of the flavor to use.
+        :param network: The ID of the network to attach.
+        :param key_name: The name of the key pair to use.
+        :param security_groups: A list of security group names to attach.
+        :param user_data: User data to pass to the server.
+        :return: A Server object
+        """
+        conn = get_openstack_conn()
+        server_params: dict[str, Any] = {
+            "name": name,
+            "flavorRef": flavor,
+            "imageRef": image,
+            "networks": [{"uuid": network}],
+            "key_name": key_name,
+            "security_groups": security_groups,
+            "user_data": user_data,
+        }
+        server_params = {
+            k: v for k, v in server_params.items() if v is not None
+        }
+
+        resp = conn.compute.create_server(**server_params)
+        # NOTE: The create_server method returns a server object with minimal information.
+        # To get the full server details, we need to fetch it again.
+        server = conn.compute.get_server(resp.id)
+
+        return Server(**server)

src/openstack_mcp_server/tools/response/compute.py

@@ -1,9 +1,34 @@
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict, Field
 
 
-class Server(BaseModel):
-    """A model to represent a Compute server."""
+class Flavor(BaseModel):
+    id: str | None = Field(default=None, exclude=True)
+    name: str = Field(validation_alias="original_name")
+    model_config = ConfigDict(validate_by_name=True)
+
+
+class Image(BaseModel):
+    id: str
+
+
+class ServerIp(BaseModel):
+    addr: str
+    version: int
+    type: str = Field(validation_alias="OS-EXT-IPS:type")
 
+    model_config = ConfigDict(validate_by_name=True)
+
+
+class ServerSecurityGroup(BaseModel):
     name: str
+
+
+class Server(BaseModel):
     id: str
-    status: str
+    name: str
+    status: str | None = None
+    flavor: Flavor | None = None
+    image: Image | None = None
+    addresses: dict[str, list[ServerIp]] | None = None
+    key_name: str | None = None
+    security_groups: list[ServerSecurityGroup] | None = None