From 19b04f792aa1725e65a5cdec1f7a8cde7cc56481 Mon Sep 17 00:00:00 2001
From: Tau <tau@nowhere>
Date: Mon, 29 Apr 2019 11:06:05 -0400
Subject: [PATCH] idz: Add SqlTitlesRepository

---
 src/idz/db/titles.ts | 51 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 src/idz/db/titles.ts

diff --git a/src/idz/db/titles.ts b/src/idz/db/titles.ts
new file mode 100644
index 0000000..fb1da77
--- /dev/null
+++ b/src/idz/db/titles.ts
@@ -0,0 +1,51 @@
+import { Client } from "pg";
+import * as sql from "sql-bricks";
+
+import { _findProfile } from "./_util";
+import { TitleCode, ExtId } from "../model/base";
+import { Profile } from "../model/profile";
+import { FlagRepository } from "../repo";
+import { generateId } from "../../db";
+
+export class SqlTitlesRepository implements FlagRepository<TitleCode> {
+  constructor(private readonly _conn: Client) {}
+
+  async loadAll(extId: ExtId<Profile>): Promise<Set<TitleCode>> {
+    const loadSql = sql
+      .select("t.title_no")
+      .from("idz.title_unlock t")
+      .join("idz.profile p", { "t.profile_id": "p.id" })
+      .where("p.ext_id", extId)
+      .toParams();
+
+    const { rows } = await this._conn.query(loadSql);
+    const result = new Set<TitleCode>();
+
+    for (const row of rows) {
+      result.add(row.title_no);
+    }
+
+    return result;
+  }
+
+  async saveAll(extId: ExtId<Profile>, flags: Set<TitleCode>): Promise<void> {
+    const profileId = await _findProfile(this._conn, extId);
+    const existing = await this.loadAll(extId);
+
+    for (const flag of flags) {
+      if (existing.has(flag)) {
+        continue;
+      }
+
+      const saveSql = sql
+        .insert("idz.title_unlock", {
+          id: generateId(),
+          profile_id: profileId,
+          title_no: flag,
+        })
+        .toParams();
+
+      await this._conn.query(saveSql);
+    }
+  }
+}