codepulse 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dfc914ce32e60fc464e2563c3da95b72889ac156e927f70ff7c0082254e5aaf2
-  data.tar.gz: c872a7873251365f85bce4713134c60b514d1854ebe56ea888cab1031bd923b8
+  metadata.gz: a8a474f7a151ae2edf2f67c74aa3f8347cef950039b6f87777455206afccdc94
+  data.tar.gz: 65af6a80141c769edbc30571a818a5af805d6a8e3366a93a8e04c9d1ab5d7d58
 SHA512:
-  metadata.gz: 531c38b79a0b09e73f427cf3d860ca16822df1b7797435375618d19feea280a8318791b3cf943dd3342b0675e5cf67cbc7c847bb35beddafb0e8faedfd9509df
-  data.tar.gz: 0be7777e2c2640f7a2ed9f3b5939cad157af98de7f825bc64e39cf27ffde1aa0324cd1e6cc8b7b1a7234eb0dadbfeb6330037f4df4b03db978a941f9662bb6bd
+  metadata.gz: c6664055a79fbc73d73cbd2293cb16ef28f3a8417baad8b97f3c8f97582e82eb0d8a5724ac69d7280d112e90f72a47c24af2d4ed133b3d4e1f45fd9df26e44ff
+  data.tar.gz: 20b9addb4cbd4bd6d8874edc4509d355e1c2a688ee25d20947b6bb590dd5bd8df8d689725f1c6b25f399bf6e917c92b8cec81f484570f9bee4a110b7ceb836c3

data/lib/codepulse/base_client.rb

@@ -1,72 +1,12 @@
 # frozen_string_literal: true
 
 require "json"
-require "uri"
 
 module Codepulse
-  # Shared GitHub API client logic for fetching PRs, reviews, and comments.
+  # Shared GitHub API client utilities.
   module BaseClient
-    REPO_FORMAT = %r{\A[^/]+/[^/]+\z}
-
-    # Fetches pull requests with pagination, then fetches full details for each.
-    def pull_requests(repository, state:, limit:)
-      ensure_repository_format(repository)
-      per_page = [limit, 100].min
-      page = 1
-      collected = []
-
-      while collected.length < limit
-        response = api_get(
-          "/repos/#{repository}/pulls",
-          state: state,
-          per_page: per_page,
-          page: page
-        )
-        break if response.empty?
-
-        collected.concat(response)
-        break if response.length < per_page
-
-        page += 1
-      end
-
-      limited = collected.first(limit)
-      fetch_pull_request_details(repository, limited)
-    end
-
-    def pull_request_reviews(repository, number)
-      ensure_repository_format(repository)
-      api_get("/repos/#{repository}/pulls/#{number}/reviews", per_page: 100)
-    end
-
-    def pull_request_comments(repository, number)
-      ensure_repository_format(repository)
-      api_get("/repos/#{repository}/pulls/#{number}/comments", per_page: 100)
-    end
-
-    def issue_comments(repository, number)
-      ensure_repository_format(repository)
-      api_get("/repos/#{repository}/issues/#{number}/comments", per_page: 100)
-    end
-
     private
 
-    def api_get(_path, _query_params = {})
-      raise NotImplementedError, "Subclasses must implement api_get"
-    end
-
-    def ensure_repository_format(repository)
-      return if repository.to_s.match?(REPO_FORMAT)
-
-      raise ConfigurationError, "Repository must be in the format owner/name"
-    end
-
-    def fetch_pull_request_details(repository, pull_requests)
-      pull_requests.map do |pull_request|
-        api_get("/repos/#{repository}/pulls/#{pull_request["number"]}")
-      end
-    end
-
     def parse_json(body)
       return {} if body.to_s.strip.empty?
 
@@ -74,11 +14,5 @@ module Codepulse
     rescue JSON::ParserError => error
       raise ApiError, "Failed to parse response: #{error.message}"
     end
-
-    def encode_query(query_params)
-      return "" if query_params.empty?
-
-      "?#{URI.encode_www_form(query_params)}"
-    end
   end
 end

data/lib/codepulse/cli.rb

@@ -122,8 +122,9 @@ module Codepulse
 
     def fetch_pull_requests(client, repo)
       limit = effective_limit
-      status "Fetching pull requests from #{repo}..."
-      client.pull_requests(repo, state: @options.fetch(:state), limit: limit)
+      business_days = @options.fetch(:business_days_back)
+      status "Fetching pull requests from #{repo} for the last #{business_days} business days..."
+      client.pull_requests_with_activity(repo, state: @options.fetch(:state), limit: limit)
     end
 
     def effective_limit
@@ -143,13 +144,12 @@ module Codepulse
       pull_requests
     end
 
-    def calculate_metrics(client, repo, pull_requests)
+    def calculate_metrics(_client, _repo, pull_requests)
       status "Calculating metrics for #{pull_requests.length} pull requests..."
-      calculator = MetricsCalculator.new(client: client)
+      calculator = MetricsCalculator.new
 
-      pull_requests.each_with_index.map do |pull_request, index|
-        status "  Analyzing PR ##{pull_request["number"]} (#{index + 1}/#{pull_requests.length})..."
-        calculator.metrics_for_pull_request(repo, pull_request)
+      pull_requests.map do |pull_request|
+        calculator.metrics_for_pull_request(pull_request)
       end
     end
 

data/lib/codepulse/gh_cli_client.rb

@@ -1,27 +1,131 @@
 # frozen_string_literal: true
 
 require "open3"
+require "json"
 
 module Codepulse
   class GhCliClient
     include BaseClient
 
     DEFAULT_COMMAND = "gh"
+    GRAPHQL_PAGE_SIZE = 50
+    ACTIVITY_PAGE_SIZE = 50
 
     def initialize(command: DEFAULT_COMMAND)
       @command = command
       verify_cli_available
     end
 
+    # Fetches PRs with reviews and comments in a single GraphQL query.
+    # Returns array of PR hashes with embedded :reviews, :review_comments, :issue_comments.
+    def pull_requests_with_activity(repository, state:, limit:)
+      owner, name = repository.split("/", 2)
+      raise ConfigurationError, "Repository must be in the format owner/name" unless owner && name
+
+      fetch_all_pull_requests(owner, name, graphql_states(state), limit)
+    end
+
     private
 
-    def api_get(path, query_params = {})
-      full_path = "#{path}#{encode_query(query_params)}"
-      stdout, stderr, status = Open3.capture3(@command, "api", full_path)
+    def fetch_all_pull_requests(owner, name, states, limit)
+      pull_requests = []
+      cursor = nil
+
+      loop do
+        batch_size = [GRAPHQL_PAGE_SIZE, limit - pull_requests.length].min
+        response = graphql_query(build_query(owner, name, states, batch_size, cursor))
+        nodes, page_info = extract_pr_data(response)
+        break if nodes.empty?
+
+        pull_requests.concat(nodes.map { |node| transform_graphql_pr(node) })
+        break if pull_requests.length >= limit || !page_info["hasNextPage"]
+
+        cursor = page_info["endCursor"]
+      end
+
+      pull_requests.first(limit)
+    end
+
+    def extract_pr_data(response)
+      pr_data = response.dig("data", "repository", "pullRequests") || {}
+      nodes = pr_data["nodes"] || []
+      page_info = pr_data["pageInfo"] || {}
+      [nodes, page_info]
+    end
+
+    def graphql_states(state)
+      case state
+      when "open" then %w[OPEN]
+      when "closed" then %w[CLOSED MERGED]
+      else %w[OPEN CLOSED MERGED]
+      end
+    end
+
+    def build_query(owner, name, states, batch_size, cursor)
+      after_clause = cursor ? ", after: \"#{cursor}\"" : ""
+      states_clause = states.join(", ")
+
+      <<~GRAPHQL
+        {
+          repository(owner: "#{owner}", name: "#{name}") {
+            pullRequests(first: #{batch_size}, states: [#{states_clause}], orderBy: {field: CREATED_AT, direction: DESC}#{after_clause}) {
+              pageInfo { hasNextPage endCursor }
+              nodes {
+                number title state createdAt mergedAt additions deletions changedFiles
+                author { login }
+                reviews(first: #{ACTIVITY_PAGE_SIZE}) { nodes { submittedAt author { login } } }
+                reviewThreads(first: #{ACTIVITY_PAGE_SIZE}) { nodes { comments(first: #{ACTIVITY_PAGE_SIZE}) { nodes { createdAt author { login } } } } }
+                comments(first: #{ACTIVITY_PAGE_SIZE}) { nodes { createdAt author { login } } }
+              }
+            }
+          }
+        }
+      GRAPHQL
+    end
+
+    def transform_graphql_pr(node)
+      {
+        "number" => node["number"],
+        "title" => node["title"],
+        "state" => node["state"]&.downcase,
+        "created_at" => node["createdAt"],
+        "merged_at" => node["mergedAt"],
+        "additions" => node["additions"],
+        "deletions" => node["deletions"],
+        "changed_files" => node["changedFiles"],
+        "user" => { "login" => node.dig("author", "login") },
+        "reviews" => transform_reviews(node),
+        "review_comments" => transform_review_comments(node),
+        "issue_comments" => transform_issue_comments(node)
+      }
+    end
+
+    def transform_reviews(node)
+      (node.dig("reviews", "nodes") || []).map do |review|
+        { "submitted_at" => review["submittedAt"], "user" => { "login" => review.dig("author", "login") } }
+      end
+    end
+
+    def transform_review_comments(node)
+      (node.dig("reviewThreads", "nodes") || []).flat_map do |thread|
+        (thread.dig("comments", "nodes") || []).map do |comment|
+          { "created_at" => comment["createdAt"], "user" => { "login" => comment.dig("author", "login") } }
+        end
+      end
+    end
+
+    def transform_issue_comments(node)
+      (node.dig("comments", "nodes") || []).map do |comment|
+        { "created_at" => comment["createdAt"], "user" => { "login" => comment.dig("author", "login") } }
+      end
+    end
+
+    def graphql_query(query)
+      stdout, stderr, status = Open3.capture3(@command, "api", "graphql", "-f", "query=#{query}")
 
       unless status.success?
         message = stderr.to_s.strip.empty? ? stdout.to_s.strip : stderr.to_s.strip
-        raise ApiError, "gh api #{full_path} failed: #{message}"
+        raise ApiError, "GraphQL query failed: #{message}"
       end
 
       parse_json(stdout)

data/lib/codepulse/metrics_calculator.rb

@@ -69,15 +69,16 @@ module Codepulse
       "allstar[bot]"
     ].freeze
 
-    def initialize(client:)
-      @client = client
+    def initialize
+      # No client needed - data is pre-fetched via GraphQL
     end
 
     # Returns a hash of metrics for a single PR.
-    def metrics_for_pull_request(repository, pull_request)
+    # Expects pull_request to include :reviews, :review_comments, :issue_comments from GraphQL.
+    def metrics_for_pull_request(pull_request)
       created_at = parse_time(pull_request["created_at"])
       merged_at = parse_time(pull_request["merged_at"])
-      pickup_event = find_pickup_event(repository, pull_request, created_at)
+      pickup_event = find_pickup_event(pull_request, created_at)
       pickup_seconds = pickup_event ? business_seconds_between(created_at, pickup_event.fetch(:timestamp)) : nil
       merge_seconds = merged_at && created_at ? business_seconds_between(created_at, merged_at) : nil
 
@@ -101,12 +102,12 @@ module Codepulse
     private
 
     # Finds the first non-author, non-bot response (review, comment, or issue comment).
-    def find_pickup_event(repository, pull_request, created_at)
-      pull_number = pull_request["number"]
+    # Uses pre-fetched data from GraphQL query.
+    def find_pickup_event(pull_request, created_at)
       author_login = pull_request.dig("user", "login")
 
       review_event = earliest_event(
-        @client.pull_request_reviews(repository, pull_number),
+        pull_request["reviews"] || [],
         author_login: author_login,
         time_key: "submitted_at",
         actor_path: %w[user login],
@@ -114,7 +115,7 @@ module Codepulse
       )
 
       review_comment_event = earliest_event(
-        @client.pull_request_comments(repository, pull_number),
+        pull_request["review_comments"] || [],
         author_login: author_login,
         time_key: "created_at",
         actor_path: %w[user login],
@@ -122,7 +123,7 @@ module Codepulse
       )
 
       issue_comment_event = earliest_event(
-        @client.issue_comments(repository, pull_number),
+        pull_request["issue_comments"] || [],
         author_login: author_login,
         time_key: "created_at",
         actor_path: %w[user login],

data/lib/codepulse.rb

@@ -9,5 +9,5 @@ require_relative "codepulse/formatter"
 require_relative "codepulse/cli"
 
 module Codepulse
-  VERSION = "0.1.3"
+  VERSION = "0.1.4"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: codepulse
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Patrick Navarro
 bindir: bin
 cert_chain: []
-date: 2025-12-27 00:00:00.000000000 Z
+date: 2025-12-30 00:00:00.000000000 Z
 dependencies: []
 description: Terminal tool to analyze GitHub pull request pickup times, merge times,
   and sizes using the gh CLI.