@metamask-previews/assets-controller 0.2.0-preview-c3cd77f → 0.2.0-preview-d01b2f93d

package/dist/data-sources/SnapDataSource.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"SnapDataSource.cjs","sourceRoot":"","sources":["../../src/data-sources/SnapDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AACA,uEAA8D;AAc9D,uDAAoE;AAGpE,iEAA0D;AAK1D,0CAA8D;AAqC9D,MAAM,GAAG,GAAG,IAAA,2BAAkB,EAAC,sBAAa,EAAE,gBAAgB,CAAC,CAAC;AAEhE,+EAA+E;AAC/E,YAAY;AACZ,+EAA+E;AAElE,QAAA,qBAAqB,GAAG,gBAAgB,CAAC;AAEtD,qDAAqD;AACxC,QAAA,kBAAkB,GAAG,mBAAmB,CAAC;AAEtD,wEAAwE;AAC3D,QAAA,iBAAiB,GAAG,kBAAkB,CAAC;AAEpD,+EAA+E;AAC/E,uBAAuB;AACvB,+EAA+E;AAE/E;;;;;;;;;GASG;AACH,SAAgB,iBAAiB,CAC/B,UAAiC;IAEjC,IAAI,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,IAAI,CACpC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,KAAK,4BAAc,CAAC,QAAQ,CACpB,CAAC;IAE1C,OAAO,MAAM,CAAC,CAAC,CAAE,MAAM,CAAC,KAAmB,CAAC,CAAC,CAAC,IAAI,CAAC;AACrD,CAAC;AAZD,8CAYC;AAED;;;;;;GAMG;AACH,SAAgB,uBAAuB,CAAC,OAAe;IACrD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACjC,OAAO,KAAK,CAAC,CAAC,CAAY,CAAC;AAC7B,CAAC;AAHD,0DAGC;AAkBD,MAAM,gBAAgB,GAAwB;IAC5C,YAAY,EAAE,EAAE;IAChB,WAAW,EAAE,EAAE;CAChB,CAAC;AAkGF,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;GAgBG;AACH,MAAa,cAAe,SAAQ,uCAGnC;IAaC,YAAY,OAA8B;QACxC,KAAK,CAAC,6BAAqB,EAAE;YAC3B,GAAG,gBAAgB;YACnB,GAAG,OAAO,CAAC,KAAK;SACjB,CAAC,CAAC;;QAhBI,4CAAoC;QAE7C,yEAAyE;QAChE,iEAEC;QAED,mEAA8C;QAEvD,6EAA6E;QACpE,6CAAkD,IAAI,GAAG,EAAE,EAAC;QAQnE,uBAAA,IAAI,6BAAc,OAAO,CAAC,SAAS,MAAA,CAAC;QAEpC,yCAAyC;QACzC,uBAAA,IAAI,kDAAmC,uBAAA,IAAI,4EAA2B,CAAC,IAAI,CACzE,IAAI,CACoD,MAAA,CAAC;QAC3D,uBAAA,IAAI,oDACF,uBAAA,IAAI,uEAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAA,CAAC;QAExC,uBAAA,IAAI,yEAAwB,MAA5B,IAAI,CAA0B,CAAC;QAC/B,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAE1B,uEAAuE;QACvE,uBAAA,IAAI,uEAAsB,MAA1B,IAAI,CAAwB,CAAC;IAC/B,CAAC;IA0ND,+EAA+E;IAC/E,QAAQ;IACR,+EAA+E;IAE/E,KAAK,CAAC,KAAK,CAAC,OAAoB;;QAC9B,kCAAkC;QAClC,gFAAgF;QAChF,IAAI,CAAC,OAAO,EAAE,QAAQ,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;YACrD,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,OAAO,GAAiB;YAC5B,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,EAAE;SACnB,CAAC;QAEF,kEAAkE;QAClE,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACvC,0DAA0D;YAC1D,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC;YACzC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,SAAS;YACX,CAAC;YAED,uEAAuE;YACvE,MAAM,2BAA2B,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CACvD,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,MAAM,CACxD,CAAC;YACF,IAAI,CAAC,2BAA2B,EAAE,CAAC;gBACjC,SAAS;YACX,CAAC;YAED,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,uBAAA,IAAI,mEAAkB,MAAtB,IAAI,EAAmB,MAAM,CAAC,CAAC;gBAE9C,kDAAkD;gBAClD,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;gBAEhE,qCAAqC;gBACrC,IAAI,CAAC,aAAa,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACjD,SAAS;gBACX,CAAC;gBAED,iDAAiD;gBACjD,MAAM,QAAQ,GACZ,MAAM,MAAM,CAAC,kBAAkB,CAC7B,SAAS,EACT,aAAgC,CACjC,CAAC;gBAEJ,oDAAoD;gBACpD,IAAI,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;oBACtE,KAAK,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;wBAC1D,MAAA,OAAO,CAAC,aAAa,EAAC,SAAS,SAAT,SAAS,IAAM,EAAE,EAAC;wBACxC,MAAM,eAAe,GAAG,OAAO,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;wBACzD,IAAI,eAAe,EAAE,CAAC;4BACnB,eAA2C,CAAC,OAAO,CAAC,GAAG;gCACtD,MAAM,EAAE,OAAO,CAAC,MAAM;6BACvB,CAAC;wBACJ,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,oDAAoD;YACtD,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,+EAA+E;IAC/E,aAAa;IACb,+EAA+E;IAE/E;;;;;;;;;;OAUG;IACH,IAAI,gBAAgB;QAClB,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;;YAC7B,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;YAE5B,6CAA6C;YAC7C,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC1D,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC1C,CAAC;YAEF,2DAA2D;YAC3D,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjC,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,IAAI,yBAAyB,GAAc,EAAE,CAAC;YAE9C,IAAI,CAAC;gBACH,6BAA6B;gBAC7B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC;oBAChC,GAAG,OAAO;oBACV,QAAQ,EAAE,eAAe;iBAC1B,CAAC,CAAC;gBAEH,8BAA8B;gBAC9B,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;oBAC3B,MAAA,OAAO,CAAC,QAAQ,EAAC,aAAa,QAAb,aAAa,GAAK,EAAE,EAAC;oBACtC,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,QAAQ,CAAC,aAAa,CACvB,EAAE,CAAC;wBACF,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,SAAS,CAAC,GAAG;4BAC1C,GAAG,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,SAAS,CAAC;4BAC5C,GAAG,eAAe;yBACnB,CAAC;oBACJ,CAAC;gBACH,CAAC;gBACD,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;oBAC5B,OAAO,CAAC,QAAQ,CAAC,cAAc,GAAG;wBAChC,GAAG,OAAO,CAAC,QAAQ,CAAC,cAAc;wBAClC,GAAG,QAAQ,CAAC,cAAc;qBAC3B,CAAC;gBACJ,CAAC;gBACD,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;oBACzB,OAAO,CAAC,QAAQ,CAAC,WAAW,GAAG;wBAC7B,GAAG,OAAO,CAAC,QAAQ,CAAC,WAAW;wBAC/B,GAAG,QAAQ,CAAC,WAAW;qBACxB,CAAC;gBACJ,CAAC;gBAED,yDAAyD;gBACzD,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,CAAC;gBACjE,yBAAyB,GAAG,eAAe,CAAC,MAAM,CAChD,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,CACxC,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC1C,yBAAyB,GAAG,EAAE,CAAC;YACjC,CAAC;YAED,sCAAsC;YACtC,IAAI,WAAW,GAAG,OAAO,CAAC;YAC1B,IAAI,yBAAyB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzC,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAC7C,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,yBAAyB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC1D,CAAC;gBACF,WAAW,GAAG;oBACZ,GAAG,OAAO;oBACV,OAAO,EAAE;wBACP,GAAG,OAAO;wBACV,QAAQ,EAAE,eAAe;qBAC1B;iBACF,CAAC;YACJ,CAAC;YAED,uBAAuB;YACvB,OAAO,IAAI,CAAC,WAAW,CAAC,CAAC;QAC3B,CAAC,CAAC;IACJ,CAAC;IAED,+EAA+E;IAC/E,4CAA4C;IAC5C,+EAA+E;IAE/E,KAAK,CAAC,SAAS,CAAC,mBAAwC;QACtD,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC;QAElE,8CAA8C;QAC9C,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC;YACvB,OAAO;QACT,CAAC;QAED,sCAAsC;QACtC,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC1D,uBAAA,IAAI,yEAAwB,MAA5B,IAAI,EAAyB,OAAO,CAAC,CACtC,CAAC;QAEF,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO;QACT,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC9D,IAAI,QAAQ,EAAE,CAAC;gBACb,QAAQ,CAAC,MAAM,GAAG,eAAe,CAAC;gBAClC,uDAAuD;gBACvD,IAAI,CAAC,KAAK,CAAC;oBACT,GAAG,OAAO;oBACV,QAAQ,EAAE,eAAe;iBAC1B,CAAC;qBACC,IAAI,CAAC,KAAK,EAAE,aAAa,EAAE,EAAE;oBAC5B,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC9D,MAAM,uBAAA,IAAI,iCAAW,CAAC,IAAI,CACxB,+BAA+B,EAC/B,aAAa,EACb,6BAAqB,CACtB,CAAC;oBACJ,CAAC;oBACD,OAAO,aAAa,CAAC;gBACvB,CAAC,CAAC;qBACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBACf,GAAG,CAAC,kCAAkC,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;gBACrE,CAAC,CAAC,CAAC;gBACL,OAAO;YACT,CAAC;QACH,CAAC;QAED,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;QAEvC,gFAAgF;QAChF,iEAAiE;QACjE,sEAAsE;QAEtE,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,EAAE;YAC3C,OAAO,EAAE,GAAG,EAAE;gBACZ,iDAAiD;YACnD,CAAC;YACD,MAAM,EAAE,eAAe;SACxB,CAAC,CAAC;QAEH,wCAAwC;QACxC,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC;gBACrC,GAAG,OAAO;gBACV,QAAQ,EAAE,eAAe;aAC1B,CAAC,CAAC;YAEH,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9D,MAAM,uBAAA,IAAI,iCAAW,CAAC,IAAI,CACxB,+BAA+B,EAC/B,aAAa,EACb,6BAAqB,CACtB,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,sBAAsB,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAiCD,+EAA+E;IAC/E,UAAU;IACV,+EAA+E;IAE/E,OAAO;QACL,8DAA8D;QAC9D,MAAM,SAAS,GAAG,uBAAA,IAAI,iCAAkB,CAAC;QAEzC,uCAAuC;QACvC,IAAI,CAAC;YACH,SAAS,CAAC,WAAW,CACnB,2CAA2C,EAC3C,uBAAA,IAAI,sDAAgC,CACrC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,gDAAgD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,sCAAsC;QACtC,IAAI,CAAC;YACH,SAAS,CAAC,WAAW,CACnB,kCAAkC,EAClC,uBAAA,IAAI,wDAAkC,CACvC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,+CAA+C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAClE,CAAC;QAED,gCAAgC;QAChC,KAAK,MAAM,CAAC,cAAc,CAAC,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACxD,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;gBAC1C,wBAAwB;YAC1B,CAAC,CAAC,CAAC;QACL,CAAC;QAED,6BAA6B;QAC7B,uBAAA,IAAI,0CAAoB,CAAC,KAAK,EAAE,CAAC;IACnC,CAAC;CACF;AAtjBD,wCAsjBC;;IA3gBG,kEAAkE;IAClE,qEAAqE;IACrE,gDAAgD;IAChD,uBAAA,IAAI,iCAAW,CAAC,SAAS,CACvB,2CAA2C,EAC3C,uBAAA,IAAI,sDAAgC,CACrC,CAAC;IAEF,0EAA0E;IAC1E,kDAAkD;IAClD,uBAAA,IAAI,iCAAW,CAAC,SAAS,CACvB,kCAAkC,EAClC,uBAAA,IAAI,wDAAkC,CACvC,CAAC;AACJ,CAAC,iGASC,OAA2C;IAE3C,4DAA4D;IAC5D,IAAI,aAAqE,CAAC;IAE1E,KAAK,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACnE,IAAI,aAA8D,CAAC;QAEnE,KAAK,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACxD,MAAM,OAAO,GAAG,uBAAuB,CAAC,OAAO,CAAC,CAAC;YACjD,IAAI,uBAAA,IAAI,yEAAwB,MAA5B,IAAI,EAAyB,OAAO,CAAC,EAAE,CAAC;gBAC1C,aAAa,KAAb,aAAa,GAAK,EAAE,EAAC;gBACrB,aAAa,CAAC,OAAwB,CAAC,GAAG;oBACxC,MAAM,EAAE,OAAO,CAAC,MAAM;iBACvB,CAAC;YACJ,CAAC;QACH,CAAC;QAED,IAAI,aAAa,EAAE,CAAC;YAClB,aAAa,KAAb,aAAa,GAAK,EAAE,EAAC;YACrB,aAAa,CAAC,SAAS,CAAC,GAAG,aAAa,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,8CAA8C;IAC9C,IAAI,aAAa,EAAE,CAAC;QAClB,MAAM,QAAQ,GAAiB,EAAE,aAAa,EAAE,CAAC;QACjD,uBAAA,IAAI,iCAAW;aACZ,IAAI,CAAC,+BAA+B,EAAE,QAAQ,EAAE,6BAAqB,CAAC;aACtE,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;AACH,CAAC,2FAQuB,OAAgB;IACtC,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AACnD,CAAC;IAGC,8DAA8D;IAC9D,MAAM,SAAS,GAAG,uBAAA,IAAI,iCAAkB,CAAC;IAEzC,SAAS,CAAC,qBAAqB,CAC7B,oCAAoC,EACpC,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAC5B,CAAC;IAEF,SAAS,CAAC,qBAAqB,CAC7B,gCAAgC,EAChC,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,CACnC,CAAC;IAEF,SAAS,CAAC,qBAAqB,CAC7B,sBAAsB,EACtB,KAAK,EAAE,OAAoB,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CACpD,CAAC;IAEF,SAAS,CAAC,qBAAqB,CAC7B,0BAA0B,EAC1B,KAAK,EAAE,OAA4B,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAChE,CAAC;IAEF,SAAS,CAAC,qBAAqB,CAC7B,4BAA4B,EAC5B,KAAK,EAAE,cAAsB,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CACnE,CAAC;AACJ,CAAC;IAaC,IAAI,CAAC;QACH,OAAO,uBAAA,IAAI,iCAAW,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;IACjE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;QAC3C,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC,mFASC,MAAc;IAEd,IAAI,CAAC;QACH,OAAO,uBAAA,IAAI,iCAAW,CAAC,IAAI,CACzB,qCAAqC,EACrC,MAAM,CACqC,CAAC;IAChD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,oCAAoC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7D,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;IAkBC,IAAI,CAAC;QACH,MAAM,aAAa,GAAG,uBAAA,IAAI,mEAAkB,MAAtB,IAAI,CAAoB,CAAC;QAC/C,MAAM,WAAW,GAA4B,EAAE,CAAC;QAChD,MAAM,eAAe,GAAc,EAAE,CAAC;QAEtC,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;YACjC,MAAM,WAAW,GAAG,uBAAA,IAAI,qEAAoB,MAAxB,IAAI,EAAqB,IAAI,CAAC,EAAE,CAAC,CAAC;YACtD,8DAA8D;YAC9D,IAAI,CAAC,WAAW,EAAE,CAAC,0BAAkB,CAAC,EAAE,CAAC;gBACvC,SAAS;YACX,CAAC;YAED,0EAA0E;YAC1E,8CAA8C;YAC9C,MAAM,gBAAgB,GAAG,WAAW,CAAC,yBAAiB,CAAC,CAAC;YACxD,MAAM,QAAQ,GAAG,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;YAErD,+EAA+E;YAC/E,IAAI,QAAQ,EAAE,CAAC;gBACb,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;oBAC/B,IAAI,CAAC,CAAC,OAAO,IAAI,WAAW,CAAC,EAAE,CAAC;wBAC9B,WAAW,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,EAAE,CAAC;wBAC/B,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBAChC,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,WAAW,CAAC;QAErC,2BAA2B;QAC3B,IAAI,CAAC;YACH,IAAI,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC,aAAa,EAAE,EAAE;gBACzD,uBAAA,IAAI,iCAAW,CAAC,IAAI,CAClB,qCAAqC,EACrC,6BAAqB,EACrB,aAAa,CACd,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,kEAAkE;QACpE,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,+BAA+B,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAChD,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,EAAE,CAAC;QAC5B,IAAI,CAAC;YACH,IAAI,CAAC,kBAAkB,CAAC,EAAE,EAAE,CAAC,aAAa,EAAE,EAAE;gBAC5C,uBAAA,IAAI,iCAAW,CAAC,IAAI,CAClB,qCAAqC,EACrC,6BAAqB,EACrB,aAAa,CACd,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,kEAAkE;QACpE,CAAC;IACH,CAAC;AACH,CAAC,+EAgQiB,MAAc;IAC9B,MAAM,YAAY,GAAG,uBAAA,IAAI,0CAAoB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC1D,IAAI,YAAY,EAAE,CAAC;QACjB,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,mCAAa,CAAC;QAC/B,IAAI,EAAE,KAAK,EAAE,OAAuB,EAAiB,EAAE,CACrD,CAAC,MAAM,uBAAA,IAAI,iCAAW,CAAC,IAAI,CAAC,8BAA8B,EAAE;YAC1D,MAAM,EAAE,MAAgB;YACxB,MAAM,EAAE,UAAU;YAClB,OAAO,EAAE,yBAAW,CAAC,gBAAgB;YACrC,OAAO;SACR,CAAC,CAAS;KACd,CAAC,CAAC;IAEH,uBAAA,IAAI,0CAAoB,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7C,OAAO,MAAM,CAAC;AAChB,CAAC;AA0CH,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E,SAAgB,oBAAoB,CAClC,OAA8B;IAE9B,OAAO,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;AACrC,CAAC;AAJD,oDAIC","sourcesContent":["import type { Balance, CaipAssetType } from '@metamask/keyring-api';\nimport { KeyringClient } from '@metamask/keyring-snap-client';\nimport type { Messenger } from '@metamask/messenger';\nimport type {\n  Caveat,\n  GetPermissions,\n  PermissionConstraint,\n  PermissionControllerStateChange,\n  SubjectPermissions,\n} from '@metamask/permission-controller';\nimport type {\n  GetRunnableSnaps,\n  HandleSnapRequest,\n} from '@metamask/snaps-controllers';\nimport type { Snap, SnapId } from '@metamask/snaps-sdk';\nimport { HandlerType, SnapCaveatType } from '@metamask/snaps-utils';\nimport type { Json, JsonRpcRequest } from '@metamask/utils';\n\nimport { AbstractDataSource } from './AbstractDataSource';\nimport type {\n  DataSourceState,\n  SubscriptionRequest,\n} from './AbstractDataSource';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport type {\n  AssetBalance,\n  ChainId,\n  Caip19AssetId,\n  DataRequest,\n  DataResponse,\n  Middleware,\n} from '../types';\n\n// ============================================================================\n// SNAP KEYRING EVENT TYPES\n// ============================================================================\n\n/**\n * Payload for AccountsController:accountBalancesUpdated event.\n * Re-published from SnapKeyring:accountBalancesUpdated.\n */\nexport type AccountBalancesUpdatedEventPayload = {\n  balances: {\n    [accountId: string]: {\n      [assetId: string]: {\n        amount: string;\n        unit: string;\n      };\n    };\n  };\n};\n\n/**\n * Event from AccountsController when snap balances are updated.\n */\nexport type AccountsControllerAccountBalancesUpdatedEvent = {\n  type: 'AccountsController:accountBalancesUpdated';\n  payload: [AccountBalancesUpdatedEventPayload];\n};\n\nconst log = createModuleLogger(projectLogger, 'SnapDataSource');\n\n// ============================================================================\n// CONSTANTS\n// ============================================================================\n\nexport const SNAP_DATA_SOURCE_NAME = 'SnapDataSource';\n\n/** The permission name for snap keyring endowment */\nexport const KEYRING_PERMISSION = 'endowment:keyring';\n\n/** The permission name for snap assets endowment (contains chainIds) */\nexport const ASSETS_PERMISSION = 'endowment:assets';\n\n// ============================================================================\n// PERMISSION UTILITIES\n// ============================================================================\n\n/**\n * Getter function to get the chainIds caveat from a permission.\n *\n * This does basic validation of the caveat, but does not validate the type or\n * value of the namespaces object itself, as this is handled by the\n * `PermissionsController` when the permission is requested.\n *\n * @param permission - The permission to get the `chainIds` caveat from.\n * @returns An array of `chainIds` that the snap supports, or null if none.\n */\nexport function getChainIdsCaveat(\n  permission?: PermissionConstraint,\n): ChainId[] | null {\n  if (!permission?.caveats) {\n    return null;\n  }\n\n  const caveat = permission.caveats.find(\n    (permCaveat) => permCaveat.type === SnapCaveatType.ChainIds,\n  ) as Caveat<string, string[]> | undefined;\n\n  return caveat ? (caveat.value as ChainId[]) : null;\n}\n\n/**\n * Extract chain ID from a CAIP-19 asset ID.\n * e.g., \"solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp/slip44:501\" -> \"solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp\"\n *\n * @param assetId - The CAIP-19 asset ID to extract chain from.\n * @returns The CAIP-2 chain ID portion of the asset ID.\n */\nexport function extractChainFromAssetId(assetId: string): ChainId {\n  const parts = assetId.split('/');\n  return parts[0] as ChainId;\n}\n\n// ============================================================================\n// STATE\n// ============================================================================\n\n/**\n * State for the SnapDataSource.\n * Uses dynamic snap discovery - chains are populated from PermissionController.\n */\nexport type SnapDataSourceState = {\n  /**\n   * Mapping of chain IDs to snap IDs that support them.\n   * Used to filter which accounts to process for a given chain request.\n   */\n  chainToSnap: Record<ChainId, string>;\n} & DataSourceState;\n\nconst defaultSnapState: SnapDataSourceState = {\n  activeChains: [],\n  chainToSnap: {},\n};\n\n// ============================================================================\n// MESSENGER TYPES\n// ============================================================================\n\nexport type SnapDataSourceGetAssetsMiddlewareAction = {\n  type: 'SnapDataSource:getAssetsMiddleware';\n  handler: () => Middleware;\n};\n\nexport type SnapDataSourceGetActiveChainsAction = {\n  type: 'SnapDataSource:getActiveChains';\n  handler: () => Promise<ChainId[]>;\n};\n\nexport type SnapDataSourceFetchAction = {\n  type: 'SnapDataSource:fetch';\n  handler: (request: DataRequest) => Promise<DataResponse>;\n};\n\nexport type SnapDataSourceSubscribeAction = {\n  type: 'SnapDataSource:subscribe';\n  handler: (request: SubscriptionRequest) => Promise<void>;\n};\n\nexport type SnapDataSourceUnsubscribeAction = {\n  type: 'SnapDataSource:unsubscribe';\n  handler: (subscriptionId: string) => Promise<void>;\n};\n\nexport type SnapDataSourceActions =\n  | SnapDataSourceGetAssetsMiddlewareAction\n  | SnapDataSourceGetActiveChainsAction\n  | SnapDataSourceFetchAction\n  | SnapDataSourceSubscribeAction\n  | SnapDataSourceUnsubscribeAction;\n\nexport type SnapDataSourceActiveChainsChangedEvent = {\n  type: 'SnapDataSource:activeChainsUpdated';\n  payload: [ChainId[]];\n};\n\nexport type SnapDataSourceAssetsUpdatedEvent = {\n  type: 'SnapDataSource:assetsUpdated';\n  payload: [DataResponse, string | undefined];\n};\n\nexport type SnapDataSourceEvents =\n  | SnapDataSourceActiveChainsChangedEvent\n  | SnapDataSourceAssetsUpdatedEvent;\n\n/**\n * Allowed events that SnapDataSource can subscribe to.\n */\nexport type SnapDataSourceAllowedEvents =\n  | AccountsControllerAccountBalancesUpdatedEvent\n  | PermissionControllerStateChange;\n\n// Actions to report to AssetsController\ntype AssetsControllerActiveChainsUpdateAction = {\n  type: 'AssetsController:activeChainsUpdate';\n  handler: (dataSourceId: string, activeChains: ChainId[]) => void;\n};\n\ntype AssetsControllerAssetsUpdateAction = {\n  type: 'AssetsController:assetsUpdate';\n  handler: (response: DataResponse, sourceId: string) => Promise<void>;\n};\n\nexport type SnapDataSourceAllowedActions =\n  | AssetsControllerActiveChainsUpdateAction\n  | AssetsControllerAssetsUpdateAction\n  | GetRunnableSnaps\n  | HandleSnapRequest\n  | GetPermissions;\n\nexport type SnapDataSourceMessenger = Messenger<\n  typeof SNAP_DATA_SOURCE_NAME,\n  SnapDataSourceActions | SnapDataSourceAllowedActions,\n  SnapDataSourceEvents | SnapDataSourceAllowedEvents\n>;\n\n// ============================================================================\n// OPTIONS\n// ============================================================================\n\nexport type SnapDataSourceOptions = {\n  /** Messenger for this data source */\n  messenger: SnapDataSourceMessenger;\n  /** Configured networks to support (defaults to all snap networks) */\n  configuredNetworks?: ChainId[];\n  /** Default polling interval in ms for subscriptions */\n  pollInterval?: number;\n  /** Initial state */\n  state?: Partial<SnapDataSourceState>;\n};\n\n// ============================================================================\n// SNAP DATA SOURCE\n// ============================================================================\n\n/**\n * Unified Snap data source that routes requests to the appropriate wallet snap\n * based on the chain ID prefix.\n *\n * @example\n * ```typescript\n * const snapDataSource = new SnapDataSource({\n *   messenger,\n * });\n *\n * // Fetch will automatically route to the correct snap\n * await snapDataSource.fetch({\n *   chainIds: ['solana:mainnet', 'bip122:000000000019d6689c085ae165831e93'],\n *   accountIds: ['account1'],\n * });\n * ```\n */\nexport class SnapDataSource extends AbstractDataSource<\n  typeof SNAP_DATA_SOURCE_NAME,\n  SnapDataSourceState\n> {\n  readonly #messenger: SnapDataSourceMessenger;\n\n  /** Bound handler for snap keyring balance updates, stored for cleanup */\n  readonly #handleSnapBalancesUpdatedBound: (\n    payload: AccountBalancesUpdatedEventPayload,\n  ) => void;\n\n  readonly #handlePermissionStateChangeBound: () => void;\n\n  /** Cache of KeyringClient instances per snap ID to avoid re-instantiation */\n  readonly #keyringClientCache: Map<string, KeyringClient> = new Map();\n\n  constructor(options: SnapDataSourceOptions) {\n    super(SNAP_DATA_SOURCE_NAME, {\n      ...defaultSnapState,\n      ...options.state,\n    });\n\n    this.#messenger = options.messenger;\n\n    // Bind handlers for cleanup in destroy()\n    this.#handleSnapBalancesUpdatedBound = this.#handleSnapBalancesUpdated.bind(\n      this,\n    ) as (payload: AccountBalancesUpdatedEventPayload) => void;\n    this.#handlePermissionStateChangeBound =\n      this.#discoverKeyringSnaps.bind(this);\n\n    this.#registerActionHandlers();\n    this.#subscribeToEvents();\n\n    // Discover keyring-capable snaps and populate activeChains dynamically\n    this.#discoverKeyringSnaps();\n  }\n\n  /**\n   * Subscribe to all events needed by SnapDataSource.\n   * Groups snap keyring events and permission change events.\n   */\n  #subscribeToEvents(): void {\n    // Subscribe to snap keyring events for real-time balance updates.\n    // The snaps emit AccountBalancesUpdated events when balances change,\n    // which are re-published by AccountsController.\n    this.#messenger.subscribe(\n      'AccountsController:accountBalancesUpdated',\n      this.#handleSnapBalancesUpdatedBound,\n    );\n\n    // Subscribe to permission changes to detect new keyring snaps at runtime.\n    // Re-runs snap discovery when permissions change.\n    this.#messenger.subscribe(\n      'PermissionController:stateChange',\n      this.#handlePermissionStateChangeBound,\n    );\n  }\n\n  /**\n   * Handle snap balance updates from the keyring.\n   * Transforms the payload and publishes to AssetsController.\n   *\n   * @param payload - The balance update payload from AccountsController.\n   */\n  #handleSnapBalancesUpdated(\n    payload: AccountBalancesUpdatedEventPayload,\n  ): void {\n    // Transform the snap keyring payload to DataResponse format\n    let assetsBalance: NonNullable<DataResponse['assetsBalance']> | undefined;\n\n    for (const [accountId, assets] of Object.entries(payload.balances)) {\n      let accountAssets: Record<Caip19AssetId, AssetBalance> | undefined;\n\n      for (const [assetId, balance] of Object.entries(assets)) {\n        const chainId = extractChainFromAssetId(assetId);\n        if (this.#isChainSupportedBySnap(chainId)) {\n          accountAssets ??= {};\n          accountAssets[assetId as Caip19AssetId] = {\n            amount: balance.amount,\n          };\n        }\n      }\n\n      if (accountAssets) {\n        assetsBalance ??= {};\n        assetsBalance[accountId] = accountAssets;\n      }\n    }\n\n    // Only report if we have snap-related updates\n    if (assetsBalance) {\n      const response: DataResponse = { assetsBalance };\n      this.#messenger\n        .call('AssetsController:assetsUpdate', response, SNAP_DATA_SOURCE_NAME)\n        .catch(console.error);\n    }\n  }\n\n  /**\n   * Check if a chain ID is supported by any discovered snap.\n   *\n   * @param chainId - The CAIP-2 chain ID to check.\n   * @returns True if we have a snap that supports this chain.\n   */\n  #isChainSupportedBySnap(chainId: ChainId): boolean {\n    return this.state.activeChains.includes(chainId);\n  }\n\n  #registerActionHandlers(): void {\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const messenger = this.#messenger as any;\n\n    messenger.registerActionHandler(\n      'SnapDataSource:getAssetsMiddleware',\n      () => this.assetsMiddleware,\n    );\n\n    messenger.registerActionHandler(\n      'SnapDataSource:getActiveChains',\n      async () => this.getActiveChains(),\n    );\n\n    messenger.registerActionHandler(\n      'SnapDataSource:fetch',\n      async (request: DataRequest) => this.fetch(request),\n    );\n\n    messenger.registerActionHandler(\n      'SnapDataSource:subscribe',\n      async (request: SubscriptionRequest) => this.subscribe(request),\n    );\n\n    messenger.registerActionHandler(\n      'SnapDataSource:unsubscribe',\n      async (subscriptionId: string) => this.unsubscribe(subscriptionId),\n    );\n  }\n\n  // ============================================================================\n  // SNAP DISCOVERY (Dynamic via PermissionController)\n  // ============================================================================\n\n  /**\n   * Get all runnable snaps from SnapController.\n   * Runnable snaps are enabled and not blocked.\n   *\n   * @returns Array of runnable snaps.\n   */\n  #getRunnableSnaps(): Snap[] {\n    try {\n      return this.#messenger.call('SnapController:getRunnableSnaps');\n    } catch (error) {\n      log('Failed to get runnable snaps', error);\n      return [];\n    }\n  }\n\n  /**\n   * Get permissions for a snap from PermissionController.\n   *\n   * @param snapId - The snap ID to get permissions for.\n   * @returns The snap's permissions, or undefined if none.\n   */\n  #getSnapPermissions(\n    snapId: string,\n  ): SubjectPermissions<PermissionConstraint> | undefined {\n    try {\n      return this.#messenger.call(\n        'PermissionController:getPermissions',\n        snapId,\n      ) as SubjectPermissions<PermissionConstraint>;\n    } catch (error) {\n      log('Failed to get permissions for snap', { snapId, error });\n      return undefined;\n    }\n  }\n\n  /**\n   * Discover all snaps with keyring capabilities and their supported chains.\n   * Uses PermissionController to find snaps with endowment:keyring permission.\n   * Updates chainToSnap mapping and activeChains.\n   *\n   * Called on initialization and whenever PermissionController state changes\n   * (e.g., new snaps installed, permissions granted/revoked).\n   *\n   * @remarks\n   * **Known limitation:** If discovery fails (e.g., SnapController not ready),\n   * the data source continues with empty chainToSnap. This means no snap\n   * chains will be supported until a re-discovery is triggered by a permission\n   * change. Callers should be aware that initialization may complete with no\n   * active chains.\n   */\n  #discoverKeyringSnaps(): void {\n    try {\n      const runnableSnaps = this.#getRunnableSnaps();\n      const chainToSnap: Record<ChainId, string> = {};\n      const supportedChains: ChainId[] = [];\n\n      for (const snap of runnableSnaps) {\n        const permissions = this.#getSnapPermissions(snap.id);\n        // Must have endowment:keyring permission to be a keyring snap\n        if (!permissions?.[KEYRING_PERMISSION]) {\n          continue;\n        }\n\n        // Get chainIds caveat from the assets permission (not keyring permission)\n        // The chainIds are stored in endowment:assets\n        const assetsPermission = permissions[ASSETS_PERMISSION];\n        const chainIds = getChainIdsCaveat(assetsPermission);\n\n        // Map each chain to this snap (first snap wins if multiple support same chain)\n        if (chainIds) {\n          for (const chainId of chainIds) {\n            if (!(chainId in chainToSnap)) {\n              chainToSnap[chainId] = snap.id;\n              supportedChains.push(chainId);\n            }\n          }\n        }\n      }\n\n      // Update chainToSnap mapping\n      this.state.chainToSnap = chainToSnap;\n\n      // Notify if chains changed\n      try {\n        this.updateActiveChains(supportedChains, (updatedChains) => {\n          this.#messenger.call(\n            'AssetsController:activeChainsUpdate',\n            SNAP_DATA_SOURCE_NAME,\n            updatedChains,\n          );\n        });\n      } catch {\n        // AssetsController not ready yet - expected during initialization\n      }\n    } catch (error) {\n      log('Keyring snap discovery failed', { error });\n      this.state.chainToSnap = {};\n      try {\n        this.updateActiveChains([], (updatedChains) => {\n          this.#messenger.call(\n            'AssetsController:activeChainsUpdate',\n            SNAP_DATA_SOURCE_NAME,\n            updatedChains,\n          );\n        });\n      } catch {\n        // AssetsController not ready yet - expected during initialization\n      }\n    }\n  }\n\n  // ============================================================================\n  // FETCH\n  // ============================================================================\n\n  async fetch(request: DataRequest): Promise<DataResponse> {\n    // Guard against undefined request\n    // Note: chainIds filtering is done by middleware/subscribe before calling fetch\n    if (!request?.accounts || !request?.chainIds?.length) {\n      return {};\n    }\n\n    const results: DataResponse = {\n      assetsBalance: {},\n      assetsMetadata: {},\n    };\n\n    // Fetch balances for each account using its snap ID from metadata\n    for (const account of request.accounts) {\n      // Skip accounts without snap metadata (non-snap accounts)\n      const snapId = account.metadata.snap?.id;\n      if (!snapId) {\n        continue;\n      }\n\n      // Skip accounts whose snap doesn't support any of the requested chains\n      const snapSupportsRequestedChains = request.chainIds.some(\n        (chainId) => this.state.chainToSnap[chainId] === snapId,\n      );\n      if (!snapSupportsRequestedChains) {\n        continue;\n      }\n\n      const accountId = account.id;\n      try {\n        const client = this.#getKeyringClient(snapId);\n\n        // Step 1: Get the list of assets for this account\n        const accountAssets = await client.listAccountAssets(accountId);\n\n        // If no assets, skip to next account\n        if (!accountAssets || accountAssets.length === 0) {\n          continue;\n        }\n\n        // Step 2: Get balances for those specific assets\n        const balances: Record<CaipAssetType, Balance> =\n          await client.getAccountBalances(\n            accountId,\n            accountAssets as CaipAssetType[],\n          );\n\n        // Transform keyring response to DataResponse format\n        if (balances && typeof balances === 'object' && results.assetsBalance) {\n          for (const [assetId, balance] of Object.entries(balances)) {\n            results.assetsBalance[accountId] ??= {};\n            const accountBalances = results.assetsBalance[accountId];\n            if (accountBalances) {\n              (accountBalances as Record<string, unknown>)[assetId] = {\n                amount: balance.amount,\n              };\n            }\n          }\n        }\n      } catch {\n        // Expected when account doesn't belong to this snap\n      }\n    }\n\n    return results;\n  }\n\n  // ============================================================================\n  // MIDDLEWARE\n  // ============================================================================\n\n  /**\n   * Get the middleware for fetching balances via Snaps.\n   * This middleware:\n   * - Supports multiple accounts in a single request\n   * - Filters request to only chains this data source supports\n   * - Fetches balances for those chains for all accounts\n   * - Merges response into context\n   * - Removes handled chains from request for next middleware\n   *\n   * @returns The middleware function for the assets pipeline.\n   */\n  get assetsMiddleware(): Middleware {\n    return async (context, next) => {\n      const { request } = context;\n\n      // Filter to chains this data source supports\n      const supportedChains = request.chainIds.filter((chainId) =>\n        this.state.activeChains.includes(chainId),\n      );\n\n      // If no supported chains, skip and pass to next middleware\n      if (supportedChains.length === 0) {\n        return next(context);\n      }\n\n      let successfullyHandledChains: ChainId[] = [];\n\n      try {\n        // Fetch for supported chains\n        const response = await this.fetch({\n          ...request,\n          chainIds: supportedChains,\n        });\n\n        // Merge response into context\n        if (response.assetsBalance) {\n          context.response.assetsBalance ??= {};\n          for (const [accountId, accountBalances] of Object.entries(\n            response.assetsBalance,\n          )) {\n            context.response.assetsBalance[accountId] = {\n              ...context.response.assetsBalance[accountId],\n              ...accountBalances,\n            };\n          }\n        }\n        if (response.assetsMetadata) {\n          context.response.assetsMetadata = {\n            ...context.response.assetsMetadata,\n            ...response.assetsMetadata,\n          };\n        }\n        if (response.assetsPrice) {\n          context.response.assetsPrice = {\n            ...context.response.assetsPrice,\n            ...response.assetsPrice,\n          };\n        }\n\n        // Determine successfully handled chains (exclude errors)\n        const failedChains = new Set(Object.keys(response.errors ?? {}));\n        successfullyHandledChains = supportedChains.filter(\n          (chainId) => !failedChains.has(chainId),\n        );\n      } catch (error) {\n        log('Middleware fetch failed', { error });\n        successfullyHandledChains = [];\n      }\n\n      // Prepare context for next middleware\n      let nextContext = context;\n      if (successfullyHandledChains.length > 0) {\n        const remainingChains = request.chainIds.filter(\n          (chainId) => !successfullyHandledChains.includes(chainId),\n        );\n        nextContext = {\n          ...context,\n          request: {\n            ...request,\n            chainIds: remainingChains,\n          },\n        };\n      }\n\n      // Call next middleware\n      return next(nextContext);\n    };\n  }\n\n  // ============================================================================\n  // SUBSCRIBE - Routes to appropriate snap(s)\n  // ============================================================================\n\n  async subscribe(subscriptionRequest: SubscriptionRequest): Promise<void> {\n    const { request, subscriptionId, isUpdate } = subscriptionRequest;\n\n    // Guard against undefined request or chainIds\n    if (!request?.chainIds) {\n      return;\n    }\n\n    // Filter to chains we have a snap for\n    const supportedChains = request.chainIds.filter((chainId) =>\n      this.#isChainSupportedBySnap(chainId),\n    );\n\n    if (supportedChains.length === 0) {\n      return;\n    }\n\n    if (isUpdate) {\n      const existing = this.activeSubscriptions.get(subscriptionId);\n      if (existing) {\n        existing.chains = supportedChains;\n        // Do a fetch to get latest data on subscription update\n        this.fetch({\n          ...request,\n          chainIds: supportedChains,\n        })\n          .then(async (fetchResponse) => {\n            if (Object.keys(fetchResponse.assetsBalance ?? {}).length > 0) {\n              await this.#messenger.call(\n                'AssetsController:assetsUpdate',\n                fetchResponse,\n                SNAP_DATA_SOURCE_NAME,\n              );\n            }\n            return fetchResponse;\n          })\n          .catch((error) => {\n            log('Subscription update fetch failed', { subscriptionId, error });\n          });\n        return;\n      }\n    }\n\n    await this.unsubscribe(subscriptionId);\n\n    // Snaps provide real-time updates via AccountsController:accountBalancesUpdated\n    // We only need to track the subscription and do an initial fetch\n    // No polling needed - updates come through #handleSnapBalancesUpdated\n\n    this.activeSubscriptions.set(subscriptionId, {\n      cleanup: () => {\n        // No timer to clear - we use event-based updates\n      },\n      chains: supportedChains,\n    });\n\n    // Initial fetch to get current balances\n    try {\n      const fetchResponse = await this.fetch({\n        ...request,\n        chainIds: supportedChains,\n      });\n\n      if (Object.keys(fetchResponse.assetsBalance ?? {}).length > 0) {\n        await this.#messenger.call(\n          'AssetsController:assetsUpdate',\n          fetchResponse,\n          SNAP_DATA_SOURCE_NAME,\n        );\n      }\n    } catch (error) {\n      log('Initial fetch failed', { subscriptionId, error });\n    }\n  }\n\n  // ============================================================================\n  // KEYRING CLIENT\n  // ============================================================================\n\n  /**\n   * Gets a `KeyringClient` for a Snap.\n   * Caches clients per snap ID to avoid re-instantiation across multiple calls.\n   *\n   * @param snapId - ID of the Snap to get the client for.\n   * @returns A `KeyringClient` for the Snap.\n   */\n  #getKeyringClient(snapId: string): KeyringClient {\n    const cachedClient = this.#keyringClientCache.get(snapId);\n    if (cachedClient) {\n      return cachedClient;\n    }\n\n    const client = new KeyringClient({\n      send: async (request: JsonRpcRequest): Promise<Json> =>\n        (await this.#messenger.call('SnapController:handleRequest', {\n          snapId: snapId as SnapId,\n          origin: 'metamask',\n          handler: HandlerType.OnKeyringRequest,\n          request,\n        })) as Json,\n    });\n\n    this.#keyringClientCache.set(snapId, client);\n    return client;\n  }\n\n  // ============================================================================\n  // CLEANUP\n  // ============================================================================\n\n  destroy(): void {\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const messenger = this.#messenger as any;\n\n    // Unsubscribe from snap keyring events\n    try {\n      messenger.unsubscribe(\n        'AccountsController:accountBalancesUpdated',\n        this.#handleSnapBalancesUpdatedBound,\n      );\n    } catch (error) {\n      log('Failed to unsubscribe from snap keyring events', { error });\n    }\n\n    // Unsubscribe from permission changes\n    try {\n      messenger.unsubscribe(\n        'PermissionController:stateChange',\n        this.#handlePermissionStateChangeBound,\n      );\n    } catch (error) {\n      log('Failed to unsubscribe from permission changes', { error });\n    }\n\n    // Clean up active subscriptions\n    for (const [subscriptionId] of this.activeSubscriptions) {\n      this.unsubscribe(subscriptionId).catch(() => {\n        // Ignore cleanup errors\n      });\n    }\n\n    // Clear keyring client cache\n    this.#keyringClientCache.clear();\n  }\n}\n\n// ============================================================================\n// FACTORY\n// ============================================================================\n\nexport function createSnapDataSource(\n  options: SnapDataSourceOptions,\n): SnapDataSource {\n  return new SnapDataSource(options);\n}\n"]}
+{"version":3,"file":"SnapDataSource.cjs","sourceRoot":"","sources":["../../src/data-sources/SnapDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AACA,uEAA8D;AAa9D,uDAAoE;AAGpE,iEAA0D;AAM1D,0CAA8D;AAqC9D,MAAM,GAAG,GAAG,IAAA,2BAAkB,EAAC,sBAAa,EAAE,gBAAgB,CAAC,CAAC;AAEhE,+EAA+E;AAC/E,YAAY;AACZ,+EAA+E;AAElE,QAAA,qBAAqB,GAAG,gBAAgB,CAAC;AAEtD,qDAAqD;AACxC,QAAA,kBAAkB,GAAG,mBAAmB,CAAC;AAEtD,wEAAwE;AAC3D,QAAA,iBAAiB,GAAG,kBAAkB,CAAC;AAEpD,+EAA+E;AAC/E,uBAAuB;AACvB,+EAA+E;AAE/E;;;;;;;;;GASG;AACH,SAAgB,iBAAiB,CAC/B,UAAiC;IAEjC,IAAI,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,IAAI,CACpC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,KAAK,4BAAc,CAAC,QAAQ,CACpB,CAAC;IAE1C,OAAO,MAAM,CAAC,CAAC,CAAE,MAAM,CAAC,KAAmB,CAAC,CAAC,CAAC,IAAI,CAAC;AACrD,CAAC;AAZD,8CAYC;AAED;;;;;;GAMG;AACH,SAAgB,uBAAuB,CAAC,OAAe;IACrD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACjC,OAAO,KAAK,CAAC,CAAC,CAAY,CAAC;AAC7B,CAAC;AAHD,0DAGC;AAkBD,MAAM,gBAAgB,GAAwB;IAC5C,YAAY,EAAE,EAAE;IAChB,WAAW,EAAE,EAAE;CAChB,CAAC;AAmCF,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAa,cAAe,SAAQ,uCAGnC;IAeC,YAAY,OAA8B;QACxC,KAAK,CAAC,6BAAqB,EAAE;YAC3B,GAAG,gBAAgB;YACnB,GAAG,OAAO,CAAC,KAAK;SACjB,CAAC,CAAC;;QAlBI,4CAAsC;QAEtC,wDAAoD;QAE7D,yEAAyE;QAChE,iEAEC;QAED,mEAA8C;QAEvD,6EAA6E;QACpE,6CAAkD,IAAI,GAAG,EAAE,EAAC;QAQnE,uBAAA,IAAI,6BAAc,OAAO,CAAC,SAAS,MAAA,CAAC;QACpC,uBAAA,IAAI,yCAA0B,OAAO,CAAC,qBAAqB,MAAA,CAAC;QAE5D,yCAAyC;QACzC,uBAAA,IAAI,kDAAmC,uBAAA,IAAI,4EAA2B,CAAC,IAAI,CACzE,IAAI,CACoD,MAAA,CAAC;QAC3D,uBAAA,IAAI,oDACF,uBAAA,IAAI,uEAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAA,CAAC;QAExC,uBAAA,IAAI,oEAAmB,MAAvB,IAAI,CAAqB,CAAC;QAE1B,uEAAuE;QACvE,uBAAA,IAAI,uEAAsB,MAA1B,IAAI,CAAwB,CAAC;IAC/B,CAAC;IAqLD,+EAA+E;IAC/E,QAAQ;IACR,+EAA+E;IAE/E,KAAK,CAAC,KAAK,CAAC,OAAoB;;QAC9B,kCAAkC;QAClC,gFAAgF;QAChF,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;YAC/B,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,IAAI,CAAC,OAAO,EAAE,2BAA2B,EAAE,MAAM,EAAE,CAAC;YAClD,OAAO,EAAE,aAAa,EAAE,EAAE,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC;QACnD,CAAC;QAED,MAAM,OAAO,GAAiB;YAC5B,aAAa,EAAE,EAAE;YACjB,cAAc,EAAE,EAAE;SACnB,CAAC;QAEF,kEAAkE;QAClE,KAAK,MAAM,EAAE,OAAO,EAAE,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;YAC9D,0DAA0D;YAC1D,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC;YACzC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,SAAS;YACX,CAAC;YAED,uEAAuE;YACvE,MAAM,2BAA2B,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CACvD,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,MAAM,CACxD,CAAC;YACF,IAAI,CAAC,2BAA2B,EAAE,CAAC;gBACjC,SAAS;YACX,CAAC;YAED,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,uBAAA,IAAI,mEAAkB,MAAtB,IAAI,EAAmB,MAAM,CAAC,CAAC;gBAE9C,kDAAkD;gBAClD,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;gBAEhE,qCAAqC;gBACrC,IAAI,CAAC,aAAa,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACjD,SAAS;gBACX,CAAC;gBAED,iDAAiD;gBACjD,MAAM,QAAQ,GACZ,MAAM,MAAM,CAAC,kBAAkB,CAC7B,SAAS,EACT,aAAgC,CACjC,CAAC;gBAEJ,oDAAoD;gBACpD,IAAI,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;oBACtE,KAAK,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;wBAC1D,MAAA,OAAO,CAAC,aAAa,EAAC,SAAS,SAAT,SAAS,IAAM,EAAE,EAAC;wBACxC,MAAM,eAAe,GAAG,OAAO,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;wBACzD,IAAI,eAAe,EAAE,CAAC;4BACnB,eAA2C,CAAC,OAAO,CAAC,GAAG;gCACtD,MAAM,EAAE,OAAO,CAAC,MAAM;6BACvB,CAAC;wBACJ,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,oDAAoD;YACtD,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,+EAA+E;IAC/E,aAAa;IACb,+EAA+E;IAE/E;;;;;;;;;;OAUG;IACH,IAAI,gBAAgB;QAClB,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;;YAC7B,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;YAE5B,6CAA6C;YAC7C,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC1D,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC1C,CAAC;YAEF,2DAA2D;YAC3D,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjC,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,IAAI,yBAAyB,GAAc,EAAE,CAAC;YAE9C,IAAI,CAAC;gBACH,6BAA6B;gBAC7B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC;oBAChC,GAAG,OAAO;oBACV,QAAQ,EAAE,eAAe;iBAC1B,CAAC,CAAC;gBAEH,8BAA8B;gBAC9B,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;oBAC3B,MAAA,OAAO,CAAC,QAAQ,EAAC,aAAa,QAAb,aAAa,GAAK,EAAE,EAAC;oBACtC,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,QAAQ,CAAC,aAAa,CACvB,EAAE,CAAC;wBACF,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,SAAS,CAAC,GAAG;4BAC1C,GAAG,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,SAAS,CAAC;4BAC5C,GAAG,eAAe;yBACnB,CAAC;oBACJ,CAAC;gBACH,CAAC;gBACD,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;oBAC5B,OAAO,CAAC,QAAQ,CAAC,cAAc,GAAG;wBAChC,GAAG,OAAO,CAAC,QAAQ,CAAC,cAAc;wBAClC,GAAG,QAAQ,CAAC,cAAc;qBAC3B,CAAC;gBACJ,CAAC;gBACD,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;oBACzB,OAAO,CAAC,QAAQ,CAAC,WAAW,GAAG;wBAC7B,GAAG,OAAO,CAAC,QAAQ,CAAC,WAAW;wBAC/B,GAAG,QAAQ,CAAC,WAAW;qBACxB,CAAC;gBACJ,CAAC;gBAED,yDAAyD;gBACzD,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,CAAC;gBACjE,yBAAyB,GAAG,eAAe,CAAC,MAAM,CAChD,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,CACxC,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC1C,yBAAyB,GAAG,EAAE,CAAC;YACjC,CAAC;YAED,sCAAsC;YACtC,IAAI,WAAW,GAAG,OAAO,CAAC;YAC1B,IAAI,yBAAyB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzC,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAC7C,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,yBAAyB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC1D,CAAC;gBACF,WAAW,GAAG;oBACZ,GAAG,OAAO;oBACV,OAAO,EAAE;wBACP,GAAG,OAAO;wBACV,QAAQ,EAAE,eAAe;qBAC1B;iBACF,CAAC;YACJ,CAAC;YAED,uBAAuB;YACvB,OAAO,IAAI,CAAC,WAAW,CAAC,CAAC;QAC3B,CAAC,CAAC;IACJ,CAAC;IAED,+EAA+E;IAC/E,4CAA4C;IAC5C,+EAA+E;IAE/E,KAAK,CAAC,SAAS,CAAC,mBAAwC;QACtD,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC;QAElE,8CAA8C;QAC9C,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC;YACvB,OAAO;QACT,CAAC;QAED,sCAAsC;QACtC,MAAM,eAAe,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC1D,uBAAA,IAAI,yEAAwB,MAA5B,IAAI,EAAyB,OAAO,CAAC,CACtC,CAAC;QAEF,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO;QACT,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC9D,IAAI,QAAQ,EAAE,CAAC;gBACb,QAAQ,CAAC,MAAM,GAAG,eAAe,CAAC;gBAClC,uDAAuD;gBACvD,IAAI,CAAC,KAAK,CAAC;oBACT,GAAG,OAAO;oBACV,QAAQ,EAAE,eAAe;iBAC1B,CAAC;qBACC,IAAI,CAAC,KAAK,EAAE,aAAa,EAAE,EAAE;oBAC5B,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC9D,MAAM,QAAQ,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;oBAC/C,CAAC;oBACD,OAAO,aAAa,CAAC;gBACvB,CAAC,CAAC;qBACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBACf,GAAG,CAAC,kCAAkC,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;gBACrE,CAAC,CAAC,CAAC;gBACL,OAAO;YACT,CAAC;QACH,CAAC;QAED,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;QAEvC,gFAAgF;QAChF,iEAAiE;QACjE,sEAAsE;QAEtE,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,EAAE;YAC3C,OAAO,EAAE,GAAG,EAAE;gBACZ,iDAAiD;YACnD,CAAC;YACD,MAAM,EAAE,eAAe;YACvB,cAAc,EAAE,mBAAmB,CAAC,cAAc;SACnD,CAAC,CAAC;QAEH,wCAAwC;QACxC,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC;gBACrC,GAAG,OAAO;gBACV,QAAQ,EAAE,eAAe;aAC1B,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAClE,IACE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC;gBACzD,YAAY,EACZ,CAAC;gBACD,MAAM,YAAY,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YACnD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,sBAAsB,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAqCD,+EAA+E;IAC/E,UAAU;IACV,+EAA+E;IAE/E,OAAO;QACL,8DAA8D;QAC9D,MAAM,SAAS,GAAG,uBAAA,IAAI,iCAAkB,CAAC;QAEzC,uCAAuC;QACvC,IAAI,CAAC;YACH,SAAS,CAAC,WAAW,CACnB,2CAA2C,EAC3C,uBAAA,IAAI,sDAAgC,CACrC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,gDAAgD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,sCAAsC;QACtC,IAAI,CAAC;YACH,SAAS,CAAC,WAAW,CACnB,kCAAkC,EAClC,uBAAA,IAAI,wDAAkC,CACvC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,+CAA+C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAClE,CAAC;QAED,gCAAgC;QAChC,KAAK,MAAM,CAAC,cAAc,CAAC,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACxD,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;gBAC1C,wBAAwB;YAC1B,CAAC,CAAC,CAAC;QACL,CAAC;QAED,6BAA6B;QAC7B,uBAAA,IAAI,0CAAoB,CAAC,KAAK,EAAE,CAAC;IACnC,CAAC;CACF;AAvhBD,wCAuhBC;;IA1eG,2FAA2F;IAC3F,8DAA8D;IAC9D,MAAM,SAAS,GAAG,uBAAA,IAAI,iCAAkB,CAAC;IACzC,SAAS,CAAC,SAAS,CACjB,2CAA2C,EAC3C,uBAAA,IAAI,sDAAgC,CACrC,CAAC;IACF,SAAS,CAAC,SAAS,CACjB,kCAAkC,EAClC,uBAAA,IAAI,wDAAkC,CACvC,CAAC;AACJ,CAAC,iGASC,OAA2C;IAE3C,4DAA4D;IAC5D,IAAI,aAAqE,CAAC;IAE1E,KAAK,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACnE,IAAI,aAA8D,CAAC;QAEnE,KAAK,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACxD,MAAM,OAAO,GAAG,uBAAuB,CAAC,OAAO,CAAC,CAAC;YACjD,IAAI,uBAAA,IAAI,yEAAwB,MAA5B,IAAI,EAAyB,OAAO,CAAC,EAAE,CAAC;gBAC1C,aAAa,KAAb,aAAa,GAAK,EAAE,EAAC;gBACrB,aAAa,CAAC,OAAwB,CAAC,GAAG;oBACxC,MAAM,EAAE,OAAO,CAAC,MAAM;iBACvB,CAAC;YACJ,CAAC;QACH,CAAC;QAED,IAAI,aAAa,EAAE,CAAC;YAClB,aAAa,KAAb,aAAa,GAAK,EAAE,EAAC;YACrB,aAAa,CAAC,SAAS,CAAC,GAAG,aAAa,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,8CAA8C;IAC9C,IAAI,aAAa,EAAE,CAAC;QAClB,MAAM,QAAQ,GAAiB,EAAE,aAAa,EAAE,CAAC;QACjD,KAAK,MAAM,YAAY,IAAI,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7D,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;AACH,CAAC,2FAQuB,OAAgB;IACtC,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AACnD,CAAC;IAaC,IAAI,CAAC;QACH,8DAA8D;QAC9D,OAAQ,uBAAA,IAAI,iCAAmB,CAAC,IAAI,CAClC,iCAAiC,CACxB,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;QAC3C,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC,mFASC,MAAc;IAEd,IAAI,CAAC;QACH,8DAA8D;QAC9D,OAAQ,uBAAA,IAAI,iCAAmB,CAAC,IAAI,CAClC,qCAAqC,EACrC,MAAM,CACqC,CAAC;IAChD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,oCAAoC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7D,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;IAkBC,IAAI,CAAC;QACH,MAAM,aAAa,GAAG,uBAAA,IAAI,mEAAkB,MAAtB,IAAI,CAAoB,CAAC;QAC/C,MAAM,WAAW,GAA4B,EAAE,CAAC;QAChD,MAAM,eAAe,GAAc,EAAE,CAAC;QAEtC,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;YACjC,MAAM,WAAW,GAAG,uBAAA,IAAI,qEAAoB,MAAxB,IAAI,EAAqB,IAAI,CAAC,EAAE,CAAC,CAAC;YACtD,8DAA8D;YAC9D,IAAI,CAAC,WAAW,EAAE,CAAC,0BAAkB,CAAC,EAAE,CAAC;gBACvC,SAAS;YACX,CAAC;YAED,0EAA0E;YAC1E,8CAA8C;YAC9C,MAAM,gBAAgB,GAAG,WAAW,CAAC,yBAAiB,CAAC,CAAC;YACxD,MAAM,QAAQ,GAAG,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;YAErD,+EAA+E;YAC/E,IAAI,QAAQ,EAAE,CAAC;gBACb,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;oBAC/B,IAAI,CAAC,CAAC,OAAO,IAAI,WAAW,CAAC,EAAE,CAAC;wBAC9B,WAAW,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,EAAE,CAAC;wBAC/B,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBAChC,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,WAAW,CAAC;QAErC,2BAA2B;QAC3B,IAAI,CAAC;YACH,IAAI,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC,aAAa,EAAE,EAAE;gBACzD,uBAAA,IAAI,6CAAuB,MAA3B,IAAI,EAAwB,aAAa,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,kEAAkE;QACpE,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,+BAA+B,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAChD,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,EAAE,CAAC;QAC5B,IAAI,CAAC;YACH,IAAI,CAAC,kBAAkB,CAAC,EAAE,EAAE,CAAC,aAAa,EAAE,EAAE;gBAC5C,uBAAA,IAAI,6CAAuB,MAA3B,IAAI,EAAwB,aAAa,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,kEAAkE;QACpE,CAAC;IACH,CAAC;AACH,CAAC,+EAgQiB,MAAc;IAC9B,MAAM,YAAY,GAAG,uBAAA,IAAI,0CAAoB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC1D,IAAI,YAAY,EAAE,CAAC;QACjB,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,mCAAa,CAAC;QAC/B,IAAI,EAAE,KAAK,EAAE,OAAuB,EAAiB,EAAE,CACrD,MACE,uBAAA,IAAI,iCAGL,CAAC,IAAI,CAAC,8BAA8B,EAAE;YACrC,MAAM,EAAE,MAAgB;YACxB,MAAM,EAAE,UAAU;YAClB,OAAO,EAAE,yBAAW,CAAC,gBAAgB;YACrC,OAAO;SACR,CAAC;KACL,CAAC,CAAC;IAEH,uBAAA,IAAI,0CAAoB,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7C,OAAO,MAAM,CAAC;AAChB,CAAC;AA0CH,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E,SAAgB,oBAAoB,CAClC,OAA8B;IAE9B,OAAO,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;AACrC,CAAC;AAJD,oDAIC","sourcesContent":["import type { Balance, CaipAssetType } from '@metamask/keyring-api';\nimport { KeyringClient } from '@metamask/keyring-snap-client';\nimport type {\n  Caveat,\n  GetPermissions,\n  PermissionConstraint,\n  PermissionControllerStateChange,\n  SubjectPermissions,\n} from '@metamask/permission-controller';\nimport type {\n  GetRunnableSnaps,\n  HandleSnapRequest,\n} from '@metamask/snaps-controllers';\nimport type { Snap, SnapId } from '@metamask/snaps-sdk';\nimport { HandlerType, SnapCaveatType } from '@metamask/snaps-utils';\nimport type { Json, JsonRpcRequest } from '@metamask/utils';\n\nimport { AbstractDataSource } from './AbstractDataSource';\nimport type {\n  DataSourceState,\n  SubscriptionRequest,\n} from './AbstractDataSource';\nimport type { AssetsControllerMessenger } from '../AssetsController';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport type {\n  AssetBalance,\n  ChainId,\n  Caip19AssetId,\n  DataRequest,\n  DataResponse,\n  Middleware,\n} from '../types';\n\n// ============================================================================\n// SNAP KEYRING EVENT TYPES\n// ============================================================================\n\n/**\n * Payload for AccountsController:accountBalancesUpdated event.\n * Re-published from SnapKeyring:accountBalancesUpdated.\n */\nexport type AccountBalancesUpdatedEventPayload = {\n  balances: {\n    [accountId: string]: {\n      [assetId: string]: {\n        amount: string;\n        unit: string;\n      };\n    };\n  };\n};\n\n/**\n * Event from AccountsController when snap balances are updated.\n */\nexport type AccountsControllerAccountBalancesUpdatedEvent = {\n  type: 'AccountsController:accountBalancesUpdated';\n  payload: [AccountBalancesUpdatedEventPayload];\n};\n\nconst log = createModuleLogger(projectLogger, 'SnapDataSource');\n\n// ============================================================================\n// CONSTANTS\n// ============================================================================\n\nexport const SNAP_DATA_SOURCE_NAME = 'SnapDataSource';\n\n/** The permission name for snap keyring endowment */\nexport const KEYRING_PERMISSION = 'endowment:keyring';\n\n/** The permission name for snap assets endowment (contains chainIds) */\nexport const ASSETS_PERMISSION = 'endowment:assets';\n\n// ============================================================================\n// PERMISSION UTILITIES\n// ============================================================================\n\n/**\n * Getter function to get the chainIds caveat from a permission.\n *\n * This does basic validation of the caveat, but does not validate the type or\n * value of the namespaces object itself, as this is handled by the\n * `PermissionsController` when the permission is requested.\n *\n * @param permission - The permission to get the `chainIds` caveat from.\n * @returns An array of `chainIds` that the snap supports, or null if none.\n */\nexport function getChainIdsCaveat(\n  permission?: PermissionConstraint,\n): ChainId[] | null {\n  if (!permission?.caveats) {\n    return null;\n  }\n\n  const caveat = permission.caveats.find(\n    (permCaveat) => permCaveat.type === SnapCaveatType.ChainIds,\n  ) as Caveat<string, string[]> | undefined;\n\n  return caveat ? (caveat.value as ChainId[]) : null;\n}\n\n/**\n * Extract chain ID from a CAIP-19 asset ID.\n * e.g., \"solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp/slip44:501\" -> \"solana:5eykt4UsFv8P8NJdTREpY1vzqKqZKvdp\"\n *\n * @param assetId - The CAIP-19 asset ID to extract chain from.\n * @returns The CAIP-2 chain ID portion of the asset ID.\n */\nexport function extractChainFromAssetId(assetId: string): ChainId {\n  const parts = assetId.split('/');\n  return parts[0] as ChainId;\n}\n\n// ============================================================================\n// STATE\n// ============================================================================\n\n/**\n * State for the SnapDataSource.\n * Uses dynamic snap discovery - chains are populated from PermissionController.\n */\nexport type SnapDataSourceState = {\n  /**\n   * Mapping of chain IDs to snap IDs that support them.\n   * Used to filter which accounts to process for a given chain request.\n   */\n  chainToSnap: Record<ChainId, string>;\n} & DataSourceState;\n\nconst defaultSnapState: SnapDataSourceState = {\n  activeChains: [],\n  chainToSnap: {},\n};\n\n// ============================================================================\n// MESSENGER TYPES\n// ============================================================================\n\n/**\n * Allowed events that SnapDataSource can subscribe to.\n */\nexport type SnapDataSourceAllowedEvents =\n  | AccountsControllerAccountBalancesUpdatedEvent\n  | PermissionControllerStateChange;\n\nexport type SnapDataSourceAllowedActions =\n  | GetRunnableSnaps\n  | HandleSnapRequest\n  | GetPermissions;\n\n// ============================================================================\n// OPTIONS\n// ============================================================================\n\nexport type SnapDataSourceOptions = {\n  /** The AssetsController messenger (shared by all data sources). */\n  messenger: AssetsControllerMessenger;\n  /** Called when this data source's active chains change. */\n  onActiveChainsUpdated: (chains: ChainId[]) => void;\n  /** Configured networks to support (defaults to all snap networks) */\n  configuredNetworks?: ChainId[];\n  /** Default polling interval in ms for subscriptions */\n  pollInterval?: number;\n  /** Initial state */\n  state?: Partial<SnapDataSourceState>;\n};\n\n// ============================================================================\n// SNAP DATA SOURCE\n// ============================================================================\n\n/**\n * Unified Snap data source that routes requests to the appropriate wallet snap\n * based on the chain ID prefix.\n *\n * @example\n * ```typescript\n * const snapDataSource = new SnapDataSource({\n *   messenger,\n *   onActiveChainsUpdated: (chains) => { /* ... *\\/ },\n * });\n *\n * // Fetch will automatically route to the correct snap\n * await snapDataSource.fetch({\n *   chainIds: ['solana:mainnet', 'bip122:000000000019d6689c085ae165831e93'],\n *   accountIds: ['account1'],\n * });\n * ```\n */\nexport class SnapDataSource extends AbstractDataSource<\n  typeof SNAP_DATA_SOURCE_NAME,\n  SnapDataSourceState\n> {\n  readonly #messenger: AssetsControllerMessenger;\n\n  readonly #onActiveChainsUpdated: (chains: ChainId[]) => void;\n\n  /** Bound handler for snap keyring balance updates, stored for cleanup */\n  readonly #handleSnapBalancesUpdatedBound: (\n    payload: AccountBalancesUpdatedEventPayload,\n  ) => void;\n\n  readonly #handlePermissionStateChangeBound: () => void;\n\n  /** Cache of KeyringClient instances per snap ID to avoid re-instantiation */\n  readonly #keyringClientCache: Map<string, KeyringClient> = new Map();\n\n  constructor(options: SnapDataSourceOptions) {\n    super(SNAP_DATA_SOURCE_NAME, {\n      ...defaultSnapState,\n      ...options.state,\n    });\n\n    this.#messenger = options.messenger;\n    this.#onActiveChainsUpdated = options.onActiveChainsUpdated;\n\n    // Bind handlers for cleanup in destroy()\n    this.#handleSnapBalancesUpdatedBound = this.#handleSnapBalancesUpdated.bind(\n      this,\n    ) as (payload: AccountBalancesUpdatedEventPayload) => void;\n    this.#handlePermissionStateChangeBound =\n      this.#discoverKeyringSnaps.bind(this);\n\n    this.#subscribeToEvents();\n\n    // Discover keyring-capable snaps and populate activeChains dynamically\n    this.#discoverKeyringSnaps();\n  }\n\n  /**\n   * Subscribe to all events needed by SnapDataSource.\n   * Groups snap keyring events and permission change events.\n   */\n  #subscribeToEvents(): void {\n    // Subscribe to snap keyring events and permission changes (not in AssetsControllerEvents).\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const messenger = this.#messenger as any;\n    messenger.subscribe(\n      'AccountsController:accountBalancesUpdated',\n      this.#handleSnapBalancesUpdatedBound,\n    );\n    messenger.subscribe(\n      'PermissionController:stateChange',\n      this.#handlePermissionStateChangeBound,\n    );\n  }\n\n  /**\n   * Handle snap balance updates from the keyring.\n   * Transforms the payload and publishes to AssetsController.\n   *\n   * @param payload - The balance update payload from AccountsController.\n   */\n  #handleSnapBalancesUpdated(\n    payload: AccountBalancesUpdatedEventPayload,\n  ): void {\n    // Transform the snap keyring payload to DataResponse format\n    let assetsBalance: NonNullable<DataResponse['assetsBalance']> | undefined;\n\n    for (const [accountId, assets] of Object.entries(payload.balances)) {\n      let accountAssets: Record<Caip19AssetId, AssetBalance> | undefined;\n\n      for (const [assetId, balance] of Object.entries(assets)) {\n        const chainId = extractChainFromAssetId(assetId);\n        if (this.#isChainSupportedBySnap(chainId)) {\n          accountAssets ??= {};\n          accountAssets[assetId as Caip19AssetId] = {\n            amount: balance.amount,\n          };\n        }\n      }\n\n      if (accountAssets) {\n        assetsBalance ??= {};\n        assetsBalance[accountId] = accountAssets;\n      }\n    }\n\n    // Only report if we have snap-related updates\n    if (assetsBalance) {\n      const response: DataResponse = { assetsBalance };\n      for (const subscription of this.activeSubscriptions.values()) {\n        subscription.onAssetsUpdate(response)?.catch(console.error);\n      }\n    }\n  }\n\n  /**\n   * Check if a chain ID is supported by any discovered snap.\n   *\n   * @param chainId - The CAIP-2 chain ID to check.\n   * @returns True if we have a snap that supports this chain.\n   */\n  #isChainSupportedBySnap(chainId: ChainId): boolean {\n    return this.state.activeChains.includes(chainId);\n  }\n\n  // ============================================================================\n  // SNAP DISCOVERY (Dynamic via PermissionController)\n  // ============================================================================\n\n  /**\n   * Get all runnable snaps from SnapController.\n   * Runnable snaps are enabled and not blocked.\n   *\n   * @returns Array of runnable snaps.\n   */\n  #getRunnableSnaps(): Snap[] {\n    try {\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      return (this.#messenger as any).call(\n        'SnapController:getRunnableSnaps',\n      ) as Snap[];\n    } catch (error) {\n      log('Failed to get runnable snaps', error);\n      return [];\n    }\n  }\n\n  /**\n   * Get permissions for a snap from PermissionController.\n   *\n   * @param snapId - The snap ID to get permissions for.\n   * @returns The snap's permissions, or undefined if none.\n   */\n  #getSnapPermissions(\n    snapId: string,\n  ): SubjectPermissions<PermissionConstraint> | undefined {\n    try {\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      return (this.#messenger as any).call(\n        'PermissionController:getPermissions',\n        snapId,\n      ) as SubjectPermissions<PermissionConstraint>;\n    } catch (error) {\n      log('Failed to get permissions for snap', { snapId, error });\n      return undefined;\n    }\n  }\n\n  /**\n   * Discover all snaps with keyring capabilities and their supported chains.\n   * Uses PermissionController to find snaps with endowment:keyring permission.\n   * Updates chainToSnap mapping and activeChains.\n   *\n   * Called on initialization and whenever PermissionController state changes\n   * (e.g., new snaps installed, permissions granted/revoked).\n   *\n   * @remarks\n   * **Known limitation:** If discovery fails (e.g., SnapController not ready),\n   * the data source continues with empty chainToSnap. This means no snap\n   * chains will be supported until a re-discovery is triggered by a permission\n   * change. Callers should be aware that initialization may complete with no\n   * active chains.\n   */\n  #discoverKeyringSnaps(): void {\n    try {\n      const runnableSnaps = this.#getRunnableSnaps();\n      const chainToSnap: Record<ChainId, string> = {};\n      const supportedChains: ChainId[] = [];\n\n      for (const snap of runnableSnaps) {\n        const permissions = this.#getSnapPermissions(snap.id);\n        // Must have endowment:keyring permission to be a keyring snap\n        if (!permissions?.[KEYRING_PERMISSION]) {\n          continue;\n        }\n\n        // Get chainIds caveat from the assets permission (not keyring permission)\n        // The chainIds are stored in endowment:assets\n        const assetsPermission = permissions[ASSETS_PERMISSION];\n        const chainIds = getChainIdsCaveat(assetsPermission);\n\n        // Map each chain to this snap (first snap wins if multiple support same chain)\n        if (chainIds) {\n          for (const chainId of chainIds) {\n            if (!(chainId in chainToSnap)) {\n              chainToSnap[chainId] = snap.id;\n              supportedChains.push(chainId);\n            }\n          }\n        }\n      }\n\n      // Update chainToSnap mapping\n      this.state.chainToSnap = chainToSnap;\n\n      // Notify if chains changed\n      try {\n        this.updateActiveChains(supportedChains, (updatedChains) => {\n          this.#onActiveChainsUpdated(updatedChains);\n        });\n      } catch {\n        // AssetsController not ready yet - expected during initialization\n      }\n    } catch (error) {\n      log('Keyring snap discovery failed', { error });\n      this.state.chainToSnap = {};\n      try {\n        this.updateActiveChains([], (updatedChains) => {\n          this.#onActiveChainsUpdated(updatedChains);\n        });\n      } catch {\n        // AssetsController not ready yet - expected during initialization\n      }\n    }\n  }\n\n  // ============================================================================\n  // FETCH\n  // ============================================================================\n\n  async fetch(request: DataRequest): Promise<DataResponse> {\n    // Guard against undefined request\n    // Note: chainIds filtering is done by middleware/subscribe before calling fetch\n    if (!request?.chainIds?.length) {\n      return {};\n    }\n    if (!request?.accountsWithSupportedChains?.length) {\n      return { assetsBalance: {}, assetsMetadata: {} };\n    }\n\n    const results: DataResponse = {\n      assetsBalance: {},\n      assetsMetadata: {},\n    };\n\n    // Fetch balances for each account using its snap ID from metadata\n    for (const { account } of request.accountsWithSupportedChains) {\n      // Skip accounts without snap metadata (non-snap accounts)\n      const snapId = account.metadata.snap?.id;\n      if (!snapId) {\n        continue;\n      }\n\n      // Skip accounts whose snap doesn't support any of the requested chains\n      const snapSupportsRequestedChains = request.chainIds.some(\n        (chainId) => this.state.chainToSnap[chainId] === snapId,\n      );\n      if (!snapSupportsRequestedChains) {\n        continue;\n      }\n\n      const accountId = account.id;\n      try {\n        const client = this.#getKeyringClient(snapId);\n\n        // Step 1: Get the list of assets for this account\n        const accountAssets = await client.listAccountAssets(accountId);\n\n        // If no assets, skip to next account\n        if (!accountAssets || accountAssets.length === 0) {\n          continue;\n        }\n\n        // Step 2: Get balances for those specific assets\n        const balances: Record<CaipAssetType, Balance> =\n          await client.getAccountBalances(\n            accountId,\n            accountAssets as CaipAssetType[],\n          );\n\n        // Transform keyring response to DataResponse format\n        if (balances && typeof balances === 'object' && results.assetsBalance) {\n          for (const [assetId, balance] of Object.entries(balances)) {\n            results.assetsBalance[accountId] ??= {};\n            const accountBalances = results.assetsBalance[accountId];\n            if (accountBalances) {\n              (accountBalances as Record<string, unknown>)[assetId] = {\n                amount: balance.amount,\n              };\n            }\n          }\n        }\n      } catch {\n        // Expected when account doesn't belong to this snap\n      }\n    }\n\n    return results;\n  }\n\n  // ============================================================================\n  // MIDDLEWARE\n  // ============================================================================\n\n  /**\n   * Get the middleware for fetching balances via Snaps.\n   * This middleware:\n   * - Supports multiple accounts in a single request\n   * - Filters request to only chains this data source supports\n   * - Fetches balances for those chains for all accounts\n   * - Merges response into context\n   * - Removes handled chains from request for next middleware\n   *\n   * @returns The middleware function for the assets pipeline.\n   */\n  get assetsMiddleware(): Middleware {\n    return async (context, next) => {\n      const { request } = context;\n\n      // Filter to chains this data source supports\n      const supportedChains = request.chainIds.filter((chainId) =>\n        this.state.activeChains.includes(chainId),\n      );\n\n      // If no supported chains, skip and pass to next middleware\n      if (supportedChains.length === 0) {\n        return next(context);\n      }\n\n      let successfullyHandledChains: ChainId[] = [];\n\n      try {\n        // Fetch for supported chains\n        const response = await this.fetch({\n          ...request,\n          chainIds: supportedChains,\n        });\n\n        // Merge response into context\n        if (response.assetsBalance) {\n          context.response.assetsBalance ??= {};\n          for (const [accountId, accountBalances] of Object.entries(\n            response.assetsBalance,\n          )) {\n            context.response.assetsBalance[accountId] = {\n              ...context.response.assetsBalance[accountId],\n              ...accountBalances,\n            };\n          }\n        }\n        if (response.assetsMetadata) {\n          context.response.assetsMetadata = {\n            ...context.response.assetsMetadata,\n            ...response.assetsMetadata,\n          };\n        }\n        if (response.assetsPrice) {\n          context.response.assetsPrice = {\n            ...context.response.assetsPrice,\n            ...response.assetsPrice,\n          };\n        }\n\n        // Determine successfully handled chains (exclude errors)\n        const failedChains = new Set(Object.keys(response.errors ?? {}));\n        successfullyHandledChains = supportedChains.filter(\n          (chainId) => !failedChains.has(chainId),\n        );\n      } catch (error) {\n        log('Middleware fetch failed', { error });\n        successfullyHandledChains = [];\n      }\n\n      // Prepare context for next middleware\n      let nextContext = context;\n      if (successfullyHandledChains.length > 0) {\n        const remainingChains = request.chainIds.filter(\n          (chainId) => !successfullyHandledChains.includes(chainId),\n        );\n        nextContext = {\n          ...context,\n          request: {\n            ...request,\n            chainIds: remainingChains,\n          },\n        };\n      }\n\n      // Call next middleware\n      return next(nextContext);\n    };\n  }\n\n  // ============================================================================\n  // SUBSCRIBE - Routes to appropriate snap(s)\n  // ============================================================================\n\n  async subscribe(subscriptionRequest: SubscriptionRequest): Promise<void> {\n    const { request, subscriptionId, isUpdate } = subscriptionRequest;\n\n    // Guard against undefined request or chainIds\n    if (!request?.chainIds) {\n      return;\n    }\n\n    // Filter to chains we have a snap for\n    const supportedChains = request.chainIds.filter((chainId) =>\n      this.#isChainSupportedBySnap(chainId),\n    );\n\n    if (supportedChains.length === 0) {\n      return;\n    }\n\n    if (isUpdate) {\n      const existing = this.activeSubscriptions.get(subscriptionId);\n      if (existing) {\n        existing.chains = supportedChains;\n        // Do a fetch to get latest data on subscription update\n        this.fetch({\n          ...request,\n          chainIds: supportedChains,\n        })\n          .then(async (fetchResponse) => {\n            if (Object.keys(fetchResponse.assetsBalance ?? {}).length > 0) {\n              await existing.onAssetsUpdate(fetchResponse);\n            }\n            return fetchResponse;\n          })\n          .catch((error) => {\n            log('Subscription update fetch failed', { subscriptionId, error });\n          });\n        return;\n      }\n    }\n\n    await this.unsubscribe(subscriptionId);\n\n    // Snaps provide real-time updates via AccountsController:accountBalancesUpdated\n    // We only need to track the subscription and do an initial fetch\n    // No polling needed - updates come through #handleSnapBalancesUpdated\n\n    this.activeSubscriptions.set(subscriptionId, {\n      cleanup: () => {\n        // No timer to clear - we use event-based updates\n      },\n      chains: supportedChains,\n      onAssetsUpdate: subscriptionRequest.onAssetsUpdate,\n    });\n\n    // Initial fetch to get current balances\n    try {\n      const fetchResponse = await this.fetch({\n        ...request,\n        chainIds: supportedChains,\n      });\n\n      const subscription = this.activeSubscriptions.get(subscriptionId);\n      if (\n        Object.keys(fetchResponse.assetsBalance ?? {}).length > 0 &&\n        subscription\n      ) {\n        await subscription.onAssetsUpdate(fetchResponse);\n      }\n    } catch (error) {\n      log('Initial fetch failed', { subscriptionId, error });\n    }\n  }\n\n  // ============================================================================\n  // KEYRING CLIENT\n  // ============================================================================\n\n  /**\n   * Gets a `KeyringClient` for a Snap.\n   * Caches clients per snap ID to avoid re-instantiation across multiple calls.\n   *\n   * @param snapId - ID of the Snap to get the client for.\n   * @returns A `KeyringClient` for the Snap.\n   */\n  #getKeyringClient(snapId: string): KeyringClient {\n    const cachedClient = this.#keyringClientCache.get(snapId);\n    if (cachedClient) {\n      return cachedClient;\n    }\n\n    const client = new KeyringClient({\n      send: async (request: JsonRpcRequest): Promise<Json> =>\n        await (\n          this.#messenger as unknown as {\n            call: (action: string, ...args: unknown[]) => Promise<Json> | Json;\n          }\n        ).call('SnapController:handleRequest', {\n          snapId: snapId as SnapId,\n          origin: 'metamask',\n          handler: HandlerType.OnKeyringRequest,\n          request,\n        }),\n    });\n\n    this.#keyringClientCache.set(snapId, client);\n    return client;\n  }\n\n  // ============================================================================\n  // CLEANUP\n  // ============================================================================\n\n  destroy(): void {\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const messenger = this.#messenger as any;\n\n    // Unsubscribe from snap keyring events\n    try {\n      messenger.unsubscribe(\n        'AccountsController:accountBalancesUpdated',\n        this.#handleSnapBalancesUpdatedBound,\n      );\n    } catch (error) {\n      log('Failed to unsubscribe from snap keyring events', { error });\n    }\n\n    // Unsubscribe from permission changes\n    try {\n      messenger.unsubscribe(\n        'PermissionController:stateChange',\n        this.#handlePermissionStateChangeBound,\n      );\n    } catch (error) {\n      log('Failed to unsubscribe from permission changes', { error });\n    }\n\n    // Clean up active subscriptions\n    for (const [subscriptionId] of this.activeSubscriptions) {\n      this.unsubscribe(subscriptionId).catch(() => {\n        // Ignore cleanup errors\n      });\n    }\n\n    // Clear keyring client cache\n    this.#keyringClientCache.clear();\n  }\n}\n\n// ============================================================================\n// FACTORY\n// ============================================================================\n\nexport function createSnapDataSource(\n  options: SnapDataSourceOptions,\n): SnapDataSource {\n  return new SnapDataSource(options);\n}\n"]}

package/dist/data-sources/SnapDataSource.d.cts

@@ -1,8 +1,8 @@
-import type { Messenger } from "@metamask/messenger";
 import type { GetPermissions, PermissionConstraint, PermissionControllerStateChange } from "@metamask/permission-controller";
 import type { GetRunnableSnaps, HandleSnapRequest } from "@metamask/snaps-controllers";
 import { AbstractDataSource } from "./AbstractDataSource.cjs";
 import type { DataSourceState, SubscriptionRequest } from "./AbstractDataSource.cjs";
+import type { AssetsControllerMessenger } from "../AssetsController.cjs";
 import type { ChainId, DataRequest, DataResponse, Middleware } from "../types.cjs";
 /**
  * Payload for AccountsController:accountBalancesUpdated event.
@@ -60,53 +60,16 @@ export type SnapDataSourceState = {
      */
     chainToSnap: Record<ChainId, string>;
 } & DataSourceState;
-export type SnapDataSourceGetAssetsMiddlewareAction = {
-    type: 'SnapDataSource:getAssetsMiddleware';
-    handler: () => Middleware;
-};
-export type SnapDataSourceGetActiveChainsAction = {
-    type: 'SnapDataSource:getActiveChains';
-    handler: () => Promise<ChainId[]>;
-};
-export type SnapDataSourceFetchAction = {
-    type: 'SnapDataSource:fetch';
-    handler: (request: DataRequest) => Promise<DataResponse>;
-};
-export type SnapDataSourceSubscribeAction = {
-    type: 'SnapDataSource:subscribe';
-    handler: (request: SubscriptionRequest) => Promise<void>;
-};
-export type SnapDataSourceUnsubscribeAction = {
-    type: 'SnapDataSource:unsubscribe';
-    handler: (subscriptionId: string) => Promise<void>;
-};
-export type SnapDataSourceActions = SnapDataSourceGetAssetsMiddlewareAction | SnapDataSourceGetActiveChainsAction | SnapDataSourceFetchAction | SnapDataSourceSubscribeAction | SnapDataSourceUnsubscribeAction;
-export type SnapDataSourceActiveChainsChangedEvent = {
-    type: 'SnapDataSource:activeChainsUpdated';
-    payload: [ChainId[]];
-};
-export type SnapDataSourceAssetsUpdatedEvent = {
-    type: 'SnapDataSource:assetsUpdated';
-    payload: [DataResponse, string | undefined];
-};
-export type SnapDataSourceEvents = SnapDataSourceActiveChainsChangedEvent | SnapDataSourceAssetsUpdatedEvent;
 /**
  * Allowed events that SnapDataSource can subscribe to.
  */
 export type SnapDataSourceAllowedEvents = AccountsControllerAccountBalancesUpdatedEvent | PermissionControllerStateChange;
-type AssetsControllerActiveChainsUpdateAction = {
-    type: 'AssetsController:activeChainsUpdate';
-    handler: (dataSourceId: string, activeChains: ChainId[]) => void;
-};
-type AssetsControllerAssetsUpdateAction = {
-    type: 'AssetsController:assetsUpdate';
-    handler: (response: DataResponse, sourceId: string) => Promise<void>;
-};
-export type SnapDataSourceAllowedActions = AssetsControllerActiveChainsUpdateAction | AssetsControllerAssetsUpdateAction | GetRunnableSnaps | HandleSnapRequest | GetPermissions;
-export type SnapDataSourceMessenger = Messenger<typeof SNAP_DATA_SOURCE_NAME, SnapDataSourceActions | SnapDataSourceAllowedActions, SnapDataSourceEvents | SnapDataSourceAllowedEvents>;
+export type SnapDataSourceAllowedActions = GetRunnableSnaps | HandleSnapRequest | GetPermissions;
 export type SnapDataSourceOptions = {
-    /** Messenger for this data source */
-    messenger: SnapDataSourceMessenger;
+    /** The AssetsController messenger (shared by all data sources). */
+    messenger: AssetsControllerMessenger;
+    /** Called when this data source's active chains change. */
+    onActiveChainsUpdated: (chains: ChainId[]) => void;
     /** Configured networks to support (defaults to all snap networks) */
     configuredNetworks?: ChainId[];
     /** Default polling interval in ms for subscriptions */
@@ -122,6 +85,7 @@ export type SnapDataSourceOptions = {
  * ```typescript
  * const snapDataSource = new SnapDataSource({
  *   messenger,
+ *   onActiveChainsUpdated: (chains) => { /* ... *\/ },
  * });
  *
  * // Fetch will automatically route to the correct snap
@@ -151,5 +115,4 @@ export declare class SnapDataSource extends AbstractDataSource<typeof SNAP_DATA_
     destroy(): void;
 }
 export declare function createSnapDataSource(options: SnapDataSourceOptions): SnapDataSource;
-export {};
 //# sourceMappingURL=SnapDataSource.d.cts.map

package/dist/data-sources/SnapDataSource.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"SnapDataSource.d.cts","sourceRoot":"","sources":["../../src/data-sources/SnapDataSource.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,SAAS,EAAE,4BAA4B;AACrD,OAAO,KAAK,EAEV,cAAc,EACd,oBAAoB,EACpB,+BAA+B,EAEhC,wCAAwC;AACzC,OAAO,KAAK,EACV,gBAAgB,EAChB,iBAAiB,EAClB,oCAAoC;AAKrC,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAC1D,OAAO,KAAK,EACV,eAAe,EACf,mBAAmB,EACpB,iCAA6B;AAE9B,OAAO,KAAK,EAEV,OAAO,EAEP,WAAW,EACX,YAAY,EACZ,UAAU,EACX,qBAAiB;AAMlB;;;GAGG;AACH,MAAM,MAAM,kCAAkC,GAAG;IAC/C,QAAQ,EAAE;QACR,CAAC,SAAS,EAAE,MAAM,GAAG;YACnB,CAAC,OAAO,EAAE,MAAM,GAAG;gBACjB,MAAM,EAAE,MAAM,CAAC;gBACf,IAAI,EAAE,MAAM,CAAC;aACd,CAAC;SACH,CAAC;KACH,CAAC;CACH,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,6CAA6C,GAAG;IAC1D,IAAI,EAAE,2CAA2C,CAAC;IAClD,OAAO,EAAE,CAAC,kCAAkC,CAAC,CAAC;CAC/C,CAAC;AAQF,eAAO,MAAM,qBAAqB,mBAAmB,CAAC;AAEtD,qDAAqD;AACrD,eAAO,MAAM,kBAAkB,sBAAsB,CAAC;AAEtD,wEAAwE;AACxE,eAAO,MAAM,iBAAiB,qBAAqB,CAAC;AAMpD;;;;;;;;;GASG;AACH,wBAAgB,iBAAiB,CAC/B,UAAU,CAAC,EAAE,oBAAoB,GAChC,OAAO,EAAE,GAAG,IAAI,CAUlB;AAED;;;;;;GAMG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAGhE;AAMD;;;GAGG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAChC;;;OAGG;IACH,WAAW,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;CACtC,GAAG,eAAe,CAAC;AAWpB,MAAM,MAAM,uCAAuC,GAAG;IACpD,IAAI,EAAE,oCAAoC,CAAC;IAC3C,OAAO,EAAE,MAAM,UAAU,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,mCAAmC,GAAG;IAChD,IAAI,EAAE,gCAAgC,CAAC;IACvC,OAAO,EAAE,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;CACnC,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,IAAI,EAAE,sBAAsB,CAAC;IAC7B,OAAO,EAAE,CAAC,OAAO,EAAE,WAAW,KAAK,OAAO,CAAC,YAAY,CAAC,CAAC;CAC1D,CAAC;AAEF,MAAM,MAAM,6BAA6B,GAAG;IAC1C,IAAI,EAAE,0BAA0B,CAAC;IACjC,OAAO,EAAE,CAAC,OAAO,EAAE,mBAAmB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1D,CAAC;AAEF,MAAM,MAAM,+BAA+B,GAAG;IAC5C,IAAI,EAAE,4BAA4B,CAAC;IACnC,OAAO,EAAE,CAAC,cAAc,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACpD,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAC7B,uCAAuC,GACvC,mCAAmC,GACnC,yBAAyB,GACzB,6BAA6B,GAC7B,+BAA+B,CAAC;AAEpC,MAAM,MAAM,sCAAsC,GAAG;IACnD,IAAI,EAAE,oCAAoC,CAAC;IAC3C,OAAO,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;CACtB,CAAC;AAEF,MAAM,MAAM,gCAAgC,GAAG;IAC7C,IAAI,EAAE,8BAA8B,CAAC;IACrC,OAAO,EAAE,CAAC,YAAY,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;CAC7C,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAC5B,sCAAsC,GACtC,gCAAgC,CAAC;AAErC;;GAEG;AACH,MAAM,MAAM,2BAA2B,GACnC,6CAA6C,GAC7C,+BAA+B,CAAC;AAGpC,KAAK,wCAAwC,GAAG;IAC9C,IAAI,EAAE,qCAAqC,CAAC;IAC5C,OAAO,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;CAClE,CAAC;AAEF,KAAK,kCAAkC,GAAG;IACxC,IAAI,EAAE,+BAA+B,CAAC;IACtC,OAAO,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACtE,CAAC;AAEF,MAAM,MAAM,4BAA4B,GACpC,wCAAwC,GACxC,kCAAkC,GAClC,gBAAgB,GAChB,iBAAiB,GACjB,cAAc,CAAC;AAEnB,MAAM,MAAM,uBAAuB,GAAG,SAAS,CAC7C,OAAO,qBAAqB,EAC5B,qBAAqB,GAAG,4BAA4B,EACpD,oBAAoB,GAAG,2BAA2B,CACnD,CAAC;AAMF,MAAM,MAAM,qBAAqB,GAAG;IAClC,qCAAqC;IACrC,SAAS,EAAE,uBAAuB,CAAC;IACnC,qEAAqE;IACrE,kBAAkB,CAAC,EAAE,OAAO,EAAE,CAAC;IAC/B,uDAAuD;IACvD,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,oBAAoB;IACpB,KAAK,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;CACtC,CAAC;AAMF;;;;;;;;;;;;;;;;GAgBG;AACH,qBAAa,cAAe,SAAQ,kBAAkB,CACpD,OAAO,qBAAqB,EAC5B,mBAAmB,CACpB;;gBAaa,OAAO,EAAE,qBAAqB;IAkPpC,KAAK,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IAuExD;;;;;;;;;;OAUG;IACH,IAAI,gBAAgB,IAAI,UAAU,CA4EjC;IAMK,SAAS,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IA8GxE,OAAO,IAAI,IAAI;CAkChB;AAMD,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,qBAAqB,GAC7B,cAAc,CAEhB"}
+{"version":3,"file":"SnapDataSource.d.cts","sourceRoot":"","sources":["../../src/data-sources/SnapDataSource.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAEV,cAAc,EACd,oBAAoB,EACpB,+BAA+B,EAEhC,wCAAwC;AACzC,OAAO,KAAK,EACV,gBAAgB,EAChB,iBAAiB,EAClB,oCAAoC;AAKrC,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAC1D,OAAO,KAAK,EACV,eAAe,EACf,mBAAmB,EACpB,iCAA6B;AAC9B,OAAO,KAAK,EAAE,yBAAyB,EAAE,gCAA4B;AAErE,OAAO,KAAK,EAEV,OAAO,EAEP,WAAW,EACX,YAAY,EACZ,UAAU,EACX,qBAAiB;AAMlB;;;GAGG;AACH,MAAM,MAAM,kCAAkC,GAAG;IAC/C,QAAQ,EAAE;QACR,CAAC,SAAS,EAAE,MAAM,GAAG;YACnB,CAAC,OAAO,EAAE,MAAM,GAAG;gBACjB,MAAM,EAAE,MAAM,CAAC;gBACf,IAAI,EAAE,MAAM,CAAC;aACd,CAAC;SACH,CAAC;KACH,CAAC;CACH,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,6CAA6C,GAAG;IAC1D,IAAI,EAAE,2CAA2C,CAAC;IAClD,OAAO,EAAE,CAAC,kCAAkC,CAAC,CAAC;CAC/C,CAAC;AAQF,eAAO,MAAM,qBAAqB,mBAAmB,CAAC;AAEtD,qDAAqD;AACrD,eAAO,MAAM,kBAAkB,sBAAsB,CAAC;AAEtD,wEAAwE;AACxE,eAAO,MAAM,iBAAiB,qBAAqB,CAAC;AAMpD;;;;;;;;;GASG;AACH,wBAAgB,iBAAiB,CAC/B,UAAU,CAAC,EAAE,oBAAoB,GAChC,OAAO,EAAE,GAAG,IAAI,CAUlB;AAED;;;;;;GAMG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAGhE;AAMD;;;GAGG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAChC;;;OAGG;IACH,WAAW,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;CACtC,GAAG,eAAe,CAAC;AAWpB;;GAEG;AACH,MAAM,MAAM,2BAA2B,GACnC,6CAA6C,GAC7C,+BAA+B,CAAC;AAEpC,MAAM,MAAM,4BAA4B,GACpC,gBAAgB,GAChB,iBAAiB,GACjB,cAAc,CAAC;AAMnB,MAAM,MAAM,qBAAqB,GAAG;IAClC,mEAAmE;IACnE,SAAS,EAAE,yBAAyB,CAAC;IACrC,2DAA2D;IAC3D,qBAAqB,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;IACnD,qEAAqE;IACrE,kBAAkB,CAAC,EAAE,OAAO,EAAE,CAAC;IAC/B,uDAAuD;IACvD,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,oBAAoB;IACpB,KAAK,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;CACtC,CAAC;AAMF;;;;;;;;;;;;;;;;;GAiBG;AACH,qBAAa,cAAe,SAAQ,kBAAkB,CACpD,OAAO,qBAAqB,EAC5B,mBAAmB,CACpB;;gBAea,OAAO,EAAE,qBAAqB;IA6MpC,KAAK,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IA0ExD;;;;;;;;;;OAUG;IACH,IAAI,gBAAgB,IAAI,UAAU,CA4EjC;IAMK,SAAS,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IA+GxE,OAAO,IAAI,IAAI;CAkChB;AAMD,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,qBAAqB,GAC7B,cAAc,CAEhB"}

package/dist/data-sources/SnapDataSource.d.mts

@@ -1,8 +1,8 @@
-import type { Messenger } from "@metamask/messenger";
 import type { GetPermissions, PermissionConstraint, PermissionControllerStateChange } from "@metamask/permission-controller";
 import type { GetRunnableSnaps, HandleSnapRequest } from "@metamask/snaps-controllers";
 import { AbstractDataSource } from "./AbstractDataSource.mjs";
 import type { DataSourceState, SubscriptionRequest } from "./AbstractDataSource.mjs";
+import type { AssetsControllerMessenger } from "../AssetsController.mjs";
 import type { ChainId, DataRequest, DataResponse, Middleware } from "../types.mjs";
 /**
  * Payload for AccountsController:accountBalancesUpdated event.
@@ -60,53 +60,16 @@ export type SnapDataSourceState = {
      */
     chainToSnap: Record<ChainId, string>;
 } & DataSourceState;
-export type SnapDataSourceGetAssetsMiddlewareAction = {
-    type: 'SnapDataSource:getAssetsMiddleware';
-    handler: () => Middleware;
-};
-export type SnapDataSourceGetActiveChainsAction = {
-    type: 'SnapDataSource:getActiveChains';
-    handler: () => Promise<ChainId[]>;
-};
-export type SnapDataSourceFetchAction = {
-    type: 'SnapDataSource:fetch';
-    handler: (request: DataRequest) => Promise<DataResponse>;
-};
-export type SnapDataSourceSubscribeAction = {
-    type: 'SnapDataSource:subscribe';
-    handler: (request: SubscriptionRequest) => Promise<void>;
-};
-export type SnapDataSourceUnsubscribeAction = {
-    type: 'SnapDataSource:unsubscribe';
-    handler: (subscriptionId: string) => Promise<void>;
-};
-export type SnapDataSourceActions = SnapDataSourceGetAssetsMiddlewareAction | SnapDataSourceGetActiveChainsAction | SnapDataSourceFetchAction | SnapDataSourceSubscribeAction | SnapDataSourceUnsubscribeAction;
-export type SnapDataSourceActiveChainsChangedEvent = {
-    type: 'SnapDataSource:activeChainsUpdated';
-    payload: [ChainId[]];
-};
-export type SnapDataSourceAssetsUpdatedEvent = {
-    type: 'SnapDataSource:assetsUpdated';
-    payload: [DataResponse, string | undefined];
-};
-export type SnapDataSourceEvents = SnapDataSourceActiveChainsChangedEvent | SnapDataSourceAssetsUpdatedEvent;
 /**
  * Allowed events that SnapDataSource can subscribe to.
  */
 export type SnapDataSourceAllowedEvents = AccountsControllerAccountBalancesUpdatedEvent | PermissionControllerStateChange;
-type AssetsControllerActiveChainsUpdateAction = {
-    type: 'AssetsController:activeChainsUpdate';
-    handler: (dataSourceId: string, activeChains: ChainId[]) => void;
-};
-type AssetsControllerAssetsUpdateAction = {
-    type: 'AssetsController:assetsUpdate';
-    handler: (response: DataResponse, sourceId: string) => Promise<void>;
-};
-export type SnapDataSourceAllowedActions = AssetsControllerActiveChainsUpdateAction | AssetsControllerAssetsUpdateAction | GetRunnableSnaps | HandleSnapRequest | GetPermissions;
-export type SnapDataSourceMessenger = Messenger<typeof SNAP_DATA_SOURCE_NAME, SnapDataSourceActions | SnapDataSourceAllowedActions, SnapDataSourceEvents | SnapDataSourceAllowedEvents>;
+export type SnapDataSourceAllowedActions = GetRunnableSnaps | HandleSnapRequest | GetPermissions;
 export type SnapDataSourceOptions = {
-    /** Messenger for this data source */
-    messenger: SnapDataSourceMessenger;
+    /** The AssetsController messenger (shared by all data sources). */
+    messenger: AssetsControllerMessenger;
+    /** Called when this data source's active chains change. */
+    onActiveChainsUpdated: (chains: ChainId[]) => void;
     /** Configured networks to support (defaults to all snap networks) */
     configuredNetworks?: ChainId[];
     /** Default polling interval in ms for subscriptions */
@@ -122,6 +85,7 @@ export type SnapDataSourceOptions = {
  * ```typescript
  * const snapDataSource = new SnapDataSource({
  *   messenger,
+ *   onActiveChainsUpdated: (chains) => { /* ... *\/ },
  * });
  *
  * // Fetch will automatically route to the correct snap
@@ -151,5 +115,4 @@ export declare class SnapDataSource extends AbstractDataSource<typeof SNAP_DATA_
     destroy(): void;
 }
 export declare function createSnapDataSource(options: SnapDataSourceOptions): SnapDataSource;
-export {};
 //# sourceMappingURL=SnapDataSource.d.mts.map

package/dist/data-sources/SnapDataSource.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"SnapDataSource.d.mts","sourceRoot":"","sources":["../../src/data-sources/SnapDataSource.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,SAAS,EAAE,4BAA4B;AACrD,OAAO,KAAK,EAEV,cAAc,EACd,oBAAoB,EACpB,+BAA+B,EAEhC,wCAAwC;AACzC,OAAO,KAAK,EACV,gBAAgB,EAChB,iBAAiB,EAClB,oCAAoC;AAKrC,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAC1D,OAAO,KAAK,EACV,eAAe,EACf,mBAAmB,EACpB,iCAA6B;AAE9B,OAAO,KAAK,EAEV,OAAO,EAEP,WAAW,EACX,YAAY,EACZ,UAAU,EACX,qBAAiB;AAMlB;;;GAGG;AACH,MAAM,MAAM,kCAAkC,GAAG;IAC/C,QAAQ,EAAE;QACR,CAAC,SAAS,EAAE,MAAM,GAAG;YACnB,CAAC,OAAO,EAAE,MAAM,GAAG;gBACjB,MAAM,EAAE,MAAM,CAAC;gBACf,IAAI,EAAE,MAAM,CAAC;aACd,CAAC;SACH,CAAC;KACH,CAAC;CACH,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,6CAA6C,GAAG;IAC1D,IAAI,EAAE,2CAA2C,CAAC;IAClD,OAAO,EAAE,CAAC,kCAAkC,CAAC,CAAC;CAC/C,CAAC;AAQF,eAAO,MAAM,qBAAqB,mBAAmB,CAAC;AAEtD,qDAAqD;AACrD,eAAO,MAAM,kBAAkB,sBAAsB,CAAC;AAEtD,wEAAwE;AACxE,eAAO,MAAM,iBAAiB,qBAAqB,CAAC;AAMpD;;;;;;;;;GASG;AACH,wBAAgB,iBAAiB,CAC/B,UAAU,CAAC,EAAE,oBAAoB,GAChC,OAAO,EAAE,GAAG,IAAI,CAUlB;AAED;;;;;;GAMG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAGhE;AAMD;;;GAGG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAChC;;;OAGG;IACH,WAAW,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;CACtC,GAAG,eAAe,CAAC;AAWpB,MAAM,MAAM,uCAAuC,GAAG;IACpD,IAAI,EAAE,oCAAoC,CAAC;IAC3C,OAAO,EAAE,MAAM,UAAU,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,mCAAmC,GAAG;IAChD,IAAI,EAAE,gCAAgC,CAAC;IACvC,OAAO,EAAE,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;CACnC,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,IAAI,EAAE,sBAAsB,CAAC;IAC7B,OAAO,EAAE,CAAC,OAAO,EAAE,WAAW,KAAK,OAAO,CAAC,YAAY,CAAC,CAAC;CAC1D,CAAC;AAEF,MAAM,MAAM,6BAA6B,GAAG;IAC1C,IAAI,EAAE,0BAA0B,CAAC;IACjC,OAAO,EAAE,CAAC,OAAO,EAAE,mBAAmB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1D,CAAC;AAEF,MAAM,MAAM,+BAA+B,GAAG;IAC5C,IAAI,EAAE,4BAA4B,CAAC;IACnC,OAAO,EAAE,CAAC,cAAc,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACpD,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAC7B,uCAAuC,GACvC,mCAAmC,GACnC,yBAAyB,GACzB,6BAA6B,GAC7B,+BAA+B,CAAC;AAEpC,MAAM,MAAM,sCAAsC,GAAG;IACnD,IAAI,EAAE,oCAAoC,CAAC;IAC3C,OAAO,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;CACtB,CAAC;AAEF,MAAM,MAAM,gCAAgC,GAAG;IAC7C,IAAI,EAAE,8BAA8B,CAAC;IACrC,OAAO,EAAE,CAAC,YAAY,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;CAC7C,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAC5B,sCAAsC,GACtC,gCAAgC,CAAC;AAErC;;GAEG;AACH,MAAM,MAAM,2BAA2B,GACnC,6CAA6C,GAC7C,+BAA+B,CAAC;AAGpC,KAAK,wCAAwC,GAAG;IAC9C,IAAI,EAAE,qCAAqC,CAAC;IAC5C,OAAO,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;CAClE,CAAC;AAEF,KAAK,kCAAkC,GAAG;IACxC,IAAI,EAAE,+BAA+B,CAAC;IACtC,OAAO,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACtE,CAAC;AAEF,MAAM,MAAM,4BAA4B,GACpC,wCAAwC,GACxC,kCAAkC,GAClC,gBAAgB,GAChB,iBAAiB,GACjB,cAAc,CAAC;AAEnB,MAAM,MAAM,uBAAuB,GAAG,SAAS,CAC7C,OAAO,qBAAqB,EAC5B,qBAAqB,GAAG,4BAA4B,EACpD,oBAAoB,GAAG,2BAA2B,CACnD,CAAC;AAMF,MAAM,MAAM,qBAAqB,GAAG;IAClC,qCAAqC;IACrC,SAAS,EAAE,uBAAuB,CAAC;IACnC,qEAAqE;IACrE,kBAAkB,CAAC,EAAE,OAAO,EAAE,CAAC;IAC/B,uDAAuD;IACvD,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,oBAAoB;IACpB,KAAK,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;CACtC,CAAC;AAMF;;;;;;;;;;;;;;;;GAgBG;AACH,qBAAa,cAAe,SAAQ,kBAAkB,CACpD,OAAO,qBAAqB,EAC5B,mBAAmB,CACpB;;gBAaa,OAAO,EAAE,qBAAqB;IAkPpC,KAAK,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IAuExD;;;;;;;;;;OAUG;IACH,IAAI,gBAAgB,IAAI,UAAU,CA4EjC;IAMK,SAAS,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IA8GxE,OAAO,IAAI,IAAI;CAkChB;AAMD,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,qBAAqB,GAC7B,cAAc,CAEhB"}
+{"version":3,"file":"SnapDataSource.d.mts","sourceRoot":"","sources":["../../src/data-sources/SnapDataSource.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAEV,cAAc,EACd,oBAAoB,EACpB,+BAA+B,EAEhC,wCAAwC;AACzC,OAAO,KAAK,EACV,gBAAgB,EAChB,iBAAiB,EAClB,oCAAoC;AAKrC,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAC1D,OAAO,KAAK,EACV,eAAe,EACf,mBAAmB,EACpB,iCAA6B;AAC9B,OAAO,KAAK,EAAE,yBAAyB,EAAE,gCAA4B;AAErE,OAAO,KAAK,EAEV,OAAO,EAEP,WAAW,EACX,YAAY,EACZ,UAAU,EACX,qBAAiB;AAMlB;;;GAGG;AACH,MAAM,MAAM,kCAAkC,GAAG;IAC/C,QAAQ,EAAE;QACR,CAAC,SAAS,EAAE,MAAM,GAAG;YACnB,CAAC,OAAO,EAAE,MAAM,GAAG;gBACjB,MAAM,EAAE,MAAM,CAAC;gBACf,IAAI,EAAE,MAAM,CAAC;aACd,CAAC;SACH,CAAC;KACH,CAAC;CACH,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,6CAA6C,GAAG;IAC1D,IAAI,EAAE,2CAA2C,CAAC;IAClD,OAAO,EAAE,CAAC,kCAAkC,CAAC,CAAC;CAC/C,CAAC;AAQF,eAAO,MAAM,qBAAqB,mBAAmB,CAAC;AAEtD,qDAAqD;AACrD,eAAO,MAAM,kBAAkB,sBAAsB,CAAC;AAEtD,wEAAwE;AACxE,eAAO,MAAM,iBAAiB,qBAAqB,CAAC;AAMpD;;;;;;;;;GASG;AACH,wBAAgB,iBAAiB,CAC/B,UAAU,CAAC,EAAE,oBAAoB,GAChC,OAAO,EAAE,GAAG,IAAI,CAUlB;AAED;;;;;;GAMG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAGhE;AAMD;;;GAGG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAChC;;;OAGG;IACH,WAAW,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;CACtC,GAAG,eAAe,CAAC;AAWpB;;GAEG;AACH,MAAM,MAAM,2BAA2B,GACnC,6CAA6C,GAC7C,+BAA+B,CAAC;AAEpC,MAAM,MAAM,4BAA4B,GACpC,gBAAgB,GAChB,iBAAiB,GACjB,cAAc,CAAC;AAMnB,MAAM,MAAM,qBAAqB,GAAG;IAClC,mEAAmE;IACnE,SAAS,EAAE,yBAAyB,CAAC;IACrC,2DAA2D;IAC3D,qBAAqB,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,IAAI,CAAC;IACnD,qEAAqE;IACrE,kBAAkB,CAAC,EAAE,OAAO,EAAE,CAAC;IAC/B,uDAAuD;IACvD,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,oBAAoB;IACpB,KAAK,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;CACtC,CAAC;AAMF;;;;;;;;;;;;;;;;;GAiBG;AACH,qBAAa,cAAe,SAAQ,kBAAkB,CACpD,OAAO,qBAAqB,EAC5B,mBAAmB,CACpB;;gBAea,OAAO,EAAE,qBAAqB;IA6MpC,KAAK,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IA0ExD;;;;;;;;;;OAUG;IACH,IAAI,gBAAgB,IAAI,UAAU,CA4EjC;IAMK,SAAS,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IA+GxE,OAAO,IAAI,IAAI;CAkChB;AAMD,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,qBAAqB,GAC7B,cAAc,CAEhB"}

package/dist/data-sources/SnapDataSource.mjs

@@ -9,7 +9,7 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
     return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
 };
-var _SnapDataSource_instances, _SnapDataSource_messenger, _SnapDataSource_handleSnapBalancesUpdatedBound, _SnapDataSource_handlePermissionStateChangeBound, _SnapDataSource_keyringClientCache, _SnapDataSource_subscribeToEvents, _SnapDataSource_handleSnapBalancesUpdated, _SnapDataSource_isChainSupportedBySnap, _SnapDataSource_registerActionHandlers, _SnapDataSource_getRunnableSnaps, _SnapDataSource_getSnapPermissions, _SnapDataSource_discoverKeyringSnaps, _SnapDataSource_getKeyringClient;
+var _SnapDataSource_instances, _SnapDataSource_messenger, _SnapDataSource_onActiveChainsUpdated, _SnapDataSource_handleSnapBalancesUpdatedBound, _SnapDataSource_handlePermissionStateChangeBound, _SnapDataSource_keyringClientCache, _SnapDataSource_subscribeToEvents, _SnapDataSource_handleSnapBalancesUpdated, _SnapDataSource_isChainSupportedBySnap, _SnapDataSource_getRunnableSnaps, _SnapDataSource_getSnapPermissions, _SnapDataSource_discoverKeyringSnaps, _SnapDataSource_getKeyringClient;
 import { KeyringClient } from "@metamask/keyring-snap-client";
 import { HandlerType, SnapCaveatType } from "@metamask/snaps-utils";
 import { AbstractDataSource } from "./AbstractDataSource.mjs";
@@ -69,6 +69,7 @@ const defaultSnapState = {
  * ```typescript
  * const snapDataSource = new SnapDataSource({
  *   messenger,
+ *   onActiveChainsUpdated: (chains) => { /* ... *\/ },
  * });
  *
  * // Fetch will automatically route to the correct snap
@@ -86,16 +87,17 @@ export class SnapDataSource extends AbstractDataSource {
         });
         _SnapDataSource_instances.add(this);
         _SnapDataSource_messenger.set(this, void 0);
+        _SnapDataSource_onActiveChainsUpdated.set(this, void 0);
         /** Bound handler for snap keyring balance updates, stored for cleanup */
         _SnapDataSource_handleSnapBalancesUpdatedBound.set(this, void 0);
         _SnapDataSource_handlePermissionStateChangeBound.set(this, void 0);
         /** Cache of KeyringClient instances per snap ID to avoid re-instantiation */
         _SnapDataSource_keyringClientCache.set(this, new Map());
         __classPrivateFieldSet(this, _SnapDataSource_messenger, options.messenger, "f");
+        __classPrivateFieldSet(this, _SnapDataSource_onActiveChainsUpdated, options.onActiveChainsUpdated, "f");
         // Bind handlers for cleanup in destroy()
         __classPrivateFieldSet(this, _SnapDataSource_handleSnapBalancesUpdatedBound, __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_handleSnapBalancesUpdated).bind(this), "f");
         __classPrivateFieldSet(this, _SnapDataSource_handlePermissionStateChangeBound, __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_discoverKeyringSnaps).bind(this), "f");
-        __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_registerActionHandlers).call(this);
         __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_subscribeToEvents).call(this);
         // Discover keyring-capable snaps and populate activeChains dynamically
         __classPrivateFieldGet(this, _SnapDataSource_instances, "m", _SnapDataSource_discoverKeyringSnaps).call(this);
@@ -107,15 +109,18 @@ export class SnapDataSource extends AbstractDataSource {
         var _a;
         // Guard against undefined request
         // Note: chainIds filtering is done by middleware/subscribe before calling fetch
-        if (!request?.accounts || !request?.chainIds?.length) {
+        if (!request?.chainIds?.length) {
             return {};
         }
+        if (!request?.accountsWithSupportedChains?.length) {
+            return { assetsBalance: {}, assetsMetadata: {} };
+        }
         const results = {
             assetsBalance: {},
             assetsMetadata: {},
         };
         // Fetch balances for each account using its snap ID from metadata
-        for (const account of request.accounts) {
+        for (const { account } of request.accountsWithSupportedChains) {
             // Skip accounts without snap metadata (non-snap accounts)
             const snapId = account.metadata.snap?.id;
             if (!snapId) {
@@ -258,7 +263,7 @@ export class SnapDataSource extends AbstractDataSource {
                 })
                     .then(async (fetchResponse) => {
                     if (Object.keys(fetchResponse.assetsBalance ?? {}).length > 0) {
-                        await __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('AssetsController:assetsUpdate', fetchResponse, SNAP_DATA_SOURCE_NAME);
+                        await existing.onAssetsUpdate(fetchResponse);
                     }
                     return fetchResponse;
                 })
@@ -277,6 +282,7 @@ export class SnapDataSource extends AbstractDataSource {
                 // No timer to clear - we use event-based updates
             },
             chains: supportedChains,
+            onAssetsUpdate: subscriptionRequest.onAssetsUpdate,
         });
         // Initial fetch to get current balances
         try {
@@ -284,8 +290,10 @@ export class SnapDataSource extends AbstractDataSource {
                 ...request,
                 chainIds: supportedChains,
             });
-            if (Object.keys(fetchResponse.assetsBalance ?? {}).length > 0) {
-                await __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('AssetsController:assetsUpdate', fetchResponse, SNAP_DATA_SOURCE_NAME);
+            const subscription = this.activeSubscriptions.get(subscriptionId);
+            if (Object.keys(fetchResponse.assetsBalance ?? {}).length > 0 &&
+                subscription) {
+                await subscription.onAssetsUpdate(fetchResponse);
             }
         }
         catch (error) {
@@ -322,14 +330,12 @@ export class SnapDataSource extends AbstractDataSource {
         __classPrivateFieldGet(this, _SnapDataSource_keyringClientCache, "f").clear();
     }
 }
-_SnapDataSource_messenger = new WeakMap(), _SnapDataSource_handleSnapBalancesUpdatedBound = new WeakMap(), _SnapDataSource_handlePermissionStateChangeBound = new WeakMap(), _SnapDataSource_keyringClientCache = new WeakMap(), _SnapDataSource_instances = new WeakSet(), _SnapDataSource_subscribeToEvents = function _SnapDataSource_subscribeToEvents() {
-    // Subscribe to snap keyring events for real-time balance updates.
-    // The snaps emit AccountBalancesUpdated events when balances change,
-    // which are re-published by AccountsController.
-    __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").subscribe('AccountsController:accountBalancesUpdated', __classPrivateFieldGet(this, _SnapDataSource_handleSnapBalancesUpdatedBound, "f"));
-    // Subscribe to permission changes to detect new keyring snaps at runtime.
-    // Re-runs snap discovery when permissions change.
-    __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").subscribe('PermissionController:stateChange', __classPrivateFieldGet(this, _SnapDataSource_handlePermissionStateChangeBound, "f"));
+_SnapDataSource_messenger = new WeakMap(), _SnapDataSource_onActiveChainsUpdated = new WeakMap(), _SnapDataSource_handleSnapBalancesUpdatedBound = new WeakMap(), _SnapDataSource_handlePermissionStateChangeBound = new WeakMap(), _SnapDataSource_keyringClientCache = new WeakMap(), _SnapDataSource_instances = new WeakSet(), _SnapDataSource_subscribeToEvents = function _SnapDataSource_subscribeToEvents() {
+    // Subscribe to snap keyring events and permission changes (not in AssetsControllerEvents).
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const messenger = __classPrivateFieldGet(this, _SnapDataSource_messenger, "f");
+    messenger.subscribe('AccountsController:accountBalancesUpdated', __classPrivateFieldGet(this, _SnapDataSource_handleSnapBalancesUpdatedBound, "f"));
+    messenger.subscribe('PermissionController:stateChange', __classPrivateFieldGet(this, _SnapDataSource_handlePermissionStateChangeBound, "f"));
 }, _SnapDataSource_handleSnapBalancesUpdated = function _SnapDataSource_handleSnapBalancesUpdated(payload) {
     // Transform the snap keyring payload to DataResponse format
     let assetsBalance;
@@ -352,22 +358,15 @@ _SnapDataSource_messenger = new WeakMap(), _SnapDataSource_handleSnapBalancesUpd
     // Only report if we have snap-related updates
     if (assetsBalance) {
         const response = { assetsBalance };
-        __classPrivateFieldGet(this, _SnapDataSource_messenger, "f")
-            .call('AssetsController:assetsUpdate', response, SNAP_DATA_SOURCE_NAME)
-            .catch(console.error);
+        for (const subscription of this.activeSubscriptions.values()) {
+            subscription.onAssetsUpdate(response)?.catch(console.error);
+        }
     }
 }, _SnapDataSource_isChainSupportedBySnap = function _SnapDataSource_isChainSupportedBySnap(chainId) {
     return this.state.activeChains.includes(chainId);
-}, _SnapDataSource_registerActionHandlers = function _SnapDataSource_registerActionHandlers() {
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const messenger = __classPrivateFieldGet(this, _SnapDataSource_messenger, "f");
-    messenger.registerActionHandler('SnapDataSource:getAssetsMiddleware', () => this.assetsMiddleware);
-    messenger.registerActionHandler('SnapDataSource:getActiveChains', async () => this.getActiveChains());
-    messenger.registerActionHandler('SnapDataSource:fetch', async (request) => this.fetch(request));
-    messenger.registerActionHandler('SnapDataSource:subscribe', async (request) => this.subscribe(request));
-    messenger.registerActionHandler('SnapDataSource:unsubscribe', async (subscriptionId) => this.unsubscribe(subscriptionId));
 }, _SnapDataSource_getRunnableSnaps = function _SnapDataSource_getRunnableSnaps() {
     try {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
         return __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('SnapController:getRunnableSnaps');
     }
     catch (error) {
@@ -376,6 +375,7 @@ _SnapDataSource_messenger = new WeakMap(), _SnapDataSource_handleSnapBalancesUpd
     }
 }, _SnapDataSource_getSnapPermissions = function _SnapDataSource_getSnapPermissions(snapId) {
     try {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
         return __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('PermissionController:getPermissions', snapId);
     }
     catch (error) {
@@ -412,7 +412,7 @@ _SnapDataSource_messenger = new WeakMap(), _SnapDataSource_handleSnapBalancesUpd
         // Notify if chains changed
         try {
             this.updateActiveChains(supportedChains, (updatedChains) => {
-                __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('AssetsController:activeChainsUpdate', SNAP_DATA_SOURCE_NAME, updatedChains);
+                __classPrivateFieldGet(this, _SnapDataSource_onActiveChainsUpdated, "f").call(this, updatedChains);
             });
         }
         catch {
@@ -424,7 +424,7 @@ _SnapDataSource_messenger = new WeakMap(), _SnapDataSource_handleSnapBalancesUpd
         this.state.chainToSnap = {};
         try {
             this.updateActiveChains([], (updatedChains) => {
-                __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('AssetsController:activeChainsUpdate', SNAP_DATA_SOURCE_NAME, updatedChains);
+                __classPrivateFieldGet(this, _SnapDataSource_onActiveChainsUpdated, "f").call(this, updatedChains);
             });
         }
         catch {
@@ -437,12 +437,12 @@ _SnapDataSource_messenger = new WeakMap(), _SnapDataSource_handleSnapBalancesUpd
         return cachedClient;
     }
     const client = new KeyringClient({
-        send: async (request) => (await __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('SnapController:handleRequest', {
+        send: async (request) => await __classPrivateFieldGet(this, _SnapDataSource_messenger, "f").call('SnapController:handleRequest', {
             snapId: snapId,
             origin: 'metamask',
             handler: HandlerType.OnKeyringRequest,
             request,
-        })),
+        }),
     });
     __classPrivateFieldGet(this, _SnapDataSource_keyringClientCache, "f").set(snapId, client);
     return client;