@inkeep/agents-core 0.64.10 → 0.65.1

package/dist/data-access/runtime/scheduledTriggerUsers.d.ts

@@ -0,0 +1,54 @@
+import { AgentsRunDatabaseClient } from "../../db/runtime/runtime-client.js";
+
+//#region src/data-access/runtime/scheduledTriggerUsers.d.ts
+declare const getScheduledTriggerUsers: (db: AgentsRunDatabaseClient) => (params: {
+  tenantId: string;
+  scheduledTriggerId: string;
+}) => Promise<{
+  tenantId: string;
+  scheduledTriggerId: string;
+  userId: string;
+  createdAt: string;
+}[]>;
+declare const createScheduledTriggerUser: (db: AgentsRunDatabaseClient) => (params: {
+  tenantId: string;
+  scheduledTriggerId: string;
+  userId: string;
+}) => Promise<{
+  tenantId: string;
+  createdAt: string;
+  userId: string;
+  scheduledTriggerId: string;
+}>;
+declare const deleteScheduledTriggerUser: (db: AgentsRunDatabaseClient) => (params: {
+  tenantId: string;
+  scheduledTriggerId: string;
+  userId: string;
+}) => Promise<void>;
+declare const setScheduledTriggerUsers: (db: AgentsRunDatabaseClient) => (params: {
+  tenantId: string;
+  scheduledTriggerId: string;
+  userIds: string[];
+}) => Promise<void>;
+declare const getScheduledTriggerUserCount: (db: AgentsRunDatabaseClient) => (params: {
+  tenantId: string;
+  scheduledTriggerId: string;
+}) => Promise<number>;
+declare const getTriggerIdsWithUser: (db: AgentsRunDatabaseClient) => (params: {
+  tenantId: string;
+  projectId: string;
+  userId: string;
+}) => Promise<{
+  id: string;
+}[]>;
+declare const removeUserFromProjectScheduledTriggers: (db: AgentsRunDatabaseClient) => (params: {
+  tenantId: string;
+  projectId: string;
+  userId: string;
+}) => Promise<void>;
+declare const getScheduledTriggerUsersBatch: (db: AgentsRunDatabaseClient) => (params: {
+  tenantId: string;
+  scheduledTriggerIds: string[];
+}) => Promise<Map<string, string[]>>;
+//#endregion
+export { createScheduledTriggerUser, deleteScheduledTriggerUser, getScheduledTriggerUserCount, getScheduledTriggerUsers, getScheduledTriggerUsersBatch, getTriggerIdsWithUser, removeUserFromProjectScheduledTriggers, setScheduledTriggerUsers };

package/dist/data-access/runtime/scheduledTriggerUsers.js

@@ -0,0 +1,62 @@
+import { scheduledTriggerUsers, scheduledTriggers } from "../../db/runtime/runtime-schema.js";
+import { and, asc, count, eq, inArray } from "drizzle-orm";
+
+//#region src/data-access/runtime/scheduledTriggerUsers.ts
+const getScheduledTriggerUsers = (db) => async (params) => {
+	return await db.select().from(scheduledTriggerUsers).where(and(eq(scheduledTriggerUsers.tenantId, params.tenantId), eq(scheduledTriggerUsers.scheduledTriggerId, params.scheduledTriggerId))).orderBy(asc(scheduledTriggerUsers.createdAt));
+};
+const createScheduledTriggerUser = (db) => async (params) => {
+	return (await db.insert(scheduledTriggerUsers).values({
+		tenantId: params.tenantId,
+		scheduledTriggerId: params.scheduledTriggerId,
+		userId: params.userId
+	}).onConflictDoNothing().returning())[0];
+};
+const deleteScheduledTriggerUser = (db) => async (params) => {
+	await db.delete(scheduledTriggerUsers).where(and(eq(scheduledTriggerUsers.tenantId, params.tenantId), eq(scheduledTriggerUsers.scheduledTriggerId, params.scheduledTriggerId), eq(scheduledTriggerUsers.userId, params.userId)));
+};
+const setScheduledTriggerUsers = (db) => async (params) => {
+	await db.transaction(async (tx) => {
+		await tx.delete(scheduledTriggerUsers).where(and(eq(scheduledTriggerUsers.tenantId, params.tenantId), eq(scheduledTriggerUsers.scheduledTriggerId, params.scheduledTriggerId)));
+		if (params.userIds.length > 0) await tx.insert(scheduledTriggerUsers).values(params.userIds.map((userId) => ({
+			tenantId: params.tenantId,
+			scheduledTriggerId: params.scheduledTriggerId,
+			userId
+		})));
+	});
+};
+const getScheduledTriggerUserCount = (db) => async (params) => {
+	return (await db.select({ count: count() }).from(scheduledTriggerUsers).where(and(eq(scheduledTriggerUsers.tenantId, params.tenantId), eq(scheduledTriggerUsers.scheduledTriggerId, params.scheduledTriggerId))))[0]?.count ?? 0;
+};
+const getTriggerIdsWithUser = (db) => async (params) => {
+	return db.select({ id: scheduledTriggers.id }).from(scheduledTriggers).innerJoin(scheduledTriggerUsers, and(eq(scheduledTriggerUsers.tenantId, scheduledTriggers.tenantId), eq(scheduledTriggerUsers.scheduledTriggerId, scheduledTriggers.id))).where(and(eq(scheduledTriggers.tenantId, params.tenantId), eq(scheduledTriggers.projectId, params.projectId), eq(scheduledTriggerUsers.userId, params.userId)));
+};
+const removeUserFromProjectScheduledTriggers = (db) => async (params) => {
+	await db.transaction(async (tx) => {
+		const triggerIdsWithUser = await getTriggerIdsWithUser(tx)(params);
+		if (triggerIdsWithUser.length === 0) return;
+		const triggerIds = triggerIdsWithUser.map((t) => t.id);
+		await tx.delete(scheduledTriggerUsers).where(and(eq(scheduledTriggerUsers.tenantId, params.tenantId), inArray(scheduledTriggerUsers.scheduledTriggerId, triggerIds), eq(scheduledTriggerUsers.userId, params.userId)));
+		const triggersWithRemainingUsers = await tx.select({ triggerId: scheduledTriggerUsers.scheduledTriggerId }).from(scheduledTriggerUsers).where(and(eq(scheduledTriggerUsers.tenantId, params.tenantId), inArray(scheduledTriggerUsers.scheduledTriggerId, triggerIds))).groupBy(scheduledTriggerUsers.scheduledTriggerId);
+		const triggerIdsWithRemainingUsers = new Set(triggersWithRemainingUsers.map((r) => r.triggerId));
+		const emptyTriggerIds = triggerIds.filter((id) => !triggerIdsWithRemainingUsers.has(id));
+		if (emptyTriggerIds.length > 0) await tx.update(scheduledTriggers).set({
+			enabled: false,
+			updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+		}).where(and(eq(scheduledTriggers.tenantId, params.tenantId), inArray(scheduledTriggers.id, emptyTriggerIds), eq(scheduledTriggers.enabled, true)));
+	});
+};
+const getScheduledTriggerUsersBatch = (db) => async (params) => {
+	if (params.scheduledTriggerIds.length === 0) return /* @__PURE__ */ new Map();
+	const rows = await db.select().from(scheduledTriggerUsers).where(and(eq(scheduledTriggerUsers.tenantId, params.tenantId), inArray(scheduledTriggerUsers.scheduledTriggerId, params.scheduledTriggerIds))).orderBy(asc(scheduledTriggerUsers.createdAt));
+	const result = /* @__PURE__ */ new Map();
+	for (const id of params.scheduledTriggerIds) result.set(id, []);
+	for (const row of rows) {
+		const users = result.get(row.scheduledTriggerId);
+		if (users) users.push(row.userId);
+	}
+	return result;
+};
+
+//#endregion
+export { createScheduledTriggerUser, deleteScheduledTriggerUser, getScheduledTriggerUserCount, getScheduledTriggerUsers, getScheduledTriggerUsersBatch, getTriggerIdsWithUser, removeUserFromProjectScheduledTriggers, setScheduledTriggerUsers };

package/dist/data-access/runtime/scheduledTriggers.d.ts

@@ -30,6 +30,7 @@ declare const listScheduledTriggersPaginated: (db: AgentsRunDatabaseClient) => (
     createdBy: string | null;
     nextRunAt: string | null;
     ref: string;
+    dispatchDelayMs: number | null;
     agentId: string;
     projectId: string;
     tenantId: string;

package/dist/data-access/runtime/tasks.d.ts

@@ -7,20 +7,20 @@ import { TaskInsert, TaskSelect } from "../../types/entities.js";
 
 //#region src/data-access/runtime/tasks.d.ts
 declare const createTask: (db: AgentsRunDatabaseClient) => (params: TaskInsert) => Promise<{
-  tenantId: string;
-  projectId: string;
+  status: string;
   id: string;
+  tenantId: string;
+  createdAt: string;
   metadata: TaskMetadataConfig | null;
   ref: {
     type: "commit" | "tag" | "branch";
     name: string;
     hash: string;
   } | null;
-  createdAt: string;
   updatedAt: string;
+  projectId: string;
   agentId: string;
   subAgentId: string;
-  status: string;
   contextId: string;
 }>;
 declare const getTask: (db: AgentsRunDatabaseClient) => (params: {

package/dist/db/manage/dolt-safe-jsonb.d.ts

@@ -1,4 +1,4 @@
-import * as drizzle_orm_pg_core0 from "drizzle-orm/pg-core";
+import * as drizzle_orm_pg_core220 from "drizzle-orm/pg-core";
 
 //#region src/db/manage/dolt-safe-jsonb.d.ts
 declare function encodeBackslashes(value: unknown): unknown;
@@ -7,6 +7,6 @@ declare function decodeBackslashes(value: unknown): unknown;
  * Drop-in replacement for drizzle-orm's `jsonb()`.
  * Encodes backslashes on write and decodes on read to work around the Doltgres bug.
  */
-declare function jsonb(name: string): ReturnType<typeof drizzle_orm_pg_core0.jsonb>;
+declare function jsonb(name: string): ReturnType<typeof drizzle_orm_pg_core220.jsonb>;
 //#endregion
 export { decodeBackslashes, encodeBackslashes, jsonb };