@dainprotocol/service-sdk 2.0.94 → 2.1.0

package/dist/service/types.d.ts

@@ -7,6 +7,7 @@ import { ServerType } from "@hono/node-server/.";
 import { DainPlugin } from "../plugins/types";
 import { StorageAdapter } from "@dainprotocol/oauth2-token-manager";
 import { DirectAuthSetupManager } from "./direct-auth-setup";
+import type { DainAccountContext, DainActionGrant, DainAuthContext, DainDataPermissionGrant, DainGroupContext, DainRuntimeContext, DainServiceRegistryPolicy } from "../protocol";
 export interface SimpleOAuth2Client {
     getAccessToken: (provider: string, email: string, options?: any) => Promise<string | null>;
     getValidToken: (provider: string, email: string, options?: any) => Promise<any>;
@@ -19,9 +20,23 @@ export type AgentInfo = {
     agentId: string;
     address: string;
     smartAccountPDA?: string;
+    dainAccountId?: string;
+    dainGroupId?: string;
+    account?: DainAccountContext;
+    group?: DainGroupContext;
+    auth?: DainAuthContext;
+    grants?: DainActionGrant[];
+    dataPermissions?: DainDataPermissionGrant[];
+    registryPolicy?: DainServiceRegistryPolicy;
     id: string;
     webhookUrl?: string;
 };
+export type ServiceExtraData = DainRuntimeContext & {
+    plugins?: Record<string, any>;
+    oauth2Client?: SimpleOAuth2Client;
+    app?: Hono;
+    [key: string]: any;
+};
 /**
  * Context for HITL action routing
  * Contains user and platform information for action tracking
@@ -67,6 +82,7 @@ export type Metadata = {
     contractVersion?: string;
     capabilities?: ServiceCapabilities;
     compatibility?: ContractCompatibility;
+    registryPolicy?: DainServiceRegistryPolicy;
 };
 export type ServiceCapabilities = {
     tools?: boolean;
@@ -78,6 +94,12 @@ export type ServiceCapabilities = {
     datasourcePolicy?: boolean;
     widgetPolicy?: boolean;
     toolSafety?: boolean;
+    dainAccount?: boolean;
+    registryDiscovery?: boolean;
+    dataPermissions?: boolean;
+    actionGrants?: boolean;
+    smartAccountTransactions?: boolean;
+    payments?: boolean;
     [key: string]: unknown;
 };
 export type ContractCompatibility = {
@@ -97,7 +119,16 @@ export type Identity = {
 export type ToolContext = {
     app: Hono;
     oauth2Client?: SimpleOAuth2Client;
-    extraData?: any;
+    extraData?: ServiceExtraData;
+    dainAccountId?: string;
+    dainGroupId?: string;
+    smartAccountPDA?: string;
+    account?: DainAccountContext;
+    group?: DainGroupContext;
+    auth?: DainAuthContext;
+    grants?: DainActionGrant[];
+    dataPermissions?: DainDataPermissionGrant[];
+    registryPolicy?: DainServiceRegistryPolicy;
     updateUI?: (update: {
         ui?: any;
         type?: string;
@@ -236,7 +267,7 @@ export type ToolConfig<TInput extends z.ZodType = any, TOutput extends z.ZodType
             type: "one-time" | "recurring";
         }[];
     }>;
-    handleInputError?: (error: ZodError, agentInfo: AgentInfo, extraData?: any) => Promise<{
+    handleInputError?: (error: ZodError, agentInfo: AgentInfo, extraData?: ToolContext) => Promise<{
         success?: boolean;
         text: string;
         data: z.output<TOutput>;
@@ -266,11 +297,7 @@ export interface ServiceContext<TData = any> {
     id: string;
     name: string;
     description: string;
-    getContextData: (agentInfo: AgentInfo, extraData?: {
-        plugins?: Record<string, any>;
-        oauth2Client?: SimpleOAuth2Client;
-        [key: string]: any;
-    }) => Promise<TData>;
+    getContextData: (agentInfo: AgentInfo, extraData?: ServiceExtraData) => Promise<TData>;
 }
 export type DatasourceType = "metadata" | "data" | "config" | "file" | "state" | "crypto" | "json" | "csv" | "xml" | "text";
 export type DatasourceTransport = "stream" | "poll" | "hybrid";
@@ -292,11 +319,7 @@ export interface ServiceDatasource<TInput extends z.ZodType = any, TData = any>
     scope?: DatasourceScope;
     dataClass?: string;
     input: TInput;
-    getDatasource: (agentInfo: AgentInfo, params: z.infer<TInput>, extraData?: {
-        plugins?: Record<string, any>;
-        oauth2Client?: SimpleOAuth2Client;
-        [key: string]: any;
-    }) => Promise<TData>;
+    getDatasource: (agentInfo: AgentInfo, params: z.infer<TInput>, extraData?: ServiceExtraData) => Promise<TData>;
 }
 export type ServiceWidget = {
     id: string;
@@ -309,11 +332,7 @@ export type ServiceWidget = {
      * UI runtimes clamp this interval to safe bounds.
      */
     refreshIntervalMs?: number;
-    getWidget: (agentInfo: AgentInfo, extraData?: {
-        plugins?: Record<string, any>;
-        oauth2Client?: SimpleOAuth2Client;
-        [key: string]: any;
-    }) => Promise<{
+    getWidget: (agentInfo: AgentInfo, extraData?: ServiceExtraData) => Promise<{
         text?: string;
         data?: any;
         ui?: any;
@@ -370,7 +389,7 @@ export interface ServiceAgent<TInput extends z.ZodType = any, TOutput extends z.
     output: TOutput;
 }
 export type DAINServiceConfig = {
-    metadata: Omit<Metadata, 'id' | 'orgId' | 'agentId'> & {
+    metadata: Omit<Metadata, "id" | "orgId" | "agentId"> & {
         id?: string;
         orgId?: string;
         agentId?: string;
@@ -386,14 +405,8 @@ export type DAINServiceConfig = {
     datasources?: ServiceDatasource[];
     widgets?: ServiceWidget[];
     agents?: ServiceAgent[];
-    getUserWidgets?: (agentInfo: AgentInfo, extraData?: {
-        plugins?: Record<string, any>;
-        [key: string]: any;
-    }) => Promise<string[]>;
-    homeUI?: string | ((agentInfo: AgentInfo, extraData?: {
-        plugins?: Record<string, any>;
-        [key: string]: any;
-    }) => Promise<string>);
+    getUserWidgets?: (agentInfo: AgentInfo, extraData?: ServiceExtraData) => Promise<string[]>;
+    homeUI?: string | ((agentInfo: AgentInfo, extraData?: ServiceExtraData) => Promise<string>);
     plugins?: DainPlugin[];
     oauth2?: OAuth2Config & {
         baseUrl: string;
@@ -415,8 +428,8 @@ export type DAINServiceConfig = {
     dainIdUrl?: string;
     jwtIssuer?: string;
     platformBaseUrl?: string;
-    webhookTriggers?: import('./webhooks').WebhookTriggerRegistry;
-    skills?: import('./skills').SkillRegistry;
+    webhookTriggers?: import("./webhooks").WebhookTriggerRegistry;
+    skills?: import("./skills").SkillRegistry;
     hitl?: {
         /** Enable HITL action webhook */
         enabled: boolean;
@@ -493,9 +506,9 @@ export interface DirectProviderConfig {
  * Allows type-safe handling of different authentication patterns
  */
 export type AuthProviderConfig = (OAuth2ProviderConfig & {
-    type: 'oauth2';
+    type: "oauth2";
 }) | (DirectProviderConfig & {
-    type: 'direct';
+    type: "direct";
 });
 /**
  * Provider information returned to clients
@@ -503,7 +516,7 @@ export type AuthProviderConfig = (OAuth2ProviderConfig & {
  */
 export interface ProviderInfo {
     name: string;
-    type: 'oauth2' | 'direct';
+    type: "oauth2" | "direct";
     connected: boolean;
     reason?: string;
     requiredTools?: string[];
@@ -515,14 +528,14 @@ export interface ProviderInfo {
  * OAuth2 authentication method details
  */
 export interface OAuth2AuthMethod {
-    type: 'oauth2';
+    type: "oauth2";
     authUrl: string;
 }
 /**
  * Direct configuration authentication method details
  */
 export interface DirectAuthMethod {
-    type: 'direct';
+    type: "direct";
     onboardingTool: string;
 }
 export type OAuth2Config = {
@@ -545,7 +558,7 @@ declare module "hono" {
         processes?: ProcessHandler;
     }
     interface ContextVariableMap {
-        authMethod?: 'jwt' | 'apiKey';
+        authMethod?: "jwt" | "apiKey";
         smartAccountId?: string;
         scope?: string[];
         jwtPayload?: any;
@@ -553,5 +566,6 @@ declare module "hono" {
         orgId?: string;
         address?: string;
         smartAccountPDA?: string;
+        dainRuntimeContext?: DainRuntimeContext;
     }
 }

package/package.json

@@ -1,10 +1,14 @@
 {
   "name": "@dainprotocol/service-sdk",
-  "version": "2.0.94",
+  "version": "2.1.0",
   "description": "DAIN Service SDK",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "private": false,
+  "packageManager": "bun@1.3.13",
+  "engines": {
+    "bun": ">=1.3.13"
+  },
   "publishConfig": {
     "access": "public"
   },
@@ -57,6 +61,10 @@
       "types": "./dist/lib/payments/index.d.ts",
       "default": "./dist/lib/payments/index.js"
     },
+    "./protocol": {
+      "types": "./dist/protocol/index.d.ts",
+      "default": "./dist/protocol/index.js"
+    },
     "./plugins": {
       "types": "./dist/plugins/index.d.ts",
       "default": "./dist/plugins/index.js"
@@ -103,16 +111,21 @@
       ],
       "payments": [
         "./dist/lib/payments/index.d.ts"
+      ],
+      "protocol": [
+        "./dist/protocol/index.d.ts"
       ]
     }
   },
   "scripts": {
-    "build": "tsc",
-    "build:types": "tsc --emitDeclarationOnly",
-    "test": "jest",
-    "test:watch": "jest --watch",
-    "prepublishOnly": "npm run build && npm run build:types",
-    "test:streaming": "jest src/__tests__/streaming.test.ts --runInBand"
+    "build": "bunx --bun tsc",
+    "build:types": "bunx --bun tsc --emitDeclarationOnly",
+    "test": "bun test --parallel=4",
+    "test:serial": "bun test --isolate",
+    "test:watch": "bun test --watch --isolate",
+    "test:jest": "jest --runInBand",
+    "prepublishOnly": "bun run build && bun run build:types",
+    "test:streaming": "bun test src/__tests__/streaming.test.ts --isolate"
   },
   "keywords": [
     "dain",