@agentrix/shared 2.0.14 → 2.2.0

package/dist/index.cjs

@@ -234,6 +234,8 @@ const StartTaskRequestSchema = zod.z.object({
   // Repository source type
   dataEncryptionKey: zod.z.string().optional(),
   // base64 sealed-box: app public key encrypted by machine public key
+  ownerEncryptedDataKey: zod.z.string().optional(),
+  // base64 secretbox: task data key encrypted by owner's machine AES key (local mode)
   // Multi-agent collaboration fields
   agentId: zod.z.string().optional(),
   // Agent ID to execute the task (overrides chat's first agent)
@@ -281,8 +283,8 @@ const StartTaskResponseSchema = zod.z.object({
 });
 const TaskItemSchema = zod.z.object({
   id: IdSchema,
-  type: zod.z.enum(["chat", "work"]).default("work"),
-  // Task type: "chat" (virtual task for chat events) | "work" (real work task)
+  type: zod.z.enum(["chat", "work", "shadow"]).default("work"),
+  // Task type: "chat" (virtual) | "work" (real work) | "shadow" (companion heartbeat/shadow)
   chatId: IdSchema,
   userId: zod.z.string(),
   state: zod.z.string(),
@@ -299,6 +301,7 @@ const TaskItemSchema = zod.z.object({
   // Custom task title (set by user, takes priority over title)
   agentSessionId: zod.z.string().nullable(),
   dataEncryptionKey: zod.z.string().nullable(),
+  ownerEncryptedDataKey: zod.z.string().nullable(),
   cwd: zod.z.string().nullable(),
   // Current working directory from CLI worker
   userCwd: zod.z.string().nullable(),
@@ -537,6 +540,25 @@ const SubTaskSummarySchema = zod.z.object({
 const ListSubTasksResponseSchema = zod.z.object({
   tasks: zod.z.array(SubTaskSummarySchema)
 });
+const ListRecentTasksRequestSchema = zod.z.object({
+  chatId: zod.z.string(),
+  limit: zod.z.coerce.number().int().min(1).max(50).optional().default(10),
+  status: zod.z.enum(["all", "active", "completed"]).optional().default("all")
+});
+const RecentTaskSummarySchema = zod.z.object({
+  taskId: IdSchema,
+  type: zod.z.enum(["chat", "work", "shadow"]),
+  state: zod.z.string(),
+  agentId: zod.z.string(),
+  title: zod.z.string().nullable(),
+  totalDuration: zod.z.number().nullable(),
+  createdAt: zod.z.string(),
+  updatedAt: zod.z.string()
+});
+const ListRecentTasksResponseSchema = zod.z.object({
+  tasks: zod.z.array(RecentTaskSummarySchema),
+  hasMore: zod.z.boolean()
+});
 const FindTaskByAgentRequestSchema = zod.z.object({
   parentTaskId: zod.z.string(),
   agentId: zod.z.string()
@@ -583,7 +605,10 @@ const ChatMemberInputSchema = zod.z.object({
 const CreateChatRequestSchema = zod.z.object({
   type: ChatTypeSchema.default("direct"),
   // Optional, defaults to 'direct'
-  members: zod.z.array(ChatMemberInputSchema)
+  members: zod.z.array(ChatMemberInputSchema),
+  // Encryption keys for companion chat tasks (generated by App, stored on virtual task)
+  dataEncryptionKey: zod.z.string().optional(),
+  ownerEncryptedDataKey: zod.z.string().optional()
 });
 const CreateChatResponseSchema = ChatWithMembersSchema;
 const ListChatsQuerySchema = zod.z.object({
@@ -621,7 +646,7 @@ const UpdateChatContextRequestSchema = zod.z.object({
 });
 const UpdateChatContextResponseSchema = ChatSchema;
 
-const AgentTypeSchema = zod.z.enum(["claude", "codex"]);
+const AgentTypeSchema = zod.z.enum(["claude", "codex", "companion"]);
 const DisplayConfigKeysSchema = zod.z.object({
   // PR Actions
   createPR: zod.z.string().optional(),
@@ -645,7 +670,9 @@ const DisplayConfigKeysSchema = zod.z.object({
 });
 const DisplayConfigSchema = zod.z.record(zod.z.string(), DisplayConfigKeysSchema);
 const AgentCustomConfigSchema = zod.z.object({
-  displayConfig: DisplayConfigSchema.optional()
+  displayConfig: DisplayConfigSchema.optional(),
+  machineId: zod.z.string().optional(),
+  heartbeatTaskId: zod.z.string().optional()
 });
 const AgentPermissionsSchema = zod.z.object({
   role: zod.z.string(),
@@ -656,6 +683,7 @@ const AgentPermissionsSchema = zod.z.object({
 const AgentSchema = zod.z.object({
   id: IdSchema,
   name: zod.z.string(),
+  displayName: zod.z.string().nullable(),
   type: AgentTypeSchema,
   avatar: zod.z.string().nullable(),
   userId: zod.z.string(),
@@ -692,6 +720,7 @@ const CreateAgentRequestSchema = zod.z.object({
 const CreateAgentResponseSchema = AgentSchema;
 const UpdateAgentRequestSchema = zod.z.object({
   name: zod.z.string().min(1).optional(),
+  displayName: zod.z.string().nullable().optional(),
   type: AgentTypeSchema.optional(),
   avatar: zod.z.string().nullable().optional(),
   description: zod.z.string().nullable().optional(),
@@ -1154,6 +1183,16 @@ const GitServerSchema = zod.z.object({
   baseUrl: zod.z.string().url(),
   apiUrl: zod.z.string().url(),
   oauthServerId: zod.z.string().nullable(),
+  ownerId: zod.z.string().nullable(),
+  // null = public (admin-created), userId = private (user-created)
+  authModeDefault: zod.z.string().default("oauth"),
+  // "oauth" | "local_pat"
+  executionMode: zod.z.string().default("server"),
+  // "server" | "daemon_proxy"
+  syncMode: zod.z.string().default("polling_only"),
+  // "polling_only" | "hybrid" | "webhook_only"
+  networkMode: zod.z.string().default("direct"),
+  // "direct" | "tunnel"
   enabled: zod.z.boolean(),
   createdAt: zod.z.union([DateSchema, zod.z.date().transform((d) => d.toISOString())]),
   updatedAt: zod.z.union([DateSchema, zod.z.date().transform((d) => d.toISOString())])
@@ -1216,7 +1255,7 @@ const AskUserResponseMessageSchema = zod.z.object({
 const TaskAgentInfoSchema = zod.z.object({
   id: zod.z.string(),
   name: zod.z.string(),
-  type: zod.z.enum(["claude", "codex"]).optional().default("claude"),
+  type: AgentTypeSchema.optional().default("claude"),
   description: zod.z.string().optional()
 });
 function isAskUserMessage(message) {
@@ -1225,8 +1264,14 @@ function isAskUserMessage(message) {
 function isAskUserResponseMessage(message) {
   return typeof message === "object" && message !== null && "type" in message && message.type === "ask_user_response";
 }
+function isCompanionHeartbeatMessage(message) {
+  return typeof message === "object" && message !== null && "type" in message && message.type === "companion_heartbeat";
+}
+function isCompanionReminderMessage(message) {
+  return typeof message === "object" && message !== null && "type" in message && message.type === "companion_reminder";
+}
 function isSDKMessage(message) {
-  return typeof message === "object" && message !== null && "type" in message && message.type !== "ask_user" && message.type !== "ask_user_response";
+  return typeof message === "object" && message !== null && "type" in message && message.type !== "ask_user" && message.type !== "ask_user_response" && message.type !== "companion_heartbeat" && message.type !== "companion_reminder";
 }
 function isSDKUserMessage(message) {
   return isSDKMessage(message) && message.type === "user";
@@ -1350,6 +1395,8 @@ const baseTaskSchema = EventBaseSchema.extend({
   // Existing repository ID (used to skip auto-association)
   repositorySourceType: zod.z.enum(["temporary", "directory", "git-server"]).optional().default("temporary"),
   // Repository source type (determines workspace mode)
+  gitServerId: zod.z.string().optional(),
+  // Git server ID (for local_pat mode: worker loads PAT locally)
   environmentVariables: zod.z.record(zod.z.string(), zod.z.string()).optional(),
   // Environment variables for task execution
   // Multi-agent collaboration fields
@@ -1361,8 +1408,8 @@ const baseTaskSchema = EventBaseSchema.extend({
   // Agents available for this task
   todos: zod.z.array(TaskTodoSchema).optional(),
   // Structured todos for group tasks
-  taskType: zod.z.enum(["chat", "work"]).optional().default("work"),
-  // Task type: 'chat' for main chat, 'work' for task execution
+  taskType: zod.z.enum(["chat", "work", "shadow"]).optional().default("work"),
+  // Task type: 'chat' for main chat, 'work' for task execution, 'shadow' for companion shadow
   customTitle: zod.z.string().min(1).max(200).optional()
   // Custom task title (set by user/tool)
 });
@@ -1437,6 +1484,8 @@ const TaskMessageSchema = EventBaseSchema.extend({
       message: "Invalid TaskMessagePayload format"
     }
   ).optional(),
+  messageType: zod.z.string().optional(),
+  // Message type metadata (for encrypted payload routing)
   encryptedMessage: zod.z.string().optional(),
   // base64 sealed-box payload (local mode)
   // Multi-agent collaboration fields
@@ -1491,7 +1540,9 @@ const MachineRtcRequestSchema = EventBaseSchema.extend({
   machineId: zod.z.string(),
   sessionId: zod.z.string(),
   role: zod.z.literal("app"),
-  userId: zod.z.string().optional()
+  taskId: zod.z.string().optional(),
+  userId: zod.z.string().optional(),
+  workspaceUserId: zod.z.string().optional()
 });
 const MachineRtcResponseSchema = EventBaseSchema.extend({
   machineId: zod.z.string(),
@@ -1509,7 +1560,9 @@ const RtcSignalSchema = EventBaseSchema.extend({
   sessionId: zod.z.string(),
   from: zod.z.enum(["app", "machine"]),
   signal: zod.z.any(),
-  userId: zod.z.string().optional()
+  userId: zod.z.string().optional(),
+  taskId: zod.z.string().optional(),
+  workspaceUserId: zod.z.string().optional()
 });
 const WorkspaceFileRequestSchema = EventBaseSchema.extend({
   taskId: zod.z.string(),
@@ -1604,6 +1657,8 @@ const SubTaskResultUpdatedEventSchema = EventBaseSchema.extend({
     result: zod.z.string(),
     is_error: zod.z.boolean().optional()
   }),
+  encryptedResultMessage: zod.z.string().optional(),
+  // Encrypted result payload for E2E forwarding
   // Optional artifacts summary (for displaying in parent task chat)
   artifacts: TaskArtifactsSummarySchema.optional()
 });
@@ -1662,6 +1717,15 @@ const AssociateRepoEventDataSchema = EventBaseSchema.extend({
   remoteUrl: zod.z.string()
   // Original URL for logging
 });
+const UpdateAgentInfoEventSchema = EventBaseSchema.extend({
+  taskId: zod.z.string(),
+  agentId: zod.z.string(),
+  displayName: zod.z.string().optional(),
+  avatar: zod.z.string().optional(),
+  // emoji or URL (fileId)
+  signature: zod.z.string().optional()
+  // status line / tagline
+});
 const IdOnlySchema = zod.z.object({
   id: zod.z.string()
 });
@@ -1686,6 +1750,7 @@ const SystemMessageSchema = EventBaseSchema.extend({
     "repo-removed",
     "pr-state-changed",
     "draft-agent-added",
+    "agent-updated",
     "task-added"
   ]),
   data: zod.z.union([
@@ -1703,11 +1768,53 @@ const SystemMessageSchema = EventBaseSchema.extend({
     // pr-state-changed
     DraftAgentSchema,
     // draft-agent-added
+    AgentSchema,
+    // agent-updated
     TaskItemSchema
     // task-added
   ]),
   timestamp: zod.z.string()
 });
+const DaemonGitlabOperationSchema = zod.z.enum([
+  "listRepos",
+  "listBranches",
+  "createMergeRequest",
+  "getMergeRequest",
+  "listMergeRequests",
+  "resolveGitAuthContext"
+]);
+const DaemonGitlabRequestSchema = EventBaseSchema.extend({
+  requestId: zod.z.string(),
+  userId: zod.z.string(),
+  gitServerId: zod.z.string(),
+  operation: DaemonGitlabOperationSchema,
+  payload: zod.z.record(zod.z.string(), zod.z.unknown()),
+  ttlMs: zod.z.number(),
+  nonce: zod.z.string()
+});
+const DaemonGitlabResponseSchema = EventBaseSchema.extend({
+  requestId: zod.z.string(),
+  success: zod.z.boolean(),
+  data: zod.z.unknown().optional(),
+  errorCode: zod.z.string().optional(),
+  errorMessage: zod.z.string().optional(),
+  machineId: zod.z.string(),
+  executionTimeMs: zod.z.number()
+});
+const CompanionHeartbeatRequestSchema = EventBaseSchema.extend({
+  machineId: zod.z.string(),
+  agentId: zod.z.string(),
+  chatId: zod.z.string(),
+  userId: zod.z.string(),
+  timestamp: zod.z.string()
+});
+const CompanionHeartbeatResponseSchema = EventBaseSchema.extend({
+  taskId: zod.z.string(),
+  chatId: zod.z.string()
+});
+const ResetTaskSessionSchema = EventBaseSchema.extend({
+  taskId: zod.z.string()
+});
 const EventSchemaMap = {
   // App events
   "app-alive": AppAliveEventSchema,
@@ -1748,6 +1855,8 @@ const EventSchemaMap = {
   "sub-task-result-updated": SubTaskResultUpdatedEventSchema,
   // Repository association events
   "associate-repo": AssociateRepoEventDataSchema,
+  // Agent info update events
+  "update-agent-info": UpdateAgentInfoEventSchema,
   // System message events
   "system-message": SystemMessageSchema,
   // Billing events
@@ -1763,6 +1872,13 @@ const EventSchemaMap = {
   "workspace-file-response": WorkspaceFileResponseSchema,
   // RPC events
   "rpc-call": RpcCallEventSchema,
+  // Daemon GitLab proxy events
+  "daemon-gitlab-request": DaemonGitlabRequestSchema,
+  "daemon-gitlab-response": DaemonGitlabResponseSchema,
+  // Companion heartbeat events
+  "request-companion-heartbeat": CompanionHeartbeatRequestSchema,
+  "companion-heartbeat-response": CompanionHeartbeatResponseSchema,
+  "reset-task-session": ResetTaskSessionSchema,
   // Ack events
   "event-ack": EventAckSchema
 };
@@ -1774,9 +1890,11 @@ const workerTaskEvents = [
   "worker-exit",
   "change-task-title",
   "update-task-agent-session-id",
+  "reset-task-session",
   "merge-request",
   "merge-pr",
-  "associate-repo"
+  "associate-repo",
+  "update-agent-info"
 ];
 
 function userAuth(token) {
@@ -2068,10 +2186,51 @@ async function loadSystemPrompt(claudeDir, promptFile) {
     );
   }
 }
-function replacePromptPlaceholders(template, cwd) {
-  return template.replace(/\{\{WORKING_DIR\}\}/g, cwd).replace(/\{\{PLATFORM\}\}/g, process.platform).replace(/\{\{OS_VERSION\}\}/g, `${os__namespace.type()} ${os__namespace.release()}`).replace(/\{\{DATE\}\}/g, (/* @__PURE__ */ new Date()).toISOString().split("T")[0]);
+function replacePromptPlaceholders(template, cwd, extra) {
+  const vars = {
+    WORKING_DIR: cwd,
+    PLATFORM: process.platform,
+    OS_VERSION: `${os__namespace.type()} ${os__namespace.release()}`,
+    DATE: (/* @__PURE__ */ new Date()).toISOString().split("T")[0],
+    ...extra
+  };
+  let result = template.replace(
+    /\{\{#if\s+(\w+)\s*(==|!=)\s*(\w+)\}\}([\s\S]*?)\{\{\/if\}\}/g,
+    (_match, key, op, expected, content) => {
+      const actual = vars[key] ?? "";
+      const matches = op === "==" ? actual === expected : actual !== expected;
+      return matches ? content : "";
+    }
+  );
+  for (const [key, value] of Object.entries(vars)) {
+    result = result.replace(new RegExp(`\\{\\{${key}\\}\\}`, "g"), value);
+  }
+  return result;
 }
 
+const CompanionWorkspaceFileSchema = zod.z.object({
+  name: zod.z.string(),
+  path: zod.z.string(),
+  // relative to workspace root
+  size: zod.z.number(),
+  modifiedAt: zod.z.number(),
+  isDirectory: zod.z.boolean()
+});
+const RegisterCompanionRequestSchema = zod.z.object({
+  machineId: zod.z.string().min(1)
+});
+const RegisterCompanionResponseSchema = zod.z.object({
+  agentId: zod.z.string(),
+  userId: zod.z.string(),
+  chatId: zod.z.string(),
+  created: zod.z.boolean()
+  // true = newly created, false = already existed
+});
+const CompanionEnsureResponseSchema = zod.z.object({
+  agentDir: zod.z.string(),
+  workspaceDir: zod.z.string()
+});
+
 const cryptoModule = typeof globalThis.crypto !== "undefined" ? globalThis.crypto : (async () => {
   try {
     const nodeCrypto = await import('node:crypto');
@@ -2706,6 +2865,10 @@ exports.CloudJoinResultQuerySchema = CloudJoinResultQuerySchema;
 exports.CloudJoinStatusQuerySchema = CloudJoinStatusQuerySchema;
 exports.CloudMachineSchema = CloudMachineSchema;
 exports.CloudSchema = CloudSchema;
+exports.CompanionEnsureResponseSchema = CompanionEnsureResponseSchema;
+exports.CompanionHeartbeatRequestSchema = CompanionHeartbeatRequestSchema;
+exports.CompanionHeartbeatResponseSchema = CompanionHeartbeatResponseSchema;
+exports.CompanionWorkspaceFileSchema = CompanionWorkspaceFileSchema;
 exports.ConfirmUploadRequestSchema = ConfirmUploadRequestSchema;
 exports.ConfirmUploadResponseSchema = ConfirmUploadResponseSchema;
 exports.ConsumeTransactionSchema = ConsumeTransactionSchema;
@@ -2724,6 +2887,9 @@ exports.CreateTaskShareResponseSchema = CreateTaskShareResponseSchema;
 exports.CreateTaskShareSchema = CreateTaskShareSchema;
 exports.CreditExhaustedEventSchema = CreditExhaustedEventSchema;
 exports.CreditsPackageSchema = CreditsPackageSchema;
+exports.DaemonGitlabOperationSchema = DaemonGitlabOperationSchema;
+exports.DaemonGitlabRequestSchema = DaemonGitlabRequestSchema;
+exports.DaemonGitlabResponseSchema = DaemonGitlabResponseSchema;
 exports.DateSchema = DateSchema;
 exports.DeleteAgentResponseSchema = DeleteAgentResponseSchema;
 exports.DeleteOAuthServerResponseSchema = DeleteOAuthServerResponseSchema;
@@ -2780,6 +2946,8 @@ exports.ListOAuthServersQuerySchema = ListOAuthServersQuerySchema;
 exports.ListOAuthServersResponseSchema = ListOAuthServersResponseSchema;
 exports.ListPackagesQuerySchema = ListPackagesQuerySchema;
 exports.ListPackagesResponseSchema = ListPackagesResponseSchema;
+exports.ListRecentTasksRequestSchema = ListRecentTasksRequestSchema;
+exports.ListRecentTasksResponseSchema = ListRecentTasksResponseSchema;
 exports.ListRepositoriesResponseSchema = ListRepositoriesResponseSchema;
 exports.ListSubTasksRequestSchema = ListSubTasksRequestSchema;
 exports.ListSubTasksResponseSchema = ListSubTasksResponseSchema;
@@ -2823,11 +2991,15 @@ exports.PublishDraftAgentResponseSchema = PublishDraftAgentResponseSchema;
 exports.QueryEventsRequestSchema = QueryEventsRequestSchema;
 exports.RELEVANT_DEPENDENCIES = RELEVANT_DEPENDENCIES;
 exports.RTC_CHUNK_HEADER_SIZE = RTC_CHUNK_HEADER_SIZE;
+exports.RecentTaskSummarySchema = RecentTaskSummarySchema;
 exports.RechargeResponseSchema = RechargeResponseSchema;
+exports.RegisterCompanionRequestSchema = RegisterCompanionRequestSchema;
+exports.RegisterCompanionResponseSchema = RegisterCompanionResponseSchema;
 exports.RemoveChatMemberRequestSchema = RemoveChatMemberRequestSchema;
 exports.RepositorySchema = RepositorySchema;
 exports.ResetSecretRequestSchema = ResetSecretRequestSchema;
 exports.ResetSecretResponseSchema = ResetSecretResponseSchema;
+exports.ResetTaskSessionSchema = ResetTaskSessionSchema;
 exports.ResumeTaskRequestSchema = ResumeTaskRequestSchema;
 exports.ResumeTaskResponseSchema = ResumeTaskResponseSchema;
 exports.RpcCallEventSchema = RpcCallEventSchema;
@@ -2879,6 +3051,7 @@ exports.ToggleOAuthServerResponseSchema = ToggleOAuthServerResponseSchema;
 exports.TransactionSchema = TransactionSchema;
 exports.UnarchiveTaskRequestSchema = UnarchiveTaskRequestSchema;
 exports.UnarchiveTaskResponseSchema = UnarchiveTaskResponseSchema;
+exports.UpdateAgentInfoEventSchema = UpdateAgentInfoEventSchema;
 exports.UpdateAgentRequestSchema = UpdateAgentRequestSchema;
 exports.UpdateAgentResponseSchema = UpdateAgentResponseSchema;
 exports.UpdateChatContextRequestSchema = UpdateChatContextRequestSchema;
@@ -2941,6 +3114,8 @@ exports.getAgentContext = getAgentContext;
 exports.getRandomBytes = getRandomBytes;
 exports.isAskUserMessage = isAskUserMessage;
 exports.isAskUserResponseMessage = isAskUserResponseMessage;
+exports.isCompanionHeartbeatMessage = isCompanionHeartbeatMessage;
+exports.isCompanionReminderMessage = isCompanionReminderMessage;
 exports.isSDKMessage = isSDKMessage;
 exports.isSDKUserMessage = isSDKUserMessage;
 exports.loadAgentConfig = loadAgentConfig;