improve

src/server/database/migrations/0006_clear_leech.sql

@@ -0,0 +1,2 @@
+CREATE UNIQUE INDEX `public_key_interface_unique` ON `clients_table` (`public_key`,`interface_id`);--> statement-breakpoint
+CREATE UNIQUE INDEX `users_table_email_unique` ON `users_table` (`email`);

src/server/database/migrations/meta/_journal.json

@@ -43,6 +43,13 @@
       "when": 1780035570366,
       "tag": "0005_clumsy_korg",
       "breakpoints": true
+    },
+    {
+      "idx": 6,
+      "version": "6",
+      "when": 1782122902196,
+      "tag": "0006_clear_leech",
+      "breakpoints": true
     }
   ]
 }

src/server/database/repositories/client/schema.ts

@@ -1,5 +1,5 @@
 import { sql, relations } from 'drizzle-orm';
-import { int, sqliteTable, text } from 'drizzle-orm/sqlite-core';
+import { int, sqliteTable, text, uniqueIndex } from 'drizzle-orm/sqlite-core';
 
 import { wgInterface } from '#db/repositories/interface/schema';
 import { oneTimeLink } from '#db/repositories/oneTimeLink/schema';
@@ -7,7 +7,9 @@ import { user } from '#db/repositories/user/schema';
 
 /** null means use value from userConfig */
 
-export const client = sqliteTable('clients_table', {
+export const client = sqliteTable(
+  'clients_table',
+  {
     id: int().primaryKey({ autoIncrement: true }),
     userId: int('user_id')
       .notNull()
@@ -37,7 +39,9 @@ export const client = sqliteTable('clients_table', {
       .$type<string[]>()
       .notNull(),
     // Firewall-enforced allowed IPs (null = use allowedIps)
-  firewallIps: text('firewall_ips', { mode: 'json' }).$type<string[] | null>(),
+    firewallIps: text('firewall_ips', { mode: 'json' }).$type<
+      string[] | null
+    >(),
     persistentKeepalive: int('persistent_keepalive').notNull(),
     mtu: int().notNull(),
     jC: int('j_c'),
@@ -58,7 +62,14 @@ export const client = sqliteTable('clients_table', {
       .notNull()
       .default(sql`(CURRENT_TIMESTAMP)`)
       .$onUpdate(() => sql`(CURRENT_TIMESTAMP)`),
-});
+  },
+  (table) => [
+    uniqueIndex('public_key_interface_unique').on(
+      table.publicKey,
+      table.interfaceId
+    ),
+  ]
+);
 
 export const clientsRelations = relations(client, ({ one }) => ({
   oneTimeLink: one(oneTimeLink, {

src/server/database/repositories/user/schema.ts

@@ -12,7 +12,7 @@ export const user = sqliteTable(
     username: text().notNull().unique(),
     /** `password == null` means password login disabled */
     password: text(),
-    email: text(),
+    email: text().unique(),
     name: text().notNull(),
     role: int().$type<Role>().notNull(),
     totpKey: text('totp_key'),

src/server/database/sqlite.ts

@@ -69,6 +69,7 @@ async function migrate() {
     if (e instanceof Error) {
       DB_DEBUG('Failed to migrate database:', e.message);
     }
+    throw e;
   }
 }
 

src/server/utils/Database.ts

@@ -17,9 +17,14 @@ const nullObject = new Proxy(
 // eslint-disable-next-line import/no-mutable-exports
 let provider = nullObject as never as DBServiceType;
 
-connect().then((db) => {
+connect()
+  .then((db) => {
     provider = db;
     WireGuard.Startup();
-});
+  })
+  .catch((err) => {
+    console.log('Failed to connect to Database:', err);
+    process.exit(1);
+  });
 
 export default provider;