aboutsummaryrefslogtreecommitdiff
path: root/cmd/pr-analyzer/analyzer.go
blob: a7935090076f61647026a80f92fe87e7e6e49af5 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
package main

import (
	"fmt"
	"sort"
	"time"
)

// AnalyzePRs processes a list of PRs and generates metrics for each reviewer.
func (a *Analyzer) AnalyzePRs(prs []PRData) ([]ReviewerMetrics, error) {
	reviewerPRs := make(map[string][]PRMetrics)

	for _, pr := range prs {
		metrics, err := a.analyzePR(pr)
		if err != nil {
			return nil, fmt.Errorf("analyzing PR %d: %w", pr.Number, err)
		}

		for _, reviewer := range metrics.Reviewers {
			reviewerPRs[reviewer] = append(reviewerPRs[reviewer], metrics)
		}
	}

	var results []ReviewerMetrics
	for reviewer, prs := range reviewerPRs {
		metrics := a.calculateReviewerMetrics(reviewer, prs)
		results = append(results, metrics)
	}

	sort.Slice(results, func(i, j int) bool {
		return results[i].PRsReviewed > results[j].PRsReviewed
	})

	return results, nil
}

func (a *Analyzer) analyzePR(pr PRData) (PRMetrics, error) {
	metrics := PRMetrics{
		Number:          pr.Number,
		CreatedAt:       pr.CreatedAt,
		CommentsPerFile: make(map[string]int),
		Reviewers:       make([]string, 0),
	}

	sort.Slice(pr.Reviews, func(i, j int) bool {
		return pr.Reviews[i].SubmittedAt.Before(pr.Reviews[j].SubmittedAt)
	})

	for _, review := range pr.Reviews {
		metrics.Reviewers = append(metrics.Reviewers, review.Reviewer)

		pattern := ReviewPattern{
			FirstReviewState:  review.State,
			TimeToFirstReview: review.SubmittedAt.Sub(pr.CreatedAt),
		}

		for _, comment := range review.Comments {
			metrics.CommentsPerFile[comment.Path]++
		}

		metrics.ReviewPatterns = append(metrics.ReviewPatterns, pattern)
	}

	if len(pr.Reviews) > 0 {
		metrics.TimeToFirstReview = pr.Reviews[0].SubmittedAt.Sub(pr.CreatedAt)
	}

	if len(metrics.ReviewPatterns) > 0 {
		lastReview := pr.Reviews[len(pr.Reviews)-1]
		metrics.ReviewPatterns[len(metrics.ReviewPatterns)-1].FinalReviewState = lastReview.State
		metrics.ReviewPatterns[len(metrics.ReviewPatterns)-1].TimeToFinalReview = lastReview.SubmittedAt.Sub(
			pr.CreatedAt,
		)
	}

	return metrics, nil
}

func (a *Analyzer) calculateReviewerMetrics(reviewer string, prs []PRMetrics) ReviewerMetrics {
	metrics := ReviewerMetrics{
		Username:    reviewer,
		PRsReviewed: len(prs),
	}

	var totalReviewSpeed time.Duration
	var totalComments int
	var immediateApprovals int

	for _, pr := range prs {
		var pattern ReviewPattern
		for _, p := range pr.ReviewPatterns {
			if p.FirstReviewState != "" {
				pattern = p
				break
			}
		}

		if pattern.TimeToFirstReview > 0 {
			totalReviewSpeed += pattern.TimeToFirstReview
		}

		for _, count := range pr.CommentsPerFile {
			totalComments += count
		}

		if pattern.FirstReviewState == "APPROVED" {
			immediateApprovals++
		}
	}

	if metrics.PRsReviewed > 0 {
		metrics.AverageReviewSpeed = totalReviewSpeed.Hours() / float64(metrics.PRsReviewed)
		metrics.AverageComments = float64(totalComments) / float64(metrics.PRsReviewed)
		metrics.ImmediateApprovals = float64(
			immediateApprovals,
		) / float64(
			metrics.PRsReviewed,
		) * 100
	}

	return metrics
}