@agentunion/fastaun-browser 0.3.6 → 0.4.1

package/dist/client.js

@@ -11,9 +11,6 @@ import { GatewayDiscovery } from './discovery.js';
 import { RPCTransport } from './transport.js';
 import { AuthFlow } from './auth.js';
 import { SeqTracker } from './seq-tracker.js';
-import { AuthNamespace } from './namespaces/auth.js';
-import { CustodyNamespace } from './namespaces/custody.js';
-import { MetaNamespace } from './namespaces/meta.js';
 import { CryptoProvider, uint8ToBase64, base64ToUint8, pemToArrayBuffer, p1363ToDer, certificateSha256Fingerprint, ecdsaSignDer, ecdsaVerifyDer, importCertPublicKeyEcdsa, importPrivateKeyEcdsa, } from './crypto.js';
 import { IndexedDBKeyStore } from './keystore/indexeddb.js';
 import { V2Session, V2KeyStore } from './v2/session/index.js';
@@ -21,8 +18,9 @@ import { encryptP2PMessage, encryptGroupMessage, decryptMessage, } from './v2/e2
 import { ecdsaVerifyRaw } from './v2/crypto/ecdsa.js';
 import { computeStateCommitment } from './v2/state/index.js';
 import { AUNLogger } from './logger.js';
-import { AUNError, AuthError, ConnectionError, E2EEError, PermissionError, StateError, ValidationError, } from './errors.js';
-import { isJsonObject, } from './types.js';
+import { AUNError, AuthError, ConnectionError, E2EEError, NotFoundError, PermissionError, StateError, ValidationError, } from './errors.js';
+import { isJsonObject, ConnectionState, STATE_TO_PUBLIC, } from './types.js';
+import { AID } from './aid.js';
 /**
  * 递归排序键的 JSON 序列化（Canonical JSON for AUN）
  * 等价于 Python json.dumps(sort_keys=True, separators=(",",":"), ensure_ascii=False)
@@ -171,6 +169,12 @@ const DEFAULT_SESSION_OPTIONS = {
         http: 30.0,
     },
 };
+const PROTECTED_HEADERS_METHODS = new Set([
+    'message.send',
+    'group.send',
+    'message.thought.put',
+    'group.thought.put',
+]);
 const RECONNECT_MIN_BASE_DELAY_SECONDS = 1.0;
 const RECONNECT_MAX_BASE_DELAY_SECONDS = 64.0;
 const TOKEN_REFRESH_CHECK_INTERVAL_MS = 30_000;
@@ -213,6 +217,7 @@ function reconnectSleepDelaySeconds(baseDelay, maxBaseDelay) {
 /** 对端证书缓存 TTL（秒） */
 const PEER_CERT_CACHE_TTL = 3600;
 const PEER_PREKEYS_CACHE_TTL = 3600;
+const AGENT_MD_HTTP_TIMEOUT_MS = 30_000;
 /**
  * 将 WebSocket URL 转为对应的 HTTP URL
  */
@@ -245,6 +250,29 @@ function buildCertUrl(gatewayUrl, aid, certFingerprint) {
     }
     return url.toString();
 }
+function agentMdHttpScheme(gatewayUrl) {
+    const raw = String(gatewayUrl ?? '').trim().toLowerCase();
+    return raw.startsWith('ws://') ? 'http' : 'https';
+}
+function agentMdAuthority(aid) {
+    return String(aid ?? '').trim();
+}
+async function fetchWithTimeout(input, init, timeoutMs = AGENT_MD_HTTP_TIMEOUT_MS) {
+    const controller = new AbortController();
+    const timer = globalThis.setTimeout(() => controller.abort(), timeoutMs);
+    try {
+        return await fetch(input, { ...init, signal: controller.signal });
+    }
+    catch (error) {
+        if (controller.signal.aborted) {
+            throw new AUNError(`agent.md request timed out after ${timeoutMs}ms`);
+        }
+        throw error;
+    }
+    finally {
+        globalThis.clearTimeout(timer);
+    }
+}
 /**
  * 跨域时将 Gateway URL 替换为 peer 所在域的 Gateway URL。
  *
@@ -581,6 +609,22 @@ function normalizeDeliveryModeConfig(raw, opts = {}) {
         affinity_ttl_ms: affinityTtlMs,
     };
 }
+function assertClientOptions(value, label) {
+    if (value == null)
+        return;
+    if (typeof value !== 'object' || Array.isArray(value) || value instanceof AID) {
+        throw new ValidationError(`${label} must be an options object`);
+    }
+}
+function clientOptionsConfig(options) {
+    const raw = { ...(options ?? {}) };
+    if (Object.prototype.hasOwnProperty.call(raw, 'aid')) {
+        throw new ValidationError('AUNClient options must not include aid; pass an AID object as the first argument');
+    }
+    delete raw.debug;
+    delete raw.protected_headers;
+    return raw;
+}
 /**
  * AUN Core SDK 客户端 — 浏览器版本。
  *
@@ -601,6 +645,8 @@ export class AUNClient {
     _aid = null;
     _identity = null;
     _state = 'idle';
+    _currentAid = null;
+    _instanceProtectedHeaders = null;
     _gatewayUrl = null;
     _deviceId;
     _slotId;
@@ -615,12 +661,6 @@ export class AUNClient {
     _keystore;
     _auth;
     _transport;
-    /** 认证命名空间 */
-    auth;
-    /** AID 托管命名空间 */
-    custody;
-    /** 元数据命名空间（心跳、状态、信任根管理） */
-    meta;
     // E2EE 编排状态（内存缓存）
     _certCache = new Map();
     // 后台任务 handle（浏览器 setInterval/setTimeout）
@@ -661,7 +701,7 @@ export class AUNClient {
     _localAgentMdEtag = '';
     /** gateway 在 RPC envelope._meta.agent_md_etag 注入的服务端 etag；纯观察，无下游依赖。 */
     _remoteAgentMdEtag = '';
-    /** 浏览器侧 AgentMDs 逻辑根目录，正文映射到 IndexedDB 里的 {aid}/agent.md。 */
+    /** 浏览器侧 AIDs 逻辑根目录，正文映射到 IndexedDB 里的 {aid}/agent.md。 */
     _agentMdPath = '';
     _agentMdCache = new Map();
     _agentMdFetchInflight = new Set();
@@ -686,6 +726,14 @@ export class AUNClient {
     _reconnectActive = false;
     _reconnectAbort = null;
     _serverKicked = false;
+    // 重连状态追踪（对齐 Python client.py）
+    _nextRetryAt = null;
+    _retryAttempt = 0;
+    _retryMaxAttempts = 0;
+    _lastError = null;
+    _lastErrorCode = null;
+    /** 对端 AID 缓存（aid string → AID 对象） */
+    _peerCache = new Map();
     /**
      * 缓存最近一次服务端 gateway.disconnect 信息（含 code/reason/detail），
      * 让后续 connection.state(terminal_failed) 也能携带 detail（如配额超限信息）。
@@ -699,17 +747,25 @@ export class AUNClient {
     _logKeystore;
     _logDiscovery;
     _logEvents;
-    constructor(config, _debug = false) {
-        const rawConfig = config ?? {};
+    constructor(aid) {
+        const inputAid = aid instanceof AID ? aid : null;
+        if (typeof aid === 'string') {
+            throw new ValidationError('AUNClient aid must be an AID object, not a string');
+        }
+        const options = {};
+        const rawConfig = clientOptionsConfig(options);
+        if (inputAid)
+            rawConfig.aun_path = inputAid.aunPath;
+        const _debug = false;
         this.configModel = createConfig(rawConfig);
-        const initAid = String(rawConfig.aid ?? '').trim() || null;
+        const initAid = inputAid ? inputAid.aid : null;
         this.config = {
             aun_path: this.configModel.aunPath,
             root_ca_path: this.configModel.rootCaPem,
             seed_password: this.configModel.seedPassword,
         };
         this._agentMdPath = this._agentMdDefaultRoot();
-        this._deviceId = getDeviceId();
+        this._deviceId = (inputAid?.deviceId) || getDeviceId();
         // Logger 必须最早初始化（其他子模块构造时通过 logger 输出）
         this._logger = new AUNLogger({ debug: _debug, aunPath: this.configModel.aunPath });
         this._logger.bindDeviceId(this._deviceId);
@@ -723,7 +779,7 @@ export class AUNClient {
         this._dispatcher = new EventDispatcher();
         this._discovery = new GatewayDiscovery();
         this._keystore = new IndexedDBKeyStore({ encryptionSeed: this.configModel.seedPassword ?? undefined });
-        this._slotId = '';
+        this._slotId = inputAid?.slotId || 'default';
         this._connectDeliveryMode = normalizeDeliveryModeConfig({ mode: 'fanout' });
         this._defaultConnectDeliveryMode = { ...this._connectDeliveryMode };
         this._auth = new AuthFlow({
@@ -746,9 +802,21 @@ export class AUNClient {
                 this._clientLog.debug(`agent.md meta observer skipped: ${String(exc)}`);
             });
         });
-        this.auth = new AuthNamespace(this);
-        this.custody = new CustodyNamespace(this);
-        this.meta = new MetaNamespace(this);
+        if (inputAid) {
+            if (!inputAid.isPrivateKeyValid())
+                throw new StateError('AUNClient requires an AID with a valid private key');
+            this._currentAid = inputAid;
+            this._identity = {
+                aid: inputAid.aid,
+                private_key_pem: inputAid._privateKeyPem ?? '',
+                public_key_der_b64: inputAid.publicKey,
+                cert: inputAid.certPem,
+            };
+            this._state = 'disconnected';
+        }
+        if (options?.protected_headers !== undefined) {
+            this.setProtectedHeaders(options.protected_headers);
+        }
         // 注入 logger 到各子模块（构造时未传 logger，构造后通过 setLogger 注入）
         this._auth.setLogger(this._logAuth);
         this._transport.setLogger(this._logTransport);
@@ -756,18 +824,9 @@ export class AUNClient {
         if (typeof this._discovery.setLogger === 'function') {
             this._discovery.setLogger(this._logger.for('aun_core.discovery'));
         }
-        if (typeof this.auth.setLogger === 'function') {
-            this.auth.setLogger(this._logger.for('aun_core.namespace.auth'));
-        }
-        if (typeof this.custody.setLogger === 'function') {
-            this.custody.setLogger(this._logger.for('aun_core.namespace.custody'));
-        }
         if (typeof this._keystore.setLogger === 'function') {
             this._keystore.setLogger(this._logKeystore);
         }
-        if (typeof this.meta.setLogger === 'function') {
-            this.meta.setLogger(this._logger.for('aun_core.namespace.meta'));
-        }
         // 内部订阅：推送消息 re-publish 给用户（V2 加密消息走 _raw.peer.v2.message_received）
         this._dispatcher.subscribe('_raw.message.received', (data) => {
             this._onRawMessageReceived(data);
@@ -818,17 +877,182 @@ export class AUNClient {
     get aid() {
         return this._aid;
     }
-    setAgentMdPath(root) {
+    _setAgentMdRoot(root) {
         const next = String(root ?? '').trim() || this._agentMdDefaultRoot();
         this._agentMdPath = next;
         this._agentMdCache.clear();
         return next;
     }
-    setAgentMDPath(root) {
-        return this.setAgentMdPath(root);
+    async _resolveAgentMdUrl(aid) {
+        const target = String(aid ?? '').trim();
+        if (!target)
+            throw new ValidationError('agent.md requires non-empty aid');
+        let gatewayUrl = String(this._gatewayUrl ?? '').trim();
+        if (!gatewayUrl) {
+            try {
+                gatewayUrl = await this._resolveGatewayForAid(target);
+            }
+            catch {
+                gatewayUrl = '';
+            }
+        }
+        const authority = agentMdAuthority(target);
+        return `${agentMdHttpScheme(gatewayUrl)}://${authority}/agent.md`;
+    }
+    async _ensureAgentMdUploadToken(aid, gatewayUrl) {
+        let identity = await this._auth.loadIdentityOrNone(aid);
+        if (!identity && this._identity && String(this._identity.aid ?? '') === aid) {
+            identity = this._identity;
+        }
+        if (!identity) {
+            throw new StateError('no local identity found, register or load an AID first');
+        }
+        const cachedToken = String(identity.access_token ?? '');
+        const expiresAt = this._auth.getAccessTokenExpiry(identity);
+        if (cachedToken && (expiresAt === null || expiresAt > Date.now() / 1000 + 30)) {
+            return cachedToken;
+        }
+        if (identity.refresh_token) {
+            try {
+                const refreshed = await this._auth.refreshCachedTokens(gatewayUrl, identity);
+                const refreshedToken = String(refreshed.access_token ?? '');
+                const refreshedExpiry = this._auth.getAccessTokenExpiry(refreshed);
+                if (refreshedToken && (refreshedExpiry === null || refreshedExpiry > Date.now() / 1000 + 30)) {
+                    this._identity = refreshed;
+                    return refreshedToken;
+                }
+            }
+            catch {
+                // refresh 失败时回退到完整 authenticate。
+            }
+        }
+        const result = await this._auth.authenticate(gatewayUrl, aid);
+        const token = String(result.access_token ?? '');
+        if (!token)
+            throw new StateError('authenticate did not return access_token');
+        const fallbackIdentity = {
+            ...identity,
+            access_token: token,
+            refresh_token: String(result.refresh_token ?? identity.refresh_token ?? ''),
+        };
+        const fallbackExpiresAt = Number(result.expires_at ?? identity.expires_at ?? NaN);
+        if (Number.isFinite(fallbackExpiresAt))
+            fallbackIdentity.expires_at = fallbackExpiresAt;
+        this._identity = await this._auth.loadIdentityOrNone(aid) ?? fallbackIdentity;
+        return token;
+    }
+    async _uploadAgentMd(content) {
+        const target = String(this._aid ?? this._currentAid?.aid ?? '').trim();
+        if (!target)
+            throw new StateError('uploadAgentMd requires local AID');
+        const gatewayUrl = await this._resolveGatewayForAid(target);
+        this._gatewayUrl = gatewayUrl;
+        const token = await this._ensureAgentMdUploadToken(target, gatewayUrl);
+        const response = await fetchWithTimeout(await this._resolveAgentMdUrl(target), {
+            method: 'PUT',
+            headers: {
+                Authorization: `Bearer ${token}`,
+                'Content-Type': 'text/markdown; charset=utf-8',
+            },
+            body: content,
+        });
+        if (response.status === 404) {
+            throw new NotFoundError(`agent.md endpoint not found for aid: ${target}`);
+        }
+        if (!response.ok) {
+            const message = (await response.text()).trim();
+            throw new AUNError(`upload agent.md failed: HTTP ${response.status}${message ? ` - ${message}` : ''}`);
+        }
+        const payload = await response.json();
+        if (!isJsonObject(payload))
+            throw new AUNError('upload agent.md returned invalid JSON payload');
+        return payload;
+    }
+    async _downloadAgentMd(aid) {
+        const target = String(aid ?? '').trim();
+        if (!target)
+            throw new ValidationError('downloadAgentMd requires non-empty aid');
+        const cached = this._agentMdCache.get(target);
+        const url = await this._resolveAgentMdUrl(target);
+        const response = await fetchWithTimeout(url, {
+            method: 'GET',
+            headers: { Accept: 'text/markdown' },
+            redirect: 'follow',
+        });
+        if (response.status === 304 && typeof cached?.text === 'string') {
+            return String(cached.text);
+        }
+        if (response.status === 404) {
+            throw new NotFoundError(`agent.md not found for aid: ${target}`);
+        }
+        if (!response.ok) {
+            const message = (await response.text()).trim();
+            throw new AUNError(`download agent.md failed: HTTP ${response.status}${message ? ` - ${message}` : ''}`);
+        }
+        const text = await response.text();
+        const etag = String(response.headers?.get('ETag') ?? response.headers?.get('etag') ?? '').trim();
+        const lastModified = String(response.headers?.get('Last-Modified') ?? response.headers?.get('last-modified') ?? '').trim();
+        this._agentMdCache.set(target, {
+            ...(cached ?? {}),
+            text,
+            etag,
+            lastModified,
+            remote_etag: etag,
+            last_modified: lastModified,
+        });
+        return text;
+    }
+    async _headAgentMd(aid) {
+        const target = String(aid ?? '').trim();
+        if (!target)
+            throw new ValidationError('headAgentMd requires non-empty aid');
+        const response = await fetchWithTimeout(await this._resolveAgentMdUrl(target), {
+            method: 'HEAD',
+            headers: { Accept: 'text/markdown' },
+        });
+        const etag = String(response.headers?.get('ETag') ?? response.headers?.get('etag') ?? '').trim();
+        const lastModified = String(response.headers?.get('Last-Modified') ?? response.headers?.get('last-modified') ?? '').trim();
+        if (response.status === 404) {
+            return { aid: target, found: false, etag: '', last_modified: '', status: 404 };
+        }
+        if (!response.ok) {
+            throw new AUNError(`head agent.md failed: HTTP ${response.status}`);
+        }
+        const cached = this._agentMdCache.get(target) ?? {};
+        this._agentMdCache.set(target, {
+            ...cached,
+            etag,
+            lastModified,
+            remote_etag: etag,
+            last_modified: lastModified,
+        });
+        return { aid: target, found: true, etag, last_modified: lastModified, status: response.status };
     }
-    SetAgentMDPath(root) {
-        return this.setAgentMdPath(root);
+    async _verifyAgentMd(content, aid) {
+        const target = String(aid ?? '').trim();
+        if (!target)
+            throw new ValidationError('verifyAgentMd requires non-empty aid');
+        let peer = target === this._currentAid?.aid ? this._currentAid : null;
+        if (!peer) {
+            let certPem = String(await this._keystore.loadCert(target) ?? '').trim();
+            if (!certPem) {
+                certPem = String(await this._fetchPeerCert(target) ?? '').trim();
+            }
+            if (!certPem)
+                throw new NotFoundError(`certificate not found for aid: ${target}`);
+            peer = await AID.create({
+                aid: target,
+                aunPath: this.configModel.aunPath,
+                certPem,
+                privateKeyPem: null,
+                certValid: true,
+                privateKeyValid: false,
+            });
+        }
+        const result = await peer.verifyAgentMd(content);
+        if (!result.ok)
+            throw new AUNError(result.error.message);
+        return { ...result.data, verified: result.data.status === 'verified' };
     }
     /**
      * 浏览器版本 publishAgentMd。默认从 {agentMdPath}/{self_aid}/agent.md 的等价 IndexedDB 正文读取，
@@ -856,8 +1080,12 @@ export class AUNClient {
         if (localContent === null || localContent.length === 0) {
             throw new ValidationError('publishAgentMd requires local agent.md content');
         }
-        const signed = await this.auth.signAgentMd(localContent);
-        const result = await this.auth.uploadAgentMd(signed);
+        const signedResult = await this._currentAid?.signAgentMd(localContent);
+        if (!signedResult?.ok) {
+            throw new StateError(signedResult?.error.message ?? 'publishAgentMd requires a valid local AID private key');
+        }
+        const signed = signedResult.data.signed;
+        const result = await this._uploadAgentMd(signed);
         this._localAgentMdEtag = await this._agentMdContentEtag(signed);
         const remoteEtag = isJsonObject(result) ? String(result.etag ?? '').trim() : '';
         if (remoteEtag)
@@ -877,13 +1105,13 @@ export class AUNClient {
      * 浏览器版本 fetchAgentMd。aid 缺省时取自身；下载后的正文固定写入
      * {agentMdPath}/{aid}/agent.md 的等价 IndexedDB 正文，agentmd.json 只保存元数据。
      */
-    async fetchAgentMd(aid) {
+    async _fetchAgentMdCache(aid) {
         const target = String(aid ?? this._aid ?? '').trim();
         if (!target) {
             throw new ValidationError('fetchAgentMd requires aid (or local AID)');
         }
-        const content = await this.auth.downloadAgentMd(target);
-        const signature = await this.auth.verifyAgentMd(content, { aid: target });
+        const content = await this._downloadAgentMd(target);
+        const signature = await this._verifyAgentMd(content, target);
         const isSelf = target === (this._aid ?? '');
         const localEtag = await this._agentMdContentEtag(content);
         const cacheMeta = this._agentMdAuthCacheMeta(target);
@@ -932,7 +1160,7 @@ export class AUNClient {
         return String(this._aid ?? '').trim();
     }
     _agentMdDefaultRoot() {
-        return this._joinAgentMdPath(this.configModel.aunPath || '.', 'AgentMDs');
+        return this._joinAgentMdPath(this.configModel.aunPath || '.', 'AIDs');
     }
     _joinAgentMdPath(base, name) {
         const left = String(base ?? '').trim().replace(/[\\/]+$/g, '');
@@ -1039,8 +1267,7 @@ export class AUNClient {
     }
     _agentMdAuthCacheMeta(aid) {
         try {
-            const store = this.auth._agentMdCache;
-            const record = store?.get(String(aid ?? '').trim());
+            const record = this._agentMdCache.get(String(aid ?? '').trim());
             return record && typeof record === 'object' ? { ...record } : {};
         }
         catch {
@@ -1165,7 +1392,7 @@ export class AUNClient {
             return;
         this._agentMdFetchInflight.add(target);
         try {
-            await this.fetchAgentMd(target);
+            await this._fetchAgentMdCache(target);
         }
         catch (err) {
             await this._saveAgentMdRecord(target, {
@@ -1226,7 +1453,7 @@ export class AUNClient {
         }
         await this._observeAgentMdMeta(senderAid, String(sender.etag ?? '').trim(), String(sender.last_modified ?? sender.lastModified ?? '').trim(), 'envelope');
     }
-    async checkAgentMd(aid, maxUnsyncedDays = 0) {
+    async _checkAgentMdCache(aid, maxUnsyncedDays = 0) {
         const target = String(aid ?? this._aid ?? '').trim();
         if (!target)
             throw new ValidationError('checkAgentMd requires aid (or local AID)');
@@ -1256,7 +1483,7 @@ export class AUNClient {
         const now = Date.now();
         let remote;
         try {
-            remote = await this.auth.headAgentMd(target);
+            remote = await this._headAgentMd(target);
         }
         catch (err) {
             await this._saveAgentMdRecord(target, { checked_at: now, remote_status: 'error', last_error: err instanceof Error ? err.message : String(err) });
@@ -1310,7 +1537,116 @@ export class AUNClient {
         }
     }
     get state() {
-        return this._state;
+        return this._publicState(this._state);
+    }
+    _publicState(state) {
+        return STATE_TO_PUBLIC[state] ?? state;
+    }
+    get currentAid() {
+        return this._currentAid;
+    }
+    get hasIdentity() {
+        return this._currentAid !== null && this.state !== ConnectionState.CLOSED;
+    }
+    get canSign() {
+        return this.hasIdentity && !!this._currentAid?.isPrivateKeyValid();
+    }
+    get canConnect() {
+        return this.hasIdentity && this.state !== ConnectionState.CLOSED;
+    }
+    get canSend() {
+        return this.state === ConnectionState.READY;
+    }
+    get isReady() { return this.canSend; }
+    get isOnline() {
+        return this.state === ConnectionState.READY
+            || this.state === ConnectionState.RECONNECTING
+            || this.state === ConnectionState.RETRY_BACKOFF;
+    }
+    get isClosed() { return this.state === ConnectionState.CLOSED; }
+    get aunPath() { return this.hasIdentity ? this._currentAid?.aunPath ?? this.configModel.aunPath : null; }
+    /** 下次重连时间（仅在 retry_backoff 状态时非 null，对齐 Python next_retry_at） */
+    get nextRetryAt() {
+        return this.state === ConnectionState.RETRY_BACKOFF ? this._nextRetryAt : null;
+    }
+    /** 距下次重连的剩余秒数（仅在 retry_backoff 状态时非 null，对齐 Python next_retry_in_seconds） */
+    get nextRetryInSeconds() {
+        const t = this.nextRetryAt;
+        if (t === null)
+            return null;
+        return Math.max(0, (t.getTime() - Date.now()) / 1000);
+    }
+    /** 当前重连尝试次数（对齐 Python retry_attempt） */
+    get retryAttempt() { return this._retryAttempt; }
+    /** 最大重连次数（0 = 无限，对齐 Python retry_max_attempts） */
+    get retryMaxAttempts() { return this._retryMaxAttempts; }
+    /** 最近一次错误（对齐 Python last_error） */
+    get lastError() { return this._lastError; }
+    /** 最近一次错误码（对齐 Python last_error_code） */
+    get lastErrorCode() { return this._lastErrorCode; }
+    loadIdentity(aid) {
+        if (!aid?.isPrivateKeyValid())
+            throw new StateError('loadIdentity requires an AID with a valid private key');
+        const publicState = this.state;
+        if (publicState !== ConnectionState.NO_IDENTITY && publicState !== ConnectionState.CLOSED) {
+            throw new StateError(`loadIdentity not allowed in state ${publicState}`);
+        }
+        this._currentAid = aid;
+        this._aid = aid.aid;
+        this._identity = {
+            aid: aid.aid,
+            private_key_pem: aid._privateKeyPem ?? '',
+            public_key_der_b64: aid.publicKey,
+            cert: aid.certPem,
+        };
+        this._auth._aid = aid.aid;
+        this._state = 'disconnected';
+        this._closing = false;
+    }
+    setProtectedHeaders(headers) {
+        if (!headers) {
+            this._instanceProtectedHeaders = null;
+            return;
+        }
+        const cleaned = {};
+        for (const [key, value] of Object.entries(headers)) {
+            if (key === '_auth')
+                continue;
+            cleaned[String(key)] = String(value);
+        }
+        this._instanceProtectedHeaders = Object.keys(cleaned).length ? cleaned : null;
+    }
+    getProtectedHeaders() {
+        return this._instanceProtectedHeaders ? { ...this._instanceProtectedHeaders } : null;
+    }
+    cachePeer(aid) {
+        if (!this.hasIdentity)
+            throw new StateError('cachePeer requires a loaded identity');
+        if (!aid.isCertValid())
+            throw new ValidationError('cachePeer requires an AID with a valid certificate');
+        this._peerCache.set(aid.aid, aid);
+        return aid;
+    }
+    getPeer(aid) {
+        if (!this.hasIdentity)
+            throw new StateError('getPeer requires a loaded identity');
+        return this._peerCache.get(String(aid ?? '').trim()) ?? null;
+    }
+    async lookupPeer(aid) {
+        if (!this.hasIdentity)
+            throw new StateError('lookupPeer requires a loaded identity');
+        const target = String(aid ?? '').trim();
+        if (!target)
+            throw new ValidationError('lookupPeer requires non-empty aid');
+        const cached = this._peerCache.get(target);
+        if (cached)
+            return cached;
+        throw new NotFoundError(`peer not found in cache: ${target}`);
+    }
+    peers() {
+        if (!this.hasIdentity)
+            throw new StateError('peers requires a loaded identity');
+        return [...this._peerCache.entries()].sort(([a], [b]) => a.localeCompare(b)).map(([, v]) => v);
     }
     get gatewayUrl() {
         return this._gatewayUrl;
@@ -1325,36 +1661,81 @@ export class AUNClient {
     get gatewayHealth() {
         return this._discovery.lastHealthy;
     }
-    /** 主动检查 gateway 可用性（GET /health） */
-    async checkGatewayHealth(gatewayUrl, timeout = 5000) {
+    // ── 生命周期 ──────────────────────────────────────
+    /** 仅认证当前身份，获取/刷新 token，但不建立长连接。 */
+    async authenticate(options = {}) {
         const tStart = Date.now();
-        this._clientLog.debug(`checkGatewayHealth enter: gateway=${gatewayUrl} timeout=${timeout}`);
+        const target = this._currentAid?.aid ?? this._aid ?? '';
+        if (!target || !this._currentAid?.isPrivateKeyValid()) {
+            throw new StateError('authenticate requires a loaded AID with a valid private key');
+        }
+        const publicState = this.state;
+        if (publicState !== ConnectionState.STANDBY) {
+            throw new StateError(`authenticate not allowed in state ${publicState}`);
+        }
+        if ('aid' in options || 'access_token' in options || 'token' in options || 'kite_token' in options) {
+            throw new ValidationError('authenticate options must not include aid or token fields; load an AID object first');
+        }
+        this._state = 'connecting';
         try {
-            const result = await this._discovery.checkHealth(gatewayUrl, timeout);
-            this._clientLog.debug(`checkGatewayHealth exit: elapsed=${Date.now() - tStart}ms healthy=${result}`);
+            const gateway = String(options.gateway ?? this._gatewayUrl ?? await this._resolveGatewayForAid(target)).trim();
+            const result = await this._auth.authenticate(gateway, target);
+            this._gatewayUrl = String(result.gateway ?? gateway);
+            this._identity = await this._auth.loadIdentityOrNone(target);
+            this._state = 'authenticated';
+            this._clientLog.debug(`authenticate exit: elapsed=${Date.now() - tStart}ms aid=${target}`);
             return result;
         }
         catch (err) {
-            this._clientLog.debug(`checkGatewayHealth exit (error): elapsed=${Date.now() - tStart}ms err=${err instanceof Error ? err.message : String(err)}`);
+            this._state = 'disconnected';
+            this._clientLog.debug(`authenticate exit (error): elapsed=${Date.now() - tStart}ms err=${err instanceof Error ? err.message : String(err)}`);
             throw err;
         }
     }
-    // ── 生命周期 ──────────────────────────────────────
-    /**
-     * 连接到 Gateway。
-     *
-     * @param auth - 认证参数，必须包含 access_token 和 gateway
-     * @param options - 可选的会话选项（auto_reconnect, heartbeat_interval 等）
-     */
-    async connect(auth, options) {
+    /** 连接到 Gateway；身份来自构造函数或 loadIdentity(aid)，认证由 SDK 内部自动完成。 */
+    async connect(opts) {
         const tStart = Date.now();
+        const options = {};
+        if (opts?.auto_reconnect !== undefined)
+            options.auto_reconnect = opts.auto_reconnect;
+        if (opts?.heartbeat_interval !== undefined)
+            options.heartbeat_interval = opts.heartbeat_interval;
+        if (opts?.connect_timeout !== undefined || opts?.call_timeout !== undefined) {
+            options.timeouts = {
+                ...(opts.connect_timeout !== undefined ? { connect: opts.connect_timeout } : {}),
+                ...(opts.call_timeout !== undefined ? { call: opts.call_timeout } : {}),
+            };
+        }
+        if (opts?.retry_initial_delay !== undefined || opts?.retry_max_delay !== undefined || opts?.retry_max_attempts !== undefined) {
+            options.retry = {
+                initial_delay: opts.retry_initial_delay ?? 1,
+                max_delay: opts.retry_max_delay ?? 64,
+                max_attempts: opts.retry_max_attempts ?? 0,
+            };
+        }
         this._clientLog.debug(`connect enter: state=${this._state} aid=${this._aid ?? '-'}`);
-        if (this._state !== 'idle' && this._state !== 'closed' && this._state !== 'disconnected') {
+        const target = this._currentAid?.aid ?? this._aid ?? '';
+        if (!target || !this._currentAid?.isPrivateKeyValid()) {
+            throw new StateError('connect requires a loaded AID with a valid private key');
+        }
+        const publicState = this.state;
+        const allowed = new Set([
+            ConnectionState.STANDBY,
+            ConnectionState.AUTHENTICATED,
+            ConnectionState.RETRY_BACKOFF,
+            ConnectionState.CONNECTION_FAILED,
+        ]);
+        if (!allowed.has(publicState)) {
             this._clientLog.debug(`connect exit (error): elapsed=${Date.now() - tStart}ms err=invalid_state state=${this._state}`);
-            throw new StateError(`connect not allowed in state ${this._state}`);
+            throw new StateError(`connect not allowed in state ${publicState}`);
+        }
+        // gateway 来自 authenticate() 缓存的 this._gatewayUrl；未认证则自动 authenticate()
+        if (!this._gatewayUrl) {
+            await this.authenticate();
         }
         this._state = 'connecting';
-        const params = { ...auth, ...options };
+        const gateway = String(this._gatewayUrl ?? '').trim();
+        const params = { ...options, gateway };
         const normalized = this._normalizeConnectParams(params);
         this._sessionParams = normalized;
         this._sessionOptions = this._buildSessionOptions(normalized);
@@ -1365,7 +1746,7 @@ export class AUNClient {
         for (const gw of gateways) {
             try {
                 const gwParams = { ...normalized, gateway: gw };
-                await this._connectOnce(gwParams, false);
+                await this._connectOnce(gwParams, true);
                 this._clientLog.debug(`connect exit: elapsed=${Date.now() - tStart}ms state=${this._state}`);
                 return;
             }
@@ -1380,7 +1761,7 @@ export class AUNClient {
             }
         }
         if (this._state === 'connecting' || this._state === 'authenticating') {
-            this._state = 'disconnected';
+            this._state = 'terminal_failed';
         }
         this._clientLog.debug(`connect exit (error): elapsed=${Date.now() - tStart}ms err=${lastErr instanceof Error ? lastErr.message : String(lastErr)}`);
         throw lastErr;
@@ -1402,56 +1783,9 @@ export class AUNClient {
         }
         await this._transport.close();
         this._state = 'disconnected';
-        await this._dispatcher.publish('connection.state', { state: this._state });
+        await this._dispatcher.publish('state_change', { state: this._publicState(this._state) });
         this._clientLog.debug(`disconnect exit: elapsed=${Date.now() - tStart}ms`);
     }
-    /** 列出本地所有已存储的身份摘要（仅返回有有效私钥的 AID） */
-    async listIdentities() {
-        const tStart = Date.now();
-        this._clientLog.debug('listIdentities enter');
-        try {
-            const listFn = this._keystore.listIdentities;
-            if (typeof listFn !== 'function') {
-                this._clientLog.debug(`listIdentities exit: elapsed=${Date.now() - tStart}ms count=0 reason=keystore_no_list`);
-                return [];
-            }
-            const aids = await listFn.call(this._keystore);
-            const summaries = [];
-            for (const aid of [...aids].sort()) {
-                const identity = await this._keystore.loadIdentity(aid);
-                if (!identity || !identity.private_key_pem)
-                    continue;
-                const summary = { aid };
-                // 优先从 loadMetadata 获取
-                const loadMeta = this._keystore.loadMetadata;
-                if (typeof loadMeta === 'function') {
-                    const md = await loadMeta.call(this._keystore, aid);
-                    if (md && Object.keys(md).length > 0) {
-                        summary.metadata = md;
-                    }
-                }
-                // 回退：从 identity 中提取非核心字段
-                if (!summary.metadata) {
-                    const metadata = {};
-                    for (const [key, value] of Object.entries(identity)) {
-                        if (!['aid', 'private_key_pem', 'public_key_der_b64', 'curve', 'cert'].includes(key)) {
-                            metadata[key] = value;
-                        }
-                    }
-                    if (Object.keys(metadata).length > 0) {
-                        summary.metadata = metadata;
-                    }
-                }
-                summaries.push(summary);
-            }
-            this._clientLog.debug(`listIdentities exit: elapsed=${Date.now() - tStart}ms count=${summaries.length}`);
-            return summaries;
-        }
-        catch (err) {
-            this._clientLog.debug(`listIdentities exit (error): elapsed=${Date.now() - tStart}ms err=${err instanceof Error ? err.message : String(err)}`);
-            throw err;
-        }
-    }
     /** 关闭连接 */
     async close() {
         const tStart = Date.now();
@@ -1480,7 +1814,7 @@ export class AUNClient {
         }
         await this._transport.close();
         this._state = 'closed';
-        await this._dispatcher.publish('connection.state', { state: this._state });
+        await this._dispatcher.publish('state_change', { state: this._publicState(this._state) });
         this._resetSeqTrackingState();
         this._clientLog.debug(`close exit: elapsed=${Date.now() - tStart}ms`);
     }
@@ -1515,6 +1849,10 @@ export class AUNClient {
             throw new PermissionError(`legacy E2EE method is removed in this SDK: ${method}`);
         }
         const p = { ...(params ?? {}) };
+        if (this._instanceProtectedHeaders && PROTECTED_HEADERS_METHODS.has(method)) {
+            const existing = isJsonObject(p.protected_headers) ? p.protected_headers : {};
+            p.protected_headers = { ...this._instanceProtectedHeaders, ...existing };
+        }
         if (method === 'message.send' || method === 'group.send') {
             this._normalizeOutboundMessagePayload(p, method);
         }
@@ -1545,7 +1883,7 @@ export class AUNClient {
                     throw new StateError('V2 session not initialized; encrypted message.send requires V2 (V1 E2EE removed)');
                 }
                 this._clientLog.debug('call route: message.send → V2 encrypted send');
-                return await this.sendV2(String(p.to ?? ''), p.payload ?? {}, {
+                return await this._sendV2(String(p.to ?? ''), p.payload ?? {}, {
                     messageId: String(p.message_id ?? '') || undefined,
                     timestamp: p.timestamp,
                     protectedHeaders: this._protectedHeadersFromParams(p),
@@ -1564,7 +1902,7 @@ export class AUNClient {
                     throw new StateError('V2 session not initialized; encrypted group.send requires V2 (V1 E2EE removed)');
                 }
                 this._clientLog.debug('call route: group.send → V2 encrypted send');
-                return await this.sendGroupV2(String(p.group_id ?? ''), p.payload ?? {}, {
+                return await this._sendGroupV2(String(p.group_id ?? ''), p.payload ?? {}, {
                     messageId: String(p.message_id ?? '') || undefined,
                     timestamp: p.timestamp,
                     protectedHeaders: this._protectedHeadersFromParams(p),
@@ -1612,24 +1950,24 @@ export class AUNClient {
         // message.pull：V2 就绪时走 V2 pull
         if (method === 'message.pull' && this._v2Session) {
             this._clientLog.debug('call route: message.pull → V2 pull');
-            const messages = await this.pullV2(Number(p.after_seq ?? 0) || 0, Number(p.limit ?? 50) || 50, { force: p.force === true });
+            const messages = await this._pullV2(Number(p.after_seq ?? 0) || 0, Number(p.limit ?? 50) || 50, { force: p.force === true });
             return { messages };
         }
         // message.ack：V2 就绪时走 V2 ack
         if (method === 'message.ack' && this._v2Session) {
             this._clientLog.debug('call route: message.ack → V2 ack');
-            return await this.ackV2(Number(p.seq ?? p.up_to_seq ?? 0) || undefined);
+            return await this._ackV2(Number(p.seq ?? p.up_to_seq ?? 0) || undefined);
         }
         // group.pull：V2 就绪时走 V2 pull
         if (method === 'group.pull' && this._v2Session && p.group_id) {
             this._clientLog.debug('call route: group.pull → V2 pull');
-            const messages = await this.pullGroupV2(String(p.group_id), Number(p.after_seq ?? p.after_message_seq ?? 0) || 0, Number(p.limit ?? 50) || 50);
+            const messages = await this._pullGroupV2(String(p.group_id), Number(p.after_seq ?? p.after_message_seq ?? 0) || 0, Number(p.limit ?? 50) || 50);
             return { messages };
         }
         // group.ack_messages：V2 就绪时走 V2 ack
         if (method === 'group.ack_messages' && this._v2Session && p.group_id) {
             this._clientLog.debug('call route: group.ack_messages → V2 ack');
-            return await this.ackGroupV2(String(p.group_id), Number(p.seq ?? p.msg_seq ?? p.up_to_seq ?? 0) || undefined);
+            return await this._ackGroupV2(String(p.group_id), Number(p.seq ?? p.msg_seq ?? p.up_to_seq ?? 0) || undefined);
         }
         // 关键操作自动附加客户端签名
         if (SIGNED_METHODS.has(method)) {
@@ -1754,16 +2092,6 @@ export class AUNClient {
         }
         return await this._transport.call(method, p);
     }
-    // ── 便利方法 ──────────────────────────────────────
-    async ping(params) {
-        return this.meta.ping(params);
-    }
-    async status(params) {
-        return this.meta.status(params);
-    }
-    async trustRoots(params) {
-        return this.meta.trustRoots(params);
-    }
     // ── 事件 ──────────────────────────────────────────
     /**
      * 订阅事件。
@@ -1901,7 +2229,7 @@ export class AUNClient {
             this._gapFillDone.add(dedupKey);
             try {
                 this._clientLog.debug(`_onRawGroupV2MessageCreated -> group.v2.pull group=${groupId} after_seq=${afterSeq}`);
-                const messages = await this.pullGroupV2(groupId, afterSeq, 50);
+                const messages = await this._pullGroupV2(groupId, afterSeq, 50);
                 this._clientLog.debug(`_onRawGroupV2MessageCreated pulled ${messages.length} msgs for group=${groupId}`);
             }
             finally {
@@ -3372,8 +3700,8 @@ export class AUNClient {
             }
             this._state = 'connected';
             this._connectedAt = Date.now();
-            await this._dispatcher.publish('connection.state', {
-                state: this._state,
+            await this._dispatcher.publish('state_change', {
+                state: this._publicState(this._state),
                 gateway: gatewayUrl,
             });
             if (this._seqTrackerContext !== this._currentSeqTrackerContext()) {
@@ -3383,7 +3711,7 @@ export class AUNClient {
             this._startBackgroundTasks();
             // V2 E2EE: 初始化 session 并注册设备 SPK（与 Python `_init_v2_session` 对齐）
             try {
-                await this.initV2Session();
+                await this._initV2Session();
             }
             catch (exc) {
                 this._clientLog.warn(`V2 session init failed (non-fatal): ${String(exc)}`);
@@ -3402,6 +3730,57 @@ export class AUNClient {
         const gateways = this._resolveGateways(params);
         return gateways[0];
     }
+    async _resolveGatewayForAid(aid) {
+        const target = String(aid ?? this._aid ?? '').trim();
+        if (!target)
+            throw new StateError('gateway discovery requires a loaded AID');
+        if (this._gatewayUrl)
+            return this._gatewayUrl;
+        try {
+            const getMetadata = this._keystore.getMetadata;
+            const raw = typeof getMetadata === 'function'
+                ? String(await getMetadata.call(this._keystore, target, 'gateway_url') ?? '').trim()
+                : '';
+            if (raw) {
+                const gateway = raw.startsWith('"') && raw.endsWith('"') ? String(JSON.parse(raw)).trim() : raw;
+                if (gateway) {
+                    this._gatewayUrl = gateway;
+                    return gateway;
+                }
+            }
+        }
+        catch {
+            // 缓存读取失败不影响发现流程。
+        }
+        const dotIdx = target.indexOf('.');
+        const issuerDomain = dotIdx >= 0 ? target.slice(dotIdx + 1) : target;
+        const portSuffix = '';
+        const candidates = [
+            `https://${target}${portSuffix}/.well-known/aun-gateway`,
+            `https://gateway.${issuerDomain}${portSuffix}/.well-known/aun-gateway`,
+        ];
+        let lastError = null;
+        for (const url of candidates) {
+            try {
+                const gateway = await this._discovery.discover(url);
+                this._gatewayUrl = gateway;
+                try {
+                    const setMetadata = this._keystore.setMetadata;
+                    if (typeof setMetadata === 'function') {
+                        await setMetadata.call(this._keystore, target, 'gateway_url', gateway);
+                    }
+                }
+                catch {
+                    // 缓存写入失败不影响连接。
+                }
+                return gateway;
+            }
+            catch (err) {
+                lastError = err;
+            }
+        }
+        throw lastError instanceof Error ? lastError : new ConnectionError(`gateway discovery failed for ${target}`);
+    }
     _resolveGateways(params) {
         const topology = isJsonObject(params.topology) ? params.topology : null;
         if (topology) {
@@ -3451,12 +3830,13 @@ export class AUNClient {
     _normalizeConnectParams(params) {
         const request = { ...params };
         const accessToken = String(request.access_token ?? '');
-        if (!accessToken)
-            throw new StateError('connect requires non-empty access_token');
         const gateway = String(request.gateway ?? this._gatewayUrl ?? '');
         if (!gateway)
             throw new StateError('connect requires non-empty gateway');
-        request.access_token = accessToken;
+        if (accessToken)
+            request.access_token = accessToken;
+        else
+            delete request.access_token;
         request.gateway = gateway;
         request.device_id = this._deviceId;
         request.slot_id = normalizeInstanceId(request.slot_id ?? this._slotId, 'slot_id', { allowEmpty: true });
@@ -3676,6 +4056,11 @@ export class AUNClient {
                 }
                 try {
                     identity = await this._auth.refreshCachedTokens(this._gatewayUrl, identity);
+                    // 刷新期间可能已断线，复检状态，避免写回 stale identity
+                    if (this._state !== 'connected') {
+                        scheduleRefresh();
+                        return;
+                    }
                     this._identity = identity;
                     if (this._sessionParams && identity.access_token) {
                         this._sessionParams.access_token = identity.access_token;
@@ -3815,8 +4200,8 @@ export class AUNClient {
         this._state = 'disconnected';
         // 先停止后台任务，避免心跳/token刷新在重连期间继续触发
         this._stopBackgroundTasks();
-        await this._dispatcher.publish('connection.state', {
-            state: this._state,
+        await this._dispatcher.publish('state_change', {
+            state: this._publicState(this._state),
             error,
         });
         if (!this._sessionOptions.auto_reconnect)
@@ -3830,7 +4215,7 @@ export class AUNClient {
             this._clientLog.warn(`suppress auto-reconnect: ${reason}`);
             const disconnectInfo = this._lastDisconnectInfo ?? {};
             const eventPayload = {
-                state: this._state, error, reason,
+                state: this._publicState(this._state), error, reason,
             };
             // 把服务端附带的结构化 detail（如配额超限信息）也带给应用层
             const detail = disconnectInfo.detail;
@@ -3840,7 +4225,7 @@ export class AUNClient {
             if (disconnectInfo.code !== undefined && disconnectInfo.code !== null) {
                 eventPayload.code = disconnectInfo.code;
             }
-            await this._dispatcher.publish('connection.state', eventPayload);
+            await this._dispatcher.publish('state_change', eventPayload);
             return;
         }
         // 1000 = 正常关闭, 1006 = 网络异常断开（无 close frame），其他 code = 服务端主动关闭
@@ -3858,34 +4243,51 @@ export class AUNClient {
         const maxAttempts = Number.isFinite(maxAttemptsRaw) && maxAttemptsRaw > 0 ? Math.floor(maxAttemptsRaw) : 0;
         // 服务端主动关闭时从 16s 起跳，避免重连风暴；网络断开从 initial_delay 起跳
         let delay = clampReconnectDelaySeconds(serverInitiated ? 16.0 : retry.initial_delay, serverInitiated ? 16.0 : 1.0, maxBaseDelay);
+        this._retryAttempt = 0;
+        this._retryMaxAttempts = maxAttempts;
         for (let attempt = 1; !this._reconnectAbort?.signal.aborted; attempt++) {
             // R1 fix: max_attempts 检查在循环顶部，覆盖所有路径（含 health-fail）
             if (maxAttempts > 0 && attempt > maxAttempts) {
                 this._state = 'terminal_failed';
+                this._nextRetryAt = null;
                 this._reconnectActive = false;
                 this._reconnectAbort = null;
-                await this._dispatcher.publish('connection.state', {
-                    state: this._state,
+                await this._dispatcher.publish('state_change', {
+                    state: this._publicState(this._state),
                     attempt: attempt - 1,
                     reason: 'max_attempts_exhausted',
                 });
                 return;
             }
-            this._state = 'reconnecting';
-            await this._dispatcher.publish('connection.state', {
-                state: this._state,
+            // 先进入 retry_backoff 状态（对齐 Python：先退避再重连）
+            this._retryAttempt = attempt;
+            const sleepMs = reconnectSleepDelaySeconds(delay, maxBaseDelay) * 1000;
+            this._nextRetryAt = new Date(Date.now() + sleepMs);
+            this._state = 'retry_backoff';
+            await this._dispatcher.publish('state_change', {
+                state: this._publicState(this._state),
                 attempt,
+                next_retry_at: this._nextRetryAt.getTime() / 1000,
             });
             try {
-                await this._sleep(reconnectSleepDelaySeconds(delay, maxBaseDelay) * 1000);
+                await this._sleep(sleepMs);
+                this._nextRetryAt = null;
                 if (this._reconnectAbort?.signal.aborted) {
                     this._reconnectActive = false;
                     return;
                 }
+                // 退避结束，进入 reconnecting 状态
+                this._state = 'reconnecting';
+                await this._dispatcher.publish('state_change', {
+                    state: this._publicState(this._state),
+                    attempt,
+                });
                 // 重连前先 GET /health 探测，不健康则跳过本轮
                 if (this._gatewayUrl) {
                     const healthy = await this._discovery.checkHealth(this._gatewayUrl, 5000);
                     if (!healthy) {
+                        this._lastError = new Error('gateway health check failed');
+                        this._lastErrorCode = 'gateway_unhealthy';
                         delay = Math.min(delay * 2, maxBaseDelay);
                         continue;
                     }
@@ -3895,21 +4297,27 @@ export class AUNClient {
                     throw new StateError('missing connect params for reconnect');
                 }
                 await this._connectOnce(this._sessionParams, true);
+                this._lastError = null;
+                this._lastErrorCode = null;
+                this._nextRetryAt = null;
                 this._reconnectActive = false;
                 this._reconnectAbort = null;
                 return;
             }
             catch (exc) {
+                this._lastError = exc instanceof Error ? exc : new Error(String(exc));
+                this._lastErrorCode = 'reconnect_failed';
                 await this._dispatcher.publish('connection.error', {
                     error: formatCaughtError(exc),
                     attempt,
                 });
                 if (!this._shouldRetryReconnect(exc)) {
                     this._state = 'terminal_failed';
+                    this._nextRetryAt = null;
                     this._reconnectActive = false;
                     this._reconnectAbort = null;
-                    await this._dispatcher.publish('connection.state', {
-                        state: this._state,
+                    await this._dispatcher.publish('state_change', {
+                        state: this._publicState(this._state),
                         error: formatCaughtError(exc),
                         attempt,
                     });
@@ -4259,7 +4667,7 @@ export class AUNClient {
      *
      * connect 成功后自动调用，可幂等手动调用。
      */
-    async initV2Session() {
+    async _initV2Session() {
         if (!this._aid)
             return;
         let identity = this._identity;
@@ -4564,7 +4972,7 @@ export class AUNClient {
      * @param opts 可选 messageId / timestamp（与 Python 行为一致）
      * @returns 服务端响应
      */
-    async sendV2(to, payload, opts) {
+    async _sendV2(to, payload, opts) {
         if (!this._v2Session) {
             throw new StateError('V2 session not initialized (not connected?)');
         }
@@ -4608,7 +5016,7 @@ export class AUNClient {
      * @param afterSeq 从此 seq 之后开始拉取（0/省略 = 从当前 contiguous 开始）
      * @param limit 最多拉取条数
      */
-    async pullV2(afterSeq = 0, limit = 50, opts) {
+    async _pullV2(afterSeq = 0, limit = 50, opts) {
         if (!this._v2Session) {
             throw new StateError('V2 session not initialized (not connected?)');
         }
@@ -4704,7 +5112,7 @@ export class AUNClient {
                     this._saveSeqTrackerState();
                 }
                 if (messages.length > 0 && contigAdvanced && ackSeq > 0) {
-                    this._safeAsync(this.ackV2(ackSeq).then(() => undefined));
+                    this._safeAsync(this._ackV2(ackSeq).then(() => undefined));
                 }
             }
             const nextAfter = Math.max(pageMaxSeq, nextAfterSeq);
@@ -4722,7 +5130,7 @@ export class AUNClient {
      *
      * @param upToSeq 确认到此 seq；省略则用当前 contiguous
      */
-    async ackV2(upToSeq) {
+    async _ackV2(upToSeq) {
         const ns = this._aid ? `p2p:${this._aid}` : '';
         let seq = upToSeq ?? (ns ? this._seqTracker.getContiguousSeq(ns) : 0);
         if (seq <= 0)
@@ -4959,7 +5367,7 @@ export class AUNClient {
      * @param opts 可选 messageId / timestamp
      * @returns 服务端响应
      */
-    async sendGroupV2(groupId, payload, opts) {
+    async _sendGroupV2(groupId, payload, opts) {
         if (!this._v2Session) {
             throw new StateError('V2 session not initialized (not connected?)');
         }
@@ -5018,7 +5426,7 @@ export class AUNClient {
         }
     }
     async _pullGroupV2Internal(params) {
-        await this.pullGroupV2(params.group_id, params.after_seq, params.limit);
+        await this._pullGroupV2(params.group_id, params.after_seq, params.limit);
     }
     /**
      * 拉取并解密 V2 Group 消息。
@@ -5027,7 +5435,7 @@ export class AUNClient {
      * @param afterSeq 从此 seq 之后开始拉取（0/省略 = 从当前 contiguous 开始）
      * @param limit 最多拉取条数
      */
-    async pullGroupV2(groupId, afterSeq = 0, limit = 50) {
+    async _pullGroupV2(groupId, afterSeq = 0, limit = 50) {
         if (!this._v2Session) {
             throw new StateError('V2 session not initialized (not connected?)');
         }
@@ -5126,7 +5534,7 @@ export class AUNClient {
                 this._saveSeqTrackerState();
             }
             if (messages.length > 0 && contigAdvanced && ackSeq > 0) {
-                this._safeAsync(this.ackGroupV2(gid, ackSeq).then(() => undefined));
+                this._safeAsync(this._ackGroupV2(gid, ackSeq).then(() => undefined));
             }
             const nextAfter = Math.max(pageMaxSeq, nextAfterSeq);
             if (messages.length === 0 || nextAfter <= nextAfterSeq || result.has_more === false)
@@ -5144,7 +5552,7 @@ export class AUNClient {
      * @param groupId 群 ID
      * @param upToSeq 确认到此 seq；省略则用当前 contiguous
      */
-    async ackGroupV2(groupId, upToSeq) {
+    async _ackGroupV2(groupId, upToSeq) {
         const gid = normalizeGroupId(groupId) || String(groupId ?? '').trim();
         if (!gid)
             throw new ValidationError('group.ack_messages requires group_id');
@@ -6221,7 +6629,7 @@ export class AUNClient {
         try {
             do {
                 this._v2PullPending = false;
-                await this.pullV2();
+                await this._pullV2();
                 const newContig = ns ? this._seqTracker.getContiguousSeq(ns) : -1;
                 this._clientLog.debug(`_onV2PushNotification pull done: contiguous_seq=${contigBefore}->${newContig} (push_seq=${pushSeq || 'null'})`);
             } while (this._v2PullPending);