- Added new routes for server discovery and key rotation, including challenge issuance and confirmation processes. - Introduced database schema for managing server registrations and rotation challenges. - Implemented encryption and decryption utilities for secure communication between servers. - Updated package dependencies and added new client and server plugins for social features. - Enhanced user management with additional fields and relations in the database schema.
46 lines
No EOL
1.3 KiB
TypeScript
46 lines
No EOL
1.3 KiB
TypeScript
// tests/helpers/db.ts
|
|
import db from "@/lib/db";
|
|
import { rotateChallengeTokens, serverRegistry } from "@/lib/db/schema";
|
|
import forge from "node-forge";
|
|
|
|
export function generateKeypair() {
|
|
const keypair = forge.pki.rsa.generateKeyPair(2048);
|
|
return {
|
|
publicKey: forge.pki.publicKeyToPem(keypair.publicKey),
|
|
privateKey: forge.pki.privateKeyToPem(keypair.privateKey),
|
|
}
|
|
}
|
|
|
|
export async function seedServer(url: string, publicKey: string) {
|
|
await db.insert(serverRegistry).values({
|
|
id: crypto.randomUUID(),
|
|
url,
|
|
publicKey,
|
|
lastSeen: new Date(),
|
|
createdAt: new Date(),
|
|
updatedAt: new Date(),
|
|
isHealthy: true,
|
|
}).onConflictDoNothing()
|
|
}
|
|
|
|
export async function seedChallenge(overrides?: Partial<typeof rotateChallengeTokens.$inferInsert>) {
|
|
const { publicKey: defaultNewPublicKey } = generateKeypair()
|
|
const defaults = {
|
|
id: crypto.randomUUID(),
|
|
serverUrl: "https://test-server.com",
|
|
oldKeyToken: crypto.randomUUID(),
|
|
newKeyToken: crypto.randomUUID(),
|
|
newPublicKey: defaultNewPublicKey,
|
|
attemptsLeft: 3,
|
|
createdAt: new Date(),
|
|
expiresAt: new Date(Date.now() + 1000 * 60 * 5),
|
|
}
|
|
const row = { ...defaults, ...overrides }
|
|
await db.insert(rotateChallengeTokens).values(row)
|
|
return row
|
|
}
|
|
|
|
export async function clearTables() {
|
|
await db.delete(rotateChallengeTokens)
|
|
await db.delete(serverRegistry)
|
|
} |