Adding copilot metrics

Author: austenstone

package-lock.json

@@ -358,3 +358,77 @@ func (client *Client) getWorkflowRuns(ctx context.Context, owner, repo, workflow
 
 	return workflowRuns, response.NextPage, nil
 }
+
+// GetCopilotMetrics sends a request to the GitHub REST API to get Copilot metrics for an organization
+func (client *Client) GetCopilotMetrics(ctx context.Context, organization string, opts models.ListCopilotMetricsOptions) ([]models.CopilotMetrics, *googlegithub.Response, error) {
+	u := fmt.Sprintf("orgs/%s/copilot/metrics", organization)
+
+	// Build query parameters
+	params := url.Values{}
+	if opts.Since != nil {
+		params.Add("since", opts.Since.Format("2006-01-02"))
+	}
+	if opts.Until != nil {
+		params.Add("until", opts.Until.Format("2006-01-02"))
+	}
+	if opts.Page > 0 {
+		params.Add("page", strconv.Itoa(opts.Page))
+	}
+	if opts.PerPage > 0 {
+		params.Add("per_page", strconv.Itoa(opts.PerPage))
+	}
+
+	if len(params) > 0 {
+		u += "?" + params.Encode()
+	}
+
+	req, err := client.restClient.NewRequest("GET", u, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	var metrics []models.CopilotMetrics
+	resp, err := client.restClient.Do(ctx, req, &metrics)
+	if err != nil {
+		return nil, resp, addErrorSourceToError(err, resp)
+	}
+
+	return metrics, resp, nil
+}
+
+// GetCopilotMetricsTeam sends a request to the GitHub REST API to get Copilot metrics for a team
+func (client *Client) GetCopilotMetricsTeam(ctx context.Context, organization, teamSlug string, opts models.ListCopilotMetricsTeamOptions) ([]models.CopilotMetrics, *googlegithub.Response, error) {
+	u := fmt.Sprintf("orgs/%s/team/%s/copilot/metrics", organization, teamSlug)
+
+	// Build query parameters
+	params := url.Values{}
+	if opts.Since != nil {
+		params.Add("since", opts.Since.Format("2006-01-02"))
+	}
+	if opts.Until != nil {
+		params.Add("until", opts.Until.Format("2006-01-02"))
+	}
+	if opts.Page > 0 {
+		params.Add("page", strconv.Itoa(opts.Page))
+	}
+	if opts.PerPage > 0 {
+		params.Add("per_page", strconv.Itoa(opts.PerPage))
+	}
+
+	if len(params) > 0 {
+		u += "?" + params.Encode()
+	}
+
+	req, err := client.restClient.NewRequest("GET", u, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	var metrics []models.CopilotMetrics
+	resp, err := client.restClient.Do(ctx, req, &metrics)
+	if err != nil {
+		return nil, resp, addErrorSourceToError(err, resp)
+	}
+
+	return metrics, resp, nil
+}

pkg/github/client/client.go

@@ -52,6 +52,14 @@ func (m *mockClient) ListAlertsForOrg(ctx context.Context, owner string, opts *g
 	return m.mockAlerts, m.mockResponse, nil
 }
 
+func (m *mockClient) GetCopilotMetrics(ctx context.Context, organization string, opts models.ListCopilotMetricsOptions) ([]models.CopilotMetrics, *googlegithub.Response, error) {
+	return nil, nil, nil
+}
+
+func (m *mockClient) GetCopilotMetricsTeam(ctx context.Context, organization, teamSlug string, opts models.ListCopilotMetricsTeamOptions) ([]models.CopilotMetrics, *googlegithub.Response, error) {
+	return nil, nil, nil
+}
+
 func TestGetCodeScanningAlerts(t *testing.T) {
 	var (
 		ctx  = context.Background()

pkg/github/codescanning_test.go

@@ -0,0 +1,163 @@
+package github
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"time"
+
+	"github.com/grafana/github-datasource/pkg/dfutil"
+	"github.com/grafana/github-datasource/pkg/models"
+	"github.com/grafana/grafana-plugin-sdk-go/data"
+)
+
+// CopilotMetricsResponse represents the response from GitHub's Copilot metrics API
+type CopilotMetricsResponse []models.CopilotMetrics
+
+// GetCopilotMetrics retrieves Copilot metrics for an organization
+func GetCopilotMetrics(ctx context.Context, client models.Client, opts models.ListCopilotMetricsOptions) (dfutil.Framer, error) {
+	metrics, _, err := client.GetCopilotMetrics(ctx, opts.Organization, opts)
+	if err != nil {
+		return nil, err
+	}
+
+	return copilotMetricsToDataFrame(CopilotMetricsResponse(metrics), "copilot_metrics")
+}
+
+// GetCopilotMetricsTeam retrieves Copilot metrics for a team
+func GetCopilotMetricsTeam(ctx context.Context, client models.Client, opts models.ListCopilotMetricsTeamOptions) (dfutil.Framer, error) {
+	metrics, _, err := client.GetCopilotMetricsTeam(ctx, opts.Organization, opts.TeamSlug, opts)
+	if err != nil {
+		return nil, err
+	}
+
+	return copilotMetricsToDataFrame(CopilotMetricsResponse(metrics), "copilot_metrics_team")
+}
+
+// copilotMetricsToDataFrame converts Copilot metrics to a Grafana data frame
+func copilotMetricsToDataFrame(metrics CopilotMetricsResponse, name string) (dfutil.Framer, error) {
+	return metrics, nil
+}
+
+// Frames converts the list of copilot metrics to a Grafana DataFrame
+func (c CopilotMetricsResponse) Frames() data.Frames {
+	frame := data.NewFrame("copilot_metrics")
+
+	if len(c) == 0 {
+		return data.Frames{frame}
+	}
+
+	// Create time series for the main metrics
+	dates := make([]time.Time, len(c))
+	totalActiveUsers := make([]int64, len(c))
+	totalEngagedUsers := make([]int64, len(c))
+	ideCompletionUsers := make([]int64, len(c))
+	ideChatUsers := make([]int64, len(c))
+	dotcomChatUsers := make([]int64, len(c))
+	dotcomPRUsers := make([]int64, len(c))
+
+	for i, metric := range c {
+		date, err := time.Parse("2006-01-02", metric.Date)
+		if err != nil {
+			// If date parsing fails, use a default date
+			date = time.Now().AddDate(0, 0, -i)
+		}
+
+		dates[i] = date
+		totalActiveUsers[i] = int64(metric.TotalActiveUsers)
+		totalEngagedUsers[i] = int64(metric.TotalEngagedUsers)
+		ideCompletionUsers[i] = int64(metric.CopilotIDECodeCompletions.TotalEngagedUsers)
+		ideChatUsers[i] = int64(metric.CopilotIDEChat.TotalEngagedUsers)
+		dotcomChatUsers[i] = int64(metric.CopilotDotcomChat.TotalEngagedUsers)
+		dotcomPRUsers[i] = int64(metric.CopilotDotcomPullRequests.TotalEngagedUsers)
+	}
+
+	// Add fields to the frame
+	frame.Fields = append(frame.Fields, data.NewField("time", nil, dates))
+	frame.Fields = append(frame.Fields, data.NewField("total_active_users", nil, totalActiveUsers))
+	frame.Fields = append(frame.Fields, data.NewField("total_engaged_users", nil, totalEngagedUsers))
+	frame.Fields = append(frame.Fields, data.NewField("ide_completion_users", nil, ideCompletionUsers))
+	frame.Fields = append(frame.Fields, data.NewField("ide_chat_users", nil, ideChatUsers))
+	frame.Fields = append(frame.Fields, data.NewField("dotcom_chat_users", nil, dotcomChatUsers))
+	frame.Fields = append(frame.Fields, data.NewField("dotcom_pr_users", nil, dotcomPRUsers))
+
+	// Add language breakdown data if available
+	if len(c) > 0 && len(c[0].CopilotIDECodeCompletions.Languages) > 0 {
+		langData := make(map[string][]int64)
+		for _, metric := range c {
+			for _, lang := range metric.CopilotIDECodeCompletions.Languages {
+				if langData[lang.Name] == nil {
+					langData[lang.Name] = make([]int64, len(c))
+				}
+			}
+		}
+
+		for i, metric := range c {
+			for langName := range langData {
+				found := false
+				for _, lang := range metric.CopilotIDECodeCompletions.Languages {
+					if lang.Name == langName {
+						langData[langName][i] = int64(lang.TotalEngagedUsers)
+						found = true
+						break
+					}
+				}
+				if !found {
+					langData[langName][i] = 0
+				}
+			}
+		}
+
+		for langName, users := range langData {
+			fieldName := fmt.Sprintf("language_%s_users", langName)
+			frame.Fields = append(frame.Fields, data.NewField(fieldName, nil, users))
+		}
+	}
+
+	// Add editor breakdown data if available
+	if len(c) > 0 && len(c[0].CopilotIDECodeCompletions.Editors) > 0 {
+		editorData := make(map[string][]int64)
+		for _, metric := range c {
+			for _, editor := range metric.CopilotIDECodeCompletions.Editors {
+				if editorData[editor.Name] == nil {
+					editorData[editor.Name] = make([]int64, len(c))
+				}
+			}
+		}
+
+		for i, metric := range c {
+			for editorName := range editorData {
+				found := false
+				for _, editor := range metric.CopilotIDECodeCompletions.Editors {
+					if editor.Name == editorName {
+						editorData[editorName][i] = int64(editor.TotalEngagedUsers)
+						found = true
+						break
+					}
+				}
+				if !found {
+					editorData[editorName][i] = 0
+				}
+			}
+		}
+
+		for editorName, users := range editorData {
+			fieldName := fmt.Sprintf("editor_%s_users", editorName)
+			frame.Fields = append(frame.Fields, data.NewField(fieldName, nil, users))
+		}
+	}
+
+	// Add detailed JSON for complex nested data
+	detailedData := make([]string, len(c))
+	for i, metric := range c {
+		jsonData, err := json.Marshal(metric)
+		if err != nil {
+			detailedData[i] = ""
+		} else {
+			detailedData[i] = string(jsonData)
+		}
+	}
+	frame.Fields = append(frame.Fields, data.NewField("detailed_metrics", nil, detailedData))
+
+	return data.Frames{frame}
+}

pkg/github/copilot_metrics.go

@@ -0,0 +1,39 @@
+package github
+
+import (
+	"context"
+
+	"github.com/grafana/github-datasource/pkg/dfutil"
+	"github.com/grafana/github-datasource/pkg/models"
+	"github.com/grafana/grafana-plugin-sdk-go/backend"
+)
+
+func (s *QueryHandler) handleCopilotMetricsQuery(ctx context.Context, q backend.DataQuery) backend.DataResponse {
+	query := &models.CopilotMetricsQuery{}
+	if err := UnmarshalQuery(q.JSON, query); err != nil {
+		return *err
+	}
+	return dfutil.FrameResponseWithError(s.Datasource.HandleCopilotMetricsQuery(ctx, query, q))
+}
+
+// HandleCopilotMetrics handles the plugin query for GitHub Copilot metrics
+func (s *QueryHandler) HandleCopilotMetrics(ctx context.Context, req *backend.QueryDataRequest) (*backend.QueryDataResponse, error) {
+	return &backend.QueryDataResponse{
+		Responses: processQueries(ctx, req, s.handleCopilotMetricsQuery),
+	}, nil
+}
+
+func (s *QueryHandler) handleCopilotMetricsTeamQuery(ctx context.Context, q backend.DataQuery) backend.DataResponse {
+	query := &models.CopilotMetricsTeamQuery{}
+	if err := UnmarshalQuery(q.JSON, query); err != nil {
+		return *err
+	}
+	return dfutil.FrameResponseWithError(s.Datasource.HandleCopilotMetricsTeamQuery(ctx, query, q))
+}
+
+// HandleCopilotMetricsTeam handles the plugin query for GitHub Copilot metrics for a team
+func (s *QueryHandler) HandleCopilotMetricsTeam(ctx context.Context, req *backend.QueryDataRequest) (*backend.QueryDataResponse, error) {
+	return &backend.QueryDataResponse{
+		Responses: processQueries(ctx, req, s.handleCopilotMetricsTeamQuery),
+	}, nil
+}

pkg/github/copilot_metrics_handler.go

@@ -0,0 +1,86 @@
+package github
+
+import (
+	"testing"
+
+	"github.com/grafana/github-datasource/pkg/models"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCopilotMetricsResponse_Frames(t *testing.T) {
+	// Test empty response
+	t.Run("empty response", func(t *testing.T) {
+		response := CopilotMetricsResponse{}
+		frames := response.Frames()
+		assert.Len(t, frames, 1)
+		assert.Equal(t, "copilot_metrics", frames[0].Name)
+		assert.Len(t, frames[0].Fields, 0)
+	})
+
+	// Test response with data
+	t.Run("response with data", func(t *testing.T) {
+		response := CopilotMetricsResponse{
+			{
+				Date:              "2025-01-01",
+				TotalActiveUsers:  100,
+				TotalEngagedUsers: 75,
+				CopilotIDECodeCompletions: models.CopilotIDECodeCompletions{
+					TotalEngagedUsers: 50,
+					Languages: []models.CopilotLanguageMetrics{
+						{Name: "go", TotalEngagedUsers: 25},
+						{Name: "typescript", TotalEngagedUsers: 20},
+					},
+					Editors: []models.CopilotEditorMetrics{
+						{Name: "vscode", TotalEngagedUsers: 45},
+						{Name: "neovim", TotalEngagedUsers: 5},
+					},
+				},
+				CopilotIDEChat: models.CopilotIDEChat{
+					TotalEngagedUsers: 30,
+				},
+				CopilotDotcomChat: models.CopilotDotcomChat{
+					TotalEngagedUsers: 25,
+				},
+				CopilotDotcomPullRequests: models.CopilotDotcomPullRequests{
+					TotalEngagedUsers: 15,
+				},
+			},
+		}
+
+		frames := response.Frames()
+		assert.Len(t, frames, 1)
+		frame := frames[0]
+
+		assert.Equal(t, "copilot_metrics", frame.Name)
+
+		// Check that we have the expected fields
+		fieldNames := make([]string, len(frame.Fields))
+		for i, field := range frame.Fields {
+			fieldNames[i] = field.Name
+		}
+
+		expectedFields := []string{
+			"time",
+			"total_active_users",
+			"total_engaged_users",
+			"ide_completion_users",
+			"ide_chat_users",
+			"dotcom_chat_users",
+			"dotcom_pr_users",
+			"language_go_users",
+			"language_typescript_users",
+			"editor_vscode_users",
+			"editor_neovim_users",
+			"detailed_metrics",
+		}
+
+		for _, expected := range expectedFields {
+			assert.Contains(t, fieldNames, expected, "Field %s should be present", expected)
+		}
+
+		// Check that all fields have the correct length
+		for _, field := range frame.Fields {
+			assert.Equal(t, 1, field.Len(), "Field %s should have length 1", field.Name)
+		}
+	})
+}