@naisys/hub-database 3.0.0-beta.10

package/dist/generated/prisma/internal/prismaNamespace.js

@@ -0,0 +1,248 @@
+/* !!! This is code generated by Prisma. Do not edit directly. !!! */
+/* eslint-disable */
+// biome-ignore-all lint: generated file
+// @ts-nocheck 
+/*
+ * WARNING: This is an internal file that is subject to change!
+ *
+ * 🛑 Under no circumstances should you import this file directly! 🛑
+ *
+ * All exports from this file are wrapped under a `Prisma` namespace object in the client.ts file.
+ * While this enables partial backward compatibility, it is not part of the stable public API.
+ *
+ * If you are looking for your Models, Enums, and Input Types, please import them from the respective
+ * model files in the `model` directory!
+ */
+import * as runtime from "@prisma/client/runtime/client";
+/**
+ * Prisma Errors
+ */
+export const PrismaClientKnownRequestError = runtime.PrismaClientKnownRequestError;
+export const PrismaClientUnknownRequestError = runtime.PrismaClientUnknownRequestError;
+export const PrismaClientRustPanicError = runtime.PrismaClientRustPanicError;
+export const PrismaClientInitializationError = runtime.PrismaClientInitializationError;
+export const PrismaClientValidationError = runtime.PrismaClientValidationError;
+/**
+ * Re-export of sql-template-tag
+ */
+export const sql = runtime.sqltag;
+export const empty = runtime.empty;
+export const join = runtime.join;
+export const raw = runtime.raw;
+export const Sql = runtime.Sql;
+/**
+ * Decimal.js
+ */
+export const Decimal = runtime.Decimal;
+export const getExtensionContext = runtime.Extensions.getExtensionContext;
+/**
+ * Prisma Client JS version: 7.6.0
+ * Query Engine version: 75cbdc1eb7150937890ad5465d861175c6624711
+ */
+export const prismaVersion = {
+    client: "7.6.0",
+    engine: "75cbdc1eb7150937890ad5465d861175c6624711"
+};
+export const NullTypes = {
+    DbNull: runtime.NullTypes.DbNull,
+    JsonNull: runtime.NullTypes.JsonNull,
+    AnyNull: runtime.NullTypes.AnyNull,
+};
+/**
+ * Helper for filtering JSON entries that have `null` on the database (empty on the db)
+ *
+ * @see https://www.prisma.io/docs/concepts/components/prisma-client/working-with-fields/working-with-json-fields#filtering-on-a-json-field
+ */
+export const DbNull = runtime.DbNull;
+/**
+ * Helper for filtering JSON entries that have JSON `null` values (not empty on the db)
+ *
+ * @see https://www.prisma.io/docs/concepts/components/prisma-client/working-with-fields/working-with-json-fields#filtering-on-a-json-field
+ */
+export const JsonNull = runtime.JsonNull;
+/**
+ * Helper for filtering JSON entries that are `Prisma.DbNull` or `Prisma.JsonNull`
+ *
+ * @see https://www.prisma.io/docs/concepts/components/prisma-client/working-with-fields/working-with-json-fields#filtering-on-a-json-field
+ */
+export const AnyNull = runtime.AnyNull;
+export const ModelName = {
+    context_log: 'context_log',
+    costs: 'costs',
+    mail_messages: 'mail_messages',
+    attachments: 'attachments',
+    mail_attachments: 'mail_attachments',
+    mail_recipients: 'mail_recipients',
+    users: 'users',
+    user_notifications: 'user_notifications',
+    user_hosts: 'user_hosts',
+    run_session: 'run_session',
+    schema_version: 'schema_version',
+    hosts: 'hosts',
+    config_revisions: 'config_revisions',
+    variables: 'variables',
+    models: 'models'
+};
+/**
+ * Enums
+ */
+export const TransactionIsolationLevel = runtime.makeStrictEnum({
+    Serializable: 'Serializable'
+});
+export const Context_logScalarFieldEnum = {
+    id: 'id',
+    user_id: 'user_id',
+    run_id: 'run_id',
+    session_id: 'session_id',
+    host_id: 'host_id',
+    role: 'role',
+    source: 'source',
+    type: 'type',
+    message: 'message',
+    attachment_id: 'attachment_id',
+    created_at: 'created_at'
+};
+export const CostsScalarFieldEnum = {
+    id: 'id',
+    user_id: 'user_id',
+    run_id: 'run_id',
+    session_id: 'session_id',
+    host_id: 'host_id',
+    source: 'source',
+    model: 'model',
+    cost: 'cost',
+    input_tokens: 'input_tokens',
+    output_tokens: 'output_tokens',
+    cache_write_tokens: 'cache_write_tokens',
+    cache_read_tokens: 'cache_read_tokens',
+    created_at: 'created_at'
+};
+export const Mail_messagesScalarFieldEnum = {
+    id: 'id',
+    from_user_id: 'from_user_id',
+    host_id: 'host_id',
+    kind: 'kind',
+    participants: 'participants',
+    subject: 'subject',
+    body: 'body',
+    created_at: 'created_at'
+};
+export const AttachmentsScalarFieldEnum = {
+    id: 'id',
+    public_id: 'public_id',
+    filepath: 'filepath',
+    filename: 'filename',
+    file_size: 'file_size',
+    file_hash: 'file_hash',
+    purpose: 'purpose',
+    uploaded_by: 'uploaded_by',
+    created_at: 'created_at'
+};
+export const Mail_attachmentsScalarFieldEnum = {
+    message_id: 'message_id',
+    attachment_id: 'attachment_id',
+    created_at: 'created_at'
+};
+export const Mail_recipientsScalarFieldEnum = {
+    id: 'id',
+    message_id: 'message_id',
+    user_id: 'user_id',
+    type: 'type',
+    read_at: 'read_at',
+    archived_at: 'archived_at',
+    created_at: 'created_at'
+};
+export const UsersScalarFieldEnum = {
+    id: 'id',
+    uuid: 'uuid',
+    username: 'username',
+    title: 'title',
+    api_key: 'api_key',
+    lead_user_id: 'lead_user_id',
+    config: 'config',
+    created_at: 'created_at',
+    updated_at: 'updated_at',
+    enabled: 'enabled',
+    archived: 'archived'
+};
+export const User_notificationsScalarFieldEnum = {
+    user_id: 'user_id',
+    latest_host_id: 'latest_host_id',
+    latest_log_id: 'latest_log_id',
+    latest_mail_id: 'latest_mail_id',
+    latest_chat_id: 'latest_chat_id',
+    last_active: 'last_active',
+    cost_suspended_reason: 'cost_suspended_reason',
+    spend_limit_reset_at: 'spend_limit_reset_at',
+    budget_left: 'budget_left',
+    updated_at: 'updated_at'
+};
+export const User_hostsScalarFieldEnum = {
+    user_id: 'user_id',
+    host_id: 'host_id',
+    created_at: 'created_at'
+};
+export const Run_sessionScalarFieldEnum = {
+    user_id: 'user_id',
+    run_id: 'run_id',
+    session_id: 'session_id',
+    host_id: 'host_id',
+    last_active: 'last_active',
+    model_name: 'model_name',
+    latest_log_id: 'latest_log_id',
+    total_lines: 'total_lines',
+    total_cost: 'total_cost',
+    created_at: 'created_at',
+    updated_at: 'updated_at'
+};
+export const Schema_versionScalarFieldEnum = {
+    id: 'id',
+    version: 'version',
+    updated: 'updated'
+};
+export const HostsScalarFieldEnum = {
+    id: 'id',
+    name: 'name',
+    restricted: 'restricted',
+    host_type: 'host_type',
+    last_ip: 'last_ip',
+    last_active: 'last_active',
+    created_at: 'created_at'
+};
+export const Config_revisionsScalarFieldEnum = {
+    id: 'id',
+    user_id: 'user_id',
+    config: 'config',
+    changed_by_id: 'changed_by_id',
+    created_at: 'created_at'
+};
+export const VariablesScalarFieldEnum = {
+    key: 'key',
+    value: 'value',
+    export_to_shell: 'export_to_shell',
+    created_by: 'created_by',
+    updated_by: 'updated_by',
+    created_at: 'created_at',
+    updated_at: 'updated_at'
+};
+export const ModelsScalarFieldEnum = {
+    id: 'id',
+    key: 'key',
+    type: 'type',
+    label: 'label',
+    version_name: 'version_name',
+    is_builtin: 'is_builtin',
+    is_custom: 'is_custom',
+    meta: 'meta',
+    created_at: 'created_at',
+    updated_at: 'updated_at'
+};
+export const SortOrder = {
+    asc: 'asc',
+    desc: 'desc'
+};
+export const NullsOrder = {
+    first: 'first',
+    last: 'last'
+};
+export const defineExtension = runtime.Extensions.defineExtension;

package/dist/generated/prisma/internal/prismaNamespaceBrowser.js

@@ -0,0 +1,219 @@
+/* !!! This is code generated by Prisma. Do not edit directly. !!! */
+/* eslint-disable */
+// biome-ignore-all lint: generated file
+// @ts-nocheck 
+/*
+ * WARNING: This is an internal file that is subject to change!
+ *
+ * 🛑 Under no circumstances should you import this file directly! 🛑
+ *
+ * All exports from this file are wrapped under a `Prisma` namespace object in the browser.ts file.
+ * While this enables partial backward compatibility, it is not part of the stable public API.
+ *
+ * If you are looking for your Models, Enums, and Input Types, please import them from the respective
+ * model files in the `model` directory!
+ */
+import * as runtime from "@prisma/client/runtime/index-browser";
+export const Decimal = runtime.Decimal;
+export const NullTypes = {
+    DbNull: runtime.NullTypes.DbNull,
+    JsonNull: runtime.NullTypes.JsonNull,
+    AnyNull: runtime.NullTypes.AnyNull,
+};
+/**
+ * Helper for filtering JSON entries that have `null` on the database (empty on the db)
+ *
+ * @see https://www.prisma.io/docs/concepts/components/prisma-client/working-with-fields/working-with-json-fields#filtering-on-a-json-field
+ */
+export const DbNull = runtime.DbNull;
+/**
+ * Helper for filtering JSON entries that have JSON `null` values (not empty on the db)
+ *
+ * @see https://www.prisma.io/docs/concepts/components/prisma-client/working-with-fields/working-with-json-fields#filtering-on-a-json-field
+ */
+export const JsonNull = runtime.JsonNull;
+/**
+ * Helper for filtering JSON entries that are `Prisma.DbNull` or `Prisma.JsonNull`
+ *
+ * @see https://www.prisma.io/docs/concepts/components/prisma-client/working-with-fields/working-with-json-fields#filtering-on-a-json-field
+ */
+export const AnyNull = runtime.AnyNull;
+export const ModelName = {
+    context_log: 'context_log',
+    costs: 'costs',
+    mail_messages: 'mail_messages',
+    attachments: 'attachments',
+    mail_attachments: 'mail_attachments',
+    mail_recipients: 'mail_recipients',
+    users: 'users',
+    user_notifications: 'user_notifications',
+    user_hosts: 'user_hosts',
+    run_session: 'run_session',
+    schema_version: 'schema_version',
+    hosts: 'hosts',
+    config_revisions: 'config_revisions',
+    variables: 'variables',
+    models: 'models'
+};
+/*
+ * Enums
+ */
+export const TransactionIsolationLevel = runtime.makeStrictEnum({
+    Serializable: 'Serializable'
+});
+export const Context_logScalarFieldEnum = {
+    id: 'id',
+    user_id: 'user_id',
+    run_id: 'run_id',
+    session_id: 'session_id',
+    host_id: 'host_id',
+    role: 'role',
+    source: 'source',
+    type: 'type',
+    message: 'message',
+    attachment_id: 'attachment_id',
+    created_at: 'created_at'
+};
+export const CostsScalarFieldEnum = {
+    id: 'id',
+    user_id: 'user_id',
+    run_id: 'run_id',
+    session_id: 'session_id',
+    host_id: 'host_id',
+    source: 'source',
+    model: 'model',
+    cost: 'cost',
+    input_tokens: 'input_tokens',
+    output_tokens: 'output_tokens',
+    cache_write_tokens: 'cache_write_tokens',
+    cache_read_tokens: 'cache_read_tokens',
+    created_at: 'created_at'
+};
+export const Mail_messagesScalarFieldEnum = {
+    id: 'id',
+    from_user_id: 'from_user_id',
+    host_id: 'host_id',
+    kind: 'kind',
+    participants: 'participants',
+    subject: 'subject',
+    body: 'body',
+    created_at: 'created_at'
+};
+export const AttachmentsScalarFieldEnum = {
+    id: 'id',
+    public_id: 'public_id',
+    filepath: 'filepath',
+    filename: 'filename',
+    file_size: 'file_size',
+    file_hash: 'file_hash',
+    purpose: 'purpose',
+    uploaded_by: 'uploaded_by',
+    created_at: 'created_at'
+};
+export const Mail_attachmentsScalarFieldEnum = {
+    message_id: 'message_id',
+    attachment_id: 'attachment_id',
+    created_at: 'created_at'
+};
+export const Mail_recipientsScalarFieldEnum = {
+    id: 'id',
+    message_id: 'message_id',
+    user_id: 'user_id',
+    type: 'type',
+    read_at: 'read_at',
+    archived_at: 'archived_at',
+    created_at: 'created_at'
+};
+export const UsersScalarFieldEnum = {
+    id: 'id',
+    uuid: 'uuid',
+    username: 'username',
+    title: 'title',
+    api_key: 'api_key',
+    lead_user_id: 'lead_user_id',
+    config: 'config',
+    created_at: 'created_at',
+    updated_at: 'updated_at',
+    enabled: 'enabled',
+    archived: 'archived'
+};
+export const User_notificationsScalarFieldEnum = {
+    user_id: 'user_id',
+    latest_host_id: 'latest_host_id',
+    latest_log_id: 'latest_log_id',
+    latest_mail_id: 'latest_mail_id',
+    latest_chat_id: 'latest_chat_id',
+    last_active: 'last_active',
+    cost_suspended_reason: 'cost_suspended_reason',
+    spend_limit_reset_at: 'spend_limit_reset_at',
+    budget_left: 'budget_left',
+    updated_at: 'updated_at'
+};
+export const User_hostsScalarFieldEnum = {
+    user_id: 'user_id',
+    host_id: 'host_id',
+    created_at: 'created_at'
+};
+export const Run_sessionScalarFieldEnum = {
+    user_id: 'user_id',
+    run_id: 'run_id',
+    session_id: 'session_id',
+    host_id: 'host_id',
+    last_active: 'last_active',
+    model_name: 'model_name',
+    latest_log_id: 'latest_log_id',
+    total_lines: 'total_lines',
+    total_cost: 'total_cost',
+    created_at: 'created_at',
+    updated_at: 'updated_at'
+};
+export const Schema_versionScalarFieldEnum = {
+    id: 'id',
+    version: 'version',
+    updated: 'updated'
+};
+export const HostsScalarFieldEnum = {
+    id: 'id',
+    name: 'name',
+    restricted: 'restricted',
+    host_type: 'host_type',
+    last_ip: 'last_ip',
+    last_active: 'last_active',
+    created_at: 'created_at'
+};
+export const Config_revisionsScalarFieldEnum = {
+    id: 'id',
+    user_id: 'user_id',
+    config: 'config',
+    changed_by_id: 'changed_by_id',
+    created_at: 'created_at'
+};
+export const VariablesScalarFieldEnum = {
+    key: 'key',
+    value: 'value',
+    export_to_shell: 'export_to_shell',
+    created_by: 'created_by',
+    updated_by: 'updated_by',
+    created_at: 'created_at',
+    updated_at: 'updated_at'
+};
+export const ModelsScalarFieldEnum = {
+    id: 'id',
+    key: 'key',
+    type: 'type',
+    label: 'label',
+    version_name: 'version_name',
+    is_builtin: 'is_builtin',
+    is_custom: 'is_custom',
+    meta: 'meta',
+    created_at: 'created_at',
+    updated_at: 'updated_at'
+};
+export const SortOrder = {
+    asc: 'asc',
+    desc: 'desc'
+};
+export const NullsOrder = {
+    first: 'first',
+    last: 'last'
+};

package/dist/generated/prisma/models/attachments.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/config_revisions.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/context_log.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/costs.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/hosts.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/mail_attachments.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/mail_messages.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/mail_recipients.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/models.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/run_session.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/schema_version.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/user_hosts.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/user_notifications.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/users.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models/variables.js

@@ -0,0 +1 @@
+export {};

package/dist/generated/prisma/models.js

@@ -0,0 +1 @@
+export {};

package/dist/hubDatabaseService.js

@@ -0,0 +1,38 @@
+import { deployPrismaMigrations } from "@naisys/common-node";
+import { existsSync, mkdirSync } from "fs";
+import { dirname } from "path";
+import { fileURLToPath } from "url";
+import { hubDbPath } from "./dbConfig.js";
+import { createPrismaClient } from "./prismaClient.js";
+export async function createHubDatabaseService() {
+    /** We run migration scripts if this is greater than what's in the schema_version table */
+    const HUB_DB_VERSION = 34;
+    const dbPath = hubDbPath();
+    // Ensure database directory exists
+    const databaseDir = dirname(dbPath);
+    if (!existsSync(databaseDir)) {
+        mkdirSync(databaseDir, { recursive: true });
+    }
+    await runMigrations();
+    const prisma = await createPrismaClient(dbPath);
+    async function runMigrations() {
+        const currentFilePath = fileURLToPath(import.meta.url);
+        const databasePackageDir = dirname(dirname(currentFilePath));
+        await deployPrismaMigrations({
+            packageDir: databasePackageDir,
+            databasePath: dbPath,
+            expectedVersion: HUB_DB_VERSION,
+        });
+    }
+    function getSchemaVersion() {
+        return HUB_DB_VERSION;
+    }
+    async function disconnect() {
+        await prisma.$disconnect();
+    }
+    return {
+        hubDb: prisma,
+        getSchemaVersion,
+        disconnect,
+    };
+}

package/dist/hubSessionService.js

@@ -0,0 +1,112 @@
+import { existsSync } from "fs";
+import { hubDbPath } from "./dbConfig.js";
+import { createPrismaClient } from "./prismaClient.js";
+let prisma = null;
+/**
+ * Initialize hub sessions by connecting to the shared naisys_hub.db.
+ * Idempotent — returns early if already initialized.
+ * No-ops gracefully if NAISYS_FOLDER is unset or the database doesn't exist.
+ */
+export async function createHubDatabaseClient() {
+    if (prisma)
+        return true;
+    const dbPath = hubDbPath();
+    if (!existsSync(dbPath))
+        return false;
+    prisma = await createPrismaClient(dbPath);
+    return true;
+}
+/**
+ * Find an agent (from the hub `users` table) by API key.
+ */
+export async function findAgentByApiKey(apiKey) {
+    if (!prisma)
+        return null;
+    const user = await prisma.users.findUnique({
+        where: { api_key: apiKey },
+        select: { uuid: true, username: true },
+    });
+    return user;
+}
+/**
+ * Find a hub agent by its numeric ID.
+ */
+export async function getHubAgentById(id) {
+    if (!prisma)
+        return null;
+    return prisma.users.findUnique({
+        where: { id },
+        select: { id: true, uuid: true, username: true },
+    });
+}
+/**
+ * Look up an agent's API key by UUID.
+ */
+export async function getAgentApiKeyByUuid(uuid) {
+    if (!prisma)
+        return null;
+    const user = await prisma.users.findFirst({
+        where: { uuid },
+        select: { api_key: true },
+    });
+    return user?.api_key ?? null;
+}
+/**
+ * Get the latest run_id and current session start time for a hub user by UUID.
+ */
+export async function getLatestRunInfoByUuid(uuid) {
+    if (!prisma)
+        return null;
+    const user = await prisma.users.findFirst({
+        where: { uuid },
+        select: { id: true },
+    });
+    if (!user)
+        return null;
+    const latest = await prisma.run_session.findFirst({
+        where: { user_id: user.id },
+        orderBy: [{ run_id: "desc" }, { session_id: "desc" }],
+        select: { run_id: true, created_at: true },
+    });
+    if (!latest)
+        return null;
+    return { runId: latest.run_id, sessionStart: latest.created_at };
+}
+/**
+ * Sum the cost of all cost entries for a hub user (by UUID) within a time range.
+ */
+export async function sumCostsByUuid(uuid, from, to) {
+    if (!prisma)
+        return 0;
+    const user = await prisma.users.findFirst({
+        where: { uuid },
+        select: { id: true },
+    });
+    if (!user)
+        return 0;
+    const result = await prisma.costs.aggregate({
+        _sum: { cost: true },
+        where: {
+            user_id: user.id,
+            created_at: { gte: from, lte: to },
+        },
+    });
+    return result._sum.cost ?? 0;
+}
+/**
+ * Rotate an agent's API key by UUID.
+ */
+export async function rotateAgentApiKeyByUuid(uuid, newKey) {
+    if (!prisma)
+        throw new Error("Hub database not initialized");
+    const user = await prisma.users.findFirst({
+        where: { uuid },
+        select: { id: true },
+    });
+    if (!user)
+        throw new Error("Agent not found in hub database");
+    await prisma.users.update({
+        where: { id: user.id },
+        data: { api_key: newKey },
+    });
+}

package/dist/index.js

@@ -0,0 +1,13 @@
+// Re-export Database Service
+export { createHubDatabaseService, } from "./hubDatabaseService.js";
+// Re-export Migration Helper
+export { deployPrismaMigrations } from "@naisys/common-node";
+// Re-export Hub Session Service
+export { createHubDatabaseClient, findAgentByApiKey, getAgentApiKeyByUuid, getHubAgentById, getLatestRunInfoByUuid, rotateAgentApiKeyByUuid, sumCostsByUuid, } from "./hubSessionService.js";
+// Re-export Prisma Client and all generated types
+export * from "./generated/prisma/client.js";
+export { PrismaClient } from "./generated/prisma/client.js";
+// Re-export Prisma enums
+export * from "./generated/prisma/enums.js";
+// Re-export Prisma client factory
+export { createPrismaClient } from "./prismaClient.js";