@atolis-hq/wake 0.1.15 → 0.1.16

package/dist/src/adapters/github/github-pull-request-activity-source.js

@@ -62,8 +62,10 @@ export function createGitHubPullRequestActivitySource(deps) {
         return { owner, repo, repoRef: `${owner}/${repo}`, number: Number(numberStr) };
     }
     async function discoverPullRequests(ingestedAt) {
+        const seenPrData = new Map();
+        const confirmedOpenRepos = new Set();
         if (!deps.config.sources.github.pullRequests.enabled) {
-            return [];
+            return { events: [], seenPrData, confirmedOpenRepos };
         }
         const events = [];
         for (const repoRef of deps.config.sources.github.repos) {
@@ -72,9 +74,22 @@ export function createGitHubPullRequestActivitySource(deps) {
                 continue;
             }
             try {
-                const prs = await deps.client.listPullRequests(owner, repo, deps.config.sources.github.pullRequests.maxPullRequestsPerRepo);
+                const maxResults = deps.config.sources.github.pullRequests.maxPullRequestsPerRepo;
+                const prs = await deps.client.listPullRequests(owner, repo, maxResults);
+                // listPullRequests only ever returns open PRs (state: 'open'), so an
+                // uncapped result set here is a complete picture of what's currently
+                // open — used below to confirm a watched PR has actually closed
+                // rather than merely being truncated by maxResults.
+                if (prs.length < maxResults) {
+                    confirmedOpenRepos.add(repoRef);
+                }
                 for (const pr of prs) {
                     const resourceUri = prResourceUri(repoRef, pr.number);
+                    seenPrData.set(resourceUri, {
+                        headSha: pr.head.sha,
+                        baseRef: pr.base?.ref,
+                        htmlUrl: pr.html_url,
+                    });
                     const known = await deps.resourceIndex.resolve(resourceUri);
                     if (known !== undefined) {
                         continue;
@@ -110,7 +125,7 @@ export function createGitHubPullRequestActivitySource(deps) {
                 console.error(`[github-pr-activity-source] discovery failed for ${repoRef}, skipping this tick: ${error instanceof Error ? error.message : String(error)}`);
             }
         }
-        return events;
+        return { events, seenPrData, confirmedOpenRepos };
     }
     async function pollWatchedPr(resourceUri, ingestedAt) {
         if (!deps.config.sources.github.pullRequests.enabled) {
@@ -236,7 +251,7 @@ export function createGitHubPullRequestActivitySource(deps) {
         }
         return events;
     }
-    async function pollRequiredCheckFailures(ref, resourceUri, ingestedAt) {
+    async function pollRequiredCheckFailures(ref, resourceUri, ingestedAt, cachedPr) {
         if (!deps.config.sources.github.pullRequests.enabled ||
             !deps.config.sources.github.pullRequests.checks.enabled ||
             deps.client.getRequiredStatusChecks === undefined ||
@@ -244,9 +259,25 @@ export function createGitHubPullRequestActivitySource(deps) {
             deps.client.getCombinedStatusForRef === undefined) {
             return [];
         }
-        const pr = await deps.client.getPullRequest(ref.owner, ref.repo, ref.number);
-        const headSha = pr.head.sha;
-        const baseRef = pr.base?.ref;
+        // headSha/baseRef/htmlUrl are already returned by this tick's PR
+        // discovery pass (listPullRequests) — reuse that instead of spending a
+        // dedicated getPullRequest call per watched PR. Falls back only when the
+        // watched PR wasn't present in discovery (e.g. truncated by
+        // maxPullRequestsPerRepo, or pullRequests polling failed this tick).
+        let headSha;
+        let baseRef;
+        let htmlUrl;
+        if (cachedPr?.headSha !== undefined && cachedPr.baseRef !== undefined) {
+            headSha = cachedPr.headSha;
+            baseRef = cachedPr.baseRef;
+            htmlUrl = cachedPr.htmlUrl;
+        }
+        else {
+            const pr = await deps.client.getPullRequest(ref.owner, ref.repo, ref.number);
+            headSha = pr.head.sha;
+            baseRef = pr.base?.ref;
+            htmlUrl = pr.html_url;
+        }
         if (headSha === undefined || baseRef === undefined) {
             return [];
         }
@@ -265,7 +296,7 @@ export function createGitHubPullRequestActivitySource(deps) {
                 continue;
             }
             const occurredAt = run.completed_at ?? run.started_at ?? ingestedAt;
-            const sourceUrl = run.html_url ?? run.details_url ?? pr.html_url;
+            const sourceUrl = run.html_url ?? run.details_url ?? htmlUrl;
             events.push(createUnkeyedEventEnvelope({
                 eventId: `pr-check-failed-${safeEventToken(ref.repoRef)}-${ref.number}-${safeEventToken(run.name)}-${headSha}-${safeEventToken(occurredAt)}-${run.conclusion}`,
                 streamScope: 'work-item',
@@ -306,7 +337,7 @@ export function createGitHubPullRequestActivitySource(deps) {
                 sourceEventType: 'pr.checks.failed',
                 sourceRefs: {
                     repo: ref.repoRef,
-                    sourceUrl: status.target_url ?? pr.html_url,
+                    sourceUrl: status.target_url ?? htmlUrl,
                     resourceUri,
                 },
                 occurredAt: status.updated_at,
@@ -336,15 +367,28 @@ export function createGitHubPullRequestActivitySource(deps) {
         async pollEvents(input) {
             const ingestedAt = deps.now().toISOString();
             const watched = (input?.watch ?? []).filter((ref) => ref.resourceUri.startsWith('github:pr:'));
-            const discovered = await discoverPullRequests(ingestedAt);
-            const activityBatches = await Promise.all(watched.map((ref) => pollWatchedPr(ref.resourceUri, ingestedAt)));
-            const checkBatches = await Promise.all(watched.map(async (watchRef) => {
+            const { events: discovered, seenPrData, confirmedOpenRepos, } = await discoverPullRequests(ingestedAt);
+            // A watched PR absent from this tick's complete open-PR listing has
+            // closed or merged — stop spending its 7 activity/checks calls every
+            // tick. `confirmedOpenRepos` only covers repos where discovery
+            // succeeded and wasn't truncated, so an untruncated miss or a failed
+            // repo poll leaves the PR in the active set rather than risking a
+            // false drop.
+            const activeWatched = watched.filter((ref) => {
+                const parsed = repoAndNumberFromPrUri(ref.resourceUri);
+                if (parsed === null) {
+                    return true;
+                }
+                return !confirmedOpenRepos.has(parsed.repoRef) || seenPrData.has(ref.resourceUri);
+            });
+            const activityBatches = await Promise.all(activeWatched.map((ref) => pollWatchedPr(ref.resourceUri, ingestedAt)));
+            const checkBatches = await Promise.all(activeWatched.map(async (watchRef) => {
                 const ref = repoAndNumberFromPrUri(watchRef.resourceUri);
                 if (ref === null) {
                     return [];
                 }
                 try {
-                    return await pollRequiredCheckFailures(ref, watchRef.resourceUri, ingestedAt);
+                    return await pollRequiredCheckFailures(ref, watchRef.resourceUri, ingestedAt, seenPrData.get(watchRef.resourceUri));
                 }
                 catch (error) {
                     console.error(`[github-pr-activity-source] check poll failed for ${watchRef.resourceUri}, skipping this tick: ${error instanceof Error ? error.message : String(error)}`);

package/dist/src/version.js

@@ -1 +1 @@
-export const wakeVersion = "0.1.15+ge2584a1";
+export const wakeVersion = "0.1.16+g97c5859";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@atolis-hq/wake",
-  "version": "0.1.15",
+  "version": "0.1.16",
   "description": "Local autonomous agent control plane for software development",
   "license": "Apache-2.0",
   "repository": {