diff --git a/packages/better-auth/src/plugins/organization/adapter.ts b/packages/better-auth/src/plugins/organization/adapter.ts index 54dc43d2b3..dfa44bf3cd 100644 --- a/packages/better-auth/src/plugins/organization/adapter.ts +++ b/packages/better-auth/src/plugins/organization/adapter.ts @@ -1,4 +1,4 @@ -import type { Session, User } from "../../types"; +import type { GenericEndpointContext, Session, User } from "../../types"; import { getDate } from "../../utils/date"; import type { OrganizationOptions } from "./types"; import type { @@ -378,12 +378,14 @@ export const getOrgAdapter = ( setActiveOrganization: async ( sessionToken: string, organizationId: string | null, + ctx: GenericEndpointContext, ) => { const session = await context.internalAdapter.updateSession( sessionToken, { activeOrganizationId: organizationId, }, + ctx, ); return session as Session; }, @@ -692,12 +694,17 @@ export const getOrgAdapter = ( return invitation; }, - setActiveTeam: async (sessionToken: string, teamId: string | null) => { + setActiveTeam: async ( + sessionToken: string, + teamId: string | null, + ctx: GenericEndpointContext, + ) => { const session = await context.internalAdapter.updateSession( sessionToken, { activeTeamId: teamId, }, + ctx, ); return session as Session; }, diff --git a/packages/better-auth/src/plugins/organization/routes/crud-invites.ts b/packages/better-auth/src/plugins/organization/routes/crud-invites.ts index c91723c8af..0cade57553 100644 --- a/packages/better-auth/src/plugins/organization/routes/crud-invites.ts +++ b/packages/better-auth/src/plugins/organization/routes/crud-invites.ts @@ -601,6 +601,7 @@ export const acceptInvitation = (options: O) => const updatedSession = await adapter.setActiveTeam( session.session.token, teamId, + ctx, ); await setSessionCookie(ctx, { @@ -620,6 +621,7 @@ export const acceptInvitation = (options: O) => await adapter.setActiveOrganization( session.session.token, invitation.organizationId, + ctx, ); if (!acceptedI) { return ctx.json(null, { diff --git a/packages/better-auth/src/plugins/organization/routes/crud-members.ts b/packages/better-auth/src/plugins/organization/routes/crud-members.ts index ed3797fc5d..2314854dc0 100644 --- a/packages/better-auth/src/plugins/organization/routes/crud-members.ts +++ b/packages/better-auth/src/plugins/organization/routes/crud-members.ts @@ -384,7 +384,7 @@ export const removeMember = (options: O) => session.session.activeOrganizationId === toBeRemovedMember.organizationId ) { - await adapter.setActiveOrganization(session.session.token, null); + await adapter.setActiveOrganization(session.session.token, null, ctx); } // Run afterRemoveMember hook @@ -794,7 +794,7 @@ export const leaveOrganization = (options: O) => } await adapter.deleteMember(member.id); if (session.session.activeOrganizationId === ctx.body.organizationId) { - await adapter.setActiveOrganization(session.session.token, null); + await adapter.setActiveOrganization(session.session.token, null, ctx); } return ctx.json(member); }, diff --git a/packages/better-auth/src/plugins/organization/routes/crud-org.ts b/packages/better-auth/src/plugins/organization/routes/crud-org.ts index f2395e073e..f364e7fbf2 100644 --- a/packages/better-auth/src/plugins/organization/routes/crud-org.ts +++ b/packages/better-auth/src/plugins/organization/routes/crud-org.ts @@ -304,6 +304,7 @@ export const createOrganization = ( await adapter.setActiveOrganization( ctx.context.session.session.token, organization.id, + ctx, ); } @@ -315,6 +316,7 @@ export const createOrganization = ( await adapter.setActiveTeam( ctx.context.session.session.token, teamMember.teamId, + ctx, ); } @@ -605,7 +607,7 @@ export const deleteOrganization = ( /** * If the organization is deleted, we set the active organization to null */ - await adapter.setActiveOrganization(session.session.token, null); + await adapter.setActiveOrganization(session.session.token, null, ctx); } const org = await adapter.findOrganizationById(organizationId); @@ -712,7 +714,7 @@ export const getFullOrganization = ( organizationId: organization.id, }); if (!isMember) { - await adapter.setActiveOrganization(session.session.token, null); + await adapter.setActiveOrganization(session.session.token, null, ctx); throw new APIError("FORBIDDEN", { message: ORGANIZATION_ERROR_CODES.USER_IS_NOT_A_MEMBER_OF_THE_ORGANIZATION, @@ -791,6 +793,7 @@ export const setActiveOrganization = ( const updatedSession = await adapter.setActiveOrganization( session.session.token, null, + ctx, ); await setSessionCookie(ctx, { session: updatedSession, @@ -829,7 +832,7 @@ export const setActiveOrganization = ( organizationId, }); if (!isMember) { - await adapter.setActiveOrganization(session.session.token, null); + await adapter.setActiveOrganization(session.session.token, null, ctx); throw new APIError("FORBIDDEN", { message: ORGANIZATION_ERROR_CODES.USER_IS_NOT_A_MEMBER_OF_THE_ORGANIZATION, @@ -845,6 +848,7 @@ export const setActiveOrganization = ( const updatedSession = await adapter.setActiveOrganization( session.session.token, organization.id, + ctx, ); await setSessionCookie(ctx, { session: updatedSession, diff --git a/packages/better-auth/src/plugins/organization/routes/crud-team.ts b/packages/better-auth/src/plugins/organization/routes/crud-team.ts index 48b257d654..9ddcfa0e9c 100644 --- a/packages/better-auth/src/plugins/organization/routes/crud-team.ts +++ b/packages/better-auth/src/plugins/organization/routes/crud-team.ts @@ -700,6 +700,7 @@ export const setActiveTeam = (options: O) => const updatedSession = await adapter.setActiveTeam( session.session.token, null, + ctx, ); await setSessionCookie(ctx, { @@ -745,6 +746,7 @@ export const setActiveTeam = (options: O) => const updatedSession = await adapter.setActiveTeam( session.session.token, team.id, + ctx, ); await setSessionCookie(ctx, {