events.go

package tools

import (
	"context"
	"fmt"
	"time"

	mcpnetbird "github.com/aantti/mcp-netbird"
	"github.com/mark3labs/mcp-go/server"
)

type NetbirdEvent struct {
	ID          string    `json:"id"`
	Timestamp   time.Time `json:"timestamp"`
	Activity    string    `json:"activity"`
	InitiatorID string    `json:"initiator_id"`
	TargetID    string    `json:"target_id"`
	Meta        any       `json:"meta"`
}

func eventClient() *mcpnetbird.NetbirdClient {
	if mcpnetbird.TestNetbirdClient != nil {
		return mcpnetbird.TestNetbirdClient
	}
	return mcpnetbird.NewNetbirdClient()
}

// list_netbird_events

type ListNetbirdEventsParams struct {
	Page  int `json:"page,omitempty" jsonschema:"description=Page number (1-based)"`
	Limit int `json:"limit,omitempty" jsonschema:"description=Number of events per page"`
}

func listNetbirdEvents(ctx context.Context, args ListNetbirdEventsParams) ([]NetbirdEvent, error) {
	path := "/events/audit"
	query := ""
	if args.Page > 0 {
		query += fmt.Sprintf("page=%d", args.Page)
	}
	if args.Limit > 0 {
		if query != "" {
			query += "&"
		}
		query += fmt.Sprintf("limit=%d", args.Limit)
	}
	if query != "" {
		path += "?" + query
	}

	var events []NetbirdEvent
	if err := eventClient().Get(ctx, path, &events); err != nil {
		return nil, err
	}
	return events, nil
}

var ListNetbirdEvents = mcpnetbird.MustTool(
	"list_netbird_events",
	"List Netbird audit events",
	listNetbirdEvents,
)

func AddNetbirdEventTools(mcp *server.MCPServer) {
	ListNetbirdEvents.Register(mcp)
}