@pikku/kysely 0.13.24 → 0.13.28

package/src/kysely-credential-service.ts

@@ -1,286 +0,0 @@
-import {
-  deriveKEK,
-  envelopeDecrypt,
-  envelopeEncrypt,
-  envelopeRewrap,
-  generateKEKSalt,
-} from '@pikku/core/crypto-utils'
-import type { CredentialService } from '@pikku/core/services'
-import type { Kysely } from 'kysely'
-import type { KyselyPikkuDB } from './kysely-tables.js'
-import { requirePikkuSchema } from './schema/index.js'
-import { credentialSchema } from './schema/credential.schema.js'
-
-export interface KyselyCredentialServiceConfig {
-  key: string
-  keyVersion?: number
-  previousKey?: string
-  audit?: boolean
-  auditReads?: boolean
-}
-
-export class KyselyCredentialService implements CredentialService {
-  private initialized = false
-  private kekSalts = new Map<number, string>()
-  private keks = new Map<number, CryptoKey>()
-  private key: string
-  private keyVersion: number
-  private previousKey?: string
-  private audit: boolean
-  private auditReads: boolean
-
-  constructor(
-    private db: Kysely<KyselyPikkuDB>,
-    config: KyselyCredentialServiceConfig
-  ) {
-    this.key = config.key
-    this.keyVersion = config.keyVersion ?? 1
-    this.previousKey = config.previousKey
-    this.audit = config.audit ?? false
-    this.auditReads = config.auditReads ?? false
-  }
-
-  public async init(): Promise<void> {
-    if (this.initialized) return
-    await requirePikkuSchema(this.db, credentialSchema)
-    this.initialized = true
-  }
-
-  private async logAudit(
-    name: string,
-    userId: string | undefined,
-    action: 'read' | 'write' | 'delete' | 'rotate'
-  ): Promise<void> {
-    if (!this.audit) return
-    if (action === 'read' && !this.auditReads) return
-
-    await this.db
-      .insertInto('credentialsAudit')
-      .values({
-        id: crypto.randomUUID(),
-        credentialName: name,
-        userId: userId ?? null,
-        action,
-        performedAt: new Date().toISOString() as unknown as Date,
-      })
-      .execute()
-  }
-
-  private async getKEKSalt(version: number): Promise<string> {
-    const cached = this.kekSalts.get(version)
-    if (cached) return cached
-
-    const existing = await this.db
-      .selectFrom('credentialKekSalts')
-      .select('salt')
-      .where('keyVersion', '=', version)
-      .executeTakeFirst()
-
-    let salt = existing?.salt
-    if (!salt) {
-      await this.db
-        .insertInto('credentialKekSalts')
-        .values({
-          keyVersion: version,
-          salt: generateKEKSalt(),
-          createdAt: new Date().toISOString() as unknown as Date,
-        })
-        .onConflict((oc) => oc.column('keyVersion').doNothing())
-        .execute()
-
-      const row = await this.db
-        .selectFrom('credentialKekSalts')
-        .select('salt')
-        .where('keyVersion', '=', version)
-        .executeTakeFirstOrThrow()
-      salt = row.salt
-    }
-
-    this.kekSalts.set(version, salt)
-    return salt
-  }
-
-  private async getKEK(version: number): Promise<CryptoKey> {
-    const cached = this.keks.get(version)
-    if (cached) return cached
-
-    const passphrase = version === this.keyVersion ? this.key : this.previousKey
-    if (!passphrase) {
-      throw new Error(`No KEK available for key_version ${version}`)
-    }
-
-    const kek = await deriveKEK(passphrase, await this.getKEKSalt(version))
-    this.keks.set(version, kek)
-    return kek
-  }
-
-  private whereUserId(qb: any, userId?: string) {
-    return userId
-      ? qb.where('userId', '=', userId)
-      : qb.where('userId', 'is', null)
-  }
-
-  async get<T = unknown>(name: string, userId?: string): Promise<T | null> {
-    let qb = this.db
-      .selectFrom('credentials')
-      .select(['ciphertext', 'wrappedDek', 'keyVersion'])
-      .where('name', '=', name)
-    qb = this.whereUserId(qb, userId)
-
-    const row = await qb.executeTakeFirst()
-    if (!row) return null
-
-    const kek = await this.getKEK(row.keyVersion)
-    const plaintext = await envelopeDecrypt<string>(
-      kek,
-      row.ciphertext,
-      row.wrappedDek
-    )
-    await this.logAudit(name, userId, 'read')
-
-    try {
-      return JSON.parse(plaintext) as T
-    } catch {
-      throw new Error(`Credential '${name}' contains invalid data`)
-    }
-  }
-
-  async set(name: string, value: unknown, userId?: string): Promise<void> {
-    const plaintext = JSON.stringify(value)
-    const { ciphertext, wrappedDEK } = await envelopeEncrypt(
-      await this.getKEK(this.keyVersion),
-      plaintext
-    )
-    const now = new Date().toISOString()
-    const exists = await this.has(name, userId)
-
-    if (exists) {
-      let qb = this.db
-        .updateTable('credentials')
-        .set({
-          ciphertext,
-          wrappedDek: wrappedDEK,
-          keyVersion: this.keyVersion,
-          updatedAt: now as unknown as Date,
-        })
-        .where('name', '=', name)
-      qb = this.whereUserId(qb, userId)
-      await qb.execute()
-    } else {
-      await this.db
-        .insertInto('credentials')
-        .values({
-          name,
-          userId: userId ?? null,
-          ciphertext,
-          wrappedDek: wrappedDEK,
-          keyVersion: this.keyVersion,
-          createdAt: now as unknown as Date,
-          updatedAt: now as unknown as Date,
-        })
-        .execute()
-    }
-
-    await this.logAudit(name, userId, 'write')
-  }
-
-  async delete(name: string, userId?: string): Promise<void> {
-    let qb = this.db.deleteFrom('credentials').where('name', '=', name)
-    qb = this.whereUserId(qb, userId)
-    await qb.execute()
-
-    await this.logAudit(name, userId, 'delete')
-  }
-
-  async has(name: string, userId?: string): Promise<boolean> {
-    let qb = this.db
-      .selectFrom('credentials')
-      .select('name')
-      .where('name', '=', name)
-    qb = this.whereUserId(qb, userId)
-
-    const row = await qb.executeTakeFirst()
-    return !!row
-  }
-
-  async getAll(userId: string): Promise<Record<string, unknown>> {
-    const rows = await this.db
-      .selectFrom('credentials')
-      .select(['name', 'ciphertext', 'wrappedDek', 'keyVersion'])
-      .where('userId', '=', userId)
-      .execute()
-
-    const result: Record<string, unknown> = {}
-    for (const row of rows) {
-      const kek = await this.getKEK(row.keyVersion)
-      const plaintext = await envelopeDecrypt<string>(
-        kek,
-        row.ciphertext,
-        row.wrappedDek
-      )
-      try {
-        result[row.name] = JSON.parse(plaintext)
-      } catch {
-        throw new Error(`Credential '${row.name}' contains invalid data`)
-      }
-      await this.logAudit(row.name, userId, 'read')
-    }
-
-    return result
-  }
-
-  async getUsersWithCredential(name: string): Promise<string[]> {
-    const rows = await this.db
-      .selectFrom('credentials')
-      .select('userId')
-      .where('name', '=', name)
-      .where('userId', 'is not', null)
-      .execute()
-
-    return rows.map((row) => row.userId!).filter(Boolean)
-  }
-
-  async getAllUsers(): Promise<string[]> {
-    const rows = await this.db
-      .selectFrom('credentials')
-      .select('userId')
-      .distinct()
-      .where('userId', 'is not', null)
-      .execute()
-
-    return rows.map((row) => row.userId!).filter(Boolean)
-  }
-
-  async rotateKEK(): Promise<number> {
-    if (!this.previousKey) {
-      throw new Error('No previousKey configured — nothing to rotate from')
-    }
-
-    const rows = await this.db
-      .selectFrom('credentials')
-      .select(['name', 'userId', 'wrappedDek'])
-      .where('keyVersion', '<', this.keyVersion)
-      .execute()
-
-    const oldKEK = await this.getKEK(this.keyVersion - 1)
-    const newKEK = await this.getKEK(this.keyVersion)
-
-    for (const row of rows) {
-      const newWrappedDEK = await envelopeRewrap(oldKEK, newKEK, row.wrappedDek)
-      let qb = this.db
-        .updateTable('credentials')
-        .set({
-          wrappedDek: newWrappedDEK,
-          keyVersion: this.keyVersion,
-          updatedAt: new Date().toISOString() as unknown as Date,
-        })
-        .where('name', '=', row.name)
-      qb = this.whereUserId(qb, row.userId ?? undefined)
-      await qb.execute()
-
-      await this.logAudit(row.name, row.userId ?? undefined, 'rotate')
-    }
-
-    return rows.length
-  }
-}

package/src/kysely-deployment-service.ts

@@ -1,163 +0,0 @@
-import { buildRemoteHeaders } from '@pikku/core/remote'
-import type {
-  DeploymentService,
-  DeploymentServiceConfig,
-  DeploymentConfig,
-} from '@pikku/core/services'
-import type { JWTService } from '@pikku/core/services'
-import type { SecretService } from '@pikku/core/services'
-import { getAllFunctionNames } from '@pikku/core/function'
-import type { Kysely } from 'kysely'
-import type { KyselyPikkuDB } from './kysely-tables.js'
-import { requirePikkuSchema } from './schema/index.js'
-import { deploymentSchema } from './schema/deployment.schema.js'
-
-export class KyselyDeploymentService implements DeploymentService {
-  private initialized = false
-  private heartbeatTimer?: ReturnType<typeof setInterval>
-  private deploymentConfig?: DeploymentConfig
-  private heartbeatInterval: number
-  private heartbeatTtl: number
-
-  constructor(
-    config: DeploymentServiceConfig,
-    protected db: Kysely<KyselyPikkuDB>,
-    private jwt?: JWTService,
-    private secrets?: SecretService
-  ) {
-    this.heartbeatInterval = config.heartbeatInterval ?? 10000
-    this.heartbeatTtl = config.heartbeatTtl ?? 30000
-  }
-
-  public async init(): Promise<void> {
-    if (this.initialized) return
-    await requirePikkuSchema(this.db, deploymentSchema)
-    this.initialized = true
-  }
-
-  async start(config: DeploymentConfig): Promise<void> {
-    const functions = config.functions ?? getAllFunctionNames()
-    this.deploymentConfig = { ...config, functions }
-
-    await this.db.transaction().execute(async (trx) => {
-      await trx
-        .insertInto('pikkuDeployments')
-        .values({
-          deploymentId: config.deploymentId,
-          endpoint: config.endpoint,
-          lastHeartbeat: new Date(),
-        })
-        .onConflict((oc) =>
-          oc.column('deploymentId').doUpdateSet({
-            endpoint: config.endpoint,
-            lastHeartbeat: new Date(),
-          })
-        )
-        .execute()
-
-      await trx
-        .deleteFrom('pikkuDeploymentFunctions')
-        .where('deploymentId', '=', config.deploymentId)
-        .execute()
-
-      if (functions.length > 0) {
-        await trx
-          .insertInto('pikkuDeploymentFunctions')
-          .values(
-            functions.map((fn) => ({
-              deploymentId: config.deploymentId,
-              functionName: fn,
-            }))
-          )
-          .execute()
-      }
-    })
-
-    this.heartbeatTimer = setInterval(
-      () => this.sendHeartbeat(),
-      this.heartbeatInterval
-    )
-  }
-
-  async stop(): Promise<void> {
-    if (this.heartbeatTimer) {
-      clearInterval(this.heartbeatTimer)
-      this.heartbeatTimer = undefined
-    }
-
-    if (this.deploymentConfig) {
-      await this.db
-        .deleteFrom('pikkuDeployments')
-        .where('deploymentId', '=', this.deploymentConfig.deploymentId)
-        .execute()
-    }
-  }
-
-  async invoke(
-    funcName: string,
-    data: unknown,
-    session?: unknown,
-    traceId?: string
-  ): Promise<unknown> {
-    const headers = await buildRemoteHeaders(
-      this.jwt,
-      this.secrets,
-      funcName,
-      session,
-      traceId
-    )
-    const ttlMs = this.heartbeatTtl
-    const cutoff = new Date(Date.now() - ttlMs)
-
-    const result = await this.db
-      .selectFrom('pikkuDeployments as d')
-      .innerJoin(
-        'pikkuDeploymentFunctions as f',
-        'f.deploymentId',
-        'd.deploymentId'
-      )
-      .select(['d.deploymentId', 'd.endpoint'])
-      .where('f.functionName', '=', funcName)
-      .where('d.lastHeartbeat', '>', cutoff)
-      .orderBy('d.lastHeartbeat', 'desc')
-      .limit(1)
-      .execute()
-
-    if (result.length === 0) {
-      throw new Error(`No deployment found for function '${funcName}'`)
-    }
-
-    const endpoint = result[0].endpoint
-    const url = `${endpoint}/remote/rpc/${encodeURIComponent(funcName)}`
-    const response = await fetch(url, {
-      method: 'POST',
-      headers: {
-        'content-type': 'application/json',
-        ...headers,
-      },
-      body: JSON.stringify({ data }),
-    })
-
-    if (!response.ok) {
-      throw new Error(
-        `Remote RPC call to '${funcName}' failed: ${response.status}`
-      )
-    }
-
-    return response.json()
-  }
-
-  private async sendHeartbeat(): Promise<void> {
-    if (!this.deploymentConfig) return
-
-    try {
-      await this.db
-        .updateTable('pikkuDeployments')
-        .set({ lastHeartbeat: new Date() })
-        .where('deploymentId', '=', this.deploymentConfig.deploymentId)
-        .execute()
-    } catch {
-      // Heartbeat failed, will retry on next interval
-    }
-  }
-}

package/src/kysely-eventhub-store.ts

@@ -1,53 +0,0 @@
-import { EventHubStore } from '@pikku/core/channel'
-import type { Kysely } from 'kysely'
-import type { KyselyPikkuDB } from './kysely-tables.js'
-
-export class KyselyEventHubStore extends EventHubStore {
-  private initialized = false
-
-  constructor(private db: Kysely<KyselyPikkuDB>) {
-    super()
-  }
-
-  public async init(): Promise<void> {
-    if (this.initialized) {
-      return
-    }
-    this.initialized = true
-  }
-
-  public async getChannelIdsForTopic(topic: string): Promise<string[]> {
-    const result = await this.db
-      .selectFrom('channelSubscriptions')
-      .select('channelId')
-      .where('topic', '=', topic)
-      .execute()
-
-    return result.map((row) => row.channelId)
-  }
-
-  public async subscribe(topic: string, channelId: string): Promise<boolean> {
-    try {
-      await this.db
-        .insertInto('channelSubscriptions')
-        .values({ channelId: channelId, topic: topic as string })
-        .onConflict((oc) => oc.columns(['channelId', 'topic']).doNothing())
-        .execute()
-      return true
-    } catch {
-      return false
-    }
-  }
-
-  public async unsubscribe(topic: string, channelId: string): Promise<boolean> {
-    const result = await this.db
-      .deleteFrom('channelSubscriptions')
-      .where('channelId', '=', channelId)
-      .where('topic', '=', topic as string)
-      .executeTakeFirst()
-
-    return BigInt(result.numDeletedRows) > 0n
-  }
-
-  public async close(): Promise<void> {}
-}

package/src/kysely-json.ts

@@ -1,5 +0,0 @@
-export function parseJson(val: unknown): any {
-  if (val == null) return undefined
-  if (typeof val === 'string') return JSON.parse(val)
-  return val
-}