/**
 * @license
 * Copyright 2026 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 *
 * @license
 */

import { GITHUB_OWNER, GITHUB_REPO, type MetricOutput } from '../types.js';
import { execSync } from 'node:child_process';

try {
  const query = `
  query($owner: String!, $repo: String!) {
    repository(owner: $owner, name: $repo) {
      pullRequests(last: 100, states: MERGED) {
        nodes {
          authorAssociation
          comments { totalCount }
          reviews { totalCount }
        }
      }
      issues(last: 100, states: CLOSED) {
        nodes {
          authorAssociation
          comments { totalCount }
        }
      }
    }
  }
  `;
  const output = execSync(
    `gh api graphql -F owner=${GITHUB_OWNER} -F repo=${GITHUB_REPO} -f query='${query}'`,
    { encoding: 'utf-8' },
  );
  const data = JSON.parse(output).data.repository;

  const prs = data.pullRequests.nodes;
  const issues = data.issues.nodes;

  const allItems = [
    ...prs.map(
      (p: {
        authorAssociation: string;
        comments: { totalCount: number };
        reviews?: { totalCount: number };
      }) => ({
        association: p.authorAssociation,
        touches: p.comments.totalCount + (p.reviews ? p.reviews.totalCount : 0),
      }),
    ),
    ...issues.map(
      (i: { authorAssociation: string; comments: { totalCount: number } }) => ({
        association: i.authorAssociation,
        touches: i.comments.totalCount,
      }),
    ),
  ];

  const isMaintainer = (assoc: string) =>
    ['MEMBER', 'OWNER', 'COLLABORATOR'].includes(assoc);

  const calculateAvg = (items: { touches: number; association: string }[]) =>
    items.length ? items.reduce((a, b) => a + b.touches, 0) / items.length : 0;

  const overall = calculateAvg(allItems);
  const maintainers = calculateAvg(
    allItems.filter((i) => isMaintainer(i.association)),
  );
  const community = calculateAvg(
    allItems.filter((i) => !isMaintainer(i.association)),
  );

  const timestamp = new Date().toISOString();

  process.stdout.write(
    JSON.stringify(<MetricOutput>{
      metric: 'user_touches_overall',
      value: Math.round(overall * 100) / 100,
      timestamp,
    }) + '\n',
  );
  process.stdout.write(
    JSON.stringify(<MetricOutput>{
      metric: 'user_touches_maintainers',
      value: Math.round(maintainers * 100) / 100,
      timestamp,
    }) + '\n',
  );
  process.stdout.write(
    JSON.stringify(<MetricOutput>{
      metric: 'user_touches_community',
      value: Math.round(community * 100) / 100,
      timestamp,
    }) + '\n',
  );
} catch (err) {
  process.stderr.write(err instanceof Error ? err.message : String(err));
  process.exit(1);
}