Function ensureCode

packages/console/core/src/referral.ts:32–55

(workspaceID = Actor.workspace())

  }

  async function ensureCode(workspaceID = Actor.workspace()) {
    return Database.use(async (db) => {
      const existing = await db
        .select({ code: ReferralCodeTable.code })
        .from(ReferralCodeTable)
        .where(eq(ReferralCodeTable.workspaceID, workspaceID))
        .then((rows) => rows[0])
      if (existing) return { code: existing.code }

      await db.insert(ReferralCodeTable).ignore().values({
        workspaceID,
        code: generateCode(),
      })

      const created = await db
        .select({ code: ReferralCodeTable.code })
        .from(ReferralCodeTable)
        .where(eq(ReferralCodeTable.workspaceID, workspaceID))
        .then((rows) => rows[0])
      if (created) return { code: created.code }

      throw new Error("Failed to generate referral code")
    })
  }

  export const summary = fn(z.void(), async () => {
    const workspaceID = Actor.workspace()