@aithos/sdk 0.1.0-alpha.4 → 0.1.0-alpha.6

package/dist/src/auth.js

@@ -1,100 +1,242 @@
 // SPDX-License-Identifier: Apache-2.0
 // Copyright 2026 Mathieu Colla
-// Aithos auth — sign-up, sign-in, sign-in-with-Google.
+// Aithos auth — sign-up, sign-in (email+password / Google SSO /
+// recovery file), mandate import, sign-out.
 //
-// One class, three flows, automatic session persistence. Apps shouldn't
-// need to touch Argon2id, AES-GCM, or {@link sessionStorage} directly :
+// One stateful object per app. Holds the active {@link AithosSession}
+// (JWT-backed when present), the loaded owner signers in memory (when
+// signed in as an owner), and a registry of active delegate
+// sessions. Persists across reloads via the configured
+// {@link AithosSessionStore} (JWT) and {@link AithosKeyStore} (signing
+// material).
 //
-//   const auth = new AithosAuth();
+// Strict mode (per design discussion): the two stores must agree about
+// who's signed in. If sessionStore has a JWT for one DID and keyStore
+// has an owner with a DIFFERENT DID, both are wiped. If sessionStore
+// has a JWT but keyStore has no owner at all, the JWT is wiped (a
+// JWT alone is useless without local signing capability for everything
+// past compute/wallet).
 //
-//   // Sign in (existing account, email + password)
-//   const session = await auth.signIn({ email, password });
-//
-//   // Sign up (creates an Aithos identity end-to-end)
-//   const { recoveryFile, ...session } = await auth.signUp({
-//     email, password, handle, displayName,
-//   });
-//
-//   // Sign in with Google — redirects to Google's consent screen
-//   auth.signInWithGoogle({ appState: "/dashboard" });
-//
-//   // Back on /auth/callback after Google
-//   const session = await auth.handleCallback();
-//
-//   // Anywhere — read the active session, null if signed out / expired
-//   const current = auth.getCurrentSession();
-//
-//   // Sign out — clears the session store
-//   await auth.signOut();
-//
-// Storage : by default, sessions are persisted in `sessionStorage` (web)
-// or no-op (Node / SSR). Apps with different needs pass their own
-// `AithosSessionStore` to the constructor — see ./session-store.ts.
-import { DEFAULT_KDF, buildBlobPlaintext, createBrowserIdentity, deriveAuthAndEncKeys, encryptBlob, randomNonce, randomSalt, serializeBlob, zeroize, } from "@aithos/protocol-client";
+// JWT-less sessions (recovery / mandate sign-ins) are valid: the
+// keyStore is the source of truth for "is the user signed in", the
+// JWT is auxiliary for compute/wallet.
+import { buildBlobPlaintext, buildSignedEnvelope, createBrowserIdentity, decryptBlob, DEFAULT_KDF, deriveAuthAndEncKeys, encryptBlob, parseBlob, randomNonce, randomSalt, serializeBlob, signedDidDocument, zeroize, } from "@aithos/protocol-client";
 import { loginChallenge, loginVerify, registerAccount, } from "./auth-api.js";
 import { defaultSessionStore, } from "./session-store.js";
+import { defaultKeyStore, } from "./key-store.js";
+import { parseDelegateBundle, readDelegateBundleText, } from "./internal/delegate-bundle.js";
+import { DelegateActor, DelegateRegistry, } from "./internal/delegate-state.js";
+import { OwnerSigners } from "./internal/owner-signers.js";
+import { parseRecoveryFile, readRecoveryFileText, serializeRecoveryFile, } from "./internal/recovery-file.js";
 import { AithosSDKError } from "./types.js";
 /** Default URL of the Aithos auth backend. */
 export const DEFAULT_AUTH_BASE_URL = "https://auth.aithos.be";
+/** Default URL of the Aithos primitives API (publish_identity, publish_ethos_edition, etc.). */
+export const DEFAULT_API_BASE_URL = "https://api.aithos.be";
 /* -------------------------------------------------------------------------- */
 /*  AithosAuth                                                                */
 /* -------------------------------------------------------------------------- */
-/**
- * Authenticator for the Aithos identity service. One instance per app
- * is the recommended pattern (the constructor is cheap).
- *
- * The class is **stateful** in one specific way : it owns a session store
- * that gets written on every successful auth call and read by
- * {@link getCurrentSession}. Pass a custom store at construction time
- * if you need different persistence (localStorage, IndexedDB, no-op).
- */
 export class AithosAuth {
-    /** Resolved auth base URL with a trailing slash trimmed. */
     authBaseUrl;
-    fetchImpl;
-    win;
-    store;
+    apiBaseUrl;
+    #fetchImpl;
+    #win;
+    #sessionStore;
+    #keyStore;
+    /** In-memory owner signers — populated after sign-in or `resume`. */
+    #ownerSigners = null;
+    /** Active delegate registry. */
+    #delegates = new DelegateRegistry();
     constructor(config = {}) {
         this.authBaseUrl = trimSlash(config.authBaseUrl ?? DEFAULT_AUTH_BASE_URL);
-        this.fetchImpl = config.fetch ?? globalThis.fetch.bind(globalThis);
-        this.win = config.window ?? (typeof window !== "undefined" ? window : undefined);
-        this.store = config.sessionStore ?? defaultSessionStore();
+        this.apiBaseUrl = trimSlash(config.apiBaseUrl ?? DEFAULT_API_BASE_URL);
+        this.#fetchImpl = config.fetch ?? globalThis.fetch.bind(globalThis);
+        this.#win =
+            config.window ??
+                (typeof window !== "undefined" ? window : undefined);
+        this.#sessionStore = config.sessionStore ?? defaultSessionStore();
+        this.#keyStore = config.keyStore ?? defaultKeyStore();
     }
     /* ------------------------------------------------------------------------ */
-    /*  Email + password                                                         */
+    /*  Boot-time hydration                                                      */
     /* ------------------------------------------------------------------------ */
     /**
-     * Sign in to an existing Aithos account. Two-round-trip flow under the
-     * hood :
-     *
-     *   1. POST /auth/login/challenge → server returns the salts + KDF params
-     *   2. derive auth_key from password + salt (Argon2id)
-     *   3. POST /auth/login/verify → server checks auth_key, returns JWT + blob
+     * Reload signing material and JWT session from the configured stores.
+     * Must be called once at app boot before relying on
+     * {@link getCurrentSession} / {@link getOwnerInfo} / {@link canSignAsOwner}
+     * — until then they reflect only what's been done in-memory in the
+     * current tab.
      *
-     * The returned `enc_key_b64` is empty by design : password sign-in
-     * doesn't release the vault key over the wire (it's derived locally
-     * but discarded after the call). Apps that need the seeds — most
-     * apps don't — should use the (forthcoming) `loadEthos` helper
-     * separately with the password still in hand.
+     * Strict consistency: if the JWT and the stored owner disagree about
+     * who's signed in, both are wiped and the user re-auths. JWT-less
+     * owner state (loaded from keyStore but no JWT) is a valid resumed
+     * state — the user signed in via recovery or imported a mandate at
+     * some earlier moment and never went through the JWT flow.
+     */
+    async resume() {
+        // 1. Owner side.
+        const stored = await this.#keyStore.loadOwner().catch(() => null);
+        const jwt = this.#sessionStore.get();
+        if (stored) {
+            this.#ownerSigners = OwnerSigners.fromStoredOwnerKeys(stored);
+            // JWT must match the owner DID — otherwise it's stale state.
+            if (jwt && jwt.did !== stored.did) {
+                this.#sessionStore.clear();
+            }
+        }
+        else {
+            // No owner persisted. A lingering JWT is meaningless without local
+            // signing capability — wipe it (strict mode).
+            if (jwt)
+                this.#sessionStore.clear();
+        }
+        // 2. Delegate side. Independent of owner state — a user may hold
+        //    only delegate bundles and no owner identity at all.
+        const storedDelegates = await this.#keyStore
+            .listDelegates()
+            .catch(() => []);
+        for (const d of storedDelegates) {
+            try {
+                this.#delegates.add(DelegateActor.fromStored(d));
+            }
+            catch {
+                // Skip corrupted entries silently — the keystore validators
+                // already filter most of these. A skip here means the keystore
+                // record passed validation but the seed couldn't be re-derived
+                // (e.g. zero-length, future migration); ignore and continue.
+            }
+        }
+    }
+    /* ------------------------------------------------------------------------ */
+    /*  State accessors                                                          */
+    /* ------------------------------------------------------------------------ */
+    /** JWT-backed session. Null when signed in via recovery / mandate / not at all. */
+    getCurrentSession() {
+        const session = this.#sessionStore.get();
+        if (!session)
+            return null;
+        // Belt-and-braces: the session store auto-evicts on expiry, but we
+        // also re-check here in case the in-tab clock drifted post-load.
+        return session;
+    }
+    /** Loaded owner identity. Independent of JWT presence. */
+    getOwnerInfo() {
+        if (!this.#ownerSigners)
+            return null;
+        return {
+            did: this.#ownerSigners.did,
+            handle: this.#ownerSigners.handle,
+            displayName: this.#ownerSigners.displayName,
+        };
+    }
+    getDelegates() {
+        return this.#delegates.list().map(actorToInfo);
+    }
+    canSignAsOwner() {
+        return this.#ownerSigners !== null && !this.#ownerSigners.destroyed;
+    }
+    canSignAsDelegateFor(did) {
+        const a = this.#delegates.findForSubject(did);
+        return a !== undefined && !a.destroyed;
+    }
+    /**
+     * Internal accessor used by sibling SDK namespaces (compute, wallet,
+     * ethos) when they need to sign on behalf of the owner. Returns null
+     * if no owner is loaded.
      *
-     * Persists the session in the configured store before returning.
+     * @internal
      */
+    _getOwnerSigners() {
+        return this.#ownerSigners;
+    }
+    /**
+     * Internal accessor — looks up an active delegate by mandate id.
+     * @internal
+     */
+    _getDelegateActor(mandateId) {
+        return this.#delegates.get(mandateId);
+    }
+    /**
+     * Internal accessor — finds the first active delegate whose subject
+     * matches `did`. Used by `sdk.ethos.of(did)` when the user holds a
+     * mandate for that subject.
+     * @internal
+     */
+    _findDelegateForSubject(did) {
+        return this.#delegates.findForSubject(did);
+    }
+    /* ------------------------------------------------------------------------ */
+    /*  Email + password — signIn                                                */
+    /* ------------------------------------------------------------------------ */
     async signIn(input) {
         if (!input.email || !input.password) {
             throw new AithosSDKError("auth_invalid_input", "signIn: email and password are required");
         }
-        const challenge = await loginChallenge({ fetchImpl: this.fetchImpl, authBaseUrl: this.authBaseUrl }, input.email);
+        const challenge = await loginChallenge({ fetchImpl: this.#fetchImpl, authBaseUrl: this.authBaseUrl }, input.email);
         const { authKey, encKey } = await deriveAuthAndEncKeys(input.password, challenge.authSalt, challenge.encSalt, challenge.kdf);
-        // We don't keep enc_key around for the password flow — apps that
-        // need it can call deriveAuthAndEncKeys themselves with the password
-        // still available in their UI state. Wipe to be defensive.
-        zeroize(encKey);
         let verify;
         try {
-            verify = await loginVerify({ fetchImpl: this.fetchImpl, authBaseUrl: this.authBaseUrl }, input.email, authKey);
+            verify = await loginVerify({ fetchImpl: this.#fetchImpl, authBaseUrl: this.authBaseUrl }, input.email, authKey);
         }
-        finally {
+        catch (e) {
+            // On failure, both keys must be wiped before propagating.
             zeroize(authKey);
+            zeroize(encKey);
+            throw e;
+        }
+        zeroize(authKey);
+        // Decrypt the vault blob → plaintext seeds + delegate bundles.
+        let plaintext;
+        try {
+            const blobBytes = decryptBlob(encKey, verify.blobNonce, verify.blob);
+            try {
+                plaintext = parseBlob(blobBytes);
+            }
+            finally {
+                zeroize(blobBytes);
+            }
+        }
+        catch (e) {
+            zeroize(encKey);
+            throw new AithosSDKError("auth_blob_decrypt_failed", `Could not decrypt the vault blob: ${e.message}`);
+        }
+        zeroize(encKey);
+        // Sanity check: blob's identity must agree with what verify returned.
+        if (plaintext.identity.did !== verify.did) {
+            throw new AithosSDKError("auth_blob_identity_mismatch", "vault blob's DID does not match the verified login DID", { data: { blobDid: plaintext.identity.did, verifyDid: verify.did } });
+        }
+        // Hydrate in-memory state.
+        if (this.#ownerSigners)
+            this.#ownerSigners.destroy();
+        this.#ownerSigners = OwnerSigners.fromBlobPlaintext(plaintext);
+        // Persist to keyStore — owner first, then delegates.
+        const ownerStored = {
+            version: "0.1.0-hex",
+            did: plaintext.identity.did,
+            handle: plaintext.identity.handle,
+            displayName: plaintext.identity.displayName,
+            seedsHex: plaintext.seeds,
+            savedAt: new Date().toISOString(),
+        };
+        await this.#keyStore.saveOwner(ownerStored);
+        // Replace any prior delegate set with what the blob carries.
+        await this.#keyStore.clearAllDelegates();
+        this.#delegates.destroy();
+        for (const d of plaintext.delegates) {
+            const stored = storedDelegateFromBlob(d);
+            try {
+                await this.#keyStore.saveDelegate(stored);
+            }
+            catch {
+                // Persistence failure shouldn't block the sign-in. We still load
+                // the actor in memory so the session works for the current tab.
+            }
+            try {
+                this.#delegates.add(DelegateActor.fromStored(stored));
+            }
+            catch {
+                // Skip silently — keep going on remaining delegates.
+            }
         }
         const session = {
             session: verify.session,
@@ -107,25 +249,12 @@ export class AithosAuth {
             enc_key_b64: "",
             is_first_login: false,
         };
-        this.store.set(session);
+        this.#sessionStore.set(session);
         return session;
     }
-    /**
-     * Create a new Aithos account end-to-end :
-     *
-     *   1. Generate a fresh `BrowserIdentity` (4 Ed25519/X25519 seeds)
-     *   2. Build the recovery file (plaintext JSON, the user must save it)
-     *   3. Derive auth_key + enc_key from the password (Argon2id, fresh salts)
-     *   4. Encrypt the seeds in a vault blob (AES-GCM-256)
-     *   5. POST /auth/register with everything → JWT
-     *   6. Persist the session and return it + the recovery Blob
-     *
-     * The seeds are NOT published as an Aithos ethos here : the user's
-     * profile on `app.aithos.be` won't appear until they (or another app)
-     * publishes their first edition. This matches `aithos/app`'s design,
-     * where the vault is the source of truth for keys and the published
-     * edition is a separate concern.
-     */
+    /* ------------------------------------------------------------------------ */
+    /*  Email + password — signUp                                                */
+    /* ------------------------------------------------------------------------ */
     async signUp(input) {
         if (!input.email || !input.password) {
             throw new AithosSDKError("auth_invalid_input", "signUp: email and password are required");
@@ -134,31 +263,16 @@ export class AithosAuth {
             throw new AithosSDKError("auth_invalid_handle", "signUp: handle must be 1–63 alphanumeric chars + _ -");
         }
         const displayName = input.displayName ?? input.handle;
-        // 1) Identity.
         const identity = createBrowserIdentity(input.handle, displayName);
-        // 2) Recovery file — same shape as @aithos/protocol-client emits in
-        //    `runOnboarding`, plaintext v0.1.0.
-        const recoveryJson = {
-            aithos_recovery_version: "0.1.0-plaintext",
-            handle: identity.handle,
-            display_name: identity.displayName,
-            did: identity.did,
-            seeds_hex: {
-                root: bytesToHex(identity.root.seed),
-                public: bytesToHex(identity.public.seed),
-                circle: bytesToHex(identity.circle.seed),
-                self: bytesToHex(identity.self.seed),
-            },
-            saved_at: new Date().toISOString(),
-        };
-        const recoveryFile = new Blob([JSON.stringify(recoveryJson, null, 2)], { type: "application/json" });
-        const recoveryFilename = `aithos-recovery-${identity.handle}.json`;
-        // 3) Derive password-based keys.
+        const recoverySerialized = serializeRecoveryFile(identity);
+        const recoveryFile = new Blob([recoverySerialized.text], {
+            type: "application/json",
+        });
+        // Derive password-based keys, encrypt the vault blob.
         const authSalt = randomSalt();
         const encSalt = randomSalt();
         const kdf = DEFAULT_KDF;
         const { authKey, encKey } = await deriveAuthAndEncKeys(input.password, authSalt, encSalt, kdf);
-        // 4) Build & encrypt the vault blob.
         const plaintext = buildBlobPlaintext({
             identity: {
                 did: identity.did,
@@ -176,10 +290,9 @@ export class AithosAuth {
         const blobBytes = serializeBlob(plaintext);
         const blobNonce = randomNonce();
         const blob = encryptBlob(encKey, blobNonce, blobBytes);
-        // 5) Register.
         let registerResp;
         try {
-            registerResp = await registerAccount({ fetchImpl: this.fetchImpl, authBaseUrl: this.authBaseUrl }, {
+            registerResp = await registerAccount({ fetchImpl: this.#fetchImpl, authBaseUrl: this.authBaseUrl }, {
                 email: input.email,
                 handle: identity.handle,
                 displayName: identity.displayName,
@@ -197,6 +310,33 @@ export class AithosAuth {
             zeroize(authKey);
             zeroize(encKey);
         }
+        // Bootstrap the Ethos on api.aithos.be. Without this, every subsequent
+        // write (publish_ethos_edition, etc.) errors out with -32020
+        // "subject identity not published". We do this BEFORE hydrating local
+        // state so a bootstrap failure leaves the SDK in a clean
+        // "not signed in" state — the dev shows an error, the user retries.
+        // The auth account on auth.aithos.be DOES exist at this point, but
+        // without the local hydrate the user can't act on it. Self-heal on
+        // signIn (re-attempt publish_identity if missing) is planned for a
+        // follow-up release.
+        await this.#publishIdentity(identity);
+        // Hydrate in-memory state from the fresh identity.
+        if (this.#ownerSigners)
+            this.#ownerSigners.destroy();
+        this.#ownerSigners = OwnerSigners.fromBrowserIdentity(identity);
+        await this.#keyStore.saveOwner({
+            version: "0.1.0-hex",
+            did: identity.did,
+            handle: identity.handle,
+            displayName: identity.displayName,
+            seedsHex: {
+                root: bytesToHex(identity.root.seed),
+                public: bytesToHex(identity.public.seed),
+                circle: bytesToHex(identity.circle.seed),
+                self: bytesToHex(identity.self.seed),
+            },
+            savedAt: new Date().toISOString(),
+        });
         const session = {
             session: registerResp.session,
             exp: registerResp.exp,
@@ -208,22 +348,101 @@ export class AithosAuth {
             enc_key_b64: "",
             is_first_login: false,
         };
-        this.store.set(session);
-        return { session, recoveryFile, recoveryFilename };
+        this.#sessionStore.set(session);
+        return {
+            session,
+            recoveryFile,
+            recoveryFilename: recoverySerialized.filename,
+        };
     }
     /* ------------------------------------------------------------------------ */
-    /*  Google SSO                                                               */
+    /*  Recovery file                                                            */
     /* ------------------------------------------------------------------------ */
     /**
-     * Redirect the browser to Google's OAuth consent screen. Must be called
-     * synchronously in response to a user gesture (button click) — most
-     * browsers block top-level navigation triggered from idle code.
+     * Sign in by uploading a recovery file. Hydrates the owner signers
+     * locally — no JWT is obtained on this path because the recovery
+     * file alone doesn't authenticate against the auth backend (no
+     * password, no Google session). Apps that need compute/wallet
+     * access should follow up with an email+password sign-in or with
+     * Google SSO.
      *
-     * Does not return : navigation tears the JS context down. The `never`
-     * return type tells callers any code after the call is unreachable.
+     * The recovery file is ALWAYS the file produced by `signUp` (or the
+     * equivalent one emitted by `protocol-client`'s `runOnboarding`).
+     * Both shapes are accepted.
      */
+    async signInWithRecovery(input) {
+        const text = await readRecoveryFileText(input.file);
+        const parsed = parseRecoveryFile(text);
+        // Build a StoredOwnerKeys-shape on the spot, then push to keyStore +
+        // hydrate signers.
+        const stored = {
+            version: "0.1.0-hex",
+            did: parsed.did,
+            handle: parsed.handle,
+            displayName: parsed.displayName,
+            seedsHex: parsed.seedsHex,
+            savedAt: new Date().toISOString(),
+        };
+        // If a different owner is already loaded, refuse — apps must call
+        // signOut() first. Mixing two owners in one auth instance is a
+        // nonsense state we don't want to support.
+        if (this.#ownerSigners && this.#ownerSigners.did !== parsed.did) {
+            throw new AithosSDKError("auth_owner_already_loaded", "another owner is already signed in; call signOut first", { data: { current: this.#ownerSigners.did, incoming: parsed.did } });
+        }
+        if (this.#ownerSigners)
+            this.#ownerSigners.destroy();
+        this.#ownerSigners = OwnerSigners.fromStoredOwnerKeys(stored);
+        await this.#keyStore.saveOwner(stored);
+        // Recovery flow doesn't yield a JWT — wipe any stale one to keep
+        // the two stores in sync.
+        this.#sessionStore.clear();
+        return {
+            did: parsed.did,
+            handle: parsed.handle,
+            displayName: parsed.displayName,
+        };
+    }
+    /* ------------------------------------------------------------------------ */
+    /*  Mandate import                                                           */
+    /* ------------------------------------------------------------------------ */
+    /**
+     * Import a delegate bundle (`.aithos-delegate.json`). Works in any
+     * state: with no owner loaded (delegate-only session), or alongside
+     * an existing owner (the user holds mandates for other people's
+     * ethoses while also being an owner themselves).
+     */
+    async importMandate(input) {
+        const text = await readDelegateBundleText(input.bundle);
+        const parsed = parseDelegateBundle(text);
+        const stored = {
+            version: "0.1.0-hex",
+            subjectDid: parsed.subjectDid,
+            mandateId: parsed.mandateId,
+            mandate: parsed.mandate,
+            granteeId: parsed.granteeId,
+            granteePubkeyMultibase: parsed.granteePubkeyMultibase,
+            delegateSeedHex: parsed.delegateSeedHex,
+            importedAt: new Date().toISOString(),
+        };
+        await this.#keyStore.saveDelegate(stored);
+        this.#delegates.add(DelegateActor.fromStored(stored));
+        return {
+            mandateId: stored.mandateId,
+            subjectDid: stored.subjectDid,
+            granteeId: stored.granteeId,
+            scopes: scopesFromMandate(stored.mandate),
+            expiresAt: notAfterFromMandate(stored.mandate),
+        };
+    }
+    async removeMandate(mandateId) {
+        this.#delegates.remove(mandateId);
+        await this.#keyStore.removeDelegate(mandateId);
+    }
+    /* ------------------------------------------------------------------------ */
+    /*  Google SSO                                                               */
+    /* ------------------------------------------------------------------------ */
     signInWithGoogle(opts) {
-        if (!this.win) {
+        if (!this.#win) {
             throw new AithosSDKError("auth_no_window", "AithosAuth.signInWithGoogle requires a browser window");
         }
         const url = new URL(`${this.authBaseUrl}/auth/sso/google/start`);
@@ -233,50 +452,89 @@ export class AithosAuth {
             }
             url.searchParams.set("app_state", opts.appState);
         }
-        this.win.location.assign(url.toString());
-        // Unreachable : location.assign navigates synchronously. The throw is
-        // belt-and-braces in case a caller awaits a microtask before unload.
+        this.#win.location.assign(url.toString());
         throw new AithosSDKError("auth_redirecting", "redirecting to google");
     }
-    /**
-     * Inspect the current URL for an `aithos_code` query parameter. If it's
-     * present, exchange it at the backend, persist the session, and return
-     * it. The query params are stripped from the URL via
-     * `history.replaceState` so a page refresh doesn't replay the redeem
-     * (which would 410 anyway).
-     *
-     * Returns `null` when there's no code in the URL — safe to call on every
-     * page load. Throws {@link AithosSDKError} on backend errors or when
-     * the URL carries `aithos_error=…`.
-     */
     async handleCallback() {
-        if (!this.win)
+        if (!this.#win)
             return null;
-        const here = new URL(this.win.location.href);
+        const here = new URL(this.#win.location.href);
         const error = here.searchParams.get("aithos_error");
         const code = here.searchParams.get("aithos_code");
         const appState = here.searchParams.get("app_state");
         if (error) {
-            cleanCallbackParams(this.win, here);
+            cleanCallbackParams(this.#win, here);
             throw new AithosSDKError(`auth_${error}`, `Sign-in failed: ${error}`, { data: appState ? { app_state: appState } : undefined });
         }
         if (!code)
             return null;
         const session = await this.exchange(code);
-        cleanCallbackParams(this.win, here);
-        this.store.set(session);
+        cleanCallbackParams(this.#win, here);
+        // Hydrate signers if the SSO response carried an enc_key (Google flow
+        // gives us the AES-GCM key in plaintext, encrypted only in transit
+        // by TLS — see auth.aithos.be design doc).
+        if (session.enc_key_b64 &&
+            session.blob_b64 &&
+            session.blob_nonce_b64 &&
+            session.blob_version > 0) {
+            try {
+                const encKey = b64ToBytes(session.enc_key_b64);
+                const blob = b64ToBytes(session.blob_b64);
+                const nonce = b64ToBytes(session.blob_nonce_b64);
+                try {
+                    const blobBytes = decryptBlob(encKey, nonce, blob);
+                    try {
+                        const plaintext = parseBlob(blobBytes);
+                        if (plaintext.identity.did === session.did) {
+                            if (this.#ownerSigners)
+                                this.#ownerSigners.destroy();
+                            this.#ownerSigners = OwnerSigners.fromBlobPlaintext(plaintext);
+                            await this.#keyStore.saveOwner({
+                                version: "0.1.0-hex",
+                                did: plaintext.identity.did,
+                                handle: plaintext.identity.handle,
+                                displayName: plaintext.identity.displayName,
+                                seedsHex: plaintext.seeds,
+                                savedAt: new Date().toISOString(),
+                            });
+                            await this.#keyStore.clearAllDelegates();
+                            this.#delegates.destroy();
+                            for (const d of plaintext.delegates) {
+                                const stored = storedDelegateFromBlob(d);
+                                try {
+                                    await this.#keyStore.saveDelegate(stored);
+                                }
+                                catch {
+                                    /* keep going */
+                                }
+                                try {
+                                    this.#delegates.add(DelegateActor.fromStored(stored));
+                                }
+                                catch {
+                                    /* keep going */
+                                }
+                            }
+                        }
+                    }
+                    finally {
+                        zeroize(blobBytes);
+                    }
+                }
+                finally {
+                    zeroize(encKey);
+                }
+            }
+            catch {
+                // Decryption failure is non-fatal here: the JWT still works for
+                // compute/wallet, the user will surface the issue if they try to
+                // edit their ethos.
+            }
+        }
+        this.#sessionStore.set(session);
         return session;
     }
-    /**
-     * Programmatically redeem an `aithos_code` for a session. `handleCallback`
-     * calls this for you ; expose it directly for callers that already pulled
-     * the code out of the URL via their own router.
-     *
-     * Note : this method does NOT persist the session — it's the lower-level
-     * primitive. Use `handleCallback` for the full pipe.
-     */
     async exchange(aithosCode) {
-        const res = await this.fetchImpl(`${this.authBaseUrl}/auth/sso/exchange`, {
+        const res = await this.#fetchImpl(`${this.authBaseUrl}/auth/sso/exchange`, {
             method: "POST",
             headers: { "content-type": "application/json" },
             body: JSON.stringify({ aithos_code: aithosCode }),
@@ -303,23 +561,93 @@ export class AithosAuth {
         return (await res.json());
     }
     /* ------------------------------------------------------------------------ */
-    /*  Session lifecycle                                                        */
+    /*  Sign-out                                                                 */
     /* ------------------------------------------------------------------------ */
-    /**
-     * Read the active session from the configured store. Returns null if
-     * the user is signed out, or if the JWT has expired (the store
-     * auto-evicts expired entries — see ./session-store.ts).
-     */
-    getCurrentSession() {
-        return this.store.get();
+    async signOut() {
+        if (this.#ownerSigners)
+            this.#ownerSigners.destroy();
+        this.#ownerSigners = null;
+        this.#delegates.destroy();
+        this.#sessionStore.clear();
+        await this.#keyStore.clearOwner().catch(() => { });
+        await this.#keyStore.clearAllDelegates().catch(() => { });
     }
+    /* ------------------------------------------------------------------------ */
+    /*  Internal — Ethos bootstrap                                              */
+    /* ------------------------------------------------------------------------ */
     /**
-     * Stateless sign-out — the Aithos backend doesn't track sessions, so
-     * there's nothing to revoke server-side ; this method clears the
-     * configured session store and resolves.
+     * Provision the user's Ethos on `api.aithos.be` by signing and POSTing an
+     * `aithos.publish_identity` envelope. Required after a fresh sign-up so
+     * subsequent edition publishes (`me.publish()`) don't fail with
+     * `-32020 subject identity not published`.
+     *
+     * Retries twice with exponential backoff on transient errors (network or
+     * 5xx). Throws {@link AithosSDKError} with code `ethos_bootstrap_failed`
+     * on definitive failure — the caller is expected to abort sign-up.
+     *
+     * @internal
      */
-    async signOut() {
-        this.store.clear();
+    async #publishIdentity(identity) {
+        const url = `${this.apiBaseUrl}/mcp/primitives/write`;
+        const signedDoc = signedDidDocument(identity);
+        const params = {
+            did_document: signedDoc,
+            handle: identity.handle,
+            display_name: identity.displayName,
+        };
+        const envelope = buildSignedEnvelope({
+            iss: identity.did,
+            aud: url,
+            method: "aithos.publish_identity",
+            verificationMethod: `${identity.did}#root`,
+            params,
+            signer: identity.root,
+        });
+        const body = JSON.stringify({
+            jsonrpc: "2.0",
+            id: "publish_identity",
+            method: "aithos.publish_identity",
+            params: { ...params, _envelope: envelope },
+        });
+        // Two retries with backoff (300ms, 1500ms). Idempotent on the server
+        // side — replaying the same publish_identity for an existing DID is a
+        // no-op, so retries are safe even if the first attempt actually
+        // succeeded but the response was lost.
+        const delays = [0, 300, 1500];
+        let lastError;
+        for (const delay of delays) {
+            if (delay > 0)
+                await sleep(delay);
+            try {
+                const res = await this.#fetchImpl(url, {
+                    method: "POST",
+                    headers: { "content-type": "application/json" },
+                    body,
+                });
+                // Transport errors (5xx, no body) — retry. JSON-RPC errors come
+                // back with HTTP 200 and an `error` field.
+                if (!res.ok && res.status >= 500) {
+                    lastError = new Error(`HTTP ${res.status}`);
+                    continue;
+                }
+                const json = (await res.json());
+                if (json.error) {
+                    // JSON-RPC error: don't retry — these are deterministic
+                    // (validation, permission, identity-already-tombstoned, …).
+                    throw new AithosSDKError("ethos_bootstrap_failed", `publish_identity rejected: ${json.error.message}`, {
+                        status: res.status,
+                        data: { rpc_code: json.error.code, ...(json.error.data ?? {}) },
+                    });
+                }
+                return; // success
+            }
+            catch (e) {
+                if (e instanceof AithosSDKError)
+                    throw e;
+                lastError = e;
+            }
+        }
+        throw new AithosSDKError("ethos_bootstrap_failed", `publish_identity unreachable after ${delays.length} attempts: ${lastError?.message ?? "unknown"}`);
     }
 }
 /* -------------------------------------------------------------------------- */
@@ -328,16 +656,15 @@ export class AithosAuth {
 function trimSlash(url) {
     return url.endsWith("/") ? url.slice(0, -1) : url;
 }
+function sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
 function cleanCallbackParams(win, url) {
     url.searchParams.delete("aithos_code");
     url.searchParams.delete("aithos_error");
     url.searchParams.delete("app_state");
     win.history.replaceState(null, "", url.toString());
 }
-// Local copy of bytesToB64 — protocol-client exports it but we keep a
-// thin local wrapper to avoid a name collision with public surface and
-// to flag this is the "encode for the wire" path, not the "encode an
-// auth_key" path.
 function bytesToB64Public(bytes) {
     if (bytes.length === 0)
         return "";
@@ -346,10 +673,58 @@ function bytesToB64Public(bytes) {
         bin += String.fromCharCode(bytes[i]);
     return btoa(bin).replace(/=+$/, "");
 }
+function b64ToBytes(b64) {
+    if (!b64)
+        return new Uint8Array(0);
+    // standard b64 — pad to multiple of 4 if needed
+    const pad = b64.length % 4 === 0 ? "" : "=".repeat(4 - (b64.length % 4));
+    const bin = atob(b64 + pad);
+    const out = new Uint8Array(bin.length);
+    for (let i = 0; i < bin.length; i++)
+        out[i] = bin.charCodeAt(i);
+    return out;
+}
 function bytesToHex(b) {
     let out = "";
     for (let i = 0; i < b.length; i++)
         out += b[i].toString(16).padStart(2, "0");
     return out;
 }
+/**
+ * Project a delegate as it appears in a `BlobPlaintext` (extension-kit
+ * `StoredDelegate` shape) onto the SDK's own {@link StoredDelegateKeys}.
+ */
+function storedDelegateFromBlob(d) {
+    return {
+        version: "0.1.0-hex",
+        subjectDid: d.subjectDid,
+        mandateId: d.mandateId,
+        mandate: d.mandate,
+        granteeId: d.granteeId,
+        granteePubkeyMultibase: d.granteePubkeyMultibase,
+        delegateSeedHex: d.delegateSeedHex,
+        importedAt: new Date().toISOString(),
+    };
+}
+function actorToInfo(a) {
+    return {
+        mandateId: a.mandateId,
+        subjectDid: a.subjectDid,
+        granteeId: a.granteeId,
+        scopes: scopesFromMandate(a.mandate),
+        expiresAt: notAfterFromMandate(a.mandate),
+    };
+}
+function scopesFromMandate(m) {
+    const raw = m["scopes"];
+    if (!Array.isArray(raw))
+        return [];
+    return raw.filter((s) => typeof s === "string");
+}
+function notAfterFromMandate(m) {
+    const raw = m["not_after"];
+    if (typeof raw !== "string")
+        return null;
+    return raw;
+}
 //# sourceMappingURL=auth.js.map