Reactivate agent tool

Signed-off-by: Marek Safarik <[email protected]>

Author: Marek Safarik

backend/client.go

@@ -1,8 +1,10 @@
 package main
 
 import (
+	"bytes"
 	"crypto/tls"
 	"crypto/x509"
+	"encoding/json"
 	"fmt"
 	"log"
 	"net/http"
@@ -86,8 +88,34 @@ func (kc *KeylimeClient) Get(endpoint string) (*http.Response, error) {
 
 func (kc *KeylimeClient) Post(endpoint string, body interface{}) (*http.Response, error) {
 	url := fmt.Sprintf("%s/%s/%s", kc.baseURL, kc.apiVersion, strings.TrimPrefix(endpoint, "/"))
-	// TODO: implement body marshaling
-	return kc.httpClient.Post(url, "application/json", nil)
+	var buf bytes.Buffer
+	if body != nil {
+		if err := json.NewEncoder(&buf).Encode(body); err != nil {
+			return nil, fmt.Errorf("failed to marshal body: %w", err)
+		}
+	}
+	req, err := http.NewRequest("POST", url, &buf)
+	if err != nil {
+		return nil, err
+	}
+	req.Header.Set("Content-Type", "application/json")
+	return kc.httpClient.Do(req)
+}
+
+func (kc *KeylimeClient) Put(endpoint string, body interface{}) (*http.Response, error) {
+	url := fmt.Sprintf("%s/%s/%s", kc.baseURL, kc.apiVersion, strings.TrimPrefix(endpoint, "/"))
+	var buf bytes.Buffer
+	if body != nil {
+		if err := json.NewEncoder(&buf).Encode(body); err != nil {
+			return nil, fmt.Errorf("failed to marshal body: %w", err)
+		}
+	}
+	req, err := http.NewRequest("PUT", url, &buf)
+	if err != nil {
+		return nil, err
+	}
+	req.Header.Set("Content-Type", "application/json")
+	return kc.httpClient.Do(req)
 }
 
 func (kc *KeylimeClient) Delete(endpoint string) (*http.Response, error) {

backend/main.go

@@ -49,6 +49,7 @@ func main() {
 		mcp.AddTool(server, &mcp.Tool{Name: "Get_all_agents", Description: "Retrieves a list of all registered agent UUIDs"}, getAllAgents)
 		mcp.AddTool(server, &mcp.Tool{Name: "Get_agent_status", Description: "Retrieves the current status information for a specific agent identified by its UUID"}, getAgentStatus)
 		mcp.AddTool(server, &mcp.Tool{Name: "Get_failed_agents", Description: "Retrieves all agents currently in a failed operational state with their detailed status information including attestation history and failure reasons"}, getFailedAgents)
+		mcp.AddTool(server, &mcp.Tool{Name: "Reactivate_agent", Description: "Reactivates a failed agent identified by its UUID"}, reactivate_agent)
 		if err := server.Run(context.Background(), &mcp.StdioTransport{}); err != nil {
 			log.Fatal(err)
 		}

backend/tools.go

@@ -2,6 +2,9 @@ package main
 
 import (
 	"context"
+	"encoding/json"
+	"fmt"
+	"log"
 
 	"github.com/modelcontextprotocol/go-sdk/mcp"
 )
@@ -56,3 +59,26 @@ func getFailedAgents(ctx context.Context, req *mcp.CallToolRequest, input getFai
 
 	return nil, failedAgents, nil
 }
+
+func reactivate_agent(ctx context.Context, req *mcp.CallToolRequest, input reactivateAgentInput) (
+	*mcp.CallToolResult,
+	reactivateAgentOutput,
+	error,
+) {
+	endpoint := fmt.Sprintf("agents/%s/reactivate", input.AgentUUID)
+	resp, err := keylimeVerifierClient.Put(endpoint, nil)
+	if err != nil {
+		log.Printf("Error reactivating agent: %v", err)
+		return nil, reactivateAgentOutput{}, err
+	}
+	defer resp.Body.Close()
+
+	var response reactivateAgentOutput
+	err = json.NewDecoder(resp.Body).Decode(&response)
+	if err != nil {
+		log.Printf("Error decoding response: %v", err)
+		return nil, reactivateAgentOutput{}, err
+	}
+
+	return nil, response, nil
+}

backend/types.go

@@ -134,3 +134,13 @@ type getAgentStatusOutput struct {
 	HasMeasuredBoot             bool    `json:"has_measured_boot"`
 	HasRuntimePolicy            bool    `json:"has_runtime_policy"`
 }
+
+type reactivateAgentInput struct {
+	AgentUUID string `json:"agent_uuid"`
+}
+
+type reactivateAgentOutput struct {
+	Code    int      `json:"code"`
+	Status  string   `json:"status"`
+	Results struct{} `json:"results"`
+}