BA
Better Auth•2mo ago
Vimes

Set active org on login

I want each user to always have one active org set when they log inn, no user should ever find themselves without an active org. Docs at https://www.better-auth.com/docs/plugins/organization indicate that I can use a database hook to achive this 
export const auth = betterAuth({
  databaseHooks: {
      session: {
          create: {
              before: async(session)=>{
                  const organization = await getActiveOrganization(session.userId)
                  return {
                    data: {
                      ...session,
                      activeOrganizationId: organization.id
                    }
                  }
              }
          }
      }
  }
})
export const auth = betterAuth({
  databaseHooks: {
      session: {
          create: {
              before: async(session)=>{
                  const organization = await getActiveOrganization(session.userId)
                  return {
                    data: {
                      ...session,
                      activeOrganizationId: organization.id
                    }
                  }
              }
          }
      }
  }
})
My question is, is getActiveOrganization somehow an already made function? It seems to not exist and not work, so would I have to to create this function myself? And what benefits would I get from doing it as DB hooks instead of adding it to my login flow code? Thi seems to take in session.userId so presumably I'd have to write all the code I'd need to write anywhere else to set an active org.
Organization | Better Auth
The organization plugin allows you to manage your organization's members and teams.
3 Replies
bekacru
bekacru•2mo ago
use auth.api.getFullOrganization on the server
Vimes
VimesOP•2mo ago
What would be the best way to find a users org based on ID. Since I run it as a before-session hook I can't use headers. auth.api.getFullOrganization expects a full org ID, but I don't have the ID yet. I'd need to somehow query for the org my user is a member of before I can use headers. The only solution I've found is to use SQL/Kysley, but htis seems insecure compared to the native API
Jacob
Jacob•2mo ago
You need to use the databaseHooks / user / create / after - then do a direct DB call to make the 1st organization. I use const orgId = nanoid(25); which I assume is close to what better-auth uses. Then when the user logs in the session / create / before will pull in your active org. the function they show getActiveOrganization is something you will have to make yourself. 
async function getActiveOrganization(userId: string) {
  const userMember = await db.query.member.findFirst({
    where: and(eq(memberTable.userId, userId), eq(memberTable.role, "owner")),
    with: {
      organization: true,
    },
  });

  if (!userMember?.organization) {
    throw new Error("No active organization found");
  }

  return userMember.organization;
}
async function getActiveOrganization(userId: string) {
  const userMember = await db.query.member.findFirst({
    where: and(eq(memberTable.userId, userId), eq(memberTable.role, "owner")),
    with: {
      organization: true,
    },
  });

  if (!userMember?.organization) {
    throw new Error("No active organization found");
  }

  return userMember.organization;
}
And org create 
            const orgId = nanoid(25);

            await dbTransaction.transaction(async (tx) => {
              // Create organization
              const [newOrg] = await tx
                .insert(organizationTable)
                .values({
                  id: orgId,
                  name: firstName
                    ? `${firstName}'s Organization`
                    : "My Organization",
                  slug: orgId,
                })
                .returning();

              if (!newOrg) {
                throw new Error("Failed to create organization");
              }

              // Set this as the active organization in the session
              await tx
                .update(sessionTable)
                .set({ activeOrganizationId: newOrg.id })
                .where(eq(sessionTable.userId, user.id));
            const orgId = nanoid(25);

            await dbTransaction.transaction(async (tx) => {
              // Create organization
              const [newOrg] = await tx
                .insert(organizationTable)
                .values({
                  id: orgId,
                  name: firstName
                    ? `${firstName}'s Organization`
                    : "My Organization",
                  slug: orgId,
                })
                .returning();

              if (!newOrg) {
                throw new Error("Failed to create organization");
              }

              // Set this as the active organization in the session
              await tx
                .update(sessionTable)
                .set({ activeOrganizationId: newOrg.id })
                .where(eq(sessionTable.userId, user.id));
Im using a combo: 
  const activeMember = await auth.api.getActiveMember({
    headers: await headers(),
  });

  const activeOrg = await auth.api.getFullOrganization({
    headers: await headers(),
  });
  const activeMember = await auth.api.getActiveMember({
    headers: await headers(),
  });

  const activeOrg = await auth.api.getFullOrganization({
    headers: await headers(),
  });
That lets you get the users active org and additional details of the org like the name. If active org has been set in session then you don't need to pass the org id.

Did you find this page helpful?