Add datasource tools

Author: sd2k

cmd/main.go

@@ -18,7 +18,8 @@ func newServer() *server.MCPServer {
 		"mcp-grafana",
 		"0.1.0",
 	)
-	s.AddTool(tools.SearchDashboardsTool, tools.SearchDashboardsHandler)
+	tools.AddSearchTools(s)
+	tools.AddDatasourceTools(s)
 	return s
 }
 

tools/datasources.go

@@ -0,0 +1,85 @@
+package tools
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+
+	"github.com/mark3labs/mcp-go/mcp"
+	"github.com/mark3labs/mcp-go/server"
+
+	mcpgrafana "github.com/grafana/mcp-grafana"
+)
+
+type ListDatasourcesParams struct{}
+
+func listDatasources(ctx context.Context, args ListDatasourcesParams) (*mcp.CallToolResult, error) {
+	c := mcpgrafana.GrafanaClientFromContext(ctx)
+	datasources, err := c.Datasources.GetDataSources()
+	if err != nil {
+		return nil, fmt.Errorf("list datasources: %w", err)
+	}
+	b, err := json.Marshal(datasources.Payload)
+	if err != nil {
+		return nil, fmt.Errorf("marshal datasources: %w", err)
+	}
+	return mcp.NewToolResultText(string(b)), nil
+}
+
+var ListDatasourcesTool, ListDatasourcesHandler = mcpgrafana.MustTool(
+	"list_datasources",
+	"List datasources",
+	listDatasources,
+)
+
+type GetDatasourceByUIDParams struct {
+	UID string `json:"uid" jsonschema:"description=The uid of the datasource"`
+}
+
+func getDatasourceByUID(ctx context.Context, args GetDatasourceByUIDParams) (*mcp.CallToolResult, error) {
+	c := mcpgrafana.GrafanaClientFromContext(ctx)
+	datasource, err := c.Datasources.GetDataSourceByUID(args.UID)
+	if err != nil {
+		return nil, fmt.Errorf("get datasource by uid %s: %w", args.UID, err)
+	}
+	b, err := json.Marshal(datasource.Payload)
+	if err != nil {
+		return nil, fmt.Errorf("marshal datasource: %w", err)
+	}
+	return mcp.NewToolResultText(string(b)), nil
+}
+
+var GetDatasourceByUIDTool, GetDatasourceByUIDHandler = mcpgrafana.MustTool(
+	"get_datasource_by_uid",
+	"Get datasource by uid",
+	getDatasourceByUID,
+)
+
+type GetDatasourceByNameParams struct {
+	Name string `json:"name" jsonschema:"description=The name of the datasource"`
+}
+
+func getDatasourceByName(ctx context.Context, args GetDatasourceByNameParams) (*mcp.CallToolResult, error) {
+	c := mcpgrafana.GrafanaClientFromContext(ctx)
+	datasource, err := c.Datasources.GetDataSourceByName(args.Name)
+	if err != nil {
+		return nil, fmt.Errorf("get datasource by name %s: %w", args.Name, err)
+	}
+	b, err := json.Marshal(datasource.Payload)
+	if err != nil {
+		return nil, fmt.Errorf("marshal datasource: %w", err)
+	}
+	return mcp.NewToolResultText(string(b)), nil
+}
+
+var GetDatasourceByNameTool, GetDatasourceByNameHandler = mcpgrafana.MustTool(
+	"get_datasource_by_name",
+	"Get datasource by name",
+	getDatasourceByName,
+)
+
+func AddDatasourceTools(mcp *server.MCPServer) {
+	mcp.AddTool(ListDatasourcesTool, ListDatasourcesHandler)
+	mcp.AddTool(GetDatasourceByUIDTool, GetDatasourceByUIDHandler)
+	mcp.AddTool(GetDatasourceByNameTool, GetDatasourceByNameHandler)
+}

tools/search.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 
 	"github.com/mark3labs/mcp-go/mcp"
+	"github.com/mark3labs/mcp-go/server"
 
 	"github.com/grafana/grafana-openapi-client-go/client/search"
 	mcpgrafana "github.com/grafana/mcp-grafana"
@@ -37,3 +38,7 @@ var SearchDashboardsTool, SearchDashboardsHandler = mcpgrafana.MustTool(
 	"Search for dashboards",
 	searchDashboards,
 )
+
+func AddSearchTools(mcp *server.MCPServer) {
+	mcp.AddTool(SearchDashboardsTool, SearchDashboardsHandler)
+}