Function createFromAccount

packages/console/core/src/referral.ts:288–363

(input: { accountID: string; referralCode?: string })

  })

  export async function createFromAccount(input: { accountID: string; referralCode?: string }) {
    const referralCode = normalizeCode(input.referralCode)
    if (!referralCode) return

    return Database.transaction(async (tx) => {
      const code = await tx
        .select({ workspaceID: ReferralCodeTable.workspaceID })
        .from(ReferralCodeTable)
        .innerJoin(WorkspaceTable, eq(WorkspaceTable.id, ReferralCodeTable.workspaceID))
        .where(and(eq(ReferralCodeTable.code, referralCode), isNull(WorkspaceTable.timeDeleted)))
        .then((rows) => rows[0])
      if (!code) throw new Error("Referral code invalid")

      const existingReferral = await tx
        .select({ id: ReferralTable.id })
        .from(ReferralTable)
        .where(and(eq(ReferralTable.inviteeAccountID, input.accountID), isNull(ReferralTable.timeDeleted)))
        .then((rows) => rows[0])
      if (existingReferral) throw new Error("Referral already redeemed")

      const selfReferral = await tx
        .select({ id: UserTable.id })
        .from(UserTable)
        .where(
          and(
            eq(UserTable.workspaceID, code.workspaceID),
            eq(UserTable.accountID, input.accountID),
            isNull(UserTable.timeDeleted),
          ),
        )
        .then((rows) => rows[0])
      if (selfReferral) throw new Error("Self-referral is not allowed")

      const workspaceIDs = await tx
        .select({ workspaceID: UserTable.workspaceID })
        .from(UserTable)
        .where(and(eq(UserTable.accountID, input.accountID), isNull(UserTable.timeDeleted)))
        .then((rows) => rows.map((row) => row.workspaceID))
      if (workspaceIDs.length === 0) return

      const activeLite = await tx
        .select({ id: LiteTable.id })
        .from(LiteTable)
        .where(and(inArray(LiteTable.workspaceID, workspaceIDs), isNull(LiteTable.timeDeleted)))
        .then((rows) => rows[0])
      if (activeLite) return

      const litePayment = await tx
        .select({ id: PaymentTable.id })
        .from(PaymentTable)
        .where(
          and(
            inArray(PaymentTable.workspaceID, workspaceIDs),
            isNull(PaymentTable.timeDeleted),
            sql`JSON_UNQUOTE(JSON_EXTRACT(${PaymentTable.enrichment}, '$.type')) = 'lite'`,
          ),
        )
        .then((rows) => rows[0])