feat: persist preferred project by account key

Author: jonathanlab

apps/code/src/main/db/migrations/0004_auth_preferences.sql

@@ -0,0 +1,9 @@
+CREATE TABLE `auth_preferences` (
+	`account_key` text NOT NULL,
+	`cloud_region` text NOT NULL,
+	`last_selected_project_id` integer,
+	`created_at` text DEFAULT (CURRENT_TIMESTAMP) NOT NULL,
+	`updated_at` text DEFAULT (CURRENT_TIMESTAMP) NOT NULL
+);
+--> statement-breakpoint
+CREATE INDEX `auth_preferences_account_region_idx` ON `auth_preferences` (`account_key`,`cloud_region`);

apps/code/src/main/db/migrations/meta/_journal.json

@@ -29,6 +29,13 @@
       "when": 1774890000000,
       "tag": "0003_fair_whiplash",
       "breakpoints": true
+    },
+    {
+      "idx": 4,
+      "version": "7",
+      "when": 1774891000000,
+      "tag": "0004_auth_preferences",
+      "breakpoints": true
     }
   ]
 }

apps/code/src/main/db/repositories/auth-preference-repository.mock.ts

@@ -0,0 +1,57 @@
+import type {
+  AuthPreference,
+  IAuthPreferenceRepository,
+  PersistAuthPreferenceInput,
+} from "./auth-preference-repository";
+
+export interface MockAuthPreferenceRepository
+  extends IAuthPreferenceRepository {
+  _preferences: AuthPreference[];
+}
+
+export function createMockAuthPreferenceRepository(): MockAuthPreferenceRepository {
+  let preferences: AuthPreference[] = [];
+
+  const clone = (value: AuthPreference): AuthPreference => ({ ...value });
+
+  return {
+    get _preferences() {
+      return preferences.map(clone);
+    },
+    set _preferences(value) {
+      preferences = value.map(clone);
+    },
+    get: (accountKey, cloudRegion) => {
+      const preference = preferences.find(
+        (entry) =>
+          entry.accountKey === accountKey && entry.cloudRegion === cloudRegion,
+      );
+      return preference ? clone(preference) : null;
+    },
+    save: (input: PersistAuthPreferenceInput) => {
+      const timestamp = new Date().toISOString();
+      const existingIndex = preferences.findIndex(
+        (entry) =>
+          entry.accountKey === input.accountKey &&
+          entry.cloudRegion === input.cloudRegion,
+      );
+
+      const row: AuthPreference = {
+        accountKey: input.accountKey,
+        cloudRegion: input.cloudRegion,
+        lastSelectedProjectId: input.lastSelectedProjectId,
+        createdAt:
+          existingIndex >= 0 ? preferences[existingIndex].createdAt : timestamp,
+        updatedAt: timestamp,
+      };
+
+      if (existingIndex >= 0) {
+        preferences[existingIndex] = row;
+      } else {
+        preferences.push(row);
+      }
+
+      return clone(row);
+    },
+  };
+}

apps/code/src/main/db/repositories/auth-preference-repository.ts

@@ -0,0 +1,89 @@
+import { and, eq } from "drizzle-orm";
+import { inject, injectable } from "inversify";
+import { MAIN_TOKENS } from "../../di/tokens";
+import { authPreferences } from "../schema";
+import type { DatabaseService } from "../service";
+
+export type AuthPreference = typeof authPreferences.$inferSelect;
+export type NewAuthPreference = typeof authPreferences.$inferInsert;
+
+export interface PersistAuthPreferenceInput {
+  accountKey: string;
+  cloudRegion: "us" | "eu" | "dev";
+  lastSelectedProjectId: number | null;
+}
+
+export interface IAuthPreferenceRepository {
+  get(
+    accountKey: string,
+    cloudRegion: "us" | "eu" | "dev",
+  ): AuthPreference | null;
+  save(input: PersistAuthPreferenceInput): AuthPreference;
+}
+
+const now = () => new Date().toISOString();
+
+@injectable()
+export class AuthPreferenceRepository implements IAuthPreferenceRepository {
+  constructor(
+    @inject(MAIN_TOKENS.DatabaseService)
+    private readonly databaseService: DatabaseService,
+  ) {}
+
+  private get db() {
+    return this.databaseService.db;
+  }
+
+  get(
+    accountKey: string,
+    cloudRegion: "us" | "eu" | "dev",
+  ): AuthPreference | null {
+    return (
+      this.db
+        .select()
+        .from(authPreferences)
+        .where(
+          and(
+            eq(authPreferences.accountKey, accountKey),
+            eq(authPreferences.cloudRegion, cloudRegion),
+          ),
+        )
+        .limit(1)
+        .get() ?? null
+    );
+  }
+
+  save(input: PersistAuthPreferenceInput): AuthPreference {
+    const timestamp = now();
+    const existing = this.get(input.accountKey, input.cloudRegion);
+
+    const row: NewAuthPreference = {
+      accountKey: input.accountKey,
+      cloudRegion: input.cloudRegion,
+      lastSelectedProjectId: input.lastSelectedProjectId,
+      createdAt: existing?.createdAt ?? timestamp,
+      updatedAt: timestamp,
+    };
+
+    if (existing) {
+      this.db
+        .update(authPreferences)
+        .set(row)
+        .where(
+          and(
+            eq(authPreferences.accountKey, input.accountKey),
+            eq(authPreferences.cloudRegion, input.cloudRegion),
+          ),
+        )
+        .run();
+    } else {
+      this.db.insert(authPreferences).values(row).run();
+    }
+
+    const saved = this.get(input.accountKey, input.cloudRegion);
+    if (!saved) {
+      throw new Error("Failed to persist auth preference");
+    }
+    return saved;
+  }
+}

apps/code/src/main/db/schema.ts

@@ -86,3 +86,20 @@ export const authSessions = sqliteTable("auth_sessions", {
   createdAt: createdAt(),
   updatedAt: updatedAt(),
 });
+
+export const authPreferences = sqliteTable(
+  "auth_preferences",
+  {
+    accountKey: text().notNull(),
+    cloudRegion: text({ enum: ["us", "eu", "dev"] }).notNull(),
+    lastSelectedProjectId: integer(),
+    createdAt: createdAt(),
+    updatedAt: updatedAt(),
+  },
+  (t) => [
+    index("auth_preferences_account_region_idx").on(
+      t.accountKey,
+      t.cloudRegion,
+    ),
+  ],
+);

apps/code/src/main/di/container.ts

@@ -2,6 +2,7 @@ import "reflect-metadata";
 
 import { Container } from "inversify";
 import { ArchiveRepository } from "../db/repositories/archive-repository";
+import { AuthPreferenceRepository } from "../db/repositories/auth-preference-repository";
 import { AuthSessionRepository } from "../db/repositories/auth-session-repository";
 import { RepositoryRepository } from "../db/repositories/repository-repository";
 import { SuspensionRepositoryImpl } from "../db/repositories/suspension-repository";
@@ -52,6 +53,9 @@ export const container = new Container({
 });
 
 container.bind(MAIN_TOKENS.DatabaseService).to(DatabaseService);
+container
+  .bind(MAIN_TOKENS.AuthPreferenceRepository)
+  .to(AuthPreferenceRepository);
 container.bind(MAIN_TOKENS.AuthSessionRepository).to(AuthSessionRepository);
 container.bind(MAIN_TOKENS.RepositoryRepository).to(RepositoryRepository);
 container.bind(MAIN_TOKENS.WorkspaceRepository).to(WorkspaceRepository);

apps/code/src/main/di/tokens.ts

@@ -9,6 +9,7 @@ export const MAIN_TOKENS = Object.freeze({
   SettingsStore: Symbol.for("Main.SettingsStore"),
 
   // Database
+  AuthPreferenceRepository: Symbol.for("Main.AuthPreferenceRepository"),
   DatabaseService: Symbol.for("Main.DatabaseService"),
   AuthSessionRepository: Symbol.for("Main.AuthSessionRepository"),
   RepositoryRepository: Symbol.for("Main.RepositoryRepository"),