@mastra/github-signals 0.3.0 → 0.4.0-alpha.1

package/LICENSE.md

@@ -1,10 +1,12 @@
 Portions of this software are licensed as follows:
 
-- All content that resides under any directory named "ee/" within this
+- All content that resides under any directory named `ee/` within this
   repository, including but not limited to:
-  - `packages/core/src/auth/ee/`
-  - `packages/server/src/server/auth/ee/`
-    is licensed under the license defined in `ee/LICENSE`.
+  - `@mastra/core/auth/ee`
+  - `@mastra/core/agent-builder/ee`
+  - `@mastra/editor/ee`
+
+  is licensed under the license defined in [`ee/LICENSE`](https://github.com/mastra-ai/mastra/blob/main/ee/LICENSE).
 
 - All third-party components incorporated into the Mastra Software are
   licensed under the original license provided by the owner of the

package/dist/index.cjs

@@ -148,7 +148,8 @@ function getGithubMetadata(threadMetadata) {
 			...readString(rawSubscription.lastNotificationAt) ? { lastNotificationAt: readString(rawSubscription.lastNotificationAt) } : {},
 			...readString(rawSubscription.lastNotificationKind) ? { lastNotificationKind: readString(rawSubscription.lastNotificationKind) } : {},
 			...rawSubscription.lastNotificationPriority === "medium" || rawSubscription.lastNotificationPriority === "high" ? { lastNotificationPriority: rawSubscription.lastNotificationPriority } : {},
-			...readString(rawSubscription.lastNotificationSummary) ? { lastNotificationSummary: readString(rawSubscription.lastNotificationSummary) } : {}
+			...readString(rawSubscription.lastNotificationSummary) ? { lastNotificationSummary: readString(rawSubscription.lastNotificationSummary) } : {},
+			...readString(rawSubscription.lastNotificationDedupeKey) ? { lastNotificationDedupeKey: readString(rawSubscription.lastNotificationDedupeKey) } : {}
 		});
 	}
 	return {
@@ -278,6 +279,22 @@ function getCommentNotificationSummary(pr, snapshot) {
 	if (!snapshot.latestCommentAuthor || !snapshot.latestCommentBody) return void 0;
 	return `${snapshot.latestCommentAuthor} commented on ${pr}: ${getCommentExcerpt(snapshot.latestCommentBody)}`;
 }
+function isNoisyBotComment(input) {
+	const author = input.author?.toLowerCase();
+	const body = input.body?.toLowerCase().trim();
+	if (!author || !body) return false;
+	if (author.includes("coderabbit")) return body.includes("## review skipped") || body.includes("review_stack_entry_start") || body.includes("review change stack") || body.includes("storage.googleapis.com/coderabbit_public_assets/review-stack") || body.includes("no actionable comments were generated") || body.includes("actions performed") && body.includes("review triggered");
+	if (author === "vercel[bot]") return body.startsWith("[vc]:");
+	if (author === "socket-security[bot]") return body.includes("review the following changes in direct dependencies");
+	if (author === "dane-ai-mastra[bot]") return body.includes("mastra-pr-automation") || body.includes("## pr triage") || body.includes("## pr complexity score");
+	return false;
+}
+function hasNoisyLatestBotComment(snapshot) {
+	return isNoisyBotComment({
+		author: snapshot.latestCommentAuthor,
+		body: snapshot.latestCommentBody
+	});
+}
 const githubActivityNotificationPriority = {
 	high: 0,
 	medium: 1
@@ -297,7 +314,7 @@ function compareGithubActivityNotifications(a, b) {
 	return getGithubActivityNotificationRank(a) - getGithubActivityNotificationRank(b);
 }
 function classifyGithubCommentActivityNotification(input) {
-	if (isBotOnlyActivity(input.snapshot)) return void 0;
+	if (isBotOnlyActivity(input.snapshot) || hasNoisyLatestBotComment(input.snapshot)) return void 0;
 	const summary = getCommentNotificationSummary(`${input.subscription.owner}/${input.subscription.repo}#${input.subscription.number}`, input.snapshot);
 	if (!summary) return void 0;
 	return {
@@ -452,7 +469,7 @@ function classifyGithubActivityNotification(input) {
 		};
 	}
 	if (input.snapshot.ciState === "pending" && input.subscription.lastObservedCiState === "pending") return void 0;
-	if (isBotOnlyActivity(input.snapshot)) return void 0;
+	if (isBotOnlyActivity(input.snapshot) || hasNoisyLatestBotComment(input.snapshot)) return void 0;
 	const commentSummary = getCommentNotificationSummary(pr, input.snapshot);
 	return {
 		kind: "pull-request-activity",
@@ -852,12 +869,6 @@ var GithubSignals = class extends _mastra_core_signals.SignalProvider {
 			return false;
 		}
 		const key = this.#pollingKey(input);
-		for (const [pollingKey, state] of this.#polling.entries()) {
-			if (pollingKey === key) continue;
-			this.#invalidatePollingThread(pollingKey);
-			clearInterval(state.timer);
-			this.#polling.delete(pollingKey);
-		}
 		if (this.#polling.has(key)) return true;
 		const runPoll = (pollOptions = {}) => {
 			this.#pollThread(input, pollOptions).catch((error) => {
@@ -1013,75 +1024,127 @@ var GithubSignals = class extends _mastra_core_signals.SignalProvider {
 			threadId: context?.agent?.threadId ?? threadContext.threadId,
 			resourceId: context?.agent?.resourceId ?? threadContext.resourceId
 		});
+		const prSchema = zod.default.object({
+			number: zod.default.number().int().positive(),
+			owner: zod.default.string().optional(),
+			repo: zod.default.string().optional()
+		});
+		const subscribeSchema = zod.default.object({
+			prs: zod.default.array(prSchema).min(1),
+			mode: zod.default.enum(["working", "review"]).optional()
+		});
+		const unsubscribeSchema = zod.default.object({
+			prs: zod.default.array(prSchema).min(1).optional(),
+			all: zod.default.boolean().optional()
+		}).refine((input) => [!!input.prs, input.all === true].filter(Boolean).length === 1, { message: "Provide exactly one of prs or all." });
+		const dedupePrs = (prs) => {
+			const byKey = /* @__PURE__ */ new Map();
+			for (const pr of prs) byKey.set(`${pr.owner ?? ""}/${pr.repo ?? ""}#${pr.number}`, pr);
+			return [...byKey.values()];
+		};
+		const getSubscribeMessage = (result) => result.terminalState ? `Not subscribed to ${result.owner}/${result.repo}#${result.number} in review mode because it is already ${result.terminalState}.` : result.syncResult?.ok === false ? `Subscribed to ${result.owner}/${result.repo}#${result.number} in ${result.mode} mode, but gitcrawl sync failed: ${result.syncResult.error}` : `Subscribed to ${result.owner}/${result.repo}#${result.number} in ${result.mode} mode.`;
+		const getUnsubscribeMessage = (result) => result.removed ? `Unsubscribed from ${result.owner}/${result.repo}#${result.number}.` : `No GitHub subscription found for ${result.owner}/${result.repo}#${result.number}.`;
 		return {
 			...args.tools,
 			github_subscribe_pr: createGithubTool({
 				id: "github_subscribe_pr",
-				description: "Subscribe this thread to a GitHub pull request. Use review mode for new commits, authorized latest PR comments, review-thread-state changes including all threads resolved, and PR close or merge updates. Use working mode for all actionable PR activity. Omitted mode defaults to working. Do not subscribe for a one-off inspection.",
-				inputSchema: zod.default.object({
-					number: zod.default.number().int().positive(),
-					owner: zod.default.string().optional(),
-					repo: zod.default.string().optional(),
-					mode: zod.default.enum(["working", "review"]).optional()
-				}),
+				description: "Subscribe this thread to one or more GitHub pull requests. Use review mode for new commits, authorized latest PR comments, review-thread-state changes including all threads resolved, and PR close or merge updates. Use working mode for all actionable PR activity. Omitted mode defaults to working. Do not subscribe for a one-off inspection.",
+				inputSchema: subscribeSchema,
 				execute: async (input, context) => {
+					const subscribeInput = input;
 					const executionThreadContext = getExecutionThreadContext(context);
-					const result = await this.#subscribe({
-						id: `github-tool-subscribe-${(0, crypto.randomUUID)()}`,
-						owner: input.owner,
-						repo: input.repo,
-						number: input.number,
-						mode: normalizeGithubSubscriptionMode(input.mode),
-						threadId: executionThreadContext.threadId,
-						resourceId: executionThreadContext.resourceId
-					});
-					if (result.terminalState) return {
-						subscribed: false,
-						mode: result.mode,
-						terminalState: result.terminalState,
-						reason: "terminal",
-						owner: result.owner,
-						repo: result.repo,
-						number: result.number,
-						message: `Not subscribed to ${result.owner}/${result.repo}#${result.number} in review mode because it is already ${result.terminalState}.`
-					};
-					return {
-						subscribed: true,
-						mode: result.mode,
-						owner: result.owner,
-						repo: result.repo,
-						number: result.number,
-						syncStatus: result.syncResult?.ok === false ? "error" : result.syncResult ? "success" : void 0,
-						message: result.syncResult?.ok === false ? `Subscribed to ${result.owner}/${result.repo}#${result.number} in ${result.mode} mode, but gitcrawl sync failed: ${result.syncResult.error}` : `Subscribed to ${result.owner}/${result.repo}#${result.number} in ${result.mode} mode.`
+					const mode = normalizeGithubSubscriptionMode(subscribeInput.mode);
+					const requestedPrs = dedupePrs(subscribeInput.prs);
+					const results = [];
+					for (const pr of requestedPrs) try {
+						const result = await this.#subscribe({
+							id: `github-tool-subscribe-${(0, crypto.randomUUID)()}`,
+							owner: pr.owner,
+							repo: pr.repo,
+							number: pr.number,
+							mode,
+							threadId: executionThreadContext.threadId,
+							resourceId: executionThreadContext.resourceId
+						});
+						results.push({
+							subscribed: !result.terminalState,
+							mode: result.mode ?? mode,
+							owner: result.owner,
+							repo: result.repo,
+							number: result.number,
+							...result.terminalState ? {
+								terminalState: result.terminalState,
+								reason: "terminal"
+							} : {},
+							syncStatus: result.syncResult?.ok === false ? "error" : result.syncResult ? "success" : void 0,
+							message: getSubscribeMessage(result)
+						});
+					} catch (error) {
+						results.push({
+							subscribed: false,
+							mode,
+							owner: pr.owner ?? "",
+							repo: pr.repo ?? "",
+							number: pr.number,
+							reason: "error",
+							syncStatus: "error",
+							message: `Could not subscribe to PR #${pr.number}: ${error instanceof Error ? error.message : String(error)}`
+						});
+					}
+					const subscriptions = await this.#getSubscriptionSummaries(executionThreadContext).catch(() => []);
+					const response = {
+						subscribed: results.some((result) => result.subscribed),
+						results,
+						subscriptions
 					};
+					if (results.length === 1) Object.assign(response, results[0]);
+					return response;
 				}
 			}),
 			github_unsubscribe_pr: createGithubTool({
 				id: "github_unsubscribe_pr",
-				description: "Unsubscribe this thread from a GitHub pull request.",
-				inputSchema: zod.default.object({
-					number: zod.default.number().int().positive(),
-					owner: zod.default.string().optional(),
-					repo: zod.default.string().optional()
-				}),
+				description: "Unsubscribe this thread from one or more GitHub pull requests, or unsubscribe all current PRs.",
+				inputSchema: unsubscribeSchema,
 				execute: async (input, context) => {
 					const executionThreadContext = getExecutionThreadContext(context);
-					const result = await this.#unsubscribe({
-						id: `github-tool-unsubscribe-${(0, crypto.randomUUID)()}`,
-						owner: input.owner,
-						repo: input.repo,
-						number: input.number,
-						threadId: executionThreadContext.threadId,
-						resourceId: executionThreadContext.resourceId
-					});
-					return {
-						unsubscribed: result.removed ?? false,
-						owner: result.owner,
-						repo: result.repo,
-						number: result.number,
-						remainingSubscriptions: result.remainingSubscriptions,
-						message: result.removed ? `Unsubscribed from ${result.owner}/${result.repo}#${result.number}.` : `No GitHub subscription found for ${result.owner}/${result.repo}#${result.number}.`
+					const unsubscribeInput = input;
+					const requestedPrs = unsubscribeInput.all ? await this.#getSubscriptionSummaries(executionThreadContext) : dedupePrs(unsubscribeInput.prs ?? []);
+					const results = [];
+					for (const pr of requestedPrs) try {
+						const result = await this.#unsubscribe({
+							id: `github-tool-unsubscribe-${(0, crypto.randomUUID)()}`,
+							owner: pr.owner,
+							repo: pr.repo,
+							number: pr.number,
+							threadId: executionThreadContext.threadId,
+							resourceId: executionThreadContext.resourceId
+						});
+						results.push({
+							owner: result.owner,
+							repo: result.repo,
+							number: result.number,
+							unsubscribed: result.removed ?? false,
+							message: getUnsubscribeMessage(result)
+						});
+					} catch (error) {
+						results.push({
+							owner: pr.owner ?? "",
+							repo: pr.repo ?? "",
+							number: pr.number,
+							unsubscribed: false,
+							reason: "error",
+							message: `Could not unsubscribe from PR #${pr.number}: ${error instanceof Error ? error.message : String(error)}`
+						});
+					}
+					const subscriptions = await this.#getSubscriptionSummaries(executionThreadContext).catch(() => []);
+					const response = {
+						unsubscribed: results.some((result) => result.unsubscribed),
+						results,
+						subscriptions,
+						remainingSubscriptions: subscriptions.length
 					};
+					if (results.length === 1) Object.assign(response, results[0]);
+					return response;
 				}
 			})
 		};
@@ -1120,6 +1183,19 @@ var GithubSignals = class extends _mastra_core_signals.SignalProvider {
 	#invalidatePollingThread(key) {
 		this.#pollingThreadGenerations.set(key, (this.#pollingThreadGenerations.get(key) ?? 0) + 1);
 	}
+	#invalidatePollingForThread(input) {
+		const key = this.#pollingKey(input);
+		this.#invalidatePollingThread(key);
+		const state = this.#polling.get(key);
+		if (!state) return;
+		clearInterval(state.timer);
+		this.#polling.delete(key);
+		this.#notifyPollingChanged({
+			threadId: input.threadId,
+			resourceId: input.resourceId,
+			running: false
+		});
+	}
 	#getNotificationAgent(_input) {
 		if (this.#agent) return this.#agent;
 		const agentId = _input?.agentId ?? this.#options.agentId;
@@ -1129,6 +1205,16 @@ var GithubSignals = class extends _mastra_core_signals.SignalProvider {
 		const { loadedThread } = await this.#loadThread(input);
 		return getGithubMetadata(loadedThread.metadata).subscriptions;
 	}
+	async #getSubscriptionSummaries(input) {
+		return (await this.#getThreadSubscriptions(input)).map((subscription) => ({
+			owner: subscription.owner,
+			repo: subscription.repo,
+			number: subscription.number,
+			subscribedAt: subscription.subscribedAt,
+			updatedAt: subscription.updatedAt,
+			...subscription.lastSyncStatus ? { lastSyncStatus: subscription.lastSyncStatus } : {}
+		}));
+	}
 	#notifySubscriptionsChanged(input) {
 		this.#subscriptionsChangedHandler?.(input);
 	}
@@ -1224,6 +1310,7 @@ var GithubSignals = class extends _mastra_core_signals.SignalProvider {
 						nextSubscription.lastNotificationKind = primaryNotification.kind;
 						nextSubscription.lastNotificationPriority = primaryNotification.priority;
 						nextSubscription.lastNotificationSummary = primaryNotification.summary;
+						nextSubscription.lastNotificationDedupeKey = primaryNotification.dedupeKey;
 						if (!terminalState) shouldKeepSubscription = notifications.every((notification) => notification.kind !== "pull-request-merged");
 					}
 				}
@@ -1374,6 +1461,10 @@ var GithubSignals = class extends _mastra_core_signals.SignalProvider {
 		if (!comments.some((comment) => comment.body || comment.url || comment.updatedAt)) return snapshot;
 		for (const comment of comments) {
 			if (isCurrentGeneration && !isCurrentGeneration()) return snapshot;
+			if (isNoisyBotComment({
+				author: comment.author,
+				body: comment.body
+			})) continue;
 			if (!await this.#isAuthorizedAuthor(owner, repo, comment.author, {
 				authorType: comment.authorType,
 				isBot: comment.isBot
@@ -1479,15 +1570,20 @@ var GithubSignals = class extends _mastra_core_signals.SignalProvider {
 				if (!input.isCurrentGeneration()) return [];
 				if (!authorized) continue;
 			}
-			notificationInputs.push(this.#createGithubNotificationInput({
+			const notificationInput = this.#createGithubNotificationInput({
 				subscription: input.subscription,
 				snapshot: input.snapshot,
 				notification,
 				dedupeSuffix: input.snapshot.contentHash ?? input.snapshot.githubUpdatedAt ?? String(Date.now()),
 				previousGithubUpdatedAt: input.previousGithubUpdatedAt,
 				previousContentHash: input.previousContentHash
-			}));
-			sent.push(notification);
+			});
+			if (input.subscription.lastNotificationDedupeKey ? input.subscription.lastNotificationDedupeKey === notificationInput.dedupeKey : input.subscription.lastNotificationKind === notification.kind && input.subscription.lastNotificationSummary === notification.summary) continue;
+			notificationInputs.push(notificationInput);
+			sent.push({
+				...notification,
+				dedupeKey: notificationInput.dedupeKey
+			});
 		}
 		if (notificationInputs.length > 0) {
 			const target = {
@@ -1542,7 +1638,12 @@ var GithubSignals = class extends _mastra_core_signals.SignalProvider {
 			...existing?.lastObservedReviewStateHash ? { lastObservedReviewStateHash: existing.lastObservedReviewStateHash } : {},
 			...existing?.lastObservedCommentUrl ? { lastObservedCommentUrl: existing.lastObservedCommentUrl } : {},
 			...existing?.lastObservedCommentAuthor ? { lastObservedCommentAuthor: existing.lastObservedCommentAuthor } : {},
-			...typeof existing?.lastObservedCommentIsBot === "boolean" ? { lastObservedCommentIsBot: existing.lastObservedCommentIsBot } : {}
+			...typeof existing?.lastObservedCommentIsBot === "boolean" ? { lastObservedCommentIsBot: existing.lastObservedCommentIsBot } : {},
+			...existing?.lastNotificationAt ? { lastNotificationAt: existing.lastNotificationAt } : {},
+			...existing?.lastNotificationKind ? { lastNotificationKind: existing.lastNotificationKind } : {},
+			...existing?.lastNotificationPriority ? { lastNotificationPriority: existing.lastNotificationPriority } : {},
+			...existing?.lastNotificationSummary ? { lastNotificationSummary: existing.lastNotificationSummary } : {},
+			...existing?.lastNotificationDedupeKey ? { lastNotificationDedupeKey: existing.lastNotificationDedupeKey } : {}
 		};
 		let syncResult;
 		let baselineSnapshot;
@@ -1570,8 +1671,12 @@ var GithubSignals = class extends _mastra_core_signals.SignalProvider {
 			if (snapshot) applySnapshotCursor(subscription, snapshot);
 		} else subscription.lastSyncStatus = "skipped";
 		const terminalState = mode === "review" ? getGithubTerminalState(baselineSnapshot) : void 0;
+		const subscriptions = terminalState ? githubMetadata.subscriptions.filter((candidate) => !(candidate.owner === owner && candidate.repo === repo && candidate.number === input.number)) : existingIndex >= 0 ? githubMetadata.subscriptions.map((existingSubscription, index) => index === existingIndex ? subscription : existingSubscription) : [...githubMetadata.subscriptions, subscription];
 		if (terminalState) {
-			const subscriptions = githubMetadata.subscriptions.filter((candidate) => !(candidate.owner === owner && candidate.repo === repo && candidate.number === input.number));
+			this.#invalidatePollingForThread({
+				threadId: input.threadId,
+				resourceId: input.resourceId
+			});
 			await threadStore.saveThread({ thread: {
 				...loadedThread,
 				id: input.threadId,
@@ -1589,6 +1694,10 @@ var GithubSignals = class extends _mastra_core_signals.SignalProvider {
 				threadId: input.threadId,
 				resourceId: input.resourceId
 			});
+			else await this.startPollingForThread({
+				threadId: input.threadId,
+				resourceId: input.resourceId
+			});
 			return {
 				owner,
 				repo,
@@ -1598,7 +1707,10 @@ var GithubSignals = class extends _mastra_core_signals.SignalProvider {
 				syncResult
 			};
 		}
-		const subscriptions = [subscription];
+		this.#invalidatePollingForThread({
+			threadId: input.threadId,
+			resourceId: input.resourceId
+		});
 		await threadStore.saveThread({ thread: {
 			...loadedThread,
 			id: input.threadId,
@@ -1638,6 +1750,10 @@ var GithubSignals = class extends _mastra_core_signals.SignalProvider {
 		const subscriptions = githubMetadata.subscriptions.filter((subscription) => !(subscription.owner === owner && subscription.repo === repo && subscription.number === input.number));
 		const removed = subscriptions.length !== githubMetadata.subscriptions.length;
 		if (removed) {
+			this.#invalidatePollingForThread({
+				threadId: input.threadId,
+				resourceId: input.resourceId
+			});
 			await threadStore.saveThread({ thread: {
 				...loadedThread,
 				id: input.threadId,
@@ -1655,6 +1771,10 @@ var GithubSignals = class extends _mastra_core_signals.SignalProvider {
 				threadId: input.threadId,
 				resourceId: input.resourceId
 			});
+			else await this.startPollingForThread({
+				threadId: input.threadId,
+				resourceId: input.resourceId
+			});
 		}
 		return {
 			owner,