@inkeep/agents-core 0.47.4 → 0.48.0

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

@@ -0,0 +1,146 @@
+import { workAppSlackChannelAgentConfigs, workAppSlackUserMappings, workAppSlackWorkspaces } from "../../db/runtime/runtime-schema.js";
+import { and, eq } from "drizzle-orm";
+import { nanoid } from "nanoid";
+
+//#region src/data-access/runtime/workAppSlack.ts
+const DEFAULT_CLIENT_ID = "work-apps-slack";
+const createWorkAppSlackWorkspace = (db) => async (data) => {
+	const id = `wsw_${nanoid(21)}`;
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const [result] = await db.insert(workAppSlackWorkspaces).values({
+		id,
+		...data,
+		createdAt: now,
+		updatedAt: now
+	}).returning();
+	return result;
+};
+const findWorkAppSlackWorkspaceByTeamId = (db) => async (tenantId, slackTeamId) => {
+	return (await db.select().from(workAppSlackWorkspaces).where(and(eq(workAppSlackWorkspaces.tenantId, tenantId), eq(workAppSlackWorkspaces.slackTeamId, slackTeamId))).limit(1))[0] || null;
+};
+/**
+* Find a workspace by Slack team ID only (without tenant filter).
+* Slack team IDs are globally unique, so this is safe for resolving
+* the tenant from an incoming Slack event where tenant is unknown.
+*/
+const findWorkAppSlackWorkspaceBySlackTeamId = (db) => async (slackTeamId) => {
+	return (await db.select().from(workAppSlackWorkspaces).where(eq(workAppSlackWorkspaces.slackTeamId, slackTeamId)).limit(1))[0] || null;
+};
+/**
+* Find a workspace by its Nango connection ID.
+*
+* One Nango connection = one OAuth token = one Slack workspace.
+* The nangoConnectionId should be globally unique (not per-tenant).
+* The schema has a unique constraint on nangoConnectionId ensuring this.
+*/
+const findWorkAppSlackWorkspaceByNangoConnectionId = (db) => async (nangoConnectionId) => {
+	return (await db.select().from(workAppSlackWorkspaces).where(eq(workAppSlackWorkspaces.nangoConnectionId, nangoConnectionId)))[0] || null;
+};
+const listWorkAppSlackWorkspacesByTenant = (db) => async (tenantId) => {
+	return db.select().from(workAppSlackWorkspaces).where(eq(workAppSlackWorkspaces.tenantId, tenantId));
+};
+const updateWorkAppSlackWorkspace = (db) => async (id, data) => {
+	const [result] = await db.update(workAppSlackWorkspaces).set({
+		...data,
+		updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+	}).where(eq(workAppSlackWorkspaces.id, id)).returning();
+	return result || null;
+};
+const deleteWorkAppSlackWorkspace = (db) => async (id) => {
+	return (await db.delete(workAppSlackWorkspaces).where(eq(workAppSlackWorkspaces.id, id)).returning()).length > 0;
+};
+const deleteWorkAppSlackWorkspaceByNangoConnectionId = (db) => async (nangoConnectionId) => {
+	return (await db.delete(workAppSlackWorkspaces).where(eq(workAppSlackWorkspaces.nangoConnectionId, nangoConnectionId)).returning()).length > 0;
+};
+const findWorkAppSlackUserMapping = (db) => async (tenantId, slackUserId, slackTeamId, clientId = DEFAULT_CLIENT_ID) => {
+	return (await db.select().from(workAppSlackUserMappings).where(and(eq(workAppSlackUserMappings.tenantId, tenantId), eq(workAppSlackUserMappings.clientId, clientId), eq(workAppSlackUserMappings.slackUserId, slackUserId), eq(workAppSlackUserMappings.slackTeamId, slackTeamId))).limit(1))[0] || null;
+};
+const findWorkAppSlackUserMappingByInkeepUserId = (db) => async (inkeepUserId) => {
+	return db.select().from(workAppSlackUserMappings).where(eq(workAppSlackUserMappings.inkeepUserId, inkeepUserId));
+};
+/**
+* Find a user mapping by Slack user ID and team ID only (ignores tenant).
+* Use this when you need to find the user's tenant from their mapping.
+*/
+const findWorkAppSlackUserMappingBySlackUser = (db) => async (slackUserId, slackTeamId, clientId = DEFAULT_CLIENT_ID) => {
+	return (await db.select().from(workAppSlackUserMappings).where(and(eq(workAppSlackUserMappings.clientId, clientId), eq(workAppSlackUserMappings.slackUserId, slackUserId), eq(workAppSlackUserMappings.slackTeamId, slackTeamId))).limit(1))[0] || null;
+};
+const listWorkAppSlackUserMappingsByTeam = (db) => async (tenantId, slackTeamId) => {
+	return db.select().from(workAppSlackUserMappings).where(and(eq(workAppSlackUserMappings.tenantId, tenantId), eq(workAppSlackUserMappings.slackTeamId, slackTeamId)));
+};
+const createWorkAppSlackUserMapping = (db) => async (data) => {
+	const id = `wsum_${nanoid(21)}`;
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const [result] = await db.insert(workAppSlackUserMappings).values({
+		id,
+		clientId: data.clientId || DEFAULT_CLIENT_ID,
+		...data,
+		linkedAt: now,
+		createdAt: now,
+		updatedAt: now
+	}).returning();
+	return result;
+};
+const deleteWorkAppSlackUserMapping = (db) => async (tenantId, slackUserId, slackTeamId, clientId = DEFAULT_CLIENT_ID) => {
+	return (await db.delete(workAppSlackUserMappings).where(and(eq(workAppSlackUserMappings.tenantId, tenantId), eq(workAppSlackUserMappings.clientId, clientId), eq(workAppSlackUserMappings.slackUserId, slackUserId), eq(workAppSlackUserMappings.slackTeamId, slackTeamId))).returning()).length > 0;
+};
+const deleteAllWorkAppSlackUserMappingsByTeam = (db) => async (tenantId, slackTeamId, clientId = DEFAULT_CLIENT_ID) => {
+	return (await db.delete(workAppSlackUserMappings).where(and(eq(workAppSlackUserMappings.tenantId, tenantId), eq(workAppSlackUserMappings.clientId, clientId), eq(workAppSlackUserMappings.slackTeamId, slackTeamId))).returning()).length;
+};
+const createWorkAppSlackChannelAgentConfig = (db) => async (data) => {
+	const id = `wscac_${nanoid(21)}`;
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const [result] = await db.insert(workAppSlackChannelAgentConfigs).values({
+		id,
+		...data,
+		createdAt: now,
+		updatedAt: now
+	}).returning();
+	return result;
+};
+const findWorkAppSlackChannelAgentConfig = (db) => async (tenantId, slackTeamId, slackChannelId) => {
+	return (await db.select().from(workAppSlackChannelAgentConfigs).where(and(eq(workAppSlackChannelAgentConfigs.tenantId, tenantId), eq(workAppSlackChannelAgentConfigs.slackTeamId, slackTeamId), eq(workAppSlackChannelAgentConfigs.slackChannelId, slackChannelId))).limit(1))[0] || null;
+};
+const listWorkAppSlackChannelAgentConfigsByTeam = (db) => async (tenantId, slackTeamId) => {
+	return db.select().from(workAppSlackChannelAgentConfigs).where(and(eq(workAppSlackChannelAgentConfigs.tenantId, tenantId), eq(workAppSlackChannelAgentConfigs.slackTeamId, slackTeamId)));
+};
+/**
+* Atomic upsert using onConflictDoUpdate to avoid TOCTOU race conditions.
+* Uses the unique constraint on (tenantId, slackTeamId, slackChannelId).
+*/
+const upsertWorkAppSlackChannelAgentConfig = (db) => async (data) => {
+	const id = `wscac_${nanoid(21)}`;
+	const now = (/* @__PURE__ */ new Date()).toISOString();
+	const [result] = await db.insert(workAppSlackChannelAgentConfigs).values({
+		id,
+		...data,
+		createdAt: now,
+		updatedAt: now
+	}).onConflictDoUpdate({
+		target: [
+			workAppSlackChannelAgentConfigs.tenantId,
+			workAppSlackChannelAgentConfigs.slackTeamId,
+			workAppSlackChannelAgentConfigs.slackChannelId
+		],
+		set: {
+			projectId: data.projectId,
+			agentId: data.agentId,
+			agentName: data.agentName,
+			slackChannelName: data.slackChannelName,
+			slackChannelType: data.slackChannelType,
+			enabled: data.enabled,
+			configuredByUserId: data.configuredByUserId,
+			updatedAt: now
+		}
+	}).returning();
+	return result;
+};
+const deleteWorkAppSlackChannelAgentConfig = (db) => async (tenantId, slackTeamId, slackChannelId) => {
+	return (await db.delete(workAppSlackChannelAgentConfigs).where(and(eq(workAppSlackChannelAgentConfigs.tenantId, tenantId), eq(workAppSlackChannelAgentConfigs.slackTeamId, slackTeamId), eq(workAppSlackChannelAgentConfigs.slackChannelId, slackChannelId))).returning()).length > 0;
+};
+const deleteAllWorkAppSlackChannelAgentConfigsByTeam = (db) => async (tenantId, slackTeamId) => {
+	return (await db.delete(workAppSlackChannelAgentConfigs).where(and(eq(workAppSlackChannelAgentConfigs.tenantId, tenantId), eq(workAppSlackChannelAgentConfigs.slackTeamId, slackTeamId))).returning()).length;
+};
+
+//#endregion
+export { createWorkAppSlackChannelAgentConfig, createWorkAppSlackUserMapping, createWorkAppSlackWorkspace, deleteAllWorkAppSlackChannelAgentConfigsByTeam, deleteAllWorkAppSlackUserMappingsByTeam, deleteWorkAppSlackChannelAgentConfig, deleteWorkAppSlackUserMapping, deleteWorkAppSlackWorkspace, deleteWorkAppSlackWorkspaceByNangoConnectionId, findWorkAppSlackChannelAgentConfig, findWorkAppSlackUserMapping, findWorkAppSlackUserMappingByInkeepUserId, findWorkAppSlackUserMappingBySlackUser, findWorkAppSlackWorkspaceByNangoConnectionId, findWorkAppSlackWorkspaceBySlackTeamId, findWorkAppSlackWorkspaceByTeamId, listWorkAppSlackChannelAgentConfigsByTeam, listWorkAppSlackUserMappingsByTeam, listWorkAppSlackWorkspacesByTenant, updateWorkAppSlackWorkspace, upsertWorkAppSlackChannelAgentConfig };