feat(go): referral support (#28345)

Co-authored-by: Jack <jack@anoma.ly>

packages/console/core/src/schema/referral.sql.ts

@@ -0,0 +1,28 @@
+import { bigint, mysqlTable, primaryKey, uniqueIndex } from "drizzle-orm/mysql-core"
+import { timestamps, ulid, utc, workspaceColumns } from "../drizzle/types"
+import { workspaceIndexes } from "./workspace.sql"
+
+export const ReferralTable = mysqlTable(
+  "referral",
+  {
+    ...workspaceColumns,
+    ...timestamps,
+    inviteeAccountID: ulid("invitee_account_id").notNull(),
+  },
+  (table) => [
+    ...workspaceIndexes(table),
+    uniqueIndex("referral_invitee_account_id").on(table.inviteeAccountID),
+  ],
+)
+
+export const ReferralRewardTable = mysqlTable(
+  "referral_reward",
+  {
+    workspaceID: ulid("workspace_id").notNull(),
+    referralID: ulid("referral_id").notNull(),
+    ...timestamps,
+    amount: bigint("amount", { mode: "number" }).notNull(),
+    timeApplied: utc("time_applied"),
+  },
+  (table) => [primaryKey({ columns: [table.workspaceID, table.referralID] })],
+)

packages/console/core/src/schema/workspace.sql.ts

@@ -6,10 +6,14 @@ export const WorkspaceTable = mysqlTable(
   {
     id: ulid("id").notNull().primaryKey(),
     slug: varchar("slug", { length: 255 }),
+    referralCode: varchar("referral_code", { length: 10 }),
     name: varchar("name", { length: 255 }).notNull(),
     ...timestamps,
   },
-  (table) => [uniqueIndex("slug").on(table.slug)],
+  (table) => [
+    uniqueIndex("slug").on(table.slug),
+    uniqueIndex("referral_code").on(table.referralCode),
+  ],
 )
 
 export function workspaceIndexes(table: any) {