define schema

Author: tknkaa

apps/shared/hai.ts

@@ -1,21 +1,65 @@
+import { sql } from "drizzle-orm";
 import {
-  boolean,
+  check,
   integer,
   pgTable,
+  serial,
   text,
   timestamp,
-  varchar,
+  boolean,
+  primaryKey,
+  pgEnum,
+  index,
 } from "drizzle-orm/pg-core";
 
-export const usersTable = pgTable("users", {
-  id: integer().primaryKey().generatedAlwaysAsIdentity(),
-  name: varchar({ length: 255 }).notNull(),
-  age: integer().notNull(),
-  email: varchar({ length: 255 }).notNull().unique(),
+export const haiyama = pgTable("haiyama", {
+  id: text("id").primaryKey(),
 });
 
-// better-auth
+export const haiKindEnum = pgEnum("hai_kind", [
+  "manzu",
+  "pinzu",
+  "souzu",
+  "jihai",
+]);
 
+export const hai = pgTable("hai", {
+  id: serial("id").primaryKey(),
+  haiyamaId: text("haiyama_id")
+    .notNull()
+    .references(() => haiyama.id, { onDelete: "cascade" }),
+  kind: haiKindEnum("kind").notNull(), // "manzu" | "pinzu" | "souzu" | "jihai"
+  value: text("value").notNull(), // 1~9 or "ton" ~ "tyun"
+  order: integer("order").notNull(), // 0~17
+  index: integer("index").notNull(), // haiToIndex
+});
+
+// relation between user and haiyama
+// TODO: index
+export const kyoku = pgTable(
+  "kyoku",
+  {
+    userId: text("user_id")
+      .notNull()
+      .references(() => user.id, { onDelete: "cascade" }),
+    haiyamaId: text("haiyama_id")
+      .notNull()
+      .references(() => haiyama.id, { onDelete: "cascade" }),
+    didAgari: boolean("did_agari").notNull(),
+    agariJunme: integer("agari_junme"),
+  },
+  (table) => [
+    primaryKey({ columns: [table.userId, table.haiyamaId] }),
+    index("kyoku_user_id_idx").on(table.userId),
+    index("kyoku_haiyama_id_idx").on(table.haiyamaId),
+    check(
+      "agari_consistency",
+      sql`(${table.didAgari} = false) OR (${table.didAgari} = true AND ${table.agariJunme} IS NOT NULL)`,
+    ),
+  ],
+);
+
+// better-auth
 export const user = pgTable("user", {
   id: text("id").primaryKey(),
   name: text("name").notNull(),

rr/app/lib/db/schema.ts

@@ -0,0 +1,99 @@
+export type Hai =
+  | {
+      kind: SuhaiKind;
+      value: number;
+    }
+  | {
+      kind: "jihai";
+      value: JihaiValue;
+    };
+
+export type SuhaiKind = "manzu" | "pinzu" | "souzu";
+export type HaiKind = SuhaiKind | "jihai";
+
+export type HaiWithID = {
+  kind: HaiKind;
+  value: JihaiValue | number;
+  id: number;
+};
+
+const kindOrder: HaiKind[] = ["manzu", "pinzu", "souzu", "jihai"];
+
+const jihaiOrder: JihaiValue[] = [
+  "ton",
+  "nan",
+  "sya",
+  "pei",
+  "haku",
+  "hatsu",
+  "tyun",
+];
+
+export function haiToIndex(hai: Hai): number {
+  if (hai.kind === "jihai") {
+    return 27 + 1 + jihaiOrder.indexOf(hai.value);
+  }
+  return 9 * kindOrder.indexOf(hai.kind) + hai.value;
+}
+
+export function indexToHai(index: number): Hai {
+  if (index <= 0 || index > 34) {
+    throw new Error("invalid index");
+  }
+  const hai = constructHai(
+    kindOrder[Math.floor(index / 9)],
+    Math.floor(index / 9) === 4 ? jihaiOrder[index % 9] : index % 9,
+  );
+  return hai;
+}
+
+export type JihaiValue =
+  | "ton"
+  | "nan"
+  | "sya"
+  | "pei"
+  | "haku"
+  | "hatsu"
+  | "tyun";
+
+export function constructHai(kind: HaiKind, value: number | JihaiValue): Hai {
+  if (kind === "manzu" || kind === "pinzu" || kind === "souzu") {
+    if (typeof value === "number") {
+      return {
+        kind,
+        value,
+      };
+    }
+    throw new Error("suhai requires number");
+  }
+  if (typeof value === "string") {
+    return {
+      kind,
+      value,
+    };
+  }
+  throw new Error("jihai requires JihaiValue");
+}
+
+export function sortTehai(haiArray: Hai[]): Hai[] {
+  return haiArray.sort((a, b) => haiToIndex(a) - haiToIndex(b));
+}
+
+// To store hai in DB
+export type DBHai = {
+  haiyamaId: string;
+  kind: HaiKind;
+  value: string;
+  order: number;
+  index: number;
+};
+
+export function haiToDBHai(hai: Hai, haiyamaId: string, order: number): DBHai {
+  return {
+    haiyamaId,
+    kind: hai.kind,
+    value: String(hai.value),
+    order,
+    index: haiToIndex(hai),
+  };
+}