@ai-wendao/ant 0.4.4 → 0.4.6

package/delegation.js

@@ -1048,6 +1048,8 @@ export function startDelegationHost(ctx, readConnection, { ledger = new Delegati
   const consumers = new Map()
   let enrolledFingerprints = new Set()
   let enrolledOrigins = new Map()
+  let dmChatEnabledByFingerprint = new Map()
+  const dmDisabledNoticeAt = new Map()
   let legacyDmGrants = []
   let legacyGroupGrants = []
   let enrollmentKnown = false
@@ -1270,6 +1272,25 @@ export function startDelegationHost(ctx, readConnection, { ledger = new Delegati
       scheduleProcess(entry, 'ANT owner binding reply stalled:')
       return { accepted: true }
     }
+    if (dmChatEnabledByFingerprint.get(event.fingerprint) === false) {
+      const noticeKey = `${event.fingerprint}\0${event.actor.id}`
+      const now = Date.now()
+      for (const [key, at] of dmDisabledNoticeAt) if (now - at >= 5 * 60_000) dmDisabledNoticeAt.delete(key)
+      if (now - (dmDisabledNoticeAt.get(noticeKey) || 0) >= 5 * 60_000) {
+        dmDisabledNoticeAt.set(noticeKey, now)
+        while (dmDisabledNoticeAt.size > 500) dmDisabledNoticeAt.delete(dmDisabledNoticeAt.keys().next().value)
+        try {
+          const notice = await ctx.dshIm.replyChecked(event.botId, event.reply,
+            '机器人所有者已关闭此机器人的私聊。请联系所有者在 ANT 插件的“机器人与聊天”页重新开启；群聊设置不受影响。',
+            { expectedFingerprint: event.fingerprint, signal: signal || controller.signal })
+          if (notice?.sent !== true) dmDisabledNoticeAt.delete(noticeKey)
+        } catch (error) {
+          dmDisabledNoticeAt.delete(noticeKey)
+          logger.warn?.('ANT 私聊关闭提示暂不可用:', error?.message)
+        }
+      }
+      return { accepted: true }
+    }
     const proposedLegacy = parseDelegatedPrRequest(text)
     const matchingLegacy = proposedLegacy.kind === 'pr_review' && legacyDmGrants.some((grant) =>
       grant.bot_fingerprint === event.fingerprint && grant.pr_number === proposedLegacy.pr_number
@@ -1482,6 +1503,7 @@ export function startDelegationHost(ctx, readConnection, { ledger = new Delegati
         consumers.clear()
         enrolledFingerprints = new Set()
         enrolledOrigins = new Map()
+        dmChatEnabledByFingerprint = new Map()
         legacyDmGrants = []
         legacyGroupGrants = []
         enrollmentKnown = false
@@ -1523,6 +1545,8 @@ export function startDelegationHost(ctx, readConnection, { ledger = new Delegati
       if (enrollmentKnown) {
         enrolledFingerprints = new Set(deviceEnrolled.map((bot) => bot.bot_fingerprint))
         enrolledOrigins = new Map(deviceEnrolled.map((bot) => [bot.bot_fingerprint, bot.gitea_origin]))
+        dmChatEnabledByFingerprint = new Map(deviceEnrolled.map((bot) =>
+          [bot.bot_fingerprint, bot.dm_chat_enabled !== false]))
       }
       let groups = []
       try {
@@ -1605,6 +1629,11 @@ export function startDelegationHost(ctx, readConnection, { ledger = new Delegati
     if (refreshing) await refreshDone
     if (!stopped) await refresh()
   }
+  stop.setDmChat = (fingerprint, enabled) => {
+    if (typeof fingerprint === 'string' && typeof enabled === 'boolean') {
+      dmChatEnabledByFingerprint.set(fingerprint.toLowerCase(), enabled)
+    }
+  }
   stop.cancelTask = async (taskId, serviceUrl) => {
     await ready
     if (stopped || !ledgerReady) return false

package/index.js

@@ -25,6 +25,7 @@ const DELEGATION_BOTS_PATH = '/api/ant-teams/delegation/bots'
 const DELEGATION_GRANTS_PATH = '/api/ant-teams/delegation/grants'
 const DELEGATION_ENROLLED_BOTS_PATH = '/api/ant-teams/delegation/enrolled-bots'
 const DELEGATION_GROUPS_PATH = '/api/ant-teams/delegation/groups'
+const DELEGATION_DM_CHAT_PATH = '/api/ant-teams/delegation/dm-chat'
 const DELEGATION_OWNER_CHALLENGE_PATH = '/api/ant-teams/delegation/owner-challenge'
 const DELEGATION_STATUS_PATH = '/api/ant-teams/delegation/status'
 const DELEGATION_CARDS_PATH = '/api/ant-teams/delegation/cards'
@@ -281,7 +282,8 @@ async function fetchDelegationGrants(request, ctx, credentials, settings) {
 }
 
 async function fetchDelegationGroupSettings(request, ctx, credentials, settings, kind, getHost) {
-  const methods = kind === 'groups' ? ['GET', 'PUT'] : kind === 'challenge' ? ['POST'] : ['GET', 'POST', 'DELETE']
+  const methods = kind === 'groups' ? ['GET', 'PUT'] : kind === 'dm-chat' ? ['PUT']
+    : kind === 'challenge' ? ['POST'] : ['GET', 'POST', 'DELETE']
   if (!methods.includes(request.method)) return jsonResponse(405, { message: '请求方式不支持。' })
   let connection
   try { connection = await readMemberConnection(credentials, settings) } catch {
@@ -312,6 +314,7 @@ async function fetchDelegationGroupSettings(request, ctx, credentials, settings,
     return jsonResponse(400, { message: '机器人编号无效。' })
   }
   const path = kind === 'groups' ? '/api/me/im-delegations/groups'
+    : kind === 'dm-chat' ? '/api/me/im-delegations/dm-chat'
     : kind === 'challenge' ? '/api/me/im-delegations/owner-challenge'
       : request.method === 'DELETE'
         ? `/api/me/im-delegations/bots/${encodeURIComponent(body.bot_id)}`
@@ -331,11 +334,13 @@ async function fetchDelegationGroupSettings(request, ctx, credentials, settings,
       redirect: 'manual', signal: AbortSignal.timeout(12_000),
     })
     if (response.status === 404 && request.method !== 'DELETE') {
-      return jsonResponse(426, { message: '团队服务端尚未支持群聊委托，请联系管理员升级 ANT 服务端。' })
+      return jsonResponse(426, { message: '团队服务端尚未支持这项协作设置，请联系管理员升级 ANT 服务端。' })
     }
     if (response.status === 204) {
-      if (kind === 'groups' && request.method === 'PUT') {
-        const refresh = getHost?.()?.refresh
+      if (['groups', 'dm-chat'].includes(kind) && request.method === 'PUT') {
+        const host = getHost?.()
+        if (kind === 'dm-chat') host?.setDmChat?.(body.bot_fingerprint, body.enabled)
+        const refresh = host?.refresh
         if (refresh) await refresh().catch(() => undefined)
       }
       return new Response(null, { status: 204 })
@@ -943,6 +948,7 @@ export function apply(ctx) {
     for (const [path, methods, kind] of [
       [DELEGATION_ENROLLED_BOTS_PATH, ['GET', 'POST', 'DELETE'], 'bots'],
       [DELEGATION_GROUPS_PATH, ['GET', 'PUT'], 'groups'],
+      [DELEGATION_DM_CHAT_PATH, ['PUT'], 'dm-chat'],
       [DELEGATION_OWNER_CHALLENGE_PATH, ['POST'], 'challenge'],
     ]) {
       connectionCtx.effect(() => connectionCtx.connection.fetch.register({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-wendao/ant",
-  "version": "0.4.4",
+  "version": "0.4.6",
   "description": "ANT (AI-native Teams) connects a DeepSeek Harness Desktop profile to team identity and approved models.",
   "type": "module",
   "scripts": {