@kweaver-ai/kweaver-sdk 0.5.0 → 0.5.2

package/dist/commands/vega.js

@@ -1,6 +1,6 @@
 import { createInterface } from "node:readline";
 import { ensureValidToken, formatHttpError, with401RefreshRetry } from "../auth/oauth.js";
-import { vegaHealth, listVegaCatalogs, getVegaCatalog, createVegaCatalog, updateVegaCatalog, deleteVegaCatalogs, vegaCatalogHealthStatus, testVegaCatalogConnection, discoverVegaCatalog, listVegaCatalogResources, listVegaResources, getVegaResource, queryVegaResourceData, createVegaResource, updateVegaResource, deleteVegaResources, listVegaConnectorTypes, getVegaConnectorType, registerVegaConnectorType, updateVegaConnectorType, deleteVegaConnectorType, setVegaConnectorTypeEnabled, listVegaDiscoverTasks, getVegaDiscoverTask, } from "../api/vega.js";
+import { vegaHealth, listVegaCatalogs, getVegaCatalog, createVegaCatalog, updateVegaCatalog, deleteVegaCatalogs, vegaCatalogHealthStatus, testVegaCatalogConnection, discoverVegaCatalog, listVegaCatalogResources, listVegaResources, getVegaResource, queryVegaResourceData, createVegaResource, updateVegaResource, deleteVegaResources, listVegaConnectorTypes, getVegaConnectorType, registerVegaConnectorType, updateVegaConnectorType, deleteVegaConnectorType, setVegaConnectorTypeEnabled, listVegaDiscoverTasks, getVegaDiscoverTask, createVegaDatasetDocs, updateVegaDatasetDocs, deleteVegaDatasetDocs, deleteVegaDatasetDocsQuery, buildVegaDataset, getVegaDatasetBuildStatus, executeVegaQuery, listAllVegaResources, } from "../api/vega.js";
 import { formatCallOutput } from "./call.js";
 import { resolveBusinessDomain } from "../config/store.js";
 // ---------------------------------------------------------------------------
@@ -28,6 +28,14 @@ Subcommands:
   resource create --catalog-id <cid> --name <n> --category <cat>
   resource update <id> [--name X] [--status X] [--tags X] [-d json]
   resource delete <ids...> [-y]
+  resource list-all [--limit N] [--offset N]
+  dataset create-docs <resource-id> -d <json-array>
+  dataset update-docs <resource-id> -d <json-array>
+  dataset delete-docs <resource-id> <doc-ids...>
+  dataset delete-docs-query <resource-id> -d <filter-json>
+  dataset build <resource-id> [--mode full|incremental|realtime]
+  dataset build-status <resource-id> <task-id>
+  query execute -d <json>
   connector-type list                 List connector types
   connector-type get <type>           Get connector type details
   connector-type register -d <json>   Register a new connector type
@@ -94,6 +102,10 @@ export async function runVegaCommand(args) {
             return runVegaCatalogCommand(rest);
         if (subcommand === "resource")
             return runVegaResourceCommand(rest);
+        if (subcommand === "dataset")
+            return runVegaDatasetCommand(rest);
+        if (subcommand === "query")
+            return runVegaQueryCommand(rest);
         if (subcommand === "connector-type")
             return runVegaConnectorTypeCommand(rest);
         if (subcommand === "discovery-task")
@@ -142,19 +154,25 @@ async function runVegaStatsCommand(args) {
         return 0;
     }
     const { businessDomain, pretty } = parseCommonFlags(args);
-    const token = await ensureValidToken();
-    const body = await listVegaCatalogs({
-        baseUrl: token.baseUrl,
-        accessToken: token.accessToken,
-        limit: 100,
-        businessDomain,
-    });
-    const parsed = JSON.parse(body);
-    const entries = Array.isArray(parsed) ? parsed : (parsed.entries ?? parsed.data ?? parsed.items ?? parsed.catalogs ?? []);
-    const count = Array.isArray(entries) ? entries.length : 0;
-    const stats = { catalog_count: count };
-    console.log(pretty ? JSON.stringify(stats, null, 2) : JSON.stringify(stats));
-    return 0;
+    try {
+        const token = await ensureValidToken();
+        const body = await listVegaCatalogs({
+            baseUrl: token.baseUrl,
+            accessToken: token.accessToken,
+            limit: 100,
+            businessDomain,
+        });
+        const parsed = JSON.parse(body);
+        const entries = Array.isArray(parsed) ? parsed : (parsed.entries ?? parsed.data ?? parsed.items ?? parsed.catalogs ?? []);
+        const count = Array.isArray(entries) ? entries.length : 0;
+        const stats = { catalog_count: count };
+        console.log(pretty ? JSON.stringify(stats, null, 2) : JSON.stringify(stats));
+        return 0;
+    }
+    catch (error) {
+        console.error(formatHttpError(error));
+        return 1;
+    }
 }
 // ---------------------------------------------------------------------------
 // Top-level: inspect
@@ -652,6 +670,7 @@ async function runVegaResourceCommand(args) {
 
 Subcommands:
   list [--catalog-id X] [--category X] [--status X] [--limit N] [--offset N]
+  list-all [--limit N] [--offset N]
   get <id>
   query <id> -d <json-body>
   create --catalog-id <cid> --name <name> --category <cat>
@@ -661,6 +680,8 @@ Subcommands:
     }
     if (sub === "list")
         return await runResourceList(rest);
+    if (sub === "list-all")
+        return await runResourceListAll(rest);
     if (sub === "get")
         return await runResourceGet(rest);
     if (sub === "query")
@@ -735,6 +756,45 @@ Options:
     return 0;
 }
 // ---------------------------------------------------------------------------
+// resource list-all
+// ---------------------------------------------------------------------------
+async function runResourceListAll(args) {
+    if (args.includes("--help") || args.includes("-h")) {
+        console.log(`kweaver vega resource list-all [options]
+
+Options:
+  --limit <n>       Max results (default: 30)
+  --offset <n>      Offset
+  -bd, --biz-domain  Business domain (default: bd_public)
+  --pretty           Pretty-print JSON (default)`);
+        return 0;
+    }
+    let limit = 30;
+    let offset;
+    const { remaining, businessDomain, pretty } = parseCommonFlags(args);
+    for (let i = 0; i < remaining.length; i += 1) {
+        const arg = remaining[i];
+        if (arg === "--limit" && remaining[i + 1]) {
+            limit = parseInt(remaining[++i], 10);
+            continue;
+        }
+        if (arg === "--offset" && remaining[i + 1]) {
+            offset = parseInt(remaining[++i], 10);
+            continue;
+        }
+    }
+    const token = await ensureValidToken();
+    const body = await listAllVegaResources({
+        baseUrl: token.baseUrl,
+        accessToken: token.accessToken,
+        limit,
+        offset,
+        businessDomain,
+    });
+    console.log(formatCallOutput(body, pretty));
+    return 0;
+}
+// ---------------------------------------------------------------------------
 // resource get
 // ---------------------------------------------------------------------------
 async function runResourceGet(args) {
@@ -980,6 +1040,303 @@ Options:
     return 0;
 }
 // ---------------------------------------------------------------------------
+// Dataset router
+// ---------------------------------------------------------------------------
+async function runVegaDatasetCommand(args) {
+    const [sub, ...rest] = args;
+    if (!sub || sub === "--help" || sub === "-h") {
+        console.log(`kweaver vega dataset
+
+Subcommands:
+  create-docs <resource-id> -d <json-array>
+  update-docs <resource-id> -d <json-array>
+  delete-docs <resource-id> <doc-ids...>
+  delete-docs-query <resource-id> -d <filter-json>
+  build <resource-id> [--mode full|incremental|realtime]
+  build-status <resource-id> <task-id>`);
+        return 0;
+    }
+    if (sub === "create-docs")
+        return await runDatasetCreateDocs(rest);
+    if (sub === "update-docs")
+        return await runDatasetUpdateDocs(rest);
+    if (sub === "delete-docs")
+        return await runDatasetDeleteDocs(rest);
+    if (sub === "delete-docs-query")
+        return await runDatasetDeleteDocsQuery(rest);
+    if (sub === "build")
+        return await runDatasetBuild(rest);
+    if (sub === "build-status")
+        return await runDatasetBuildStatus(rest);
+    console.error(`Unknown dataset subcommand: ${sub}`);
+    return 1;
+}
+// ---------------------------------------------------------------------------
+// dataset create-docs
+// ---------------------------------------------------------------------------
+async function runDatasetCreateDocs(args) {
+    if (args.includes("--help") || args.includes("-h")) {
+        console.log(`kweaver vega dataset create-docs <resource-id> -d <json-array>
+
+Options:
+  -d, --data <json>   Array of documents (JSON string)`);
+        return 0;
+    }
+    let data;
+    const { remaining, businessDomain, pretty } = parseCommonFlags(args);
+    const positionals = [];
+    for (let i = 0; i < remaining.length; i += 1) {
+        const arg = remaining[i];
+        if ((arg === "-d" || arg === "--data") && remaining[i + 1]) {
+            data = remaining[++i];
+            continue;
+        }
+        if (!arg.startsWith("-"))
+            positionals.push(arg);
+    }
+    const id = positionals[0];
+    if (!id || !data) {
+        console.error("Usage: kweaver vega dataset create-docs <resource-id> -d <json-array>");
+        return 1;
+    }
+    const token = await ensureValidToken();
+    const body = await createVegaDatasetDocs({
+        baseUrl: token.baseUrl,
+        accessToken: token.accessToken,
+        id,
+        body: data,
+        businessDomain,
+    });
+    console.log(formatCallOutput(body, pretty));
+    return 0;
+}
+// ---------------------------------------------------------------------------
+// dataset update-docs
+// ---------------------------------------------------------------------------
+async function runDatasetUpdateDocs(args) {
+    if (args.includes("--help") || args.includes("-h")) {
+        console.log(`kweaver vega dataset update-docs <resource-id> -d <json-array>
+
+Options:
+  -d, --data <json>   Array of documents with ids (JSON string)`);
+        return 0;
+    }
+    let data;
+    const { remaining, businessDomain, pretty } = parseCommonFlags(args);
+    const positionals = [];
+    for (let i = 0; i < remaining.length; i += 1) {
+        const arg = remaining[i];
+        if ((arg === "-d" || arg === "--data") && remaining[i + 1]) {
+            data = remaining[++i];
+            continue;
+        }
+        if (!arg.startsWith("-"))
+            positionals.push(arg);
+    }
+    const id = positionals[0];
+    if (!id || !data) {
+        console.error("Usage: kweaver vega dataset update-docs <resource-id> -d <json-array>");
+        return 1;
+    }
+    const token = await ensureValidToken();
+    const body = await updateVegaDatasetDocs({
+        baseUrl: token.baseUrl,
+        accessToken: token.accessToken,
+        id,
+        body: data,
+        businessDomain,
+    });
+    console.log(formatCallOutput(body || "{}", pretty));
+    return 0;
+}
+// ---------------------------------------------------------------------------
+// dataset delete-docs
+// ---------------------------------------------------------------------------
+async function runDatasetDeleteDocs(args) {
+    if (args.includes("--help") || args.includes("-h")) {
+        console.log(`kweaver vega dataset delete-docs <resource-id> <doc-ids...>
+
+Positional args:
+  resource-id   The dataset resource ID
+  doc-ids       One or more document IDs (comma-joined)`);
+        return 0;
+    }
+    const { remaining, businessDomain, pretty } = parseCommonFlags(args);
+    const positionals = remaining.filter((a) => !a.startsWith("-"));
+    const id = positionals[0];
+    const docIds = positionals.slice(1);
+    if (!id || docIds.length === 0) {
+        console.error("Usage: kweaver vega dataset delete-docs <resource-id> <doc-ids...>");
+        return 1;
+    }
+    const token = await ensureValidToken();
+    const body = await deleteVegaDatasetDocs({
+        baseUrl: token.baseUrl,
+        accessToken: token.accessToken,
+        id,
+        docIds: docIds.join(","),
+        businessDomain,
+    });
+    console.log(formatCallOutput(body || "{}", pretty));
+    return 0;
+}
+// ---------------------------------------------------------------------------
+// dataset delete-docs-query
+// ---------------------------------------------------------------------------
+async function runDatasetDeleteDocsQuery(args) {
+    if (args.includes("--help") || args.includes("-h")) {
+        console.log(`kweaver vega dataset delete-docs-query <resource-id> -d <filter-json>
+
+Options:
+  -d, --data <json>   Filter condition (JSON string)`);
+        return 0;
+    }
+    let data;
+    const { remaining, businessDomain, pretty } = parseCommonFlags(args);
+    const positionals = [];
+    for (let i = 0; i < remaining.length; i += 1) {
+        const arg = remaining[i];
+        if ((arg === "-d" || arg === "--data") && remaining[i + 1]) {
+            data = remaining[++i];
+            continue;
+        }
+        if (!arg.startsWith("-"))
+            positionals.push(arg);
+    }
+    const id = positionals[0];
+    if (!id || !data) {
+        console.error("Usage: kweaver vega dataset delete-docs-query <resource-id> -d <filter-json>");
+        return 1;
+    }
+    const token = await ensureValidToken();
+    const body = await deleteVegaDatasetDocsQuery({
+        baseUrl: token.baseUrl,
+        accessToken: token.accessToken,
+        id,
+        body: data,
+        businessDomain,
+    });
+    console.log(formatCallOutput(body || "{}", pretty));
+    return 0;
+}
+// ---------------------------------------------------------------------------
+// dataset build
+// ---------------------------------------------------------------------------
+async function runDatasetBuild(args) {
+    if (args.includes("--help") || args.includes("-h")) {
+        console.log(`kweaver vega dataset build <resource-id> [options]
+
+Options:
+  --mode <mode>   Build mode: full, incremental, realtime (default: full)`);
+        return 0;
+    }
+    let mode = "full";
+    const { remaining, businessDomain, pretty } = parseCommonFlags(args);
+    const positionals = [];
+    for (let i = 0; i < remaining.length; i += 1) {
+        const arg = remaining[i];
+        if (arg === "--mode" && remaining[i + 1]) {
+            mode = remaining[++i];
+            continue;
+        }
+        if (!arg.startsWith("-"))
+            positionals.push(arg);
+    }
+    const id = positionals[0];
+    if (!id) {
+        console.error("Usage: kweaver vega dataset build <resource-id> [--mode full|incremental|realtime]");
+        return 1;
+    }
+    const token = await ensureValidToken();
+    const body = await buildVegaDataset({
+        baseUrl: token.baseUrl,
+        accessToken: token.accessToken,
+        id,
+        mode,
+        businessDomain,
+    });
+    console.log(formatCallOutput(body, pretty));
+    return 0;
+}
+// ---------------------------------------------------------------------------
+// dataset build-status
+// ---------------------------------------------------------------------------
+async function runDatasetBuildStatus(args) {
+    if (args.includes("--help") || args.includes("-h")) {
+        console.log(`kweaver vega dataset build-status <resource-id> <task-id>`);
+        return 0;
+    }
+    const { remaining, businessDomain, pretty } = parseCommonFlags(args);
+    const positionals = remaining.filter((a) => !a.startsWith("-"));
+    const id = positionals[0];
+    const taskId = positionals[1];
+    if (!id || !taskId) {
+        console.error("Usage: kweaver vega dataset build-status <resource-id> <task-id>");
+        return 1;
+    }
+    const token = await ensureValidToken();
+    const body = await getVegaDatasetBuildStatus({
+        baseUrl: token.baseUrl,
+        accessToken: token.accessToken,
+        id,
+        taskId,
+        businessDomain,
+    });
+    console.log(formatCallOutput(body, pretty));
+    return 0;
+}
+// ---------------------------------------------------------------------------
+// Query router
+// ---------------------------------------------------------------------------
+async function runVegaQueryCommand(args) {
+    const [sub, ...rest] = args;
+    if (!sub || sub === "--help" || sub === "-h") {
+        console.log(`kweaver vega query
+
+Subcommands:
+  execute -d <json>   Execute a query`);
+        return 0;
+    }
+    if (sub === "execute")
+        return await runQueryExecute(rest);
+    console.error(`Unknown query subcommand: ${sub}`);
+    return 1;
+}
+// ---------------------------------------------------------------------------
+// query execute
+// ---------------------------------------------------------------------------
+async function runQueryExecute(args) {
+    if (args.includes("--help") || args.includes("-h")) {
+        console.log(`kweaver vega query execute -d <json>
+
+Options:
+  -d, --data <json>   Query body (tables, joins, output_fields, filter_condition, sort, limit, ...)`);
+        return 0;
+    }
+    let data;
+    const { remaining, businessDomain, pretty } = parseCommonFlags(args);
+    for (let i = 0; i < remaining.length; i += 1) {
+        const arg = remaining[i];
+        if ((arg === "-d" || arg === "--data") && remaining[i + 1]) {
+            data = remaining[++i];
+            continue;
+        }
+    }
+    if (!data) {
+        console.error("Usage: kweaver vega query execute -d <json>");
+        return 1;
+    }
+    const token = await ensureValidToken();
+    const body = await executeVegaQuery({
+        baseUrl: token.baseUrl,
+        accessToken: token.accessToken,
+        body: data,
+        businessDomain,
+    });
+    console.log(formatCallOutput(body, pretty));
+    return 0;
+}
+// ---------------------------------------------------------------------------
 // Connector-type router
 // ---------------------------------------------------------------------------
 async function runVegaConnectorTypeCommand(args) {

package/dist/config/jwt.d.ts

@@ -0,0 +1,6 @@
+/**
+ * Lightweight JWT payload decoding (no signature verification).
+ * Used by the credential store and CLI to extract user identity from id_token / access_token.
+ */
+export declare function decodeJwtPayload(jwt: string): Record<string, unknown> | null;
+export declare function extractUserIdFromJwt(jwt: string): string | null;

package/dist/config/jwt.js

@@ -0,0 +1,21 @@
+/**
+ * Lightweight JWT payload decoding (no signature verification).
+ * Used by the credential store and CLI to extract user identity from id_token / access_token.
+ */
+export function decodeJwtPayload(jwt) {
+    const parts = jwt.split(".");
+    if (parts.length !== 3)
+        return null;
+    try {
+        const base64 = parts[1].replace(/-/g, "+").replace(/_/g, "/");
+        const json = Buffer.from(base64, "base64").toString("utf8");
+        return JSON.parse(json);
+    }
+    catch {
+        return null;
+    }
+}
+export function extractUserIdFromJwt(jwt) {
+    const payload = decodeJwtPayload(jwt);
+    return typeof payload?.sub === "string" ? payload.sub : null;
+}

package/dist/config/store.d.ts

@@ -10,6 +10,8 @@ export interface TokenConfig {
     obtainedAt: string;
     /** When true, skip TLS certificate verification for this platform (saved by `kweaver auth --insecure`). */
     tlsInsecure?: boolean;
+    /** Human-readable display name fetched from userinfo at login time. */
+    displayName?: string;
 }
 /** OAuth2 client registration (per platform), used for refresh_token grant. */
 export interface ClientConfig {
@@ -38,7 +40,37 @@ export interface PlatformSummary {
     hasToken: boolean;
     isCurrent: boolean;
     alias?: string;
+    /** Active user ID for this platform (from id_token sub claim). */
+    userId?: string;
+    /** Human-readable name persisted from /oauth2/userinfo at login time. */
+    displayName?: string;
 }
+/** Extract userId from a TokenConfig (try idToken, then accessToken, fallback "default"). */
+export declare function extractUserId(token: TokenConfig): string;
+/** Get the active userId for a platform. */
+export declare function getActiveUser(baseUrl: string): string | null;
+/** Set the active userId for a platform. */
+export declare function setActiveUser(baseUrl: string, userId: string): void;
+/** List all user IDs stored under a platform. */
+export declare function listUsers(baseUrl: string): string[];
+/** Load a specific user's token (not necessarily the active user). */
+export declare function loadUserTokenConfig(baseUrl: string, userId: string): TokenConfig | null;
+export interface UserProfile {
+    userId: string;
+    username?: string;
+    email?: string;
+}
+/**
+ * List all user profiles for a platform, enriched with display names.
+ *
+ * Resolution order for username:
+ *   1. ``displayName`` field persisted in token.json (set at login via /oauth2/userinfo)
+ *   2. ``preferred_username`` or ``name`` decoded from id_token JWT
+ */
+export declare function listUserProfiles(baseUrl: string): UserProfile[];
+/** Resolve a user identifier (userId, username, or email) to a userId for the given platform.
+ *  userId and username are matched case-sensitively; email is case-insensitive. */
+export declare function resolveUserId(baseUrl: string, identifier: string): string | null;
 export declare function getConfigDir(): string;
 export declare function getCurrentPlatform(): string | null;
 export declare function setCurrentPlatform(baseUrl: string): void;
@@ -47,9 +79,10 @@ export declare function deletePlatformAlias(baseUrl: string): void;
 export declare function getPlatformAlias(baseUrl: string): string | null;
 export declare function resolvePlatformIdentifier(value: string): string | null;
 export declare function loadTokenConfig(baseUrl?: string): TokenConfig | null;
-export declare function saveTokenConfig(config: TokenConfig): void;
+export declare function saveTokenConfig(config: TokenConfig, userId?: string): void;
 export declare function loadClientConfig(baseUrl?: string): ClientConfig | null;
 export declare function saveClientConfig(baseUrl: string, config: ClientConfig): void;
+export declare function deleteClientConfig(baseUrl: string): void;
 export declare function loadContextLoaderConfig(baseUrl?: string): ContextLoaderConfig | null;
 export declare function saveContextLoaderConfig(baseUrl: string, config: ContextLoaderConfig): void;
 export interface CurrentContextLoaderKn {
@@ -61,10 +94,9 @@ export declare function addContextLoaderEntry(baseUrl: string, name: string, knI
 export declare function setCurrentContextLoader(baseUrl: string, name: string): void;
 export declare function removeContextLoaderEntry(baseUrl: string, name: string): void;
 export declare function hasPlatform(baseUrl: string): boolean;
-/**
- * Remove token for a platform so the next auth will do a full login.
- */
-export declare function clearPlatformSession(baseUrl: string): void;
+export declare function clearPlatformSession(baseUrl: string, userId?: string): void;
+/** Delete a single user's profile directory under a platform. */
+export declare function deleteUser(baseUrl: string, userId: string): void;
 export declare function deletePlatform(baseUrl: string): void;
 export declare function listPlatforms(): PlatformSummary[];
 /** Per-platform config (not auth — general settings). */