@metamask-previews/assets-controller 2.0.0-preview-33dbba4f3 → 2.0.0-preview-a196307b6
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +6 -0
- package/dist/data-sources/RpcDataSource.cjs +1 -2
- package/dist/data-sources/RpcDataSource.cjs.map +1 -1
- package/dist/data-sources/RpcDataSource.d.cts.map +1 -1
- package/dist/data-sources/RpcDataSource.d.mts.map +1 -1
- package/dist/data-sources/RpcDataSource.mjs +1 -2
- package/dist/data-sources/RpcDataSource.mjs.map +1 -1
- package/dist/data-sources/StakedBalanceDataSource.cjs +6 -18
- package/dist/data-sources/StakedBalanceDataSource.cjs.map +1 -1
- package/dist/data-sources/StakedBalanceDataSource.d.cts.map +1 -1
- package/dist/data-sources/StakedBalanceDataSource.d.mts.map +1 -1
- package/dist/data-sources/StakedBalanceDataSource.mjs +6 -18
- package/dist/data-sources/StakedBalanceDataSource.mjs.map +1 -1
- package/dist/data-sources/evm-rpc-services/services/StakedBalanceFetcher.cjs +6 -6
- package/dist/data-sources/evm-rpc-services/services/StakedBalanceFetcher.cjs.map +1 -1
- package/dist/data-sources/evm-rpc-services/services/StakedBalanceFetcher.d.cts +2 -0
- package/dist/data-sources/evm-rpc-services/services/StakedBalanceFetcher.d.cts.map +1 -1
- package/dist/data-sources/evm-rpc-services/services/StakedBalanceFetcher.d.mts +2 -0
- package/dist/data-sources/evm-rpc-services/services/StakedBalanceFetcher.d.mts.map +1 -1
- package/dist/data-sources/evm-rpc-services/services/StakedBalanceFetcher.mjs +1 -1
- package/dist/data-sources/evm-rpc-services/services/StakedBalanceFetcher.mjs.map +1 -1
- package/package.json +1 -1
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"StakedBalanceDataSource.cjs","sourceRoot":"","sources":["../../src/data-sources/StakedBalanceDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2CAAqD;AACrD,wDAAwD;AAGxD,2CAKyB;AAOzB,iEAA0D;AAK1D,mEAI4B;AAE5B,0CAA8D;AAY9D,MAAM,eAAe,GAAG,yBAAyB,CAAC;AAClD,MAAM,qBAAqB,GAAG,MAAO,CAAC,CAAC,YAAY;AAEnD,wEAAwE;AACxE,MAAM,mBAAmB,GAAkB;IACzC,IAAI,EAAE,OAAO;IACb,IAAI,EAAE,iBAAiB;IACvB,MAAM,EAAE,KAAK;IACb,QAAQ,EAAE,EAAE;CACb,CAAC;AAEF,MAAM,GAAG,GAAG,IAAA,2BAAkB,EAAC,sBAAa,EAAE,eAAe,CAAC,CAAC;AA6C/D;;;;;GAKG;AACH,SAAS,gBAAgB,CAAC,OAAe;IACvC,IAAI,IAAA,yBAAiB,EAAC,OAAO,CAAC,EAAE,CAAC;QAC/B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,IAAA,qBAAa,EAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,MAAM,GAAG,GAAG,IAAA,wBAAgB,EAAC,OAAO,CAAC,CAAC,SAAS,CAAC;QAChD,OAAO,IAAA,mBAAW,EAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;IACxC,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;AAChF,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,aAAa,CACpB,OAAgB,EAChB,eAAuB;IAEvB,MAAM,WAAW,GAAG,IAAA,wBAAiB,EAAC,eAAe,CAAC,CAAC;IACvD,OAAO,GAAG,OAAO,UAAU,WAAW,EAAmB,CAAC;AAC5D,CAAC;AAED;;;;;;;;;;GAUG;AACH,MAAa,uBAAwB,SAAQ,uCAG5C;IAyBC,YAAY,OAAuC;QACjD,KAAK,CAAC,eAAe,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;;QAzBtC,qDAAsC;QAEtC,iEAIC;QAED,wDAAsB;QAEtB,mDAAkB;QAE3B,mEAAmE;QAC1D,gEAA4C;QAErD,kCAAkC;QACzB,uDAAsD,IAAI,GAAG,EAAE,EAAC;QAEzE,uDAAuD;QAC9C,iDAAyC,IAAI,GAAG,EAAE,EAAC;QAE5D,0DAA0D;QACjD,6DAA8B;QA0D9B,mEACP,SAAS,EAAC;QAEH,mEACP,SAAS,EAAC;QAEH,iEAA2D,SAAS,EAAC;QAErE,qFAEO,SAAS,EAAC;QAhExB,uBAAA,IAAI,sCAAc,OAAO,CAAC,SAAS,MAAA,CAAC;QACpC,uBAAA,IAAI,kDAA0B,OAAO,CAAC,qBAAqB,MAAA,CAAC;QAC5D,uBAAA,IAAI,yCAAiB,OAAO,CAAC,YAAY,IAAI,qBAAqB,MAAA,CAAC;QACnE,uBAAA,IAAI,oCAAY,OAAO,CAAC,OAAO,KAAK,KAAK,MAAA,CAAC;QAC1C,uBAAA,IAAI,8CAAsB,IAAA,8CAA2B,GAAe,MAAA,CAAC;QAErE,GAAG,CAAC,sCAAsC,EAAE;YAC1C,OAAO,EAAE,uBAAA,IAAI,wCAAS;YACtB,YAAY,EAAE,uBAAA,IAAI,6CAAc;SACjC,CAAC,CAAC;QAEH,mDAAmD;QACnD,uBAAA,IAAI,iDAAyB,IAAI,uCAAoB,CAAC;YACpD,eAAe,EAAE,uBAAA,IAAI,6CAAc;YACnC,kBAAkB,EAAE,CAAC,UAAU,EAA4B,EAAE,CAC3D,uBAAA,IAAI,gFAAa,MAAjB,IAAI,EAAc,UAAU,CAAC;SAChC,CAAC,MAAA,CAAC;QAEH,kEAAkE;QAClE,uBAAA,IAAI,qDAAsB,CAAC,wBAAwB,CACjD,uBAAA,IAAI,8FAA2B,CAAC,IAAI,CAAC,IAAI,CAAC,CAC3C,CAAC;QAEF,MAAM,cAAc,GAAG,uBAAA,IAAI,0CAAW,CAAC,SAAS,CAC9C,4CAA4C,EAC5C,uBAAA,IAAI,2FAAwB,CAAC,IAAI,CAAC,IAAI,CAAC,CACxC,CAAC;QACF,uBAAA,IAAI,4DACF,OAAO,cAAc,KAAK,UAAU,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS,MAAA,CAAC;QAEpE,MAAM,aAAa,GAAG,uBAAA,IAAI,0CAAW,CAAC,SAAS,CAC7C,oDAAoD,EACpD,uBAAA,IAAI,2FAAwB,CAAC,IAAI,CAAC,IAAI,CAAC,CACxC,CAAC;QACF,uBAAA,IAAI,4DACF,OAAO,aAAa,KAAK,UAAU,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,MAAA,CAAC;QAElE,MAAM,YAAY,GAAG,uBAAA,IAAI,0CAAW,CAAC,SAAS,CAC5C,+BAA+B,EAC/B,uBAAA,IAAI,yFAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CACtC,CAAC;QACF,uBAAA,IAAI,0DACF,OAAO,YAAY,KAAK,UAAU,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,MAAA,CAAC;QAEhE,MAAM,eAAe,GAAG,uBAAA,IAAI,0CAAW,CAAC,SAAS,CAC/C,yCAAyC,EACzC,uBAAA,IAAI,6GAA0C,CAAC,IAAI,CAAC,IAAI,CAAC,CAC1D,CAAC;QACF,uBAAA,IAAI,8EACF,OAAO,eAAe,KAAK,UAAU,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,SAAS,MAAA,CAAC;QAEtE,uBAAA,IAAI,2FAAwB,MAA5B,IAAI,CAA0B,CAAC;IACjC,CAAC;IAyLD;;;OAGG;IACH,KAAK,CAAC,oBAAoB;QACxB,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,uBAAA,IAAI,oFAAiB,MAArB,IAAI,CAAmB,CAAC;QAC1C,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,MAAM,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,SAAS,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAgOD;;;;;OAKG;IACH,KAAK,CAAC,KAAK,CAAC,OAAoB;;QAC9B,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;YACnB,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,QAAQ,GAAiB,EAAE,CAAC;QAClC,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAEzD,MAAM,aAAa,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CACxD,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAC7B,CAAC;QAEF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,QAAQ,GAA2D,EAAE,CAAC;QAE5E,KAAK,MAAM,EACT,OAAO,EACP,eAAe,EAAE,aAAa,GAC/B,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;YACzC,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAC5C,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,CAC9B,CAAC;YAEF,KAAK,MAAM,OAAO,IAAI,MAAM,EAAE,CAAC;gBAC7B,IAAI,CAAC;oBACH,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;oBAC7C,MAAM,eAAe,GAAG,IAAA,4CAAyB,EAAC,UAAU,CAAC,CAAC;oBAC9D,IAAI,CAAC,eAAe,EAAE,CAAC;wBACrB,SAAS;oBACX,CAAC;oBAED,MAAM,KAAK,GAA8B;wBACvC,OAAO,EAAE,UAAU;wBACnB,SAAS,EAAE,OAAO,CAAC,EAAE;wBACrB,cAAc,EAAE,OAAO,CAAC,OAAc;qBACvC,CAAC;oBAEF,MAAM,MAAM,GACV,MAAM,uBAAA,IAAI,qDAAsB,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;oBAE7D,qEAAqE;oBACrE,QAAQ,MAAC,OAAO,CAAC,EAAE,MAAnB,QAAQ,OAAiB,EAAE,EAAC;oBAC5B,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;oBACxD,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC5D,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,GAAG,CAAC,gCAAgC,EAAE;wBACpC,OAAO;wBACP,SAAS,EAAE,OAAO,CAAC,EAAE;wBACrB,KAAK;qBACN,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrC,QAAQ,CAAC,aAAa,GAAG,QAAQ,CAAC;YAClC,4DAA4D;YAC5D,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAiB,CAAC;YAC1C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACtD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;oBACnD,QAAQ,CAAC,GAAG,CAAC,OAAwB,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;YACD,QAAQ,CAAC,UAAU,GAAG,EAAE,CAAC;YACzB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,mBAAmB,CAAC;YACrD,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;;;OAMG;IACH,IAAI,gBAAgB;QAClB,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;;YAC7B,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;gBACnB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YACD,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;YAE5B,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC3C,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAClC,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAEhD,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;oBAC7B,MAAA,OAAO,CAAC,QAAQ,EAAC,UAAU,QAAV,UAAU,GAAK,EAAE,EAAC;oBACnC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,aAAa,CAAC,UAAU,CAAC,CAAC;gBACvE,CAAC;gBACD,IAAI,aAAa,CAAC,aAAa,EAAE,CAAC;oBAChC,MAAA,OAAO,CAAC,QAAQ,EAAC,aAAa,QAAb,aAAa,GAAK,EAAE,EAAC;oBACtC,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,aAAa,CAAC,aAAa,CAC5B,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;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC5C,CAAC;YAED,gEAAgE;YAChE,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,SAAS,CAAC,mBAAwC;QACtD,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC;QAElE,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACzD,MAAM,iBAAiB,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5D,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAC7B,CAAC;QAEF,GAAG,CAAC,qBAAqB,EAAE;YACzB,cAAc;YACd,QAAQ;YACR,iBAAiB;SAClB,CAAC,CAAC;QAEH,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,GAAG,CAAC,gCAAgC,CAAC,CAAC;YACtC,OAAO;QACT,CAAC;QAED,8DAA8D;QAC9D,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,uBAAA,IAAI,oDAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC/D,IAAI,QAAQ,EAAE,CAAC;gBACb,GAAG,CAAC,qDAAqD,EAAE;oBACzD,cAAc;iBACf,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,qDAAqD;QACrD,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;QAEvC,4EAA4E;QAC5E,MAAM,2BAA2B,GAC/B,OAAO,CAAC,2BAA2B;aAChC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,CAAC;YACtC,OAAO;YACP,eAAe,EAAE,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAClD,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,CAChC;SACF,CAAC,CAAC;aACF,MAAM,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAEjE,MAAM,QAAQ,GAAG,2BAA2B,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC3E,MAAM,aAAa,GAAa,EAAE,CAAC;QAEnC,uFAAuF;QACvF,uBAAA,IAAI,oDAAqB,CAAC,GAAG,CAAC,cAAc,EAAE;YAC5C,aAAa;YACb,MAAM,EAAE,iBAAiB;YACzB,QAAQ;YACR,2BAA2B;YAC3B,cAAc,EAAE,mBAAmB,CAAC,cAAc;SACnD,CAAC,CAAC;QAEH,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,EAAE;YAC3C,OAAO,EAAE,GAAG,EAAE;gBACZ,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;oBAClC,uBAAA,IAAI,qDAAsB,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;gBAC9D,CAAC;gBACD,uBAAA,IAAI,oDAAqB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,EAAE,iBAAiB;YACzB,OAAO;YACP,cAAc,EAAE,mBAAmB,CAAC,cAAc;SACnD,CAAC,CAAC;QAEH,sEAAsE;QACtE,KAAK,MAAM,EACT,OAAO,EACP,eAAe,EAAE,aAAa,GAC/B,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;YACzC,MAAM,gBAAgB,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAC1D,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,CAC9B,CAAC;YAEF,KAAK,MAAM,OAAO,IAAI,gBAAgB,EAAE,CAAC;gBACvC,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;gBAE7C,MAAM,KAAK,GAA8B;oBACvC,OAAO,EAAE,UAAU;oBACnB,SAAS,EAAE,OAAO,CAAC,EAAE;oBACrB,cAAc,EAAE,OAAO,CAAC,OAAc;iBACvC,CAAC;gBAEF,MAAM,YAAY,GAAG,uBAAA,IAAI,qDAAsB,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpE,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,6EAA6E;QAC7E,IAAI,CAAC;YACH,MAAM,cAAc,GAAgB;gBAClC,2BAA2B;gBAC3B,QAAQ,EAAE,iBAAiB;gBAC3B,SAAS,EAAE,CAAC,SAAS,CAAC;aACvB,CAAC;YACF,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YACzD,IACE,eAAe,CAAC,aAAa;gBAC7B,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC,MAAM,GAAG,CAAC,EACrD,CAAC;gBACD,mBAAmB;qBAChB,cAAc,EAAE,CAAC,eAAe,CAAC;oBAClC,EAAE,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBAChB,GAAG,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;gBACzD,CAAC,CAAC,CAAC;YACP,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,qCAAqC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACxD,CAAC;QAED,GAAG,CAAC,sBAAsB,EAAE;YAC1B,cAAc;YACd,MAAM,EAAE,iBAAiB;YACzB,YAAY,EAAE,aAAa,CAAC,MAAM;SACnC,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,WAAW,CAAC,cAAsB;QACtC,MAAM,YAAY,GAAG,uBAAA,IAAI,oDAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACnE,IAAI,YAAY,EAAE,CAAC;YACjB,KAAK,MAAM,KAAK,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;gBAC/C,uBAAA,IAAI,qDAAsB,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;YAC9D,CAAC;YACD,uBAAA,IAAI,oDAAqB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,KAAK,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,OAAO;QACL,uBAAA,IAAI,gEAAiC,EAAE,KAAvC,IAAI,CAAqC,CAAC;QAC1C,uBAAA,IAAI,gEAAiC,EAAE,KAAvC,IAAI,CAAqC,CAAC;QAC1C,uBAAA,IAAI,8DAA+B,EAAE,KAArC,IAAI,CAAmC,CAAC;QACxC,uBAAA,IAAI,kFAAmD,EAAE,KAAzD,IAAI,CAAuD,CAAC;QAC5D,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YAC9D,KAAK,MAAM,KAAK,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;gBAC/C,uBAAA,IAAI,qDAAsB,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;QACD,uBAAA,IAAI,oDAAqB,CAAC,KAAK,EAAE,CAAC;QAClC,uBAAA,IAAI,8CAAe,CAAC,KAAK,EAAE,CAAC;QAC5B,KAAK,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC;CACF;AAvxBD,0DAuxBC;;IAjrBG,uBAAA,IAAI,8CAAe,CAAC,KAAK,EAAE,CAAC;IAC5B,GAAG,CAAC,mDAAmD,CAAC,CAAC;AAC3D,CAAC,iJASC,KAAuC;IAEvC,MAAM,EAAE,iBAAiB,EAAE,GAAG,KAAK,IAAI,EAAE,CAAC;IAC1C,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,OAAO;IACT,CAAC;IACD,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,iBAAiB,CAAC,CAAC;AAChE,CAAC,2IAasC,OAGtC;IACC,MAAM,UAAU,GAAG,OAAO,EAAE,OAAO,CAAC;IACpC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,eAAe,GAAG,IAAA,4CAAyB,EAAC,UAAU,CAAC,CAAC;IAC9D,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,aAAa,GAAG,eAAe,CAAC,WAAW,EAAE,CAAC;IACpD,MAAM,IAAI,GAAG,OAAO,CAAC,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;IACnD,MAAM,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE,EAAE,EAAE,WAAW,EAAE,CAAC;IAC/C,OAAO,IAAI,KAAK,aAAa,IAAI,EAAE,KAAK,aAAa,CAAC;AACxD,CAAC,6GAYuB,OAGvB;IACC,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;QACnB,OAAO;IACT,CAAC;IACD,IAAI,CAAC,uBAAA,IAAI,0GAAuC,MAA3C,IAAI,EAAwC,OAAO,CAAC,EAAE,CAAC;QAC1D,OAAO;IACT,CAAC;IACD,MAAM,UAAU,GAAG,OAAO,EAAE,OAAO,CAAC;IACpC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO;IACT,CAAC;IACD,MAAM,WAAW,GAAG,UAAU,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,EAAa,CAAC;IACpE,MAAM,SAAS,GAAG,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,CAAC,WAAW,CAAC,CAAC,CAAC;IAC7D,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACpE,GAAG,CAAC,oDAAoD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC,6GASC,OAA0E;IAE1E,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;QACnB,OAAO;IACT,CAAC;IACD,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAU,CAAC;IAC5C,KAAK,MAAM,IAAI,IAAI,OAAO,IAAI,EAAE,EAAE,CAAC;QACjC,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC;YACnB,SAAS;QACX,CAAC;QACD,IAAI,uBAAA,IAAI,0GAAuC,MAA3C,IAAI,EAAwC,IAAI,CAAC,EAAE,CAAC;YACtD,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IACD,MAAM,YAAY,GAAG,CAAC,GAAG,iBAAiB,CAAC,CAAC,GAAG,CAC7C,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,EAAa,CAChE,CAAC;IACF,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO;IACT,CAAC;IACD,MAAM,SAAS,GAAG,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,YAAY,CAAC,CAAC;IAC5D,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACpE,GAAG,CAAC,8DAA8D,EAAE;gBAClE,KAAK;aACN,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC,2GASC,QAAmB;IAEnB,MAAM,SAAS,GAAqD,EAAE,CAAC;IACvE,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC;IACnC,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9D,KAAK,MAAM,EACT,OAAO,EACP,eAAe,GAChB,IAAI,YAAY,CAAC,2BAA2B,EAAE,CAAC;YAC9C,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;gBACtC,IAAI,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC1B,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;gBACvC,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;IAQC,MAAM,SAAS,GAAqD,EAAE,CAAC;IACvE,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9D,KAAK,MAAM,EACT,OAAO,EACP,eAAe,GAChB,IAAI,YAAY,CAAC,2BAA2B,EAAE,CAAC;YAC9C,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;gBACtC,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YACvC,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAgBD;;;;;GAKG;AACH,KAAK,wEACH,SAA2D;IAE3D,MAAM,UAAU,GAAyC,EAAE,CAAC;IAC5D,MAAM,aAAa,GAGf,EAAE,CAAC;IAEP,KAAK,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,SAAS,EAAE,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,eAAe,GAAG,IAAA,4CAAyB,EAAC,UAAU,CAAC,CAAC;YAC9D,IAAI,CAAC,eAAe,EAAE,CAAC;gBACrB,SAAS;YACX,CAAC;YAED,MAAM,KAAK,GAA8B;gBACvC,OAAO,EAAE,UAAU;gBACnB,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,cAAc,EAAE,OAAO,CAAC,OAAc;aACvC,CAAC;YAEF,MAAM,MAAM,GACV,MAAM,uBAAA,IAAI,qDAAsB,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAC7D,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;YAExD,UAAU,CAAC,OAAO,CAAC,GAAG,mBAAmB,CAAC;YAC1C,MAAM,QAAQ,GAAG,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3C,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,GAAG;gBAC1B,GAAG,QAAQ;gBACX,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE;aACrC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,uDAAuD,EAAE;gBAC3D,OAAO;gBACP,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,KAAK;aACN,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1C,MAAM,QAAQ,GAAiB,EAAE,UAAU,EAAE,aAAa,EAAE,CAAC;QAC7D,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YAC9D,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;gBAC9D,GAAG,CAAC,0DAA0D,EAAE;oBAC9D,KAAK;iBACN,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;AACH,CAAC;IAQC,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,uBAAA,IAAI,0CAAW,CAAC,IAAI,CAChC,sCAAsC,CACvC,CAAC;QACF,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EACF,KAAK,EAAE,iBAAiB,IAAI,EAAE,CAC/B,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,iDAAiD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAClE,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,EAAE,CAAC,CAAC;IACjD,CAAC;AACH,CAAC,yIAUC,iBAA0D;IAE1D,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;QACnB,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC9C,IAAI,CAAC,kBAAkB,CAAC,EAAE,EAAE,CAAC,aAAa,EAAE,EAAE,CAC5C,uBAAA,IAAI,sDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;QACF,OAAO;IACT,CAAC;IAED,MAAM,YAAY,GAAc,EAAE,CAAC;IACnC,MAAM,SAAS,GAAG,iBAAiB,CAAC,MAAM,CAAC;IAC3C,IAAI,SAAS,EAAE,CAAC;QACd,KAAK,MAAM,KAAK,IAAI,uBAAA,IAAI,kDAAmB,EAAE,CAAC;YAC5C,IAAI,CAAC,IAAA,qBAAa,EAAC,KAAK,CAAC,EAAE,CAAC;gBAC1B,SAAS;YACX,CAAC;YACD,MAAM,EAAE,SAAS,EAAE,GAAG,IAAA,wBAAgB,EAAC,KAAK,CAAC,CAAC;YAC9C,MAAM,UAAU,GAAG,IAAA,mBAAW,EAAC,QAAQ,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC;YACxD,IAAI,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC1B,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAC9C,IAAI,CAAC,kBAAkB,CAAC,YAAY,EAAE,CAAC,aAAa,EAAE,EAAE,CACtD,uBAAA,IAAI,sDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;AACJ,CAAC,uFASY,UAAe;IAC1B,MAAM,MAAM,GAAG,uBAAA,IAAI,8CAAe,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACnD,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,CAAC;QACH,MAAM,YAAY,GAAG,uBAAA,IAAI,0CAAW,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAExE,MAAM,EAAE,8BAA8B,EAAE,GAAG,YAAY,CAAC;QACxD,IAAI,CAAC,8BAA8B,EAAE,CAAC;YACpC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,WAAW,GAAG,8BAA8B,CAAC,UAAU,CAAC,CAAC;QAC/D,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,6EAA6E;QAC7E,MAAM,EAAE,YAAY,EAAE,uBAAuB,EAAE,GAAG,WAAW,CAAC;QAC9D,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/C,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,KAAK,GACT,OAAO,uBAAuB,KAAK,QAAQ;YAC3C,uBAAuB,IAAI,CAAC;YAC5B,uBAAuB,GAAG,YAAY,CAAC,MAAM;YAC3C,CAAC,CAAC,uBAAuB;YACzB,CAAC,CAAC,CAAC,CAAC;QACR,MAAM,eAAe,GAAG,YAAY,CAAC,KAAK,CAEzC,CAAC;QACF,MAAM,eAAe,GAAG,eAAe,EAAE,eAAe,CAAC;QACzD,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,aAAa,GAAG,uBAAA,IAAI,0CAAW,CAAC,IAAI,CACxC,wCAAwC,EACxC,eAAe,CAChB,CAAC;QAEF,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,CAAC;YAC7B,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,wBAAY,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC1D,uBAAA,IAAI,8CAAe,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC9C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,kCAAkC,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/D,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC,mHAS0B,MAAgC;IACzD,MAAM,eAAe,GAAG,IAAA,4CAAyB,EAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAClE,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO;IACT,CAAC;IACD,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IACpD,MAAM,WAAW,GAAG,UAAU,cAAc,EAAa,CAAC;IAC1D,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC;IAE5D,MAAM,QAAQ,GAAiB;QAC7B,UAAU,EAAE;YACV,CAAC,OAAO,CAAC,EAAE,mBAAmB;SAC/B;QACD,aAAa,EAAE;YACb,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE;gBAClB,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE;aAC7C;SACF;KACF,CAAC;IAEF,GAAG,CAAC,uBAAuB,EAAE;QAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,OAAO,EAAE,WAAW;QACpB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM;KAC9B,CAAC,CAAC;IAEH,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9D,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YAC9D,GAAG,CAAC,wCAAwC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC","sourcesContent":["import { toChecksumAddress } from '@ethereumjs/util';\nimport { Web3Provider } from '@ethersproject/providers';\nimport type { InternalAccount } from '@metamask/keyring-internal-api';\nimport type { NetworkEnablementControllerState } from '@metamask/network-enablement-controller';\nimport {\n isStrictHexString,\n isCaipChainId,\n numberToHex,\n parseCaipChainId,\n} from '@metamask/utils';\nimport type { Hex } from '@metamask/utils';\n\nimport type {\n DataSourceState,\n SubscriptionRequest,\n} from './AbstractDataSource';\nimport { AbstractDataSource } from './AbstractDataSource';\nimport type {\n StakedBalancePollingInput,\n StakedBalanceFetchResult,\n} from './evm-rpc-services';\nimport {\n StakedBalanceFetcher,\n getStakingContractAddress,\n getSupportedStakingChainIds,\n} from './evm-rpc-services';\nimport type { AssetsControllerMessenger } from '../AssetsController';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport type {\n AccountId,\n ChainId,\n Caip19AssetId,\n AssetBalance,\n AssetMetadata,\n DataRequest,\n DataResponse,\n Middleware,\n} from '../types';\n\nconst CONTROLLER_NAME = 'StakedBalanceDataSource';\nconst DEFAULT_POLL_INTERVAL = 180_000; // 3 minutes\n\n/** Metadata for staked ETH (same symbol and decimals as native ETH). */\nconst STAKED_ETH_METADATA: AssetMetadata = {\n type: 'erc20',\n name: 'staked ethereum',\n symbol: 'ETH',\n decimals: 18,\n};\n\nconst log = createModuleLogger(projectLogger, CONTROLLER_NAME);\n\n/** Optional configuration for StakedBalanceDataSource. */\nexport type StakedBalanceDataSourceConfig = {\n /** Whether staked balance fetching is enabled (default: true). */\n enabled?: boolean;\n /** Polling interval in ms (default: 180s / 3 min). */\n pollInterval?: number;\n};\n\nexport type StakedBalanceDataSourceOptions = StakedBalanceDataSourceConfig & {\n /** The AssetsController messenger (for accessing NetworkController). */\n messenger: AssetsControllerMessenger;\n /** Called when active chains are updated. */\n onActiveChainsUpdated: (\n dataSourceName: string,\n chains: ChainId[],\n previousChains: ChainId[],\n ) => void;\n};\n\n/** Per-account supported chains (same shape as in DataRequest). */\ntype AccountWithSupportedChains = {\n account: InternalAccount;\n supportedChains: ChainId[];\n};\n\n/**\n * Subscription data stored for each active subscription.\n * Stores accountsWithSupportedChains so refresh paths use the same per-account\n * scope as normal subscription setup (avoids querying unsupported chains/accounts).\n */\ntype SubscriptionData = {\n /** Polling tokens from StakedBalanceFetcher. */\n pollingTokens: string[];\n /** Chain IDs being polled (union of all account chains). */\n chains: ChainId[];\n /** Accounts being polled. */\n accounts: InternalAccount[];\n /** Per-account supported chains; used by refreshStakedBalance and transaction handlers. */\n accountsWithSupportedChains: AccountWithSupportedChains[];\n /** Callback to report asset updates. */\n onAssetsUpdate: (response: DataResponse) => void | Promise<void>;\n};\n\n/**\n * Convert CAIP chain ID or hex chain ID to hex chain ID.\n *\n * @param chainId - CAIP chain ID or hex chain ID.\n * @returns Hex chain ID.\n */\nfunction caipChainIdToHex(chainId: string): Hex {\n if (isStrictHexString(chainId)) {\n return chainId;\n }\n\n if (isCaipChainId(chainId)) {\n const ref = parseCaipChainId(chainId).reference;\n return numberToHex(parseInt(ref, 10));\n }\n\n throw new Error('caipChainIdToHex - Failed to provide CAIP-2 or Hex chainId');\n}\n\n/**\n * Build the CAIP-19 asset ID for staked balance (same format as ERC20).\n * Uses the staking contract address (checksummed) so it is consistent with\n * other token assets.\n * Example: \"eip155:1/erc20:0x4fef9d741011476750a243ac70b9789a63dd47df\"\n *\n * @param chainId - CAIP-2 chain ID (e.g. \"eip155:1\").\n * @param contractAddress - Staking contract address (hex).\n * @returns The staked asset CAIP-19 ID with checksummed address.\n */\nfunction stakedAssetId(\n chainId: ChainId,\n contractAddress: string,\n): Caip19AssetId {\n const checksummed = toChecksumAddress(contractAddress);\n return `${chainId}/erc20:${checksummed}` as Caip19AssetId;\n}\n\n/**\n * Data source for fetching staked ETH balances via on-chain staking contracts.\n *\n * Delegates to {@link StakedBalanceFetcher} for the actual RPC calls\n * (getShares + convertToAssets on ERC-4626-style staking contracts).\n * Reports balances as CAIP-19 asset IDs using the ERC20 format with the\n * staking contract address (e.g. \"eip155:1/erc20:0x4fef9d741011476750a243ac70b9789a63dd47df\").\n *\n * Only supports chains with known staking contracts (mainnet, Hoodi).\n * Polling is managed by StakedBalanceFetcher via startPolling/stopPollingByPollingToken.\n */\nexport class StakedBalanceDataSource extends AbstractDataSource<\n typeof CONTROLLER_NAME,\n DataSourceState\n> {\n readonly #messenger: AssetsControllerMessenger;\n\n readonly #onActiveChainsUpdated: (\n dataSourceName: string,\n chains: ChainId[],\n previousChains: ChainId[],\n ) => void;\n\n readonly #pollInterval: number;\n\n readonly #enabled: boolean;\n\n /** The StakedBalanceFetcher that handles polling and RPC calls. */\n readonly #stakedBalanceFetcher: StakedBalanceFetcher;\n\n /** Active subscriptions by ID. */\n readonly #activeSubscriptions: Map<string, SubscriptionData> = new Map();\n\n /** Cache of Web3Provider instances by hex chain ID. */\n readonly #providerCache: Map<Hex, Web3Provider> = new Map();\n\n /** CAIP-2 chain IDs that have known staking contracts. */\n readonly #supportedChainIds: ChainId[];\n\n constructor(options: StakedBalanceDataSourceOptions) {\n super(CONTROLLER_NAME, { activeChains: [] });\n this.#messenger = options.messenger;\n this.#onActiveChainsUpdated = options.onActiveChainsUpdated;\n this.#pollInterval = options.pollInterval ?? DEFAULT_POLL_INTERVAL;\n this.#enabled = options.enabled !== false;\n this.#supportedChainIds = getSupportedStakingChainIds() as ChainId[];\n\n log('Initializing StakedBalanceDataSource', {\n enabled: this.#enabled,\n pollInterval: this.#pollInterval,\n });\n\n // Create StakedBalanceFetcher with provider getter\n this.#stakedBalanceFetcher = new StakedBalanceFetcher({\n pollingInterval: this.#pollInterval,\n getNetworkProvider: (hexChainId): Web3Provider | undefined =>\n this.#getProvider(hexChainId),\n });\n\n // Wire the callback so polling results flow back to subscriptions\n this.#stakedBalanceFetcher.setOnStakedBalanceUpdate(\n this.#handleStakedBalanceUpdate.bind(this),\n );\n\n const unsubConfirmed = this.#messenger.subscribe(\n 'TransactionController:transactionConfirmed',\n this.#onTransactionConfirmed.bind(this),\n );\n this.#unsubscribeTransactionConfirmed =\n typeof unsubConfirmed === 'function' ? unsubConfirmed : undefined;\n\n const unsubIncoming = this.#messenger.subscribe(\n 'TransactionController:incomingTransactionsReceived',\n this.#onIncomingTransactions.bind(this),\n );\n this.#unsubscribeIncomingTransactions =\n typeof unsubIncoming === 'function' ? unsubIncoming : undefined;\n\n const unsubNetwork = this.#messenger.subscribe(\n 'NetworkController:stateChange',\n this.#onNetworkStateChange.bind(this),\n );\n this.#unsubscribeNetworkStateChange =\n typeof unsubNetwork === 'function' ? unsubNetwork : undefined;\n\n const unsubEnablement = this.#messenger.subscribe(\n 'NetworkEnablementController:stateChange',\n this.#onNetworkEnablementControllerStateChange.bind(this),\n );\n this.#unsubscribeNetworkEnablementControllerStateChange =\n typeof unsubEnablement === 'function' ? unsubEnablement : undefined;\n\n this.#initializeActiveChains();\n }\n\n readonly #unsubscribeTransactionConfirmed: (() => void) | undefined =\n undefined;\n\n readonly #unsubscribeIncomingTransactions: (() => void) | undefined =\n undefined;\n\n readonly #unsubscribeNetworkStateChange: (() => void) | undefined = undefined;\n\n readonly #unsubscribeNetworkEnablementControllerStateChange:\n | (() => void)\n | undefined = undefined;\n\n /**\n * When NetworkController state changes (e.g. RPC endpoints or network clients\n * reconfigured), clear the provider cache so subsequent fetches use fresh\n * providers.\n */\n #onNetworkStateChange(): void {\n this.#providerCache.clear();\n log('Provider cache cleared after network state change');\n }\n\n /**\n * When NetworkEnablementController state changes (user enables/disables\n * networks), recompute active chains so we only fetch for enabled staking chains.\n *\n * @param state - The new NetworkEnablementController state.\n */\n #onNetworkEnablementControllerStateChange(\n state: NetworkEnablementControllerState,\n ): void {\n const { enabledNetworkMap } = state ?? {};\n if (!enabledNetworkMap) {\n return;\n }\n this.#initializeActiveChainsFromEnabledMap(enabledNetworkMap);\n }\n\n /**\n * Returns true if the transaction involves the staking contract (from or to)\n * for the payload's chain, so we only refresh staked balance when relevant.\n *\n * @param payload - Transaction payload.\n * @param payload.chainId - Hex chain ID.\n * @param payload.txParams - Optional transaction params.\n * @param payload.txParams.from - Sender address.\n * @param payload.txParams.to - Recipient address.\n * @returns True if txParams.from or txParams.to matches the staking contract address.\n */\n #isTransactionInvolvingStakingContract(payload: {\n chainId?: string;\n txParams?: { from?: string; to?: string };\n }): boolean {\n const hexChainId = payload?.chainId;\n if (!hexChainId) {\n return false;\n }\n const contractAddress = getStakingContractAddress(hexChainId);\n if (!contractAddress) {\n return false;\n }\n const contractLower = contractAddress.toLowerCase();\n const from = payload.txParams?.from?.toLowerCase();\n const to = payload.txParams?.to?.toLowerCase();\n return from === contractLower || to === contractLower;\n }\n\n /**\n * When a transaction is confirmed, refresh staked balance only if the\n * transaction is from or to the staking contract for that chain.\n *\n * @param payload - From TransactionController:transactionConfirmed.\n * @param payload.chainId - Hex chain ID of the transaction.\n * @param payload.txParams - Optional transaction params.\n * @param payload.txParams.from - Sender address.\n * @param payload.txParams.to - Recipient address.\n */\n #onTransactionConfirmed(payload: {\n chainId?: string;\n txParams?: { from?: string; to?: string };\n }): void {\n if (!this.#enabled) {\n return;\n }\n if (!this.#isTransactionInvolvingStakingContract(payload)) {\n return;\n }\n const hexChainId = payload?.chainId;\n if (!hexChainId) {\n return;\n }\n const caipChainId = `eip155:${parseInt(hexChainId, 16)}` as ChainId;\n const toRefresh = this.#getToRefreshForChains([caipChainId]);\n if (toRefresh.length > 0) {\n this.#refreshStakedBalanceAfterTransaction(toRefresh).catch((error) => {\n log('Failed to refresh staked balance after transaction', { error });\n });\n }\n }\n\n /**\n * When incoming transactions are received, refresh staked balance only for\n * chains where at least one transaction is from or to the staking contract.\n *\n * @param payload - From TransactionController:incomingTransactionsReceived (array of { chainId?, txParams? }).\n */\n #onIncomingTransactions(\n payload: { chainId?: string; txParams?: { from?: string; to?: string } }[],\n ): void {\n if (!this.#enabled) {\n return;\n }\n const chainIdsToRefresh = new Set<string>();\n for (const item of payload ?? []) {\n if (!item?.chainId) {\n continue;\n }\n if (this.#isTransactionInvolvingStakingContract(item)) {\n chainIdsToRefresh.add(item.chainId);\n }\n }\n const caipChainIds = [...chainIdsToRefresh].map(\n (hexChainId) => `eip155:${parseInt(hexChainId, 16)}` as ChainId,\n );\n if (caipChainIds.length === 0) {\n return;\n }\n const toRefresh = this.#getToRefreshForChains(caipChainIds);\n if (toRefresh.length > 0) {\n this.#refreshStakedBalanceAfterTransaction(toRefresh).catch((error) => {\n log('Failed to refresh staked balance after incoming transactions', {\n error,\n });\n });\n }\n }\n\n /**\n * Build toRefresh list for subscribed (account, chainId) pairs for the given chains.\n *\n * @param chainIds - CAIP-2 chain IDs to target.\n * @returns Pairs of account and chainId to refresh.\n */\n #getToRefreshForChains(\n chainIds: ChainId[],\n ): { account: InternalAccount; chainId: ChainId }[] {\n const toRefresh: { account: InternalAccount; chainId: ChainId }[] = [];\n const chainSet = new Set(chainIds);\n for (const subscription of this.#activeSubscriptions.values()) {\n for (const {\n account,\n supportedChains,\n } of subscription.accountsWithSupportedChains) {\n for (const chainId of supportedChains) {\n if (chainSet.has(chainId)) {\n toRefresh.push({ account, chainId });\n }\n }\n }\n }\n return toRefresh;\n }\n\n /**\n * Build toRefresh list for all subscribed (account, chainId) pairs.\n *\n * @returns Pairs of account and chainId to refresh.\n */\n #getToRefreshAll(): { account: InternalAccount; chainId: ChainId }[] {\n const toRefresh: { account: InternalAccount; chainId: ChainId }[] = [];\n for (const subscription of this.#activeSubscriptions.values()) {\n for (const {\n account,\n supportedChains,\n } of subscription.accountsWithSupportedChains) {\n for (const chainId of supportedChains) {\n toRefresh.push({ account, chainId });\n }\n }\n }\n return toRefresh;\n }\n\n /**\n * Refresh staked balance for all currently subscribed accounts and chains, then\n * push updates to the controller. Can be called from UI or after transaction events.\n */\n async refreshStakedBalance(): Promise<void> {\n if (!this.#enabled) {\n return;\n }\n const toRefresh = this.#getToRefreshAll();\n if (toRefresh.length > 0) {\n await this.#refreshStakedBalanceAfterTransaction(toRefresh);\n }\n }\n\n /**\n * Fetch staked balance for the given account/chain pairs and push a single\n * DataResponse to all active subscriptions.\n *\n * @param toRefresh - List of { account, chainId } to refresh.\n */\n async #refreshStakedBalanceAfterTransaction(\n toRefresh: { account: InternalAccount; chainId: ChainId }[],\n ): Promise<void> {\n const assetsInfo: Record<Caip19AssetId, AssetMetadata> = {};\n const assetsBalance: Record<\n AccountId,\n Record<Caip19AssetId, AssetBalance>\n > = {};\n\n for (const { account, chainId } of toRefresh) {\n try {\n const hexChainId = caipChainIdToHex(chainId);\n const contractAddress = getStakingContractAddress(hexChainId);\n if (!contractAddress) {\n continue;\n }\n\n const input: StakedBalancePollingInput = {\n chainId: hexChainId,\n accountId: account.id,\n accountAddress: account.address as Hex,\n };\n\n const result =\n await this.#stakedBalanceFetcher.fetchStakedBalance(input);\n const assetId = stakedAssetId(chainId, contractAddress);\n\n assetsInfo[assetId] = STAKED_ETH_METADATA;\n const existing = assetsBalance[account.id];\n assetsBalance[account.id] = {\n ...existing,\n [assetId]: { amount: result.amount },\n };\n } catch (error) {\n log('Failed to fetch staked balance in transaction refresh', {\n chainId,\n accountId: account.id,\n error,\n });\n }\n }\n\n if (Object.keys(assetsBalance).length > 0) {\n const response: DataResponse = { assetsInfo, assetsBalance };\n for (const subscription of this.#activeSubscriptions.values()) {\n subscription.onAssetsUpdate(response)?.catch((error: unknown) => {\n log('Failed to report staked balance update after transaction', {\n error,\n });\n });\n }\n }\n }\n\n /**\n * Set active chains from NetworkEnablementController state.\n * Only staking-supported chains that are enabled in the network enablement map\n * are active (e.g. if mainnet is not selected we do not fetch).\n */\n #initializeActiveChains(): void {\n try {\n const state = this.#messenger.call(\n 'NetworkEnablementController:getState',\n );\n this.#initializeActiveChainsFromEnabledMap(\n state?.enabledNetworkMap ?? {},\n );\n } catch (error) {\n log('Failed to get NetworkEnablementController state', { error });\n this.#initializeActiveChainsFromEnabledMap({});\n }\n }\n\n /**\n * Compute active chains as the intersection of supported staking chains and\n * enabled networks, then update state. Uses the same EIP-155 storage key\n * convention as NetworkEnablementController (hex for eip155).\n *\n * @param enabledNetworkMap - The enabled network map from NetworkEnablementController.\n */\n #initializeActiveChainsFromEnabledMap(\n enabledNetworkMap: Record<string, Record<string, boolean>>,\n ): void {\n if (!this.#enabled) {\n const previous = [...this.state.activeChains];\n this.updateActiveChains([], (updatedChains) =>\n this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n );\n return;\n }\n\n const activeChains: ChainId[] = [];\n const eip155Map = enabledNetworkMap.eip155;\n if (eip155Map) {\n for (const caip2 of this.#supportedChainIds) {\n if (!isCaipChainId(caip2)) {\n continue;\n }\n const { reference } = parseCaipChainId(caip2);\n const storageKey = numberToHex(parseInt(reference, 10));\n if (eip155Map[storageKey]) {\n activeChains.push(caip2);\n }\n }\n }\n\n const previous = [...this.state.activeChains];\n this.updateActiveChains(activeChains, (updatedChains) =>\n this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n );\n }\n\n /**\n * Get or create a Web3Provider for the given hex chain ID.\n * Uses the same messenger-cast pattern as RpcDataSource.\n *\n * @param hexChainId - The hex chain ID.\n * @returns Web3Provider instance, or undefined if not available.\n */\n #getProvider(hexChainId: Hex): Web3Provider | undefined {\n const cached = this.#providerCache.get(hexChainId);\n if (cached) {\n return cached;\n }\n\n try {\n const networkState = this.#messenger.call('NetworkController:getState');\n\n const { networkConfigurationsByChainId } = networkState;\n if (!networkConfigurationsByChainId) {\n return undefined;\n }\n\n const chainConfig = networkConfigurationsByChainId[hexChainId];\n if (!chainConfig) {\n return undefined;\n }\n\n // Use the network's configured default RPC endpoint (same as RpcDataSource).\n const { rpcEndpoints, defaultRpcEndpointIndex } = chainConfig;\n if (!rpcEndpoints || rpcEndpoints.length === 0) {\n return undefined;\n }\n\n const index =\n typeof defaultRpcEndpointIndex === 'number' &&\n defaultRpcEndpointIndex >= 0 &&\n defaultRpcEndpointIndex < rpcEndpoints.length\n ? defaultRpcEndpointIndex\n : 0;\n const defaultEndpoint = rpcEndpoints[index] as {\n networkClientId?: string;\n };\n const networkClientId = defaultEndpoint?.networkClientId;\n if (!networkClientId) {\n return undefined;\n }\n\n const networkClient = this.#messenger.call(\n 'NetworkController:getNetworkClientById',\n networkClientId,\n );\n\n if (!networkClient?.provider) {\n return undefined;\n }\n\n const provider = new Web3Provider(networkClient.provider);\n this.#providerCache.set(hexChainId, provider);\n return provider;\n } catch (error) {\n log('Failed to get provider for chain', { hexChainId, error });\n return undefined;\n }\n }\n\n /**\n * Handle a staked balance update from StakedBalanceFetcher.\n * Converts the result into a DataResponse and forwards it to all active\n * subscriptions.\n *\n * @param result - The staked balance fetch result.\n */\n #handleStakedBalanceUpdate(result: StakedBalanceFetchResult): void {\n const contractAddress = getStakingContractAddress(result.chainId);\n if (!contractAddress) {\n return;\n }\n const chainIdDecimal = parseInt(result.chainId, 16);\n const caipChainId = `eip155:${chainIdDecimal}` as ChainId;\n const assetId = stakedAssetId(caipChainId, contractAddress);\n\n const response: DataResponse = {\n assetsInfo: {\n [assetId]: STAKED_ETH_METADATA,\n },\n assetsBalance: {\n [result.accountId]: {\n [assetId]: { amount: result.balance.amount },\n },\n },\n };\n\n log('Staked balance update', {\n accountId: result.accountId,\n chainId: caipChainId,\n amount: result.balance.amount,\n });\n\n for (const subscription of this.#activeSubscriptions.values()) {\n subscription.onAssetsUpdate(response)?.catch((error: unknown) => {\n log('Failed to report staked balance update', { error });\n });\n }\n }\n\n /**\n * Fetch staked balances for all accounts on supported chains.\n *\n * @param request - The data request with accounts and chains.\n * @returns DataResponse with staked balance data.\n */\n async fetch(request: DataRequest): Promise<DataResponse> {\n if (!this.#enabled) {\n return {};\n }\n const response: DataResponse = {};\n const activeChainsSet = new Set(this.state.activeChains);\n\n const chainsToFetch = request.chainIds.filter((chainId) =>\n activeChainsSet.has(chainId),\n );\n\n if (chainsToFetch.length === 0) {\n return response;\n }\n\n const balances: Record<AccountId, Record<Caip19AssetId, AssetBalance>> = {};\n\n for (const {\n account,\n supportedChains: accountChains,\n } of request.accountsWithSupportedChains) {\n const chains = chainsToFetch.filter((chain) =>\n accountChains.includes(chain),\n );\n\n for (const chainId of chains) {\n try {\n const hexChainId = caipChainIdToHex(chainId);\n const contractAddress = getStakingContractAddress(hexChainId);\n if (!contractAddress) {\n continue;\n }\n\n const input: StakedBalancePollingInput = {\n chainId: hexChainId,\n accountId: account.id,\n accountAddress: account.address as Hex,\n };\n\n const result =\n await this.#stakedBalanceFetcher.fetchStakedBalance(input);\n\n // Include zero amounts so merged updates clear prior non-zero state.\n balances[account.id] ??= {};\n const assetId = stakedAssetId(chainId, contractAddress);\n balances[account.id][assetId] = { amount: result.amount };\n } catch (error) {\n log('Failed to fetch staked balance', {\n chainId,\n accountId: account.id,\n error,\n });\n }\n }\n }\n\n if (Object.keys(balances).length > 0) {\n response.assetsBalance = balances;\n // Add metadata for each staked asset ID present in balances\n const assetIds = new Set<Caip19AssetId>();\n for (const accountBalances of Object.values(balances)) {\n for (const assetId of Object.keys(accountBalances)) {\n assetIds.add(assetId as Caip19AssetId);\n }\n }\n response.assetsInfo = {};\n for (const assetId of assetIds) {\n response.assetsInfo[assetId] = STAKED_ETH_METADATA;\n }\n }\n\n return response;\n }\n\n /**\n * Assets middleware for the fetch pipeline.\n * Fetches staked balances and merges them into the response, then passes\n * all chains to the next middleware (staked balance doesn't claim chains).\n *\n * @returns The middleware function for the assets pipeline.\n */\n get assetsMiddleware(): Middleware {\n return async (context, next) => {\n if (!this.#enabled) {\n return next(context);\n }\n const { request } = context;\n\n if (!request.dataTypes.includes('balance')) {\n return next(context);\n }\n\n if (request.chainIds.length === 0) {\n return next(context);\n }\n\n try {\n const fetchResponse = await this.fetch(request);\n\n if (fetchResponse.assetsInfo) {\n context.response.assetsInfo ??= {};\n Object.assign(context.response.assetsInfo, fetchResponse.assetsInfo);\n }\n if (fetchResponse.assetsBalance) {\n context.response.assetsBalance ??= {};\n for (const [accountId, accountBalances] of Object.entries(\n fetchResponse.assetsBalance,\n )) {\n context.response.assetsBalance[accountId] = {\n ...context.response.assetsBalance[accountId],\n ...accountBalances,\n };\n }\n }\n } catch (error) {\n log('Middleware fetch failed', { error });\n }\n\n // Pass all chains through (staked balance doesn't claim chains)\n return next(context);\n };\n }\n\n /**\n * Subscribe to staked balance updates with polling.\n * Starts polling via StakedBalanceFetcher for each account/chain combination.\n *\n * @param subscriptionRequest - The subscription request details.\n */\n async subscribe(subscriptionRequest: SubscriptionRequest): Promise<void> {\n const { request, subscriptionId, isUpdate } = subscriptionRequest;\n\n const activeChainsSet = new Set(this.state.activeChains);\n const chainsToSubscribe = request.chainIds.filter((chainId) =>\n activeChainsSet.has(chainId),\n );\n\n log('Subscribe requested', {\n subscriptionId,\n isUpdate,\n chainsToSubscribe,\n });\n\n if (chainsToSubscribe.length === 0) {\n log('No staking chains to subscribe');\n return;\n }\n\n // Handle subscription update - restart polling for new chains\n if (isUpdate) {\n const existing = this.#activeSubscriptions.get(subscriptionId);\n if (existing) {\n log('Updating existing subscription - restarting polling', {\n subscriptionId,\n });\n }\n }\n\n // Clean up existing subscription (stops old polling)\n await this.unsubscribe(subscriptionId);\n\n // Build subscription data first so it is available when the first poll runs\n const accountsWithSupportedChains: AccountWithSupportedChains[] =\n request.accountsWithSupportedChains\n .map(({ account, supportedChains }) => ({\n account,\n supportedChains: chainsToSubscribe.filter((chain) =>\n supportedChains.includes(chain),\n ),\n }))\n .filter(({ supportedChains }) => supportedChains.length > 0);\n\n const accounts = accountsWithSupportedChains.map((entry) => entry.account);\n const pollingTokens: string[] = [];\n\n // Store subscription before startPolling so first poll (setTimeout 0) has the callback\n this.#activeSubscriptions.set(subscriptionId, {\n pollingTokens,\n chains: chainsToSubscribe,\n accounts,\n accountsWithSupportedChains,\n onAssetsUpdate: subscriptionRequest.onAssetsUpdate,\n });\n\n this.activeSubscriptions.set(subscriptionId, {\n cleanup: () => {\n for (const token of pollingTokens) {\n this.#stakedBalanceFetcher.stopPollingByPollingToken(token);\n }\n this.#activeSubscriptions.delete(subscriptionId);\n },\n chains: chainsToSubscribe,\n request,\n onAssetsUpdate: subscriptionRequest.onAssetsUpdate,\n });\n\n // Start polling for each account/chain (first poll runs on next tick)\n for (const {\n account,\n supportedChains: accountChains,\n } of request.accountsWithSupportedChains) {\n const chainsForAccount = chainsToSubscribe.filter((chain) =>\n accountChains.includes(chain),\n );\n\n for (const chainId of chainsForAccount) {\n const hexChainId = caipChainIdToHex(chainId);\n\n const input: StakedBalancePollingInput = {\n chainId: hexChainId,\n accountId: account.id,\n accountAddress: account.address as Hex,\n };\n\n const pollingToken = this.#stakedBalanceFetcher.startPolling(input);\n pollingTokens.push(pollingToken);\n }\n }\n\n // Immediate initial fetch so state is updated without waiting for first poll\n try {\n const initialRequest: DataRequest = {\n accountsWithSupportedChains,\n chainIds: chainsToSubscribe,\n dataTypes: ['balance'],\n };\n const initialResponse = await this.fetch(initialRequest);\n if (\n initialResponse.assetsBalance &&\n Object.keys(initialResponse.assetsBalance).length > 0\n ) {\n subscriptionRequest\n .onAssetsUpdate?.(initialResponse)\n ?.catch((error) => {\n log('Initial staked balance update failed', { error });\n });\n }\n } catch (error) {\n log('Initial staked balance fetch failed', { error });\n }\n\n log('Subscription SUCCESS', {\n subscriptionId,\n chains: chainsToSubscribe,\n pollingCount: pollingTokens.length,\n });\n }\n\n /**\n * Unsubscribe from staked balance updates and stop polling.\n *\n * @param subscriptionId - The subscription ID to unsubscribe.\n */\n async unsubscribe(subscriptionId: string): Promise<void> {\n const subscription = this.#activeSubscriptions.get(subscriptionId);\n if (subscription) {\n for (const token of subscription.pollingTokens) {\n this.#stakedBalanceFetcher.stopPollingByPollingToken(token);\n }\n this.#activeSubscriptions.delete(subscriptionId);\n }\n\n await super.unsubscribe(subscriptionId);\n }\n\n /**\n * Destroy the data source and clean up all resources.\n */\n destroy(): void {\n this.#unsubscribeTransactionConfirmed?.();\n this.#unsubscribeIncomingTransactions?.();\n this.#unsubscribeNetworkStateChange?.();\n this.#unsubscribeNetworkEnablementControllerStateChange?.();\n for (const subscription of this.#activeSubscriptions.values()) {\n for (const token of subscription.pollingTokens) {\n this.#stakedBalanceFetcher.stopPollingByPollingToken(token);\n }\n }\n this.#activeSubscriptions.clear();\n this.#providerCache.clear();\n super.destroy();\n }\n}\n"]}
|
|
1
|
+
{"version":3,"file":"StakedBalanceDataSource.cjs","sourceRoot":"","sources":["../../src/data-sources/StakedBalanceDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2CAAqD;AACrD,wDAAwD;AAGxD,2CAKyB;AAOzB,iEAA0D;AAK1D,mEAI4B;AAE5B,0CAA8D;AAY9D,MAAM,eAAe,GAAG,yBAAyB,CAAC;AAClD,MAAM,qBAAqB,GAAG,MAAO,CAAC,CAAC,YAAY;AAEnD,wEAAwE;AACxE,MAAM,mBAAmB,GAAkB;IACzC,IAAI,EAAE,OAAO;IACb,IAAI,EAAE,iBAAiB;IACvB,MAAM,EAAE,KAAK;IACb,QAAQ,EAAE,EAAE;CACb,CAAC;AAEF,MAAM,GAAG,GAAG,IAAA,2BAAkB,EAAC,sBAAa,EAAE,eAAe,CAAC,CAAC;AA6C/D;;;;;GAKG;AACH,SAAS,gBAAgB,CAAC,OAAe;IACvC,IAAI,IAAA,yBAAiB,EAAC,OAAO,CAAC,EAAE,CAAC;QAC/B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,IAAA,qBAAa,EAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,MAAM,GAAG,GAAG,IAAA,wBAAgB,EAAC,OAAO,CAAC,CAAC,SAAS,CAAC;QAChD,OAAO,IAAA,mBAAW,EAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;IACxC,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;AAChF,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,aAAa,CACpB,OAAgB,EAChB,eAAuB;IAEvB,MAAM,WAAW,GAAG,IAAA,wBAAiB,EAAC,eAAe,CAAC,CAAC;IACvD,OAAO,GAAG,OAAO,UAAU,WAAW,EAAmB,CAAC;AAC5D,CAAC;AAED;;;;;;;;;;GAUG;AACH,MAAa,uBAAwB,SAAQ,uCAG5C;IAyBC,YAAY,OAAuC;QACjD,KAAK,CAAC,eAAe,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;;QAzBtC,qDAAsC;QAEtC,iEAIC;QAED,wDAAsB;QAEtB,mDAAkB;QAE3B,mEAAmE;QAC1D,gEAA4C;QAErD,kCAAkC;QACzB,uDAAsD,IAAI,GAAG,EAAE,EAAC;QAEzE,uDAAuD;QAC9C,iDAAyC,IAAI,GAAG,EAAE,EAAC;QAE5D,0DAA0D;QACjD,6DAA8B;QAIrC,uBAAA,IAAI,sCAAc,OAAO,CAAC,SAAS,MAAA,CAAC;QACpC,uBAAA,IAAI,kDAA0B,OAAO,CAAC,qBAAqB,MAAA,CAAC;QAC5D,uBAAA,IAAI,yCAAiB,OAAO,CAAC,YAAY,IAAI,qBAAqB,MAAA,CAAC;QACnE,uBAAA,IAAI,oCAAY,OAAO,CAAC,OAAO,KAAK,KAAK,MAAA,CAAC;QAC1C,uBAAA,IAAI,8CAAsB,IAAA,8CAA2B,GAAe,MAAA,CAAC;QAErE,GAAG,CAAC,sCAAsC,EAAE;YAC1C,OAAO,EAAE,uBAAA,IAAI,wCAAS;YACtB,YAAY,EAAE,uBAAA,IAAI,6CAAc;SACjC,CAAC,CAAC;QAEH,mDAAmD;QACnD,uBAAA,IAAI,iDAAyB,IAAI,uCAAoB,CAAC;YACpD,eAAe,EAAE,uBAAA,IAAI,6CAAc;YACnC,kBAAkB,EAAE,CAAC,UAAU,EAA4B,EAAE,CAC3D,uBAAA,IAAI,gFAAa,MAAjB,IAAI,EAAc,UAAU,CAAC;SAChC,CAAC,MAAA,CAAC;QAEH,kEAAkE;QAClE,uBAAA,IAAI,qDAAsB,CAAC,wBAAwB,CACjD,uBAAA,IAAI,8FAA2B,CAAC,IAAI,CAAC,IAAI,CAAC,CAC3C,CAAC;QAEF,uBAAA,IAAI,0CAAW,CAAC,SAAS,CACvB,4CAA4C,EAC5C,uBAAA,IAAI,2FAAwB,CAAC,IAAI,CAAC,IAAI,CAAC,CACxC,CAAC;QAEF,uBAAA,IAAI,0CAAW,CAAC,SAAS,CACvB,oDAAoD,EACpD,uBAAA,IAAI,2FAAwB,CAAC,IAAI,CAAC,IAAI,CAAC,CACxC,CAAC;QAEF,uBAAA,IAAI,0CAAW,CAAC,SAAS,CACvB,+BAA+B,EAC/B,uBAAA,IAAI,yFAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CACtC,CAAC;QAEF,uBAAA,IAAI,0CAAW,CAAC,SAAS,CACvB,yCAAyC,EACzC,uBAAA,IAAI,6GAA0C,CAAC,IAAI,CAAC,IAAI,CAAC,CAC1D,CAAC;QAEF,uBAAA,IAAI,2FAAwB,MAA5B,IAAI,CAA0B,CAAC;IACjC,CAAC;IA6KD;;;OAGG;IACH,KAAK,CAAC,oBAAoB;QACxB,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,uBAAA,IAAI,oFAAiB,MAArB,IAAI,CAAmB,CAAC;QAC1C,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,MAAM,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,SAAS,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAgOD;;;;;OAKG;IACH,KAAK,CAAC,KAAK,CAAC,OAAoB;;QAC9B,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;YACnB,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,QAAQ,GAAiB,EAAE,CAAC;QAClC,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAEzD,MAAM,aAAa,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CACxD,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAC7B,CAAC;QAEF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,QAAQ,GAA2D,EAAE,CAAC;QAE5E,KAAK,MAAM,EACT,OAAO,EACP,eAAe,EAAE,aAAa,GAC/B,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;YACzC,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAC5C,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,CAC9B,CAAC;YAEF,KAAK,MAAM,OAAO,IAAI,MAAM,EAAE,CAAC;gBAC7B,IAAI,CAAC;oBACH,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;oBAC7C,MAAM,eAAe,GAAG,IAAA,4CAAyB,EAAC,UAAU,CAAC,CAAC;oBAC9D,IAAI,CAAC,eAAe,EAAE,CAAC;wBACrB,SAAS;oBACX,CAAC;oBAED,MAAM,KAAK,GAA8B;wBACvC,OAAO,EAAE,UAAU;wBACnB,SAAS,EAAE,OAAO,CAAC,EAAE;wBACrB,cAAc,EAAE,OAAO,CAAC,OAAc;qBACvC,CAAC;oBAEF,MAAM,MAAM,GACV,MAAM,uBAAA,IAAI,qDAAsB,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;oBAE7D,qEAAqE;oBACrE,QAAQ,MAAC,OAAO,CAAC,EAAE,MAAnB,QAAQ,OAAiB,EAAE,EAAC;oBAC5B,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;oBACxD,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC5D,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,GAAG,CAAC,gCAAgC,EAAE;wBACpC,OAAO;wBACP,SAAS,EAAE,OAAO,CAAC,EAAE;wBACrB,KAAK;qBACN,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrC,QAAQ,CAAC,aAAa,GAAG,QAAQ,CAAC;YAClC,4DAA4D;YAC5D,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAiB,CAAC;YAC1C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACtD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;oBACnD,QAAQ,CAAC,GAAG,CAAC,OAAwB,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;YACD,QAAQ,CAAC,UAAU,GAAG,EAAE,CAAC;YACzB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,mBAAmB,CAAC;YACrD,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;;;OAMG;IACH,IAAI,gBAAgB;QAClB,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;;YAC7B,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;gBACnB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YACD,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;YAE5B,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC3C,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAClC,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAEhD,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;oBAC7B,MAAA,OAAO,CAAC,QAAQ,EAAC,UAAU,QAAV,UAAU,GAAK,EAAE,EAAC;oBACnC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,aAAa,CAAC,UAAU,CAAC,CAAC;gBACvE,CAAC;gBACD,IAAI,aAAa,CAAC,aAAa,EAAE,CAAC;oBAChC,MAAA,OAAO,CAAC,QAAQ,EAAC,aAAa,QAAb,aAAa,GAAK,EAAE,EAAC;oBACtC,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,aAAa,CAAC,aAAa,CAC5B,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;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC5C,CAAC;YAED,gEAAgE;YAChE,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,SAAS,CAAC,mBAAwC;QACtD,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC;QAElE,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACzD,MAAM,iBAAiB,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5D,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAC7B,CAAC;QAEF,GAAG,CAAC,qBAAqB,EAAE;YACzB,cAAc;YACd,QAAQ;YACR,iBAAiB;SAClB,CAAC,CAAC;QAEH,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,GAAG,CAAC,gCAAgC,CAAC,CAAC;YACtC,OAAO;QACT,CAAC;QAED,8DAA8D;QAC9D,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,uBAAA,IAAI,oDAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC/D,IAAI,QAAQ,EAAE,CAAC;gBACb,GAAG,CAAC,qDAAqD,EAAE;oBACzD,cAAc;iBACf,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,qDAAqD;QACrD,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;QAEvC,4EAA4E;QAC5E,MAAM,2BAA2B,GAC/B,OAAO,CAAC,2BAA2B;aAChC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,CAAC;YACtC,OAAO;YACP,eAAe,EAAE,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAClD,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,CAChC;SACF,CAAC,CAAC;aACF,MAAM,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAEjE,MAAM,QAAQ,GAAG,2BAA2B,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC3E,MAAM,aAAa,GAAa,EAAE,CAAC;QAEnC,uFAAuF;QACvF,uBAAA,IAAI,oDAAqB,CAAC,GAAG,CAAC,cAAc,EAAE;YAC5C,aAAa;YACb,MAAM,EAAE,iBAAiB;YACzB,QAAQ;YACR,2BAA2B;YAC3B,cAAc,EAAE,mBAAmB,CAAC,cAAc;SACnD,CAAC,CAAC;QAEH,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,EAAE;YAC3C,OAAO,EAAE,GAAG,EAAE;gBACZ,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;oBAClC,uBAAA,IAAI,qDAAsB,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;gBAC9D,CAAC;gBACD,uBAAA,IAAI,oDAAqB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,EAAE,iBAAiB;YACzB,OAAO;YACP,cAAc,EAAE,mBAAmB,CAAC,cAAc;SACnD,CAAC,CAAC;QAEH,sEAAsE;QACtE,KAAK,MAAM,EACT,OAAO,EACP,eAAe,EAAE,aAAa,GAC/B,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;YACzC,MAAM,gBAAgB,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAC1D,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,CAC9B,CAAC;YAEF,KAAK,MAAM,OAAO,IAAI,gBAAgB,EAAE,CAAC;gBACvC,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;gBAE7C,MAAM,KAAK,GAA8B;oBACvC,OAAO,EAAE,UAAU;oBACnB,SAAS,EAAE,OAAO,CAAC,EAAE;oBACrB,cAAc,EAAE,OAAO,CAAC,OAAc;iBACvC,CAAC;gBAEF,MAAM,YAAY,GAAG,uBAAA,IAAI,qDAAsB,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpE,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,6EAA6E;QAC7E,IAAI,CAAC;YACH,MAAM,cAAc,GAAgB;gBAClC,2BAA2B;gBAC3B,QAAQ,EAAE,iBAAiB;gBAC3B,SAAS,EAAE,CAAC,SAAS,CAAC;aACvB,CAAC;YACF,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YACzD,IACE,eAAe,CAAC,aAAa;gBAC7B,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC,MAAM,GAAG,CAAC,EACrD,CAAC;gBACD,mBAAmB;qBAChB,cAAc,EAAE,CAAC,eAAe,CAAC;oBAClC,EAAE,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBAChB,GAAG,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;gBACzD,CAAC,CAAC,CAAC;YACP,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,qCAAqC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACxD,CAAC;QAED,GAAG,CAAC,sBAAsB,EAAE;YAC1B,cAAc;YACd,MAAM,EAAE,iBAAiB;YACzB,YAAY,EAAE,aAAa,CAAC,MAAM;SACnC,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,WAAW,CAAC,cAAsB;QACtC,MAAM,YAAY,GAAG,uBAAA,IAAI,oDAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACnE,IAAI,YAAY,EAAE,CAAC;YACjB,KAAK,MAAM,KAAK,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;gBAC/C,uBAAA,IAAI,qDAAsB,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;YAC9D,CAAC;YACD,uBAAA,IAAI,oDAAqB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,KAAK,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,OAAO;QACL,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YAC9D,KAAK,MAAM,KAAK,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;gBAC/C,uBAAA,IAAI,qDAAsB,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;QACD,uBAAA,IAAI,oDAAqB,CAAC,KAAK,EAAE,CAAC;QAClC,uBAAA,IAAI,8CAAe,CAAC,KAAK,EAAE,CAAC;QAC5B,KAAK,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC;CACF;AA/vBD,0DA+vBC;;IA7qBG,uBAAA,IAAI,8CAAe,CAAC,KAAK,EAAE,CAAC;IAC5B,GAAG,CAAC,mDAAmD,CAAC,CAAC;AAC3D,CAAC,iJASC,KAAuC;IAEvC,MAAM,EAAE,iBAAiB,EAAE,GAAG,KAAK,IAAI,EAAE,CAAC;IAC1C,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,OAAO;IACT,CAAC;IACD,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,iBAAiB,CAAC,CAAC;AAChE,CAAC,2IAasC,OAGtC;IACC,MAAM,UAAU,GAAG,OAAO,EAAE,OAAO,CAAC;IACpC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,eAAe,GAAG,IAAA,4CAAyB,EAAC,UAAU,CAAC,CAAC;IAC9D,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,aAAa,GAAG,eAAe,CAAC,WAAW,EAAE,CAAC;IACpD,MAAM,IAAI,GAAG,OAAO,CAAC,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;IACnD,MAAM,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE,EAAE,EAAE,WAAW,EAAE,CAAC;IAC/C,OAAO,IAAI,KAAK,aAAa,IAAI,EAAE,KAAK,aAAa,CAAC;AACxD,CAAC,6GAYuB,OAGvB;IACC,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;QACnB,OAAO;IACT,CAAC;IACD,IAAI,CAAC,uBAAA,IAAI,0GAAuC,MAA3C,IAAI,EAAwC,OAAO,CAAC,EAAE,CAAC;QAC1D,OAAO;IACT,CAAC;IACD,MAAM,UAAU,GAAG,OAAO,EAAE,OAAO,CAAC;IACpC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO;IACT,CAAC;IACD,MAAM,WAAW,GAAG,UAAU,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,EAAa,CAAC;IACpE,MAAM,SAAS,GAAG,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,CAAC,WAAW,CAAC,CAAC,CAAC;IAC7D,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACpE,GAAG,CAAC,oDAAoD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC,6GASC,OAA0E;IAE1E,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;QACnB,OAAO;IACT,CAAC;IACD,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAU,CAAC;IAC5C,KAAK,MAAM,IAAI,IAAI,OAAO,IAAI,EAAE,EAAE,CAAC;QACjC,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC;YACnB,SAAS;QACX,CAAC;QACD,IAAI,uBAAA,IAAI,0GAAuC,MAA3C,IAAI,EAAwC,IAAI,CAAC,EAAE,CAAC;YACtD,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IACD,MAAM,YAAY,GAAG,CAAC,GAAG,iBAAiB,CAAC,CAAC,GAAG,CAC7C,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,EAAa,CAChE,CAAC;IACF,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO;IACT,CAAC;IACD,MAAM,SAAS,GAAG,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,YAAY,CAAC,CAAC;IAC5D,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACpE,GAAG,CAAC,8DAA8D,EAAE;gBAClE,KAAK;aACN,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC,2GASC,QAAmB;IAEnB,MAAM,SAAS,GAAqD,EAAE,CAAC;IACvE,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC;IACnC,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9D,KAAK,MAAM,EACT,OAAO,EACP,eAAe,GAChB,IAAI,YAAY,CAAC,2BAA2B,EAAE,CAAC;YAC9C,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;gBACtC,IAAI,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC1B,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;gBACvC,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;IAQC,MAAM,SAAS,GAAqD,EAAE,CAAC;IACvE,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9D,KAAK,MAAM,EACT,OAAO,EACP,eAAe,GAChB,IAAI,YAAY,CAAC,2BAA2B,EAAE,CAAC;YAC9C,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;gBACtC,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YACvC,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAgBD;;;;;GAKG;AACH,KAAK,wEACH,SAA2D;IAE3D,MAAM,UAAU,GAAyC,EAAE,CAAC;IAC5D,MAAM,aAAa,GAGf,EAAE,CAAC;IAEP,KAAK,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,SAAS,EAAE,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,eAAe,GAAG,IAAA,4CAAyB,EAAC,UAAU,CAAC,CAAC;YAC9D,IAAI,CAAC,eAAe,EAAE,CAAC;gBACrB,SAAS;YACX,CAAC;YAED,MAAM,KAAK,GAA8B;gBACvC,OAAO,EAAE,UAAU;gBACnB,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,cAAc,EAAE,OAAO,CAAC,OAAc;aACvC,CAAC;YAEF,MAAM,MAAM,GACV,MAAM,uBAAA,IAAI,qDAAsB,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAC7D,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;YAExD,UAAU,CAAC,OAAO,CAAC,GAAG,mBAAmB,CAAC;YAC1C,MAAM,QAAQ,GAAG,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3C,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,GAAG;gBAC1B,GAAG,QAAQ;gBACX,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE;aACrC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,uDAAuD,EAAE;gBAC3D,OAAO;gBACP,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,KAAK;aACN,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1C,MAAM,QAAQ,GAAiB,EAAE,UAAU,EAAE,aAAa,EAAE,CAAC;QAC7D,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YAC9D,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;gBAC9D,GAAG,CAAC,0DAA0D,EAAE;oBAC9D,KAAK;iBACN,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;AACH,CAAC;IAQC,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,uBAAA,IAAI,0CAAW,CAAC,IAAI,CAChC,sCAAsC,CACvC,CAAC;QACF,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EACF,KAAK,EAAE,iBAAiB,IAAI,EAAE,CAC/B,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,iDAAiD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAClE,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,EAAE,CAAC,CAAC;IACjD,CAAC;AACH,CAAC,yIAUC,iBAA0D;IAE1D,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;QACnB,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC9C,IAAI,CAAC,kBAAkB,CAAC,EAAE,EAAE,CAAC,aAAa,EAAE,EAAE,CAC5C,uBAAA,IAAI,sDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;QACF,OAAO;IACT,CAAC;IAED,MAAM,YAAY,GAAc,EAAE,CAAC;IACnC,MAAM,SAAS,GAAG,iBAAiB,CAAC,MAAM,CAAC;IAC3C,IAAI,SAAS,EAAE,CAAC;QACd,KAAK,MAAM,KAAK,IAAI,uBAAA,IAAI,kDAAmB,EAAE,CAAC;YAC5C,IAAI,CAAC,IAAA,qBAAa,EAAC,KAAK,CAAC,EAAE,CAAC;gBAC1B,SAAS;YACX,CAAC;YACD,MAAM,EAAE,SAAS,EAAE,GAAG,IAAA,wBAAgB,EAAC,KAAK,CAAC,CAAC;YAC9C,MAAM,UAAU,GAAG,IAAA,mBAAW,EAAC,QAAQ,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC;YACxD,IAAI,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC1B,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAC9C,IAAI,CAAC,kBAAkB,CAAC,YAAY,EAAE,CAAC,aAAa,EAAE,EAAE,CACtD,uBAAA,IAAI,sDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;AACJ,CAAC,uFASY,UAAe;IAC1B,MAAM,MAAM,GAAG,uBAAA,IAAI,8CAAe,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACnD,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,CAAC;QACH,MAAM,YAAY,GAAG,uBAAA,IAAI,0CAAW,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAExE,MAAM,EAAE,8BAA8B,EAAE,GAAG,YAAY,CAAC;QACxD,IAAI,CAAC,8BAA8B,EAAE,CAAC;YACpC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,WAAW,GAAG,8BAA8B,CAAC,UAAU,CAAC,CAAC;QAC/D,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,6EAA6E;QAC7E,MAAM,EAAE,YAAY,EAAE,uBAAuB,EAAE,GAAG,WAAW,CAAC;QAC9D,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/C,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,KAAK,GACT,OAAO,uBAAuB,KAAK,QAAQ;YAC3C,uBAAuB,IAAI,CAAC;YAC5B,uBAAuB,GAAG,YAAY,CAAC,MAAM;YAC3C,CAAC,CAAC,uBAAuB;YACzB,CAAC,CAAC,CAAC,CAAC;QACR,MAAM,eAAe,GAAG,YAAY,CAAC,KAAK,CAEzC,CAAC;QACF,MAAM,eAAe,GAAG,eAAe,EAAE,eAAe,CAAC;QACzD,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,aAAa,GAAG,uBAAA,IAAI,0CAAW,CAAC,IAAI,CACxC,wCAAwC,EACxC,eAAe,CAChB,CAAC;QAEF,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,CAAC;YAC7B,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,wBAAY,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC1D,uBAAA,IAAI,8CAAe,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC9C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,kCAAkC,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/D,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC,mHAS0B,MAAgC;IACzD,MAAM,eAAe,GAAG,IAAA,4CAAyB,EAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAClE,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO;IACT,CAAC;IACD,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IACpD,MAAM,WAAW,GAAG,UAAU,cAAc,EAAa,CAAC;IAC1D,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC;IAE5D,MAAM,QAAQ,GAAiB;QAC7B,UAAU,EAAE;YACV,CAAC,OAAO,CAAC,EAAE,mBAAmB;SAC/B;QACD,aAAa,EAAE;YACb,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE;gBAClB,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE;aAC7C;SACF;KACF,CAAC;IAEF,GAAG,CAAC,uBAAuB,EAAE;QAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,OAAO,EAAE,WAAW;QACpB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM;KAC9B,CAAC,CAAC;IAEH,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9D,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YAC9D,GAAG,CAAC,wCAAwC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC","sourcesContent":["import { toChecksumAddress } from '@ethereumjs/util';\nimport { Web3Provider } from '@ethersproject/providers';\nimport type { InternalAccount } from '@metamask/keyring-internal-api';\nimport type { NetworkEnablementControllerState } from '@metamask/network-enablement-controller';\nimport {\n isStrictHexString,\n isCaipChainId,\n numberToHex,\n parseCaipChainId,\n} from '@metamask/utils';\nimport type { Hex } from '@metamask/utils';\n\nimport type {\n DataSourceState,\n SubscriptionRequest,\n} from './AbstractDataSource';\nimport { AbstractDataSource } from './AbstractDataSource';\nimport type {\n StakedBalancePollingInput,\n StakedBalanceFetchResult,\n} from './evm-rpc-services';\nimport {\n StakedBalanceFetcher,\n getStakingContractAddress,\n getSupportedStakingChainIds,\n} from './evm-rpc-services';\nimport type { AssetsControllerMessenger } from '../AssetsController';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport type {\n AccountId,\n ChainId,\n Caip19AssetId,\n AssetBalance,\n AssetMetadata,\n DataRequest,\n DataResponse,\n Middleware,\n} from '../types';\n\nconst CONTROLLER_NAME = 'StakedBalanceDataSource';\nconst DEFAULT_POLL_INTERVAL = 180_000; // 3 minutes\n\n/** Metadata for staked ETH (same symbol and decimals as native ETH). */\nconst STAKED_ETH_METADATA: AssetMetadata = {\n type: 'erc20',\n name: 'staked ethereum',\n symbol: 'ETH',\n decimals: 18,\n};\n\nconst log = createModuleLogger(projectLogger, CONTROLLER_NAME);\n\n/** Optional configuration for StakedBalanceDataSource. */\nexport type StakedBalanceDataSourceConfig = {\n /** Whether staked balance fetching is enabled (default: true). */\n enabled?: boolean;\n /** Polling interval in ms (default: 180s / 3 min). */\n pollInterval?: number;\n};\n\nexport type StakedBalanceDataSourceOptions = StakedBalanceDataSourceConfig & {\n /** The AssetsController messenger (for accessing NetworkController). */\n messenger: AssetsControllerMessenger;\n /** Called when active chains are updated. */\n onActiveChainsUpdated: (\n dataSourceName: string,\n chains: ChainId[],\n previousChains: ChainId[],\n ) => void;\n};\n\n/** Per-account supported chains (same shape as in DataRequest). */\ntype AccountWithSupportedChains = {\n account: InternalAccount;\n supportedChains: ChainId[];\n};\n\n/**\n * Subscription data stored for each active subscription.\n * Stores accountsWithSupportedChains so refresh paths use the same per-account\n * scope as normal subscription setup (avoids querying unsupported chains/accounts).\n */\ntype SubscriptionData = {\n /** Polling tokens from StakedBalanceFetcher. */\n pollingTokens: string[];\n /** Chain IDs being polled (union of all account chains). */\n chains: ChainId[];\n /** Accounts being polled. */\n accounts: InternalAccount[];\n /** Per-account supported chains; used by refreshStakedBalance and transaction handlers. */\n accountsWithSupportedChains: AccountWithSupportedChains[];\n /** Callback to report asset updates. */\n onAssetsUpdate: (response: DataResponse) => void | Promise<void>;\n};\n\n/**\n * Convert CAIP chain ID or hex chain ID to hex chain ID.\n *\n * @param chainId - CAIP chain ID or hex chain ID.\n * @returns Hex chain ID.\n */\nfunction caipChainIdToHex(chainId: string): Hex {\n if (isStrictHexString(chainId)) {\n return chainId;\n }\n\n if (isCaipChainId(chainId)) {\n const ref = parseCaipChainId(chainId).reference;\n return numberToHex(parseInt(ref, 10));\n }\n\n throw new Error('caipChainIdToHex - Failed to provide CAIP-2 or Hex chainId');\n}\n\n/**\n * Build the CAIP-19 asset ID for staked balance (same format as ERC20).\n * Uses the staking contract address (checksummed) so it is consistent with\n * other token assets.\n * Example: \"eip155:1/erc20:0x4fef9d741011476750a243ac70b9789a63dd47df\"\n *\n * @param chainId - CAIP-2 chain ID (e.g. \"eip155:1\").\n * @param contractAddress - Staking contract address (hex).\n * @returns The staked asset CAIP-19 ID with checksummed address.\n */\nfunction stakedAssetId(\n chainId: ChainId,\n contractAddress: string,\n): Caip19AssetId {\n const checksummed = toChecksumAddress(contractAddress);\n return `${chainId}/erc20:${checksummed}` as Caip19AssetId;\n}\n\n/**\n * Data source for fetching staked ETH balances via on-chain staking contracts.\n *\n * Delegates to {@link StakedBalanceFetcher} for the actual RPC calls\n * (getShares + convertToAssets on ERC-4626-style staking contracts).\n * Reports balances as CAIP-19 asset IDs using the ERC20 format with the\n * staking contract address (e.g. \"eip155:1/erc20:0x4fef9d741011476750a243ac70b9789a63dd47df\").\n *\n * Only supports chains with known staking contracts (mainnet, Hoodi).\n * Polling is managed by StakedBalanceFetcher via startPolling/stopPollingByPollingToken.\n */\nexport class StakedBalanceDataSource extends AbstractDataSource<\n typeof CONTROLLER_NAME,\n DataSourceState\n> {\n readonly #messenger: AssetsControllerMessenger;\n\n readonly #onActiveChainsUpdated: (\n dataSourceName: string,\n chains: ChainId[],\n previousChains: ChainId[],\n ) => void;\n\n readonly #pollInterval: number;\n\n readonly #enabled: boolean;\n\n /** The StakedBalanceFetcher that handles polling and RPC calls. */\n readonly #stakedBalanceFetcher: StakedBalanceFetcher;\n\n /** Active subscriptions by ID. */\n readonly #activeSubscriptions: Map<string, SubscriptionData> = new Map();\n\n /** Cache of Web3Provider instances by hex chain ID. */\n readonly #providerCache: Map<Hex, Web3Provider> = new Map();\n\n /** CAIP-2 chain IDs that have known staking contracts. */\n readonly #supportedChainIds: ChainId[];\n\n constructor(options: StakedBalanceDataSourceOptions) {\n super(CONTROLLER_NAME, { activeChains: [] });\n this.#messenger = options.messenger;\n this.#onActiveChainsUpdated = options.onActiveChainsUpdated;\n this.#pollInterval = options.pollInterval ?? DEFAULT_POLL_INTERVAL;\n this.#enabled = options.enabled !== false;\n this.#supportedChainIds = getSupportedStakingChainIds() as ChainId[];\n\n log('Initializing StakedBalanceDataSource', {\n enabled: this.#enabled,\n pollInterval: this.#pollInterval,\n });\n\n // Create StakedBalanceFetcher with provider getter\n this.#stakedBalanceFetcher = new StakedBalanceFetcher({\n pollingInterval: this.#pollInterval,\n getNetworkProvider: (hexChainId): Web3Provider | undefined =>\n this.#getProvider(hexChainId),\n });\n\n // Wire the callback so polling results flow back to subscriptions\n this.#stakedBalanceFetcher.setOnStakedBalanceUpdate(\n this.#handleStakedBalanceUpdate.bind(this),\n );\n\n this.#messenger.subscribe(\n 'TransactionController:transactionConfirmed',\n this.#onTransactionConfirmed.bind(this),\n );\n\n this.#messenger.subscribe(\n 'TransactionController:incomingTransactionsReceived',\n this.#onIncomingTransactions.bind(this),\n );\n\n this.#messenger.subscribe(\n 'NetworkController:stateChange',\n this.#onNetworkStateChange.bind(this),\n );\n\n this.#messenger.subscribe(\n 'NetworkEnablementController:stateChange',\n this.#onNetworkEnablementControllerStateChange.bind(this),\n );\n\n this.#initializeActiveChains();\n }\n\n /**\n * When NetworkController state changes (e.g. RPC endpoints or network clients\n * reconfigured), clear the provider cache so subsequent fetches use fresh\n * providers.\n */\n #onNetworkStateChange(): void {\n this.#providerCache.clear();\n log('Provider cache cleared after network state change');\n }\n\n /**\n * When NetworkEnablementController state changes (user enables/disables\n * networks), recompute active chains so we only fetch for enabled staking chains.\n *\n * @param state - The new NetworkEnablementController state.\n */\n #onNetworkEnablementControllerStateChange(\n state: NetworkEnablementControllerState,\n ): void {\n const { enabledNetworkMap } = state ?? {};\n if (!enabledNetworkMap) {\n return;\n }\n this.#initializeActiveChainsFromEnabledMap(enabledNetworkMap);\n }\n\n /**\n * Returns true if the transaction involves the staking contract (from or to)\n * for the payload's chain, so we only refresh staked balance when relevant.\n *\n * @param payload - Transaction payload.\n * @param payload.chainId - Hex chain ID.\n * @param payload.txParams - Optional transaction params.\n * @param payload.txParams.from - Sender address.\n * @param payload.txParams.to - Recipient address.\n * @returns True if txParams.from or txParams.to matches the staking contract address.\n */\n #isTransactionInvolvingStakingContract(payload: {\n chainId?: string;\n txParams?: { from?: string; to?: string };\n }): boolean {\n const hexChainId = payload?.chainId;\n if (!hexChainId) {\n return false;\n }\n const contractAddress = getStakingContractAddress(hexChainId);\n if (!contractAddress) {\n return false;\n }\n const contractLower = contractAddress.toLowerCase();\n const from = payload.txParams?.from?.toLowerCase();\n const to = payload.txParams?.to?.toLowerCase();\n return from === contractLower || to === contractLower;\n }\n\n /**\n * When a transaction is confirmed, refresh staked balance only if the\n * transaction is from or to the staking contract for that chain.\n *\n * @param payload - From TransactionController:transactionConfirmed.\n * @param payload.chainId - Hex chain ID of the transaction.\n * @param payload.txParams - Optional transaction params.\n * @param payload.txParams.from - Sender address.\n * @param payload.txParams.to - Recipient address.\n */\n #onTransactionConfirmed(payload: {\n chainId?: string;\n txParams?: { from?: string; to?: string };\n }): void {\n if (!this.#enabled) {\n return;\n }\n if (!this.#isTransactionInvolvingStakingContract(payload)) {\n return;\n }\n const hexChainId = payload?.chainId;\n if (!hexChainId) {\n return;\n }\n const caipChainId = `eip155:${parseInt(hexChainId, 16)}` as ChainId;\n const toRefresh = this.#getToRefreshForChains([caipChainId]);\n if (toRefresh.length > 0) {\n this.#refreshStakedBalanceAfterTransaction(toRefresh).catch((error) => {\n log('Failed to refresh staked balance after transaction', { error });\n });\n }\n }\n\n /**\n * When incoming transactions are received, refresh staked balance only for\n * chains where at least one transaction is from or to the staking contract.\n *\n * @param payload - From TransactionController:incomingTransactionsReceived (array of { chainId?, txParams? }).\n */\n #onIncomingTransactions(\n payload: { chainId?: string; txParams?: { from?: string; to?: string } }[],\n ): void {\n if (!this.#enabled) {\n return;\n }\n const chainIdsToRefresh = new Set<string>();\n for (const item of payload ?? []) {\n if (!item?.chainId) {\n continue;\n }\n if (this.#isTransactionInvolvingStakingContract(item)) {\n chainIdsToRefresh.add(item.chainId);\n }\n }\n const caipChainIds = [...chainIdsToRefresh].map(\n (hexChainId) => `eip155:${parseInt(hexChainId, 16)}` as ChainId,\n );\n if (caipChainIds.length === 0) {\n return;\n }\n const toRefresh = this.#getToRefreshForChains(caipChainIds);\n if (toRefresh.length > 0) {\n this.#refreshStakedBalanceAfterTransaction(toRefresh).catch((error) => {\n log('Failed to refresh staked balance after incoming transactions', {\n error,\n });\n });\n }\n }\n\n /**\n * Build toRefresh list for subscribed (account, chainId) pairs for the given chains.\n *\n * @param chainIds - CAIP-2 chain IDs to target.\n * @returns Pairs of account and chainId to refresh.\n */\n #getToRefreshForChains(\n chainIds: ChainId[],\n ): { account: InternalAccount; chainId: ChainId }[] {\n const toRefresh: { account: InternalAccount; chainId: ChainId }[] = [];\n const chainSet = new Set(chainIds);\n for (const subscription of this.#activeSubscriptions.values()) {\n for (const {\n account,\n supportedChains,\n } of subscription.accountsWithSupportedChains) {\n for (const chainId of supportedChains) {\n if (chainSet.has(chainId)) {\n toRefresh.push({ account, chainId });\n }\n }\n }\n }\n return toRefresh;\n }\n\n /**\n * Build toRefresh list for all subscribed (account, chainId) pairs.\n *\n * @returns Pairs of account and chainId to refresh.\n */\n #getToRefreshAll(): { account: InternalAccount; chainId: ChainId }[] {\n const toRefresh: { account: InternalAccount; chainId: ChainId }[] = [];\n for (const subscription of this.#activeSubscriptions.values()) {\n for (const {\n account,\n supportedChains,\n } of subscription.accountsWithSupportedChains) {\n for (const chainId of supportedChains) {\n toRefresh.push({ account, chainId });\n }\n }\n }\n return toRefresh;\n }\n\n /**\n * Refresh staked balance for all currently subscribed accounts and chains, then\n * push updates to the controller. Can be called from UI or after transaction events.\n */\n async refreshStakedBalance(): Promise<void> {\n if (!this.#enabled) {\n return;\n }\n const toRefresh = this.#getToRefreshAll();\n if (toRefresh.length > 0) {\n await this.#refreshStakedBalanceAfterTransaction(toRefresh);\n }\n }\n\n /**\n * Fetch staked balance for the given account/chain pairs and push a single\n * DataResponse to all active subscriptions.\n *\n * @param toRefresh - List of { account, chainId } to refresh.\n */\n async #refreshStakedBalanceAfterTransaction(\n toRefresh: { account: InternalAccount; chainId: ChainId }[],\n ): Promise<void> {\n const assetsInfo: Record<Caip19AssetId, AssetMetadata> = {};\n const assetsBalance: Record<\n AccountId,\n Record<Caip19AssetId, AssetBalance>\n > = {};\n\n for (const { account, chainId } of toRefresh) {\n try {\n const hexChainId = caipChainIdToHex(chainId);\n const contractAddress = getStakingContractAddress(hexChainId);\n if (!contractAddress) {\n continue;\n }\n\n const input: StakedBalancePollingInput = {\n chainId: hexChainId,\n accountId: account.id,\n accountAddress: account.address as Hex,\n };\n\n const result =\n await this.#stakedBalanceFetcher.fetchStakedBalance(input);\n const assetId = stakedAssetId(chainId, contractAddress);\n\n assetsInfo[assetId] = STAKED_ETH_METADATA;\n const existing = assetsBalance[account.id];\n assetsBalance[account.id] = {\n ...existing,\n [assetId]: { amount: result.amount },\n };\n } catch (error) {\n log('Failed to fetch staked balance in transaction refresh', {\n chainId,\n accountId: account.id,\n error,\n });\n }\n }\n\n if (Object.keys(assetsBalance).length > 0) {\n const response: DataResponse = { assetsInfo, assetsBalance };\n for (const subscription of this.#activeSubscriptions.values()) {\n subscription.onAssetsUpdate(response)?.catch((error: unknown) => {\n log('Failed to report staked balance update after transaction', {\n error,\n });\n });\n }\n }\n }\n\n /**\n * Set active chains from NetworkEnablementController state.\n * Only staking-supported chains that are enabled in the network enablement map\n * are active (e.g. if mainnet is not selected we do not fetch).\n */\n #initializeActiveChains(): void {\n try {\n const state = this.#messenger.call(\n 'NetworkEnablementController:getState',\n );\n this.#initializeActiveChainsFromEnabledMap(\n state?.enabledNetworkMap ?? {},\n );\n } catch (error) {\n log('Failed to get NetworkEnablementController state', { error });\n this.#initializeActiveChainsFromEnabledMap({});\n }\n }\n\n /**\n * Compute active chains as the intersection of supported staking chains and\n * enabled networks, then update state. Uses the same EIP-155 storage key\n * convention as NetworkEnablementController (hex for eip155).\n *\n * @param enabledNetworkMap - The enabled network map from NetworkEnablementController.\n */\n #initializeActiveChainsFromEnabledMap(\n enabledNetworkMap: Record<string, Record<string, boolean>>,\n ): void {\n if (!this.#enabled) {\n const previous = [...this.state.activeChains];\n this.updateActiveChains([], (updatedChains) =>\n this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n );\n return;\n }\n\n const activeChains: ChainId[] = [];\n const eip155Map = enabledNetworkMap.eip155;\n if (eip155Map) {\n for (const caip2 of this.#supportedChainIds) {\n if (!isCaipChainId(caip2)) {\n continue;\n }\n const { reference } = parseCaipChainId(caip2);\n const storageKey = numberToHex(parseInt(reference, 10));\n if (eip155Map[storageKey]) {\n activeChains.push(caip2);\n }\n }\n }\n\n const previous = [...this.state.activeChains];\n this.updateActiveChains(activeChains, (updatedChains) =>\n this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n );\n }\n\n /**\n * Get or create a Web3Provider for the given hex chain ID.\n * Uses the same messenger-cast pattern as RpcDataSource.\n *\n * @param hexChainId - The hex chain ID.\n * @returns Web3Provider instance, or undefined if not available.\n */\n #getProvider(hexChainId: Hex): Web3Provider | undefined {\n const cached = this.#providerCache.get(hexChainId);\n if (cached) {\n return cached;\n }\n\n try {\n const networkState = this.#messenger.call('NetworkController:getState');\n\n const { networkConfigurationsByChainId } = networkState;\n if (!networkConfigurationsByChainId) {\n return undefined;\n }\n\n const chainConfig = networkConfigurationsByChainId[hexChainId];\n if (!chainConfig) {\n return undefined;\n }\n\n // Use the network's configured default RPC endpoint (same as RpcDataSource).\n const { rpcEndpoints, defaultRpcEndpointIndex } = chainConfig;\n if (!rpcEndpoints || rpcEndpoints.length === 0) {\n return undefined;\n }\n\n const index =\n typeof defaultRpcEndpointIndex === 'number' &&\n defaultRpcEndpointIndex >= 0 &&\n defaultRpcEndpointIndex < rpcEndpoints.length\n ? defaultRpcEndpointIndex\n : 0;\n const defaultEndpoint = rpcEndpoints[index] as {\n networkClientId?: string;\n };\n const networkClientId = defaultEndpoint?.networkClientId;\n if (!networkClientId) {\n return undefined;\n }\n\n const networkClient = this.#messenger.call(\n 'NetworkController:getNetworkClientById',\n networkClientId,\n );\n\n if (!networkClient?.provider) {\n return undefined;\n }\n\n const provider = new Web3Provider(networkClient.provider);\n this.#providerCache.set(hexChainId, provider);\n return provider;\n } catch (error) {\n log('Failed to get provider for chain', { hexChainId, error });\n return undefined;\n }\n }\n\n /**\n * Handle a staked balance update from StakedBalanceFetcher.\n * Converts the result into a DataResponse and forwards it to all active\n * subscriptions.\n *\n * @param result - The staked balance fetch result.\n */\n #handleStakedBalanceUpdate(result: StakedBalanceFetchResult): void {\n const contractAddress = getStakingContractAddress(result.chainId);\n if (!contractAddress) {\n return;\n }\n const chainIdDecimal = parseInt(result.chainId, 16);\n const caipChainId = `eip155:${chainIdDecimal}` as ChainId;\n const assetId = stakedAssetId(caipChainId, contractAddress);\n\n const response: DataResponse = {\n assetsInfo: {\n [assetId]: STAKED_ETH_METADATA,\n },\n assetsBalance: {\n [result.accountId]: {\n [assetId]: { amount: result.balance.amount },\n },\n },\n };\n\n log('Staked balance update', {\n accountId: result.accountId,\n chainId: caipChainId,\n amount: result.balance.amount,\n });\n\n for (const subscription of this.#activeSubscriptions.values()) {\n subscription.onAssetsUpdate(response)?.catch((error: unknown) => {\n log('Failed to report staked balance update', { error });\n });\n }\n }\n\n /**\n * Fetch staked balances for all accounts on supported chains.\n *\n * @param request - The data request with accounts and chains.\n * @returns DataResponse with staked balance data.\n */\n async fetch(request: DataRequest): Promise<DataResponse> {\n if (!this.#enabled) {\n return {};\n }\n const response: DataResponse = {};\n const activeChainsSet = new Set(this.state.activeChains);\n\n const chainsToFetch = request.chainIds.filter((chainId) =>\n activeChainsSet.has(chainId),\n );\n\n if (chainsToFetch.length === 0) {\n return response;\n }\n\n const balances: Record<AccountId, Record<Caip19AssetId, AssetBalance>> = {};\n\n for (const {\n account,\n supportedChains: accountChains,\n } of request.accountsWithSupportedChains) {\n const chains = chainsToFetch.filter((chain) =>\n accountChains.includes(chain),\n );\n\n for (const chainId of chains) {\n try {\n const hexChainId = caipChainIdToHex(chainId);\n const contractAddress = getStakingContractAddress(hexChainId);\n if (!contractAddress) {\n continue;\n }\n\n const input: StakedBalancePollingInput = {\n chainId: hexChainId,\n accountId: account.id,\n accountAddress: account.address as Hex,\n };\n\n const result =\n await this.#stakedBalanceFetcher.fetchStakedBalance(input);\n\n // Include zero amounts so merged updates clear prior non-zero state.\n balances[account.id] ??= {};\n const assetId = stakedAssetId(chainId, contractAddress);\n balances[account.id][assetId] = { amount: result.amount };\n } catch (error) {\n log('Failed to fetch staked balance', {\n chainId,\n accountId: account.id,\n error,\n });\n }\n }\n }\n\n if (Object.keys(balances).length > 0) {\n response.assetsBalance = balances;\n // Add metadata for each staked asset ID present in balances\n const assetIds = new Set<Caip19AssetId>();\n for (const accountBalances of Object.values(balances)) {\n for (const assetId of Object.keys(accountBalances)) {\n assetIds.add(assetId as Caip19AssetId);\n }\n }\n response.assetsInfo = {};\n for (const assetId of assetIds) {\n response.assetsInfo[assetId] = STAKED_ETH_METADATA;\n }\n }\n\n return response;\n }\n\n /**\n * Assets middleware for the fetch pipeline.\n * Fetches staked balances and merges them into the response, then passes\n * all chains to the next middleware (staked balance doesn't claim chains).\n *\n * @returns The middleware function for the assets pipeline.\n */\n get assetsMiddleware(): Middleware {\n return async (context, next) => {\n if (!this.#enabled) {\n return next(context);\n }\n const { request } = context;\n\n if (!request.dataTypes.includes('balance')) {\n return next(context);\n }\n\n if (request.chainIds.length === 0) {\n return next(context);\n }\n\n try {\n const fetchResponse = await this.fetch(request);\n\n if (fetchResponse.assetsInfo) {\n context.response.assetsInfo ??= {};\n Object.assign(context.response.assetsInfo, fetchResponse.assetsInfo);\n }\n if (fetchResponse.assetsBalance) {\n context.response.assetsBalance ??= {};\n for (const [accountId, accountBalances] of Object.entries(\n fetchResponse.assetsBalance,\n )) {\n context.response.assetsBalance[accountId] = {\n ...context.response.assetsBalance[accountId],\n ...accountBalances,\n };\n }\n }\n } catch (error) {\n log('Middleware fetch failed', { error });\n }\n\n // Pass all chains through (staked balance doesn't claim chains)\n return next(context);\n };\n }\n\n /**\n * Subscribe to staked balance updates with polling.\n * Starts polling via StakedBalanceFetcher for each account/chain combination.\n *\n * @param subscriptionRequest - The subscription request details.\n */\n async subscribe(subscriptionRequest: SubscriptionRequest): Promise<void> {\n const { request, subscriptionId, isUpdate } = subscriptionRequest;\n\n const activeChainsSet = new Set(this.state.activeChains);\n const chainsToSubscribe = request.chainIds.filter((chainId) =>\n activeChainsSet.has(chainId),\n );\n\n log('Subscribe requested', {\n subscriptionId,\n isUpdate,\n chainsToSubscribe,\n });\n\n if (chainsToSubscribe.length === 0) {\n log('No staking chains to subscribe');\n return;\n }\n\n // Handle subscription update - restart polling for new chains\n if (isUpdate) {\n const existing = this.#activeSubscriptions.get(subscriptionId);\n if (existing) {\n log('Updating existing subscription - restarting polling', {\n subscriptionId,\n });\n }\n }\n\n // Clean up existing subscription (stops old polling)\n await this.unsubscribe(subscriptionId);\n\n // Build subscription data first so it is available when the first poll runs\n const accountsWithSupportedChains: AccountWithSupportedChains[] =\n request.accountsWithSupportedChains\n .map(({ account, supportedChains }) => ({\n account,\n supportedChains: chainsToSubscribe.filter((chain) =>\n supportedChains.includes(chain),\n ),\n }))\n .filter(({ supportedChains }) => supportedChains.length > 0);\n\n const accounts = accountsWithSupportedChains.map((entry) => entry.account);\n const pollingTokens: string[] = [];\n\n // Store subscription before startPolling so first poll (setTimeout 0) has the callback\n this.#activeSubscriptions.set(subscriptionId, {\n pollingTokens,\n chains: chainsToSubscribe,\n accounts,\n accountsWithSupportedChains,\n onAssetsUpdate: subscriptionRequest.onAssetsUpdate,\n });\n\n this.activeSubscriptions.set(subscriptionId, {\n cleanup: () => {\n for (const token of pollingTokens) {\n this.#stakedBalanceFetcher.stopPollingByPollingToken(token);\n }\n this.#activeSubscriptions.delete(subscriptionId);\n },\n chains: chainsToSubscribe,\n request,\n onAssetsUpdate: subscriptionRequest.onAssetsUpdate,\n });\n\n // Start polling for each account/chain (first poll runs on next tick)\n for (const {\n account,\n supportedChains: accountChains,\n } of request.accountsWithSupportedChains) {\n const chainsForAccount = chainsToSubscribe.filter((chain) =>\n accountChains.includes(chain),\n );\n\n for (const chainId of chainsForAccount) {\n const hexChainId = caipChainIdToHex(chainId);\n\n const input: StakedBalancePollingInput = {\n chainId: hexChainId,\n accountId: account.id,\n accountAddress: account.address as Hex,\n };\n\n const pollingToken = this.#stakedBalanceFetcher.startPolling(input);\n pollingTokens.push(pollingToken);\n }\n }\n\n // Immediate initial fetch so state is updated without waiting for first poll\n try {\n const initialRequest: DataRequest = {\n accountsWithSupportedChains,\n chainIds: chainsToSubscribe,\n dataTypes: ['balance'],\n };\n const initialResponse = await this.fetch(initialRequest);\n if (\n initialResponse.assetsBalance &&\n Object.keys(initialResponse.assetsBalance).length > 0\n ) {\n subscriptionRequest\n .onAssetsUpdate?.(initialResponse)\n ?.catch((error) => {\n log('Initial staked balance update failed', { error });\n });\n }\n } catch (error) {\n log('Initial staked balance fetch failed', { error });\n }\n\n log('Subscription SUCCESS', {\n subscriptionId,\n chains: chainsToSubscribe,\n pollingCount: pollingTokens.length,\n });\n }\n\n /**\n * Unsubscribe from staked balance updates and stop polling.\n *\n * @param subscriptionId - The subscription ID to unsubscribe.\n */\n async unsubscribe(subscriptionId: string): Promise<void> {\n const subscription = this.#activeSubscriptions.get(subscriptionId);\n if (subscription) {\n for (const token of subscription.pollingTokens) {\n this.#stakedBalanceFetcher.stopPollingByPollingToken(token);\n }\n this.#activeSubscriptions.delete(subscriptionId);\n }\n\n await super.unsubscribe(subscriptionId);\n }\n\n /**\n * Destroy the data source and clean up all resources.\n */\n destroy(): void {\n for (const subscription of this.#activeSubscriptions.values()) {\n for (const token of subscription.pollingTokens) {\n this.#stakedBalanceFetcher.stopPollingByPollingToken(token);\n }\n }\n this.#activeSubscriptions.clear();\n this.#providerCache.clear();\n super.destroy();\n }\n}\n"]}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"StakedBalanceDataSource.d.cts","sourceRoot":"","sources":["../../src/data-sources/StakedBalanceDataSource.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EACV,eAAe,EACf,mBAAmB,EACpB,iCAA6B;AAC9B,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAU1D,OAAO,KAAK,EAAE,yBAAyB,EAAE,gCAA4B;AAErE,OAAO,KAAK,EAEV,OAAO,EAIP,WAAW,EACX,YAAY,EACZ,UAAU,EACX,qBAAiB;AAElB,QAAA,MAAM,eAAe,4BAA4B,CAAC;AAalD,0DAA0D;AAC1D,MAAM,MAAM,6BAA6B,GAAG;IAC1C,kEAAkE;IAClE,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,sDAAsD;IACtD,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,8BAA8B,GAAG,6BAA6B,GAAG;IAC3E,wEAAwE;IACxE,SAAS,EAAE,yBAAyB,CAAC;IACrC,6CAA6C;IAC7C,qBAAqB,EAAE,CACrB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,OAAO,EAAE,EACjB,cAAc,EAAE,OAAO,EAAE,KACtB,IAAI,CAAC;CACX,CAAC;AA+DF;;;;;;;;;;GAUG;AACH,qBAAa,uBAAwB,SAAQ,kBAAkB,CAC7D,OAAO,eAAe,EACtB,eAAe,CAChB;;gBAyBa,OAAO,EAAE,8BAA8B;
|
|
1
|
+
{"version":3,"file":"StakedBalanceDataSource.d.cts","sourceRoot":"","sources":["../../src/data-sources/StakedBalanceDataSource.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EACV,eAAe,EACf,mBAAmB,EACpB,iCAA6B;AAC9B,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAU1D,OAAO,KAAK,EAAE,yBAAyB,EAAE,gCAA4B;AAErE,OAAO,KAAK,EAEV,OAAO,EAIP,WAAW,EACX,YAAY,EACZ,UAAU,EACX,qBAAiB;AAElB,QAAA,MAAM,eAAe,4BAA4B,CAAC;AAalD,0DAA0D;AAC1D,MAAM,MAAM,6BAA6B,GAAG;IAC1C,kEAAkE;IAClE,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,sDAAsD;IACtD,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,8BAA8B,GAAG,6BAA6B,GAAG;IAC3E,wEAAwE;IACxE,SAAS,EAAE,yBAAyB,CAAC;IACrC,6CAA6C;IAC7C,qBAAqB,EAAE,CACrB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,OAAO,EAAE,EACjB,cAAc,EAAE,OAAO,EAAE,KACtB,IAAI,CAAC;CACX,CAAC;AA+DF;;;;;;;;;;GAUG;AACH,qBAAa,uBAAwB,SAAQ,kBAAkB,CAC7D,OAAO,eAAe,EACtB,eAAe,CAChB;;gBAyBa,OAAO,EAAE,8BAA8B;IA2NnD;;;OAGG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAwO3C;;;;;OAKG;IACG,KAAK,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IA0ExD;;;;;;OAMG;IACH,IAAI,gBAAgB,IAAI,UAAU,CAwCjC;IAED;;;;;OAKG;IACG,SAAS,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAuHxE;;;;OAIG;IACG,WAAW,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAYxD;;OAEG;IACH,OAAO,IAAI,IAAI;CAUhB"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"StakedBalanceDataSource.d.mts","sourceRoot":"","sources":["../../src/data-sources/StakedBalanceDataSource.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EACV,eAAe,EACf,mBAAmB,EACpB,iCAA6B;AAC9B,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAU1D,OAAO,KAAK,EAAE,yBAAyB,EAAE,gCAA4B;AAErE,OAAO,KAAK,EAEV,OAAO,EAIP,WAAW,EACX,YAAY,EACZ,UAAU,EACX,qBAAiB;AAElB,QAAA,MAAM,eAAe,4BAA4B,CAAC;AAalD,0DAA0D;AAC1D,MAAM,MAAM,6BAA6B,GAAG;IAC1C,kEAAkE;IAClE,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,sDAAsD;IACtD,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,8BAA8B,GAAG,6BAA6B,GAAG;IAC3E,wEAAwE;IACxE,SAAS,EAAE,yBAAyB,CAAC;IACrC,6CAA6C;IAC7C,qBAAqB,EAAE,CACrB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,OAAO,EAAE,EACjB,cAAc,EAAE,OAAO,EAAE,KACtB,IAAI,CAAC;CACX,CAAC;AA+DF;;;;;;;;;;GAUG;AACH,qBAAa,uBAAwB,SAAQ,kBAAkB,CAC7D,OAAO,eAAe,EACtB,eAAe,CAChB;;gBAyBa,OAAO,EAAE,8BAA8B;
|
|
1
|
+
{"version":3,"file":"StakedBalanceDataSource.d.mts","sourceRoot":"","sources":["../../src/data-sources/StakedBalanceDataSource.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EACV,eAAe,EACf,mBAAmB,EACpB,iCAA6B;AAC9B,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAU1D,OAAO,KAAK,EAAE,yBAAyB,EAAE,gCAA4B;AAErE,OAAO,KAAK,EAEV,OAAO,EAIP,WAAW,EACX,YAAY,EACZ,UAAU,EACX,qBAAiB;AAElB,QAAA,MAAM,eAAe,4BAA4B,CAAC;AAalD,0DAA0D;AAC1D,MAAM,MAAM,6BAA6B,GAAG;IAC1C,kEAAkE;IAClE,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,sDAAsD;IACtD,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,MAAM,MAAM,8BAA8B,GAAG,6BAA6B,GAAG;IAC3E,wEAAwE;IACxE,SAAS,EAAE,yBAAyB,CAAC;IACrC,6CAA6C;IAC7C,qBAAqB,EAAE,CACrB,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,OAAO,EAAE,EACjB,cAAc,EAAE,OAAO,EAAE,KACtB,IAAI,CAAC;CACX,CAAC;AA+DF;;;;;;;;;;GAUG;AACH,qBAAa,uBAAwB,SAAQ,kBAAkB,CAC7D,OAAO,eAAe,EACtB,eAAe,CAChB;;gBAyBa,OAAO,EAAE,8BAA8B;IA2NnD;;;OAGG;IACG,oBAAoB,IAAI,OAAO,CAAC,IAAI,CAAC;IAwO3C;;;;;OAKG;IACG,KAAK,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IA0ExD;;;;;;OAMG;IACH,IAAI,gBAAgB,IAAI,UAAU,CAwCjC;IAED;;;;;OAKG;IACG,SAAS,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAuHxE;;;;OAIG;IACG,WAAW,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAYxD;;OAEG;IACH,OAAO,IAAI,IAAI;CAUhB"}
|
|
@@ -9,7 +9,7 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
|
|
|
9
9
|
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");
|
|
10
10
|
return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
|
|
11
11
|
};
|
|
12
|
-
var _StakedBalanceDataSource_instances, _StakedBalanceDataSource_messenger, _StakedBalanceDataSource_onActiveChainsUpdated, _StakedBalanceDataSource_pollInterval, _StakedBalanceDataSource_enabled, _StakedBalanceDataSource_stakedBalanceFetcher, _StakedBalanceDataSource_activeSubscriptions, _StakedBalanceDataSource_providerCache, _StakedBalanceDataSource_supportedChainIds,
|
|
12
|
+
var _StakedBalanceDataSource_instances, _StakedBalanceDataSource_messenger, _StakedBalanceDataSource_onActiveChainsUpdated, _StakedBalanceDataSource_pollInterval, _StakedBalanceDataSource_enabled, _StakedBalanceDataSource_stakedBalanceFetcher, _StakedBalanceDataSource_activeSubscriptions, _StakedBalanceDataSource_providerCache, _StakedBalanceDataSource_supportedChainIds, _StakedBalanceDataSource_onNetworkStateChange, _StakedBalanceDataSource_onNetworkEnablementControllerStateChange, _StakedBalanceDataSource_isTransactionInvolvingStakingContract, _StakedBalanceDataSource_onTransactionConfirmed, _StakedBalanceDataSource_onIncomingTransactions, _StakedBalanceDataSource_getToRefreshForChains, _StakedBalanceDataSource_getToRefreshAll, _StakedBalanceDataSource_refreshStakedBalanceAfterTransaction, _StakedBalanceDataSource_initializeActiveChains, _StakedBalanceDataSource_initializeActiveChainsFromEnabledMap, _StakedBalanceDataSource_getProvider, _StakedBalanceDataSource_handleStakedBalanceUpdate;
|
|
13
13
|
import { toChecksumAddress } from "@ethereumjs/util";
|
|
14
14
|
import { Web3Provider } from "@ethersproject/providers";
|
|
15
15
|
import { isStrictHexString, isCaipChainId, numberToHex, parseCaipChainId } from "@metamask/utils";
|
|
@@ -83,10 +83,6 @@ export class StakedBalanceDataSource extends AbstractDataSource {
|
|
|
83
83
|
_StakedBalanceDataSource_providerCache.set(this, new Map());
|
|
84
84
|
/** CAIP-2 chain IDs that have known staking contracts. */
|
|
85
85
|
_StakedBalanceDataSource_supportedChainIds.set(this, void 0);
|
|
86
|
-
_StakedBalanceDataSource_unsubscribeTransactionConfirmed.set(this, undefined);
|
|
87
|
-
_StakedBalanceDataSource_unsubscribeIncomingTransactions.set(this, undefined);
|
|
88
|
-
_StakedBalanceDataSource_unsubscribeNetworkStateChange.set(this, undefined);
|
|
89
|
-
_StakedBalanceDataSource_unsubscribeNetworkEnablementControllerStateChange.set(this, undefined);
|
|
90
86
|
__classPrivateFieldSet(this, _StakedBalanceDataSource_messenger, options.messenger, "f");
|
|
91
87
|
__classPrivateFieldSet(this, _StakedBalanceDataSource_onActiveChainsUpdated, options.onActiveChainsUpdated, "f");
|
|
92
88
|
__classPrivateFieldSet(this, _StakedBalanceDataSource_pollInterval, options.pollInterval ?? DEFAULT_POLL_INTERVAL, "f");
|
|
@@ -103,14 +99,10 @@ export class StakedBalanceDataSource extends AbstractDataSource {
|
|
|
103
99
|
}), "f");
|
|
104
100
|
// Wire the callback so polling results flow back to subscriptions
|
|
105
101
|
__classPrivateFieldGet(this, _StakedBalanceDataSource_stakedBalanceFetcher, "f").setOnStakedBalanceUpdate(__classPrivateFieldGet(this, _StakedBalanceDataSource_instances, "m", _StakedBalanceDataSource_handleStakedBalanceUpdate).bind(this));
|
|
106
|
-
|
|
107
|
-
|
|
108
|
-
|
|
109
|
-
|
|
110
|
-
const unsubNetwork = __classPrivateFieldGet(this, _StakedBalanceDataSource_messenger, "f").subscribe('NetworkController:stateChange', __classPrivateFieldGet(this, _StakedBalanceDataSource_instances, "m", _StakedBalanceDataSource_onNetworkStateChange).bind(this));
|
|
111
|
-
__classPrivateFieldSet(this, _StakedBalanceDataSource_unsubscribeNetworkStateChange, typeof unsubNetwork === 'function' ? unsubNetwork : undefined, "f");
|
|
112
|
-
const unsubEnablement = __classPrivateFieldGet(this, _StakedBalanceDataSource_messenger, "f").subscribe('NetworkEnablementController:stateChange', __classPrivateFieldGet(this, _StakedBalanceDataSource_instances, "m", _StakedBalanceDataSource_onNetworkEnablementControllerStateChange).bind(this));
|
|
113
|
-
__classPrivateFieldSet(this, _StakedBalanceDataSource_unsubscribeNetworkEnablementControllerStateChange, typeof unsubEnablement === 'function' ? unsubEnablement : undefined, "f");
|
|
102
|
+
__classPrivateFieldGet(this, _StakedBalanceDataSource_messenger, "f").subscribe('TransactionController:transactionConfirmed', __classPrivateFieldGet(this, _StakedBalanceDataSource_instances, "m", _StakedBalanceDataSource_onTransactionConfirmed).bind(this));
|
|
103
|
+
__classPrivateFieldGet(this, _StakedBalanceDataSource_messenger, "f").subscribe('TransactionController:incomingTransactionsReceived', __classPrivateFieldGet(this, _StakedBalanceDataSource_instances, "m", _StakedBalanceDataSource_onIncomingTransactions).bind(this));
|
|
104
|
+
__classPrivateFieldGet(this, _StakedBalanceDataSource_messenger, "f").subscribe('NetworkController:stateChange', __classPrivateFieldGet(this, _StakedBalanceDataSource_instances, "m", _StakedBalanceDataSource_onNetworkStateChange).bind(this));
|
|
105
|
+
__classPrivateFieldGet(this, _StakedBalanceDataSource_messenger, "f").subscribe('NetworkEnablementController:stateChange', __classPrivateFieldGet(this, _StakedBalanceDataSource_instances, "m", _StakedBalanceDataSource_onNetworkEnablementControllerStateChange).bind(this));
|
|
114
106
|
__classPrivateFieldGet(this, _StakedBalanceDataSource_instances, "m", _StakedBalanceDataSource_initializeActiveChains).call(this);
|
|
115
107
|
}
|
|
116
108
|
/**
|
|
@@ -349,10 +341,6 @@ export class StakedBalanceDataSource extends AbstractDataSource {
|
|
|
349
341
|
* Destroy the data source and clean up all resources.
|
|
350
342
|
*/
|
|
351
343
|
destroy() {
|
|
352
|
-
__classPrivateFieldGet(this, _StakedBalanceDataSource_unsubscribeTransactionConfirmed, "f")?.call(this);
|
|
353
|
-
__classPrivateFieldGet(this, _StakedBalanceDataSource_unsubscribeIncomingTransactions, "f")?.call(this);
|
|
354
|
-
__classPrivateFieldGet(this, _StakedBalanceDataSource_unsubscribeNetworkStateChange, "f")?.call(this);
|
|
355
|
-
__classPrivateFieldGet(this, _StakedBalanceDataSource_unsubscribeNetworkEnablementControllerStateChange, "f")?.call(this);
|
|
356
344
|
for (const subscription of __classPrivateFieldGet(this, _StakedBalanceDataSource_activeSubscriptions, "f").values()) {
|
|
357
345
|
for (const token of subscription.pollingTokens) {
|
|
358
346
|
__classPrivateFieldGet(this, _StakedBalanceDataSource_stakedBalanceFetcher, "f").stopPollingByPollingToken(token);
|
|
@@ -363,7 +351,7 @@ export class StakedBalanceDataSource extends AbstractDataSource {
|
|
|
363
351
|
super.destroy();
|
|
364
352
|
}
|
|
365
353
|
}
|
|
366
|
-
_StakedBalanceDataSource_messenger = new WeakMap(), _StakedBalanceDataSource_onActiveChainsUpdated = new WeakMap(), _StakedBalanceDataSource_pollInterval = new WeakMap(), _StakedBalanceDataSource_enabled = new WeakMap(), _StakedBalanceDataSource_stakedBalanceFetcher = new WeakMap(), _StakedBalanceDataSource_activeSubscriptions = new WeakMap(), _StakedBalanceDataSource_providerCache = new WeakMap(), _StakedBalanceDataSource_supportedChainIds = new WeakMap(),
|
|
354
|
+
_StakedBalanceDataSource_messenger = new WeakMap(), _StakedBalanceDataSource_onActiveChainsUpdated = new WeakMap(), _StakedBalanceDataSource_pollInterval = new WeakMap(), _StakedBalanceDataSource_enabled = new WeakMap(), _StakedBalanceDataSource_stakedBalanceFetcher = new WeakMap(), _StakedBalanceDataSource_activeSubscriptions = new WeakMap(), _StakedBalanceDataSource_providerCache = new WeakMap(), _StakedBalanceDataSource_supportedChainIds = new WeakMap(), _StakedBalanceDataSource_instances = new WeakSet(), _StakedBalanceDataSource_onNetworkStateChange = function _StakedBalanceDataSource_onNetworkStateChange() {
|
|
367
355
|
__classPrivateFieldGet(this, _StakedBalanceDataSource_providerCache, "f").clear();
|
|
368
356
|
log('Provider cache cleared after network state change');
|
|
369
357
|
}, _StakedBalanceDataSource_onNetworkEnablementControllerStateChange = function _StakedBalanceDataSource_onNetworkEnablementControllerStateChange(state) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"StakedBalanceDataSource.mjs","sourceRoot":"","sources":["../../src/data-sources/StakedBalanceDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,iBAAiB,EAAE,yBAAyB;AACrD,OAAO,EAAE,YAAY,EAAE,iCAAiC;AAGxD,OAAO,EACL,iBAAiB,EACjB,aAAa,EACb,WAAW,EACX,gBAAgB,EACjB,wBAAwB;AAOzB,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAK1D,OAAO,EACL,oBAAoB,EACpB,yBAAyB,EACzB,2BAA2B,EAC5B,qCAA2B;AAE5B,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,sBAAkB;AAY9D,MAAM,eAAe,GAAG,yBAAyB,CAAC;AAClD,MAAM,qBAAqB,GAAG,MAAO,CAAC,CAAC,YAAY;AAEnD,wEAAwE;AACxE,MAAM,mBAAmB,GAAkB;IACzC,IAAI,EAAE,OAAO;IACb,IAAI,EAAE,iBAAiB;IACvB,MAAM,EAAE,KAAK;IACb,QAAQ,EAAE,EAAE;CACb,CAAC;AAEF,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;AA6C/D;;;;;GAKG;AACH,SAAS,gBAAgB,CAAC,OAAe;IACvC,IAAI,iBAAiB,CAAC,OAAO,CAAC,EAAE,CAAC;QAC/B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,MAAM,GAAG,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC;QAChD,OAAO,WAAW,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;IACxC,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;AAChF,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,aAAa,CACpB,OAAgB,EAChB,eAAuB;IAEvB,MAAM,WAAW,GAAG,iBAAiB,CAAC,eAAe,CAAC,CAAC;IACvD,OAAO,GAAG,OAAO,UAAU,WAAW,EAAmB,CAAC;AAC5D,CAAC;AAED;;;;;;;;;;GAUG;AACH,MAAM,OAAO,uBAAwB,SAAQ,kBAG5C;IAyBC,YAAY,OAAuC;QACjD,KAAK,CAAC,eAAe,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;;QAzBtC,qDAAsC;QAEtC,iEAIC;QAED,wDAAsB;QAEtB,mDAAkB;QAE3B,mEAAmE;QAC1D,gEAA4C;QAErD,kCAAkC;QACzB,uDAAsD,IAAI,GAAG,EAAE,EAAC;QAEzE,uDAAuD;QAC9C,iDAAyC,IAAI,GAAG,EAAE,EAAC;QAE5D,0DAA0D;QACjD,6DAA8B;QA0D9B,mEACP,SAAS,EAAC;QAEH,mEACP,SAAS,EAAC;QAEH,iEAA2D,SAAS,EAAC;QAErE,qFAEO,SAAS,EAAC;QAhExB,uBAAA,IAAI,sCAAc,OAAO,CAAC,SAAS,MAAA,CAAC;QACpC,uBAAA,IAAI,kDAA0B,OAAO,CAAC,qBAAqB,MAAA,CAAC;QAC5D,uBAAA,IAAI,yCAAiB,OAAO,CAAC,YAAY,IAAI,qBAAqB,MAAA,CAAC;QACnE,uBAAA,IAAI,oCAAY,OAAO,CAAC,OAAO,KAAK,KAAK,MAAA,CAAC;QAC1C,uBAAA,IAAI,8CAAsB,2BAA2B,EAAe,MAAA,CAAC;QAErE,GAAG,CAAC,sCAAsC,EAAE;YAC1C,OAAO,EAAE,uBAAA,IAAI,wCAAS;YACtB,YAAY,EAAE,uBAAA,IAAI,6CAAc;SACjC,CAAC,CAAC;QAEH,mDAAmD;QACnD,uBAAA,IAAI,iDAAyB,IAAI,oBAAoB,CAAC;YACpD,eAAe,EAAE,uBAAA,IAAI,6CAAc;YACnC,kBAAkB,EAAE,CAAC,UAAU,EAA4B,EAAE,CAC3D,uBAAA,IAAI,gFAAa,MAAjB,IAAI,EAAc,UAAU,CAAC;SAChC,CAAC,MAAA,CAAC;QAEH,kEAAkE;QAClE,uBAAA,IAAI,qDAAsB,CAAC,wBAAwB,CACjD,uBAAA,IAAI,8FAA2B,CAAC,IAAI,CAAC,IAAI,CAAC,CAC3C,CAAC;QAEF,MAAM,cAAc,GAAG,uBAAA,IAAI,0CAAW,CAAC,SAAS,CAC9C,4CAA4C,EAC5C,uBAAA,IAAI,2FAAwB,CAAC,IAAI,CAAC,IAAI,CAAC,CACxC,CAAC;QACF,uBAAA,IAAI,4DACF,OAAO,cAAc,KAAK,UAAU,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS,MAAA,CAAC;QAEpE,MAAM,aAAa,GAAG,uBAAA,IAAI,0CAAW,CAAC,SAAS,CAC7C,oDAAoD,EACpD,uBAAA,IAAI,2FAAwB,CAAC,IAAI,CAAC,IAAI,CAAC,CACxC,CAAC;QACF,uBAAA,IAAI,4DACF,OAAO,aAAa,KAAK,UAAU,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,MAAA,CAAC;QAElE,MAAM,YAAY,GAAG,uBAAA,IAAI,0CAAW,CAAC,SAAS,CAC5C,+BAA+B,EAC/B,uBAAA,IAAI,yFAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CACtC,CAAC;QACF,uBAAA,IAAI,0DACF,OAAO,YAAY,KAAK,UAAU,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,MAAA,CAAC;QAEhE,MAAM,eAAe,GAAG,uBAAA,IAAI,0CAAW,CAAC,SAAS,CAC/C,yCAAyC,EACzC,uBAAA,IAAI,6GAA0C,CAAC,IAAI,CAAC,IAAI,CAAC,CAC1D,CAAC;QACF,uBAAA,IAAI,8EACF,OAAO,eAAe,KAAK,UAAU,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,SAAS,MAAA,CAAC;QAEtE,uBAAA,IAAI,2FAAwB,MAA5B,IAAI,CAA0B,CAAC;IACjC,CAAC;IAyLD;;;OAGG;IACH,KAAK,CAAC,oBAAoB;QACxB,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,uBAAA,IAAI,oFAAiB,MAArB,IAAI,CAAmB,CAAC;QAC1C,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,MAAM,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,SAAS,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAgOD;;;;;OAKG;IACH,KAAK,CAAC,KAAK,CAAC,OAAoB;;QAC9B,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;YACnB,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,QAAQ,GAAiB,EAAE,CAAC;QAClC,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAEzD,MAAM,aAAa,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CACxD,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAC7B,CAAC;QAEF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,QAAQ,GAA2D,EAAE,CAAC;QAE5E,KAAK,MAAM,EACT,OAAO,EACP,eAAe,EAAE,aAAa,GAC/B,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;YACzC,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAC5C,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,CAC9B,CAAC;YAEF,KAAK,MAAM,OAAO,IAAI,MAAM,EAAE,CAAC;gBAC7B,IAAI,CAAC;oBACH,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;oBAC7C,MAAM,eAAe,GAAG,yBAAyB,CAAC,UAAU,CAAC,CAAC;oBAC9D,IAAI,CAAC,eAAe,EAAE,CAAC;wBACrB,SAAS;oBACX,CAAC;oBAED,MAAM,KAAK,GAA8B;wBACvC,OAAO,EAAE,UAAU;wBACnB,SAAS,EAAE,OAAO,CAAC,EAAE;wBACrB,cAAc,EAAE,OAAO,CAAC,OAAc;qBACvC,CAAC;oBAEF,MAAM,MAAM,GACV,MAAM,uBAAA,IAAI,qDAAsB,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;oBAE7D,qEAAqE;oBACrE,QAAQ,MAAC,OAAO,CAAC,EAAE,MAAnB,QAAQ,OAAiB,EAAE,EAAC;oBAC5B,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;oBACxD,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC5D,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,GAAG,CAAC,gCAAgC,EAAE;wBACpC,OAAO;wBACP,SAAS,EAAE,OAAO,CAAC,EAAE;wBACrB,KAAK;qBACN,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrC,QAAQ,CAAC,aAAa,GAAG,QAAQ,CAAC;YAClC,4DAA4D;YAC5D,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAiB,CAAC;YAC1C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACtD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;oBACnD,QAAQ,CAAC,GAAG,CAAC,OAAwB,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;YACD,QAAQ,CAAC,UAAU,GAAG,EAAE,CAAC;YACzB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,mBAAmB,CAAC;YACrD,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;;;OAMG;IACH,IAAI,gBAAgB;QAClB,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;;YAC7B,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;gBACnB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YACD,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;YAE5B,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC3C,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAClC,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAEhD,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;oBAC7B,MAAA,OAAO,CAAC,QAAQ,EAAC,UAAU,QAAV,UAAU,GAAK,EAAE,EAAC;oBACnC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,aAAa,CAAC,UAAU,CAAC,CAAC;gBACvE,CAAC;gBACD,IAAI,aAAa,CAAC,aAAa,EAAE,CAAC;oBAChC,MAAA,OAAO,CAAC,QAAQ,EAAC,aAAa,QAAb,aAAa,GAAK,EAAE,EAAC;oBACtC,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,aAAa,CAAC,aAAa,CAC5B,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;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC5C,CAAC;YAED,gEAAgE;YAChE,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,SAAS,CAAC,mBAAwC;QACtD,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC;QAElE,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACzD,MAAM,iBAAiB,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5D,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAC7B,CAAC;QAEF,GAAG,CAAC,qBAAqB,EAAE;YACzB,cAAc;YACd,QAAQ;YACR,iBAAiB;SAClB,CAAC,CAAC;QAEH,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,GAAG,CAAC,gCAAgC,CAAC,CAAC;YACtC,OAAO;QACT,CAAC;QAED,8DAA8D;QAC9D,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,uBAAA,IAAI,oDAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC/D,IAAI,QAAQ,EAAE,CAAC;gBACb,GAAG,CAAC,qDAAqD,EAAE;oBACzD,cAAc;iBACf,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,qDAAqD;QACrD,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;QAEvC,4EAA4E;QAC5E,MAAM,2BAA2B,GAC/B,OAAO,CAAC,2BAA2B;aAChC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,CAAC;YACtC,OAAO;YACP,eAAe,EAAE,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAClD,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,CAChC;SACF,CAAC,CAAC;aACF,MAAM,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAEjE,MAAM,QAAQ,GAAG,2BAA2B,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC3E,MAAM,aAAa,GAAa,EAAE,CAAC;QAEnC,uFAAuF;QACvF,uBAAA,IAAI,oDAAqB,CAAC,GAAG,CAAC,cAAc,EAAE;YAC5C,aAAa;YACb,MAAM,EAAE,iBAAiB;YACzB,QAAQ;YACR,2BAA2B;YAC3B,cAAc,EAAE,mBAAmB,CAAC,cAAc;SACnD,CAAC,CAAC;QAEH,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,EAAE;YAC3C,OAAO,EAAE,GAAG,EAAE;gBACZ,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;oBAClC,uBAAA,IAAI,qDAAsB,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;gBAC9D,CAAC;gBACD,uBAAA,IAAI,oDAAqB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,EAAE,iBAAiB;YACzB,OAAO;YACP,cAAc,EAAE,mBAAmB,CAAC,cAAc;SACnD,CAAC,CAAC;QAEH,sEAAsE;QACtE,KAAK,MAAM,EACT,OAAO,EACP,eAAe,EAAE,aAAa,GAC/B,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;YACzC,MAAM,gBAAgB,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAC1D,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,CAC9B,CAAC;YAEF,KAAK,MAAM,OAAO,IAAI,gBAAgB,EAAE,CAAC;gBACvC,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;gBAE7C,MAAM,KAAK,GAA8B;oBACvC,OAAO,EAAE,UAAU;oBACnB,SAAS,EAAE,OAAO,CAAC,EAAE;oBACrB,cAAc,EAAE,OAAO,CAAC,OAAc;iBACvC,CAAC;gBAEF,MAAM,YAAY,GAAG,uBAAA,IAAI,qDAAsB,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpE,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,6EAA6E;QAC7E,IAAI,CAAC;YACH,MAAM,cAAc,GAAgB;gBAClC,2BAA2B;gBAC3B,QAAQ,EAAE,iBAAiB;gBAC3B,SAAS,EAAE,CAAC,SAAS,CAAC;aACvB,CAAC;YACF,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YACzD,IACE,eAAe,CAAC,aAAa;gBAC7B,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC,MAAM,GAAG,CAAC,EACrD,CAAC;gBACD,mBAAmB;qBAChB,cAAc,EAAE,CAAC,eAAe,CAAC;oBAClC,EAAE,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBAChB,GAAG,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;gBACzD,CAAC,CAAC,CAAC;YACP,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,qCAAqC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACxD,CAAC;QAED,GAAG,CAAC,sBAAsB,EAAE;YAC1B,cAAc;YACd,MAAM,EAAE,iBAAiB;YACzB,YAAY,EAAE,aAAa,CAAC,MAAM;SACnC,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,WAAW,CAAC,cAAsB;QACtC,MAAM,YAAY,GAAG,uBAAA,IAAI,oDAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACnE,IAAI,YAAY,EAAE,CAAC;YACjB,KAAK,MAAM,KAAK,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;gBAC/C,uBAAA,IAAI,qDAAsB,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;YAC9D,CAAC;YACD,uBAAA,IAAI,oDAAqB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,KAAK,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,OAAO;QACL,uBAAA,IAAI,gEAAiC,EAAE,KAAvC,IAAI,CAAqC,CAAC;QAC1C,uBAAA,IAAI,gEAAiC,EAAE,KAAvC,IAAI,CAAqC,CAAC;QAC1C,uBAAA,IAAI,8DAA+B,EAAE,KAArC,IAAI,CAAmC,CAAC;QACxC,uBAAA,IAAI,kFAAmD,EAAE,KAAzD,IAAI,CAAuD,CAAC;QAC5D,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YAC9D,KAAK,MAAM,KAAK,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;gBAC/C,uBAAA,IAAI,qDAAsB,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;QACD,uBAAA,IAAI,oDAAqB,CAAC,KAAK,EAAE,CAAC;QAClC,uBAAA,IAAI,8CAAe,CAAC,KAAK,EAAE,CAAC;QAC5B,KAAK,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC;CACF;;IAjrBG,uBAAA,IAAI,8CAAe,CAAC,KAAK,EAAE,CAAC;IAC5B,GAAG,CAAC,mDAAmD,CAAC,CAAC;AAC3D,CAAC,iJASC,KAAuC;IAEvC,MAAM,EAAE,iBAAiB,EAAE,GAAG,KAAK,IAAI,EAAE,CAAC;IAC1C,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,OAAO;IACT,CAAC;IACD,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,iBAAiB,CAAC,CAAC;AAChE,CAAC,2IAasC,OAGtC;IACC,MAAM,UAAU,GAAG,OAAO,EAAE,OAAO,CAAC;IACpC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,eAAe,GAAG,yBAAyB,CAAC,UAAU,CAAC,CAAC;IAC9D,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,aAAa,GAAG,eAAe,CAAC,WAAW,EAAE,CAAC;IACpD,MAAM,IAAI,GAAG,OAAO,CAAC,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;IACnD,MAAM,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE,EAAE,EAAE,WAAW,EAAE,CAAC;IAC/C,OAAO,IAAI,KAAK,aAAa,IAAI,EAAE,KAAK,aAAa,CAAC;AACxD,CAAC,6GAYuB,OAGvB;IACC,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;QACnB,OAAO;IACT,CAAC;IACD,IAAI,CAAC,uBAAA,IAAI,0GAAuC,MAA3C,IAAI,EAAwC,OAAO,CAAC,EAAE,CAAC;QAC1D,OAAO;IACT,CAAC;IACD,MAAM,UAAU,GAAG,OAAO,EAAE,OAAO,CAAC;IACpC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO;IACT,CAAC;IACD,MAAM,WAAW,GAAG,UAAU,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,EAAa,CAAC;IACpE,MAAM,SAAS,GAAG,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,CAAC,WAAW,CAAC,CAAC,CAAC;IAC7D,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACpE,GAAG,CAAC,oDAAoD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC,6GASC,OAA0E;IAE1E,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;QACnB,OAAO;IACT,CAAC;IACD,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAU,CAAC;IAC5C,KAAK,MAAM,IAAI,IAAI,OAAO,IAAI,EAAE,EAAE,CAAC;QACjC,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC;YACnB,SAAS;QACX,CAAC;QACD,IAAI,uBAAA,IAAI,0GAAuC,MAA3C,IAAI,EAAwC,IAAI,CAAC,EAAE,CAAC;YACtD,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IACD,MAAM,YAAY,GAAG,CAAC,GAAG,iBAAiB,CAAC,CAAC,GAAG,CAC7C,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,EAAa,CAChE,CAAC;IACF,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO;IACT,CAAC;IACD,MAAM,SAAS,GAAG,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,YAAY,CAAC,CAAC;IAC5D,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACpE,GAAG,CAAC,8DAA8D,EAAE;gBAClE,KAAK;aACN,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC,2GASC,QAAmB;IAEnB,MAAM,SAAS,GAAqD,EAAE,CAAC;IACvE,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC;IACnC,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9D,KAAK,MAAM,EACT,OAAO,EACP,eAAe,GAChB,IAAI,YAAY,CAAC,2BAA2B,EAAE,CAAC;YAC9C,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;gBACtC,IAAI,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC1B,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;gBACvC,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;IAQC,MAAM,SAAS,GAAqD,EAAE,CAAC;IACvE,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9D,KAAK,MAAM,EACT,OAAO,EACP,eAAe,GAChB,IAAI,YAAY,CAAC,2BAA2B,EAAE,CAAC;YAC9C,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;gBACtC,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YACvC,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAgBD;;;;;GAKG;AACH,KAAK,wEACH,SAA2D;IAE3D,MAAM,UAAU,GAAyC,EAAE,CAAC;IAC5D,MAAM,aAAa,GAGf,EAAE,CAAC;IAEP,KAAK,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,SAAS,EAAE,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,eAAe,GAAG,yBAAyB,CAAC,UAAU,CAAC,CAAC;YAC9D,IAAI,CAAC,eAAe,EAAE,CAAC;gBACrB,SAAS;YACX,CAAC;YAED,MAAM,KAAK,GAA8B;gBACvC,OAAO,EAAE,UAAU;gBACnB,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,cAAc,EAAE,OAAO,CAAC,OAAc;aACvC,CAAC;YAEF,MAAM,MAAM,GACV,MAAM,uBAAA,IAAI,qDAAsB,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAC7D,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;YAExD,UAAU,CAAC,OAAO,CAAC,GAAG,mBAAmB,CAAC;YAC1C,MAAM,QAAQ,GAAG,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3C,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,GAAG;gBAC1B,GAAG,QAAQ;gBACX,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE;aACrC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,uDAAuD,EAAE;gBAC3D,OAAO;gBACP,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,KAAK;aACN,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1C,MAAM,QAAQ,GAAiB,EAAE,UAAU,EAAE,aAAa,EAAE,CAAC;QAC7D,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YAC9D,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;gBAC9D,GAAG,CAAC,0DAA0D,EAAE;oBAC9D,KAAK;iBACN,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;AACH,CAAC;IAQC,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,uBAAA,IAAI,0CAAW,CAAC,IAAI,CAChC,sCAAsC,CACvC,CAAC;QACF,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EACF,KAAK,EAAE,iBAAiB,IAAI,EAAE,CAC/B,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,iDAAiD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAClE,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,EAAE,CAAC,CAAC;IACjD,CAAC;AACH,CAAC,yIAUC,iBAA0D;IAE1D,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;QACnB,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC9C,IAAI,CAAC,kBAAkB,CAAC,EAAE,EAAE,CAAC,aAAa,EAAE,EAAE,CAC5C,uBAAA,IAAI,sDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;QACF,OAAO;IACT,CAAC;IAED,MAAM,YAAY,GAAc,EAAE,CAAC;IACnC,MAAM,SAAS,GAAG,iBAAiB,CAAC,MAAM,CAAC;IAC3C,IAAI,SAAS,EAAE,CAAC;QACd,KAAK,MAAM,KAAK,IAAI,uBAAA,IAAI,kDAAmB,EAAE,CAAC;YAC5C,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1B,SAAS;YACX,CAAC;YACD,MAAM,EAAE,SAAS,EAAE,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;YAC9C,MAAM,UAAU,GAAG,WAAW,CAAC,QAAQ,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC;YACxD,IAAI,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC1B,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAC9C,IAAI,CAAC,kBAAkB,CAAC,YAAY,EAAE,CAAC,aAAa,EAAE,EAAE,CACtD,uBAAA,IAAI,sDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;AACJ,CAAC,uFASY,UAAe;IAC1B,MAAM,MAAM,GAAG,uBAAA,IAAI,8CAAe,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACnD,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,CAAC;QACH,MAAM,YAAY,GAAG,uBAAA,IAAI,0CAAW,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAExE,MAAM,EAAE,8BAA8B,EAAE,GAAG,YAAY,CAAC;QACxD,IAAI,CAAC,8BAA8B,EAAE,CAAC;YACpC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,WAAW,GAAG,8BAA8B,CAAC,UAAU,CAAC,CAAC;QAC/D,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,6EAA6E;QAC7E,MAAM,EAAE,YAAY,EAAE,uBAAuB,EAAE,GAAG,WAAW,CAAC;QAC9D,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/C,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,KAAK,GACT,OAAO,uBAAuB,KAAK,QAAQ;YAC3C,uBAAuB,IAAI,CAAC;YAC5B,uBAAuB,GAAG,YAAY,CAAC,MAAM;YAC3C,CAAC,CAAC,uBAAuB;YACzB,CAAC,CAAC,CAAC,CAAC;QACR,MAAM,eAAe,GAAG,YAAY,CAAC,KAAK,CAEzC,CAAC;QACF,MAAM,eAAe,GAAG,eAAe,EAAE,eAAe,CAAC;QACzD,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,aAAa,GAAG,uBAAA,IAAI,0CAAW,CAAC,IAAI,CACxC,wCAAwC,EACxC,eAAe,CAChB,CAAC;QAEF,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,CAAC;YAC7B,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC1D,uBAAA,IAAI,8CAAe,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC9C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,kCAAkC,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/D,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC,mHAS0B,MAAgC;IACzD,MAAM,eAAe,GAAG,yBAAyB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAClE,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO;IACT,CAAC;IACD,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IACpD,MAAM,WAAW,GAAG,UAAU,cAAc,EAAa,CAAC;IAC1D,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC;IAE5D,MAAM,QAAQ,GAAiB;QAC7B,UAAU,EAAE;YACV,CAAC,OAAO,CAAC,EAAE,mBAAmB;SAC/B;QACD,aAAa,EAAE;YACb,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE;gBAClB,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE;aAC7C;SACF;KACF,CAAC;IAEF,GAAG,CAAC,uBAAuB,EAAE;QAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,OAAO,EAAE,WAAW;QACpB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM;KAC9B,CAAC,CAAC;IAEH,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9D,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YAC9D,GAAG,CAAC,wCAAwC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC","sourcesContent":["import { toChecksumAddress } from '@ethereumjs/util';\nimport { Web3Provider } from '@ethersproject/providers';\nimport type { InternalAccount } from '@metamask/keyring-internal-api';\nimport type { NetworkEnablementControllerState } from '@metamask/network-enablement-controller';\nimport {\n isStrictHexString,\n isCaipChainId,\n numberToHex,\n parseCaipChainId,\n} from '@metamask/utils';\nimport type { Hex } from '@metamask/utils';\n\nimport type {\n DataSourceState,\n SubscriptionRequest,\n} from './AbstractDataSource';\nimport { AbstractDataSource } from './AbstractDataSource';\nimport type {\n StakedBalancePollingInput,\n StakedBalanceFetchResult,\n} from './evm-rpc-services';\nimport {\n StakedBalanceFetcher,\n getStakingContractAddress,\n getSupportedStakingChainIds,\n} from './evm-rpc-services';\nimport type { AssetsControllerMessenger } from '../AssetsController';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport type {\n AccountId,\n ChainId,\n Caip19AssetId,\n AssetBalance,\n AssetMetadata,\n DataRequest,\n DataResponse,\n Middleware,\n} from '../types';\n\nconst CONTROLLER_NAME = 'StakedBalanceDataSource';\nconst DEFAULT_POLL_INTERVAL = 180_000; // 3 minutes\n\n/** Metadata for staked ETH (same symbol and decimals as native ETH). */\nconst STAKED_ETH_METADATA: AssetMetadata = {\n type: 'erc20',\n name: 'staked ethereum',\n symbol: 'ETH',\n decimals: 18,\n};\n\nconst log = createModuleLogger(projectLogger, CONTROLLER_NAME);\n\n/** Optional configuration for StakedBalanceDataSource. */\nexport type StakedBalanceDataSourceConfig = {\n /** Whether staked balance fetching is enabled (default: true). */\n enabled?: boolean;\n /** Polling interval in ms (default: 180s / 3 min). */\n pollInterval?: number;\n};\n\nexport type StakedBalanceDataSourceOptions = StakedBalanceDataSourceConfig & {\n /** The AssetsController messenger (for accessing NetworkController). */\n messenger: AssetsControllerMessenger;\n /** Called when active chains are updated. */\n onActiveChainsUpdated: (\n dataSourceName: string,\n chains: ChainId[],\n previousChains: ChainId[],\n ) => void;\n};\n\n/** Per-account supported chains (same shape as in DataRequest). */\ntype AccountWithSupportedChains = {\n account: InternalAccount;\n supportedChains: ChainId[];\n};\n\n/**\n * Subscription data stored for each active subscription.\n * Stores accountsWithSupportedChains so refresh paths use the same per-account\n * scope as normal subscription setup (avoids querying unsupported chains/accounts).\n */\ntype SubscriptionData = {\n /** Polling tokens from StakedBalanceFetcher. */\n pollingTokens: string[];\n /** Chain IDs being polled (union of all account chains). */\n chains: ChainId[];\n /** Accounts being polled. */\n accounts: InternalAccount[];\n /** Per-account supported chains; used by refreshStakedBalance and transaction handlers. */\n accountsWithSupportedChains: AccountWithSupportedChains[];\n /** Callback to report asset updates. */\n onAssetsUpdate: (response: DataResponse) => void | Promise<void>;\n};\n\n/**\n * Convert CAIP chain ID or hex chain ID to hex chain ID.\n *\n * @param chainId - CAIP chain ID or hex chain ID.\n * @returns Hex chain ID.\n */\nfunction caipChainIdToHex(chainId: string): Hex {\n if (isStrictHexString(chainId)) {\n return chainId;\n }\n\n if (isCaipChainId(chainId)) {\n const ref = parseCaipChainId(chainId).reference;\n return numberToHex(parseInt(ref, 10));\n }\n\n throw new Error('caipChainIdToHex - Failed to provide CAIP-2 or Hex chainId');\n}\n\n/**\n * Build the CAIP-19 asset ID for staked balance (same format as ERC20).\n * Uses the staking contract address (checksummed) so it is consistent with\n * other token assets.\n * Example: \"eip155:1/erc20:0x4fef9d741011476750a243ac70b9789a63dd47df\"\n *\n * @param chainId - CAIP-2 chain ID (e.g. \"eip155:1\").\n * @param contractAddress - Staking contract address (hex).\n * @returns The staked asset CAIP-19 ID with checksummed address.\n */\nfunction stakedAssetId(\n chainId: ChainId,\n contractAddress: string,\n): Caip19AssetId {\n const checksummed = toChecksumAddress(contractAddress);\n return `${chainId}/erc20:${checksummed}` as Caip19AssetId;\n}\n\n/**\n * Data source for fetching staked ETH balances via on-chain staking contracts.\n *\n * Delegates to {@link StakedBalanceFetcher} for the actual RPC calls\n * (getShares + convertToAssets on ERC-4626-style staking contracts).\n * Reports balances as CAIP-19 asset IDs using the ERC20 format with the\n * staking contract address (e.g. \"eip155:1/erc20:0x4fef9d741011476750a243ac70b9789a63dd47df\").\n *\n * Only supports chains with known staking contracts (mainnet, Hoodi).\n * Polling is managed by StakedBalanceFetcher via startPolling/stopPollingByPollingToken.\n */\nexport class StakedBalanceDataSource extends AbstractDataSource<\n typeof CONTROLLER_NAME,\n DataSourceState\n> {\n readonly #messenger: AssetsControllerMessenger;\n\n readonly #onActiveChainsUpdated: (\n dataSourceName: string,\n chains: ChainId[],\n previousChains: ChainId[],\n ) => void;\n\n readonly #pollInterval: number;\n\n readonly #enabled: boolean;\n\n /** The StakedBalanceFetcher that handles polling and RPC calls. */\n readonly #stakedBalanceFetcher: StakedBalanceFetcher;\n\n /** Active subscriptions by ID. */\n readonly #activeSubscriptions: Map<string, SubscriptionData> = new Map();\n\n /** Cache of Web3Provider instances by hex chain ID. */\n readonly #providerCache: Map<Hex, Web3Provider> = new Map();\n\n /** CAIP-2 chain IDs that have known staking contracts. */\n readonly #supportedChainIds: ChainId[];\n\n constructor(options: StakedBalanceDataSourceOptions) {\n super(CONTROLLER_NAME, { activeChains: [] });\n this.#messenger = options.messenger;\n this.#onActiveChainsUpdated = options.onActiveChainsUpdated;\n this.#pollInterval = options.pollInterval ?? DEFAULT_POLL_INTERVAL;\n this.#enabled = options.enabled !== false;\n this.#supportedChainIds = getSupportedStakingChainIds() as ChainId[];\n\n log('Initializing StakedBalanceDataSource', {\n enabled: this.#enabled,\n pollInterval: this.#pollInterval,\n });\n\n // Create StakedBalanceFetcher with provider getter\n this.#stakedBalanceFetcher = new StakedBalanceFetcher({\n pollingInterval: this.#pollInterval,\n getNetworkProvider: (hexChainId): Web3Provider | undefined =>\n this.#getProvider(hexChainId),\n });\n\n // Wire the callback so polling results flow back to subscriptions\n this.#stakedBalanceFetcher.setOnStakedBalanceUpdate(\n this.#handleStakedBalanceUpdate.bind(this),\n );\n\n const unsubConfirmed = this.#messenger.subscribe(\n 'TransactionController:transactionConfirmed',\n this.#onTransactionConfirmed.bind(this),\n );\n this.#unsubscribeTransactionConfirmed =\n typeof unsubConfirmed === 'function' ? unsubConfirmed : undefined;\n\n const unsubIncoming = this.#messenger.subscribe(\n 'TransactionController:incomingTransactionsReceived',\n this.#onIncomingTransactions.bind(this),\n );\n this.#unsubscribeIncomingTransactions =\n typeof unsubIncoming === 'function' ? unsubIncoming : undefined;\n\n const unsubNetwork = this.#messenger.subscribe(\n 'NetworkController:stateChange',\n this.#onNetworkStateChange.bind(this),\n );\n this.#unsubscribeNetworkStateChange =\n typeof unsubNetwork === 'function' ? unsubNetwork : undefined;\n\n const unsubEnablement = this.#messenger.subscribe(\n 'NetworkEnablementController:stateChange',\n this.#onNetworkEnablementControllerStateChange.bind(this),\n );\n this.#unsubscribeNetworkEnablementControllerStateChange =\n typeof unsubEnablement === 'function' ? unsubEnablement : undefined;\n\n this.#initializeActiveChains();\n }\n\n readonly #unsubscribeTransactionConfirmed: (() => void) | undefined =\n undefined;\n\n readonly #unsubscribeIncomingTransactions: (() => void) | undefined =\n undefined;\n\n readonly #unsubscribeNetworkStateChange: (() => void) | undefined = undefined;\n\n readonly #unsubscribeNetworkEnablementControllerStateChange:\n | (() => void)\n | undefined = undefined;\n\n /**\n * When NetworkController state changes (e.g. RPC endpoints or network clients\n * reconfigured), clear the provider cache so subsequent fetches use fresh\n * providers.\n */\n #onNetworkStateChange(): void {\n this.#providerCache.clear();\n log('Provider cache cleared after network state change');\n }\n\n /**\n * When NetworkEnablementController state changes (user enables/disables\n * networks), recompute active chains so we only fetch for enabled staking chains.\n *\n * @param state - The new NetworkEnablementController state.\n */\n #onNetworkEnablementControllerStateChange(\n state: NetworkEnablementControllerState,\n ): void {\n const { enabledNetworkMap } = state ?? {};\n if (!enabledNetworkMap) {\n return;\n }\n this.#initializeActiveChainsFromEnabledMap(enabledNetworkMap);\n }\n\n /**\n * Returns true if the transaction involves the staking contract (from or to)\n * for the payload's chain, so we only refresh staked balance when relevant.\n *\n * @param payload - Transaction payload.\n * @param payload.chainId - Hex chain ID.\n * @param payload.txParams - Optional transaction params.\n * @param payload.txParams.from - Sender address.\n * @param payload.txParams.to - Recipient address.\n * @returns True if txParams.from or txParams.to matches the staking contract address.\n */\n #isTransactionInvolvingStakingContract(payload: {\n chainId?: string;\n txParams?: { from?: string; to?: string };\n }): boolean {\n const hexChainId = payload?.chainId;\n if (!hexChainId) {\n return false;\n }\n const contractAddress = getStakingContractAddress(hexChainId);\n if (!contractAddress) {\n return false;\n }\n const contractLower = contractAddress.toLowerCase();\n const from = payload.txParams?.from?.toLowerCase();\n const to = payload.txParams?.to?.toLowerCase();\n return from === contractLower || to === contractLower;\n }\n\n /**\n * When a transaction is confirmed, refresh staked balance only if the\n * transaction is from or to the staking contract for that chain.\n *\n * @param payload - From TransactionController:transactionConfirmed.\n * @param payload.chainId - Hex chain ID of the transaction.\n * @param payload.txParams - Optional transaction params.\n * @param payload.txParams.from - Sender address.\n * @param payload.txParams.to - Recipient address.\n */\n #onTransactionConfirmed(payload: {\n chainId?: string;\n txParams?: { from?: string; to?: string };\n }): void {\n if (!this.#enabled) {\n return;\n }\n if (!this.#isTransactionInvolvingStakingContract(payload)) {\n return;\n }\n const hexChainId = payload?.chainId;\n if (!hexChainId) {\n return;\n }\n const caipChainId = `eip155:${parseInt(hexChainId, 16)}` as ChainId;\n const toRefresh = this.#getToRefreshForChains([caipChainId]);\n if (toRefresh.length > 0) {\n this.#refreshStakedBalanceAfterTransaction(toRefresh).catch((error) => {\n log('Failed to refresh staked balance after transaction', { error });\n });\n }\n }\n\n /**\n * When incoming transactions are received, refresh staked balance only for\n * chains where at least one transaction is from or to the staking contract.\n *\n * @param payload - From TransactionController:incomingTransactionsReceived (array of { chainId?, txParams? }).\n */\n #onIncomingTransactions(\n payload: { chainId?: string; txParams?: { from?: string; to?: string } }[],\n ): void {\n if (!this.#enabled) {\n return;\n }\n const chainIdsToRefresh = new Set<string>();\n for (const item of payload ?? []) {\n if (!item?.chainId) {\n continue;\n }\n if (this.#isTransactionInvolvingStakingContract(item)) {\n chainIdsToRefresh.add(item.chainId);\n }\n }\n const caipChainIds = [...chainIdsToRefresh].map(\n (hexChainId) => `eip155:${parseInt(hexChainId, 16)}` as ChainId,\n );\n if (caipChainIds.length === 0) {\n return;\n }\n const toRefresh = this.#getToRefreshForChains(caipChainIds);\n if (toRefresh.length > 0) {\n this.#refreshStakedBalanceAfterTransaction(toRefresh).catch((error) => {\n log('Failed to refresh staked balance after incoming transactions', {\n error,\n });\n });\n }\n }\n\n /**\n * Build toRefresh list for subscribed (account, chainId) pairs for the given chains.\n *\n * @param chainIds - CAIP-2 chain IDs to target.\n * @returns Pairs of account and chainId to refresh.\n */\n #getToRefreshForChains(\n chainIds: ChainId[],\n ): { account: InternalAccount; chainId: ChainId }[] {\n const toRefresh: { account: InternalAccount; chainId: ChainId }[] = [];\n const chainSet = new Set(chainIds);\n for (const subscription of this.#activeSubscriptions.values()) {\n for (const {\n account,\n supportedChains,\n } of subscription.accountsWithSupportedChains) {\n for (const chainId of supportedChains) {\n if (chainSet.has(chainId)) {\n toRefresh.push({ account, chainId });\n }\n }\n }\n }\n return toRefresh;\n }\n\n /**\n * Build toRefresh list for all subscribed (account, chainId) pairs.\n *\n * @returns Pairs of account and chainId to refresh.\n */\n #getToRefreshAll(): { account: InternalAccount; chainId: ChainId }[] {\n const toRefresh: { account: InternalAccount; chainId: ChainId }[] = [];\n for (const subscription of this.#activeSubscriptions.values()) {\n for (const {\n account,\n supportedChains,\n } of subscription.accountsWithSupportedChains) {\n for (const chainId of supportedChains) {\n toRefresh.push({ account, chainId });\n }\n }\n }\n return toRefresh;\n }\n\n /**\n * Refresh staked balance for all currently subscribed accounts and chains, then\n * push updates to the controller. Can be called from UI or after transaction events.\n */\n async refreshStakedBalance(): Promise<void> {\n if (!this.#enabled) {\n return;\n }\n const toRefresh = this.#getToRefreshAll();\n if (toRefresh.length > 0) {\n await this.#refreshStakedBalanceAfterTransaction(toRefresh);\n }\n }\n\n /**\n * Fetch staked balance for the given account/chain pairs and push a single\n * DataResponse to all active subscriptions.\n *\n * @param toRefresh - List of { account, chainId } to refresh.\n */\n async #refreshStakedBalanceAfterTransaction(\n toRefresh: { account: InternalAccount; chainId: ChainId }[],\n ): Promise<void> {\n const assetsInfo: Record<Caip19AssetId, AssetMetadata> = {};\n const assetsBalance: Record<\n AccountId,\n Record<Caip19AssetId, AssetBalance>\n > = {};\n\n for (const { account, chainId } of toRefresh) {\n try {\n const hexChainId = caipChainIdToHex(chainId);\n const contractAddress = getStakingContractAddress(hexChainId);\n if (!contractAddress) {\n continue;\n }\n\n const input: StakedBalancePollingInput = {\n chainId: hexChainId,\n accountId: account.id,\n accountAddress: account.address as Hex,\n };\n\n const result =\n await this.#stakedBalanceFetcher.fetchStakedBalance(input);\n const assetId = stakedAssetId(chainId, contractAddress);\n\n assetsInfo[assetId] = STAKED_ETH_METADATA;\n const existing = assetsBalance[account.id];\n assetsBalance[account.id] = {\n ...existing,\n [assetId]: { amount: result.amount },\n };\n } catch (error) {\n log('Failed to fetch staked balance in transaction refresh', {\n chainId,\n accountId: account.id,\n error,\n });\n }\n }\n\n if (Object.keys(assetsBalance).length > 0) {\n const response: DataResponse = { assetsInfo, assetsBalance };\n for (const subscription of this.#activeSubscriptions.values()) {\n subscription.onAssetsUpdate(response)?.catch((error: unknown) => {\n log('Failed to report staked balance update after transaction', {\n error,\n });\n });\n }\n }\n }\n\n /**\n * Set active chains from NetworkEnablementController state.\n * Only staking-supported chains that are enabled in the network enablement map\n * are active (e.g. if mainnet is not selected we do not fetch).\n */\n #initializeActiveChains(): void {\n try {\n const state = this.#messenger.call(\n 'NetworkEnablementController:getState',\n );\n this.#initializeActiveChainsFromEnabledMap(\n state?.enabledNetworkMap ?? {},\n );\n } catch (error) {\n log('Failed to get NetworkEnablementController state', { error });\n this.#initializeActiveChainsFromEnabledMap({});\n }\n }\n\n /**\n * Compute active chains as the intersection of supported staking chains and\n * enabled networks, then update state. Uses the same EIP-155 storage key\n * convention as NetworkEnablementController (hex for eip155).\n *\n * @param enabledNetworkMap - The enabled network map from NetworkEnablementController.\n */\n #initializeActiveChainsFromEnabledMap(\n enabledNetworkMap: Record<string, Record<string, boolean>>,\n ): void {\n if (!this.#enabled) {\n const previous = [...this.state.activeChains];\n this.updateActiveChains([], (updatedChains) =>\n this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n );\n return;\n }\n\n const activeChains: ChainId[] = [];\n const eip155Map = enabledNetworkMap.eip155;\n if (eip155Map) {\n for (const caip2 of this.#supportedChainIds) {\n if (!isCaipChainId(caip2)) {\n continue;\n }\n const { reference } = parseCaipChainId(caip2);\n const storageKey = numberToHex(parseInt(reference, 10));\n if (eip155Map[storageKey]) {\n activeChains.push(caip2);\n }\n }\n }\n\n const previous = [...this.state.activeChains];\n this.updateActiveChains(activeChains, (updatedChains) =>\n this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n );\n }\n\n /**\n * Get or create a Web3Provider for the given hex chain ID.\n * Uses the same messenger-cast pattern as RpcDataSource.\n *\n * @param hexChainId - The hex chain ID.\n * @returns Web3Provider instance, or undefined if not available.\n */\n #getProvider(hexChainId: Hex): Web3Provider | undefined {\n const cached = this.#providerCache.get(hexChainId);\n if (cached) {\n return cached;\n }\n\n try {\n const networkState = this.#messenger.call('NetworkController:getState');\n\n const { networkConfigurationsByChainId } = networkState;\n if (!networkConfigurationsByChainId) {\n return undefined;\n }\n\n const chainConfig = networkConfigurationsByChainId[hexChainId];\n if (!chainConfig) {\n return undefined;\n }\n\n // Use the network's configured default RPC endpoint (same as RpcDataSource).\n const { rpcEndpoints, defaultRpcEndpointIndex } = chainConfig;\n if (!rpcEndpoints || rpcEndpoints.length === 0) {\n return undefined;\n }\n\n const index =\n typeof defaultRpcEndpointIndex === 'number' &&\n defaultRpcEndpointIndex >= 0 &&\n defaultRpcEndpointIndex < rpcEndpoints.length\n ? defaultRpcEndpointIndex\n : 0;\n const defaultEndpoint = rpcEndpoints[index] as {\n networkClientId?: string;\n };\n const networkClientId = defaultEndpoint?.networkClientId;\n if (!networkClientId) {\n return undefined;\n }\n\n const networkClient = this.#messenger.call(\n 'NetworkController:getNetworkClientById',\n networkClientId,\n );\n\n if (!networkClient?.provider) {\n return undefined;\n }\n\n const provider = new Web3Provider(networkClient.provider);\n this.#providerCache.set(hexChainId, provider);\n return provider;\n } catch (error) {\n log('Failed to get provider for chain', { hexChainId, error });\n return undefined;\n }\n }\n\n /**\n * Handle a staked balance update from StakedBalanceFetcher.\n * Converts the result into a DataResponse and forwards it to all active\n * subscriptions.\n *\n * @param result - The staked balance fetch result.\n */\n #handleStakedBalanceUpdate(result: StakedBalanceFetchResult): void {\n const contractAddress = getStakingContractAddress(result.chainId);\n if (!contractAddress) {\n return;\n }\n const chainIdDecimal = parseInt(result.chainId, 16);\n const caipChainId = `eip155:${chainIdDecimal}` as ChainId;\n const assetId = stakedAssetId(caipChainId, contractAddress);\n\n const response: DataResponse = {\n assetsInfo: {\n [assetId]: STAKED_ETH_METADATA,\n },\n assetsBalance: {\n [result.accountId]: {\n [assetId]: { amount: result.balance.amount },\n },\n },\n };\n\n log('Staked balance update', {\n accountId: result.accountId,\n chainId: caipChainId,\n amount: result.balance.amount,\n });\n\n for (const subscription of this.#activeSubscriptions.values()) {\n subscription.onAssetsUpdate(response)?.catch((error: unknown) => {\n log('Failed to report staked balance update', { error });\n });\n }\n }\n\n /**\n * Fetch staked balances for all accounts on supported chains.\n *\n * @param request - The data request with accounts and chains.\n * @returns DataResponse with staked balance data.\n */\n async fetch(request: DataRequest): Promise<DataResponse> {\n if (!this.#enabled) {\n return {};\n }\n const response: DataResponse = {};\n const activeChainsSet = new Set(this.state.activeChains);\n\n const chainsToFetch = request.chainIds.filter((chainId) =>\n activeChainsSet.has(chainId),\n );\n\n if (chainsToFetch.length === 0) {\n return response;\n }\n\n const balances: Record<AccountId, Record<Caip19AssetId, AssetBalance>> = {};\n\n for (const {\n account,\n supportedChains: accountChains,\n } of request.accountsWithSupportedChains) {\n const chains = chainsToFetch.filter((chain) =>\n accountChains.includes(chain),\n );\n\n for (const chainId of chains) {\n try {\n const hexChainId = caipChainIdToHex(chainId);\n const contractAddress = getStakingContractAddress(hexChainId);\n if (!contractAddress) {\n continue;\n }\n\n const input: StakedBalancePollingInput = {\n chainId: hexChainId,\n accountId: account.id,\n accountAddress: account.address as Hex,\n };\n\n const result =\n await this.#stakedBalanceFetcher.fetchStakedBalance(input);\n\n // Include zero amounts so merged updates clear prior non-zero state.\n balances[account.id] ??= {};\n const assetId = stakedAssetId(chainId, contractAddress);\n balances[account.id][assetId] = { amount: result.amount };\n } catch (error) {\n log('Failed to fetch staked balance', {\n chainId,\n accountId: account.id,\n error,\n });\n }\n }\n }\n\n if (Object.keys(balances).length > 0) {\n response.assetsBalance = balances;\n // Add metadata for each staked asset ID present in balances\n const assetIds = new Set<Caip19AssetId>();\n for (const accountBalances of Object.values(balances)) {\n for (const assetId of Object.keys(accountBalances)) {\n assetIds.add(assetId as Caip19AssetId);\n }\n }\n response.assetsInfo = {};\n for (const assetId of assetIds) {\n response.assetsInfo[assetId] = STAKED_ETH_METADATA;\n }\n }\n\n return response;\n }\n\n /**\n * Assets middleware for the fetch pipeline.\n * Fetches staked balances and merges them into the response, then passes\n * all chains to the next middleware (staked balance doesn't claim chains).\n *\n * @returns The middleware function for the assets pipeline.\n */\n get assetsMiddleware(): Middleware {\n return async (context, next) => {\n if (!this.#enabled) {\n return next(context);\n }\n const { request } = context;\n\n if (!request.dataTypes.includes('balance')) {\n return next(context);\n }\n\n if (request.chainIds.length === 0) {\n return next(context);\n }\n\n try {\n const fetchResponse = await this.fetch(request);\n\n if (fetchResponse.assetsInfo) {\n context.response.assetsInfo ??= {};\n Object.assign(context.response.assetsInfo, fetchResponse.assetsInfo);\n }\n if (fetchResponse.assetsBalance) {\n context.response.assetsBalance ??= {};\n for (const [accountId, accountBalances] of Object.entries(\n fetchResponse.assetsBalance,\n )) {\n context.response.assetsBalance[accountId] = {\n ...context.response.assetsBalance[accountId],\n ...accountBalances,\n };\n }\n }\n } catch (error) {\n log('Middleware fetch failed', { error });\n }\n\n // Pass all chains through (staked balance doesn't claim chains)\n return next(context);\n };\n }\n\n /**\n * Subscribe to staked balance updates with polling.\n * Starts polling via StakedBalanceFetcher for each account/chain combination.\n *\n * @param subscriptionRequest - The subscription request details.\n */\n async subscribe(subscriptionRequest: SubscriptionRequest): Promise<void> {\n const { request, subscriptionId, isUpdate } = subscriptionRequest;\n\n const activeChainsSet = new Set(this.state.activeChains);\n const chainsToSubscribe = request.chainIds.filter((chainId) =>\n activeChainsSet.has(chainId),\n );\n\n log('Subscribe requested', {\n subscriptionId,\n isUpdate,\n chainsToSubscribe,\n });\n\n if (chainsToSubscribe.length === 0) {\n log('No staking chains to subscribe');\n return;\n }\n\n // Handle subscription update - restart polling for new chains\n if (isUpdate) {\n const existing = this.#activeSubscriptions.get(subscriptionId);\n if (existing) {\n log('Updating existing subscription - restarting polling', {\n subscriptionId,\n });\n }\n }\n\n // Clean up existing subscription (stops old polling)\n await this.unsubscribe(subscriptionId);\n\n // Build subscription data first so it is available when the first poll runs\n const accountsWithSupportedChains: AccountWithSupportedChains[] =\n request.accountsWithSupportedChains\n .map(({ account, supportedChains }) => ({\n account,\n supportedChains: chainsToSubscribe.filter((chain) =>\n supportedChains.includes(chain),\n ),\n }))\n .filter(({ supportedChains }) => supportedChains.length > 0);\n\n const accounts = accountsWithSupportedChains.map((entry) => entry.account);\n const pollingTokens: string[] = [];\n\n // Store subscription before startPolling so first poll (setTimeout 0) has the callback\n this.#activeSubscriptions.set(subscriptionId, {\n pollingTokens,\n chains: chainsToSubscribe,\n accounts,\n accountsWithSupportedChains,\n onAssetsUpdate: subscriptionRequest.onAssetsUpdate,\n });\n\n this.activeSubscriptions.set(subscriptionId, {\n cleanup: () => {\n for (const token of pollingTokens) {\n this.#stakedBalanceFetcher.stopPollingByPollingToken(token);\n }\n this.#activeSubscriptions.delete(subscriptionId);\n },\n chains: chainsToSubscribe,\n request,\n onAssetsUpdate: subscriptionRequest.onAssetsUpdate,\n });\n\n // Start polling for each account/chain (first poll runs on next tick)\n for (const {\n account,\n supportedChains: accountChains,\n } of request.accountsWithSupportedChains) {\n const chainsForAccount = chainsToSubscribe.filter((chain) =>\n accountChains.includes(chain),\n );\n\n for (const chainId of chainsForAccount) {\n const hexChainId = caipChainIdToHex(chainId);\n\n const input: StakedBalancePollingInput = {\n chainId: hexChainId,\n accountId: account.id,\n accountAddress: account.address as Hex,\n };\n\n const pollingToken = this.#stakedBalanceFetcher.startPolling(input);\n pollingTokens.push(pollingToken);\n }\n }\n\n // Immediate initial fetch so state is updated without waiting for first poll\n try {\n const initialRequest: DataRequest = {\n accountsWithSupportedChains,\n chainIds: chainsToSubscribe,\n dataTypes: ['balance'],\n };\n const initialResponse = await this.fetch(initialRequest);\n if (\n initialResponse.assetsBalance &&\n Object.keys(initialResponse.assetsBalance).length > 0\n ) {\n subscriptionRequest\n .onAssetsUpdate?.(initialResponse)\n ?.catch((error) => {\n log('Initial staked balance update failed', { error });\n });\n }\n } catch (error) {\n log('Initial staked balance fetch failed', { error });\n }\n\n log('Subscription SUCCESS', {\n subscriptionId,\n chains: chainsToSubscribe,\n pollingCount: pollingTokens.length,\n });\n }\n\n /**\n * Unsubscribe from staked balance updates and stop polling.\n *\n * @param subscriptionId - The subscription ID to unsubscribe.\n */\n async unsubscribe(subscriptionId: string): Promise<void> {\n const subscription = this.#activeSubscriptions.get(subscriptionId);\n if (subscription) {\n for (const token of subscription.pollingTokens) {\n this.#stakedBalanceFetcher.stopPollingByPollingToken(token);\n }\n this.#activeSubscriptions.delete(subscriptionId);\n }\n\n await super.unsubscribe(subscriptionId);\n }\n\n /**\n * Destroy the data source and clean up all resources.\n */\n destroy(): void {\n this.#unsubscribeTransactionConfirmed?.();\n this.#unsubscribeIncomingTransactions?.();\n this.#unsubscribeNetworkStateChange?.();\n this.#unsubscribeNetworkEnablementControllerStateChange?.();\n for (const subscription of this.#activeSubscriptions.values()) {\n for (const token of subscription.pollingTokens) {\n this.#stakedBalanceFetcher.stopPollingByPollingToken(token);\n }\n }\n this.#activeSubscriptions.clear();\n this.#providerCache.clear();\n super.destroy();\n }\n}\n"]}
|
|
1
|
+
{"version":3,"file":"StakedBalanceDataSource.mjs","sourceRoot":"","sources":["../../src/data-sources/StakedBalanceDataSource.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EAAE,iBAAiB,EAAE,yBAAyB;AACrD,OAAO,EAAE,YAAY,EAAE,iCAAiC;AAGxD,OAAO,EACL,iBAAiB,EACjB,aAAa,EACb,WAAW,EACX,gBAAgB,EACjB,wBAAwB;AAOzB,OAAO,EAAE,kBAAkB,EAAE,iCAA6B;AAK1D,OAAO,EACL,oBAAoB,EACpB,yBAAyB,EACzB,2BAA2B,EAC5B,qCAA2B;AAE5B,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,sBAAkB;AAY9D,MAAM,eAAe,GAAG,yBAAyB,CAAC;AAClD,MAAM,qBAAqB,GAAG,MAAO,CAAC,CAAC,YAAY;AAEnD,wEAAwE;AACxE,MAAM,mBAAmB,GAAkB;IACzC,IAAI,EAAE,OAAO;IACb,IAAI,EAAE,iBAAiB;IACvB,MAAM,EAAE,KAAK;IACb,QAAQ,EAAE,EAAE;CACb,CAAC;AAEF,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;AA6C/D;;;;;GAKG;AACH,SAAS,gBAAgB,CAAC,OAAe;IACvC,IAAI,iBAAiB,CAAC,OAAO,CAAC,EAAE,CAAC;QAC/B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,MAAM,GAAG,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC;QAChD,OAAO,WAAW,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;IACxC,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;AAChF,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,aAAa,CACpB,OAAgB,EAChB,eAAuB;IAEvB,MAAM,WAAW,GAAG,iBAAiB,CAAC,eAAe,CAAC,CAAC;IACvD,OAAO,GAAG,OAAO,UAAU,WAAW,EAAmB,CAAC;AAC5D,CAAC;AAED;;;;;;;;;;GAUG;AACH,MAAM,OAAO,uBAAwB,SAAQ,kBAG5C;IAyBC,YAAY,OAAuC;QACjD,KAAK,CAAC,eAAe,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;;QAzBtC,qDAAsC;QAEtC,iEAIC;QAED,wDAAsB;QAEtB,mDAAkB;QAE3B,mEAAmE;QAC1D,gEAA4C;QAErD,kCAAkC;QACzB,uDAAsD,IAAI,GAAG,EAAE,EAAC;QAEzE,uDAAuD;QAC9C,iDAAyC,IAAI,GAAG,EAAE,EAAC;QAE5D,0DAA0D;QACjD,6DAA8B;QAIrC,uBAAA,IAAI,sCAAc,OAAO,CAAC,SAAS,MAAA,CAAC;QACpC,uBAAA,IAAI,kDAA0B,OAAO,CAAC,qBAAqB,MAAA,CAAC;QAC5D,uBAAA,IAAI,yCAAiB,OAAO,CAAC,YAAY,IAAI,qBAAqB,MAAA,CAAC;QACnE,uBAAA,IAAI,oCAAY,OAAO,CAAC,OAAO,KAAK,KAAK,MAAA,CAAC;QAC1C,uBAAA,IAAI,8CAAsB,2BAA2B,EAAe,MAAA,CAAC;QAErE,GAAG,CAAC,sCAAsC,EAAE;YAC1C,OAAO,EAAE,uBAAA,IAAI,wCAAS;YACtB,YAAY,EAAE,uBAAA,IAAI,6CAAc;SACjC,CAAC,CAAC;QAEH,mDAAmD;QACnD,uBAAA,IAAI,iDAAyB,IAAI,oBAAoB,CAAC;YACpD,eAAe,EAAE,uBAAA,IAAI,6CAAc;YACnC,kBAAkB,EAAE,CAAC,UAAU,EAA4B,EAAE,CAC3D,uBAAA,IAAI,gFAAa,MAAjB,IAAI,EAAc,UAAU,CAAC;SAChC,CAAC,MAAA,CAAC;QAEH,kEAAkE;QAClE,uBAAA,IAAI,qDAAsB,CAAC,wBAAwB,CACjD,uBAAA,IAAI,8FAA2B,CAAC,IAAI,CAAC,IAAI,CAAC,CAC3C,CAAC;QAEF,uBAAA,IAAI,0CAAW,CAAC,SAAS,CACvB,4CAA4C,EAC5C,uBAAA,IAAI,2FAAwB,CAAC,IAAI,CAAC,IAAI,CAAC,CACxC,CAAC;QAEF,uBAAA,IAAI,0CAAW,CAAC,SAAS,CACvB,oDAAoD,EACpD,uBAAA,IAAI,2FAAwB,CAAC,IAAI,CAAC,IAAI,CAAC,CACxC,CAAC;QAEF,uBAAA,IAAI,0CAAW,CAAC,SAAS,CACvB,+BAA+B,EAC/B,uBAAA,IAAI,yFAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CACtC,CAAC;QAEF,uBAAA,IAAI,0CAAW,CAAC,SAAS,CACvB,yCAAyC,EACzC,uBAAA,IAAI,6GAA0C,CAAC,IAAI,CAAC,IAAI,CAAC,CAC1D,CAAC;QAEF,uBAAA,IAAI,2FAAwB,MAA5B,IAAI,CAA0B,CAAC;IACjC,CAAC;IA6KD;;;OAGG;IACH,KAAK,CAAC,oBAAoB;QACxB,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,uBAAA,IAAI,oFAAiB,MAArB,IAAI,CAAmB,CAAC;QAC1C,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,MAAM,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,SAAS,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAgOD;;;;;OAKG;IACH,KAAK,CAAC,KAAK,CAAC,OAAoB;;QAC9B,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;YACnB,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,QAAQ,GAAiB,EAAE,CAAC;QAClC,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAEzD,MAAM,aAAa,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CACxD,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAC7B,CAAC;QAEF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,QAAQ,GAA2D,EAAE,CAAC;QAE5E,KAAK,MAAM,EACT,OAAO,EACP,eAAe,EAAE,aAAa,GAC/B,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;YACzC,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAC5C,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,CAC9B,CAAC;YAEF,KAAK,MAAM,OAAO,IAAI,MAAM,EAAE,CAAC;gBAC7B,IAAI,CAAC;oBACH,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;oBAC7C,MAAM,eAAe,GAAG,yBAAyB,CAAC,UAAU,CAAC,CAAC;oBAC9D,IAAI,CAAC,eAAe,EAAE,CAAC;wBACrB,SAAS;oBACX,CAAC;oBAED,MAAM,KAAK,GAA8B;wBACvC,OAAO,EAAE,UAAU;wBACnB,SAAS,EAAE,OAAO,CAAC,EAAE;wBACrB,cAAc,EAAE,OAAO,CAAC,OAAc;qBACvC,CAAC;oBAEF,MAAM,MAAM,GACV,MAAM,uBAAA,IAAI,qDAAsB,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;oBAE7D,qEAAqE;oBACrE,QAAQ,MAAC,OAAO,CAAC,EAAE,MAAnB,QAAQ,OAAiB,EAAE,EAAC;oBAC5B,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;oBACxD,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;gBAC5D,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,GAAG,CAAC,gCAAgC,EAAE;wBACpC,OAAO;wBACP,SAAS,EAAE,OAAO,CAAC,EAAE;wBACrB,KAAK;qBACN,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrC,QAAQ,CAAC,aAAa,GAAG,QAAQ,CAAC;YAClC,4DAA4D;YAC5D,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAiB,CAAC;YAC1C,KAAK,MAAM,eAAe,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACtD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;oBACnD,QAAQ,CAAC,GAAG,CAAC,OAAwB,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;YACD,QAAQ,CAAC,UAAU,GAAG,EAAE,CAAC;YACzB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,mBAAmB,CAAC;YACrD,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;;;;;OAMG;IACH,IAAI,gBAAgB;QAClB,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;;YAC7B,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;gBACnB,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YACD,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;YAE5B,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC3C,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAClC,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAEhD,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;oBAC7B,MAAA,OAAO,CAAC,QAAQ,EAAC,UAAU,QAAV,UAAU,GAAK,EAAE,EAAC;oBACnC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,aAAa,CAAC,UAAU,CAAC,CAAC;gBACvE,CAAC;gBACD,IAAI,aAAa,CAAC,aAAa,EAAE,CAAC;oBAChC,MAAA,OAAO,CAAC,QAAQ,EAAC,aAAa,QAAb,aAAa,GAAK,EAAE,EAAC;oBACtC,KAAK,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CACvD,aAAa,CAAC,aAAa,CAC5B,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;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC5C,CAAC;YAED,gEAAgE;YAChE,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,SAAS,CAAC,mBAAwC;QACtD,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC;QAElE,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACzD,MAAM,iBAAiB,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5D,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,CAC7B,CAAC;QAEF,GAAG,CAAC,qBAAqB,EAAE;YACzB,cAAc;YACd,QAAQ;YACR,iBAAiB;SAClB,CAAC,CAAC;QAEH,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,GAAG,CAAC,gCAAgC,CAAC,CAAC;YACtC,OAAO;QACT,CAAC;QAED,8DAA8D;QAC9D,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,uBAAA,IAAI,oDAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC/D,IAAI,QAAQ,EAAE,CAAC;gBACb,GAAG,CAAC,qDAAqD,EAAE;oBACzD,cAAc;iBACf,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,qDAAqD;QACrD,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;QAEvC,4EAA4E;QAC5E,MAAM,2BAA2B,GAC/B,OAAO,CAAC,2BAA2B;aAChC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,CAAC;YACtC,OAAO;YACP,eAAe,EAAE,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAClD,eAAe,CAAC,QAAQ,CAAC,KAAK,CAAC,CAChC;SACF,CAAC,CAAC;aACF,MAAM,CAAC,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAEjE,MAAM,QAAQ,GAAG,2BAA2B,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC3E,MAAM,aAAa,GAAa,EAAE,CAAC;QAEnC,uFAAuF;QACvF,uBAAA,IAAI,oDAAqB,CAAC,GAAG,CAAC,cAAc,EAAE;YAC5C,aAAa;YACb,MAAM,EAAE,iBAAiB;YACzB,QAAQ;YACR,2BAA2B;YAC3B,cAAc,EAAE,mBAAmB,CAAC,cAAc;SACnD,CAAC,CAAC;QAEH,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,cAAc,EAAE;YAC3C,OAAO,EAAE,GAAG,EAAE;gBACZ,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;oBAClC,uBAAA,IAAI,qDAAsB,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;gBAC9D,CAAC;gBACD,uBAAA,IAAI,oDAAqB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,EAAE,iBAAiB;YACzB,OAAO;YACP,cAAc,EAAE,mBAAmB,CAAC,cAAc;SACnD,CAAC,CAAC;QAEH,sEAAsE;QACtE,KAAK,MAAM,EACT,OAAO,EACP,eAAe,EAAE,aAAa,GAC/B,IAAI,OAAO,CAAC,2BAA2B,EAAE,CAAC;YACzC,MAAM,gBAAgB,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAC1D,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,CAC9B,CAAC;YAEF,KAAK,MAAM,OAAO,IAAI,gBAAgB,EAAE,CAAC;gBACvC,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;gBAE7C,MAAM,KAAK,GAA8B;oBACvC,OAAO,EAAE,UAAU;oBACnB,SAAS,EAAE,OAAO,CAAC,EAAE;oBACrB,cAAc,EAAE,OAAO,CAAC,OAAc;iBACvC,CAAC;gBAEF,MAAM,YAAY,GAAG,uBAAA,IAAI,qDAAsB,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpE,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,6EAA6E;QAC7E,IAAI,CAAC;YACH,MAAM,cAAc,GAAgB;gBAClC,2BAA2B;gBAC3B,QAAQ,EAAE,iBAAiB;gBAC3B,SAAS,EAAE,CAAC,SAAS,CAAC;aACvB,CAAC;YACF,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YACzD,IACE,eAAe,CAAC,aAAa;gBAC7B,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC,MAAM,GAAG,CAAC,EACrD,CAAC;gBACD,mBAAmB;qBAChB,cAAc,EAAE,CAAC,eAAe,CAAC;oBAClC,EAAE,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBAChB,GAAG,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;gBACzD,CAAC,CAAC,CAAC;YACP,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,qCAAqC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACxD,CAAC;QAED,GAAG,CAAC,sBAAsB,EAAE;YAC1B,cAAc;YACd,MAAM,EAAE,iBAAiB;YACzB,YAAY,EAAE,aAAa,CAAC,MAAM;SACnC,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,WAAW,CAAC,cAAsB;QACtC,MAAM,YAAY,GAAG,uBAAA,IAAI,oDAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACnE,IAAI,YAAY,EAAE,CAAC;YACjB,KAAK,MAAM,KAAK,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;gBAC/C,uBAAA,IAAI,qDAAsB,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;YAC9D,CAAC;YACD,uBAAA,IAAI,oDAAqB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,KAAK,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,OAAO;QACL,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YAC9D,KAAK,MAAM,KAAK,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;gBAC/C,uBAAA,IAAI,qDAAsB,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;QACD,uBAAA,IAAI,oDAAqB,CAAC,KAAK,EAAE,CAAC;QAClC,uBAAA,IAAI,8CAAe,CAAC,KAAK,EAAE,CAAC;QAC5B,KAAK,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC;CACF;;IA7qBG,uBAAA,IAAI,8CAAe,CAAC,KAAK,EAAE,CAAC;IAC5B,GAAG,CAAC,mDAAmD,CAAC,CAAC;AAC3D,CAAC,iJASC,KAAuC;IAEvC,MAAM,EAAE,iBAAiB,EAAE,GAAG,KAAK,IAAI,EAAE,CAAC;IAC1C,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,OAAO;IACT,CAAC;IACD,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,iBAAiB,CAAC,CAAC;AAChE,CAAC,2IAasC,OAGtC;IACC,MAAM,UAAU,GAAG,OAAO,EAAE,OAAO,CAAC;IACpC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,eAAe,GAAG,yBAAyB,CAAC,UAAU,CAAC,CAAC;IAC9D,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,aAAa,GAAG,eAAe,CAAC,WAAW,EAAE,CAAC;IACpD,MAAM,IAAI,GAAG,OAAO,CAAC,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;IACnD,MAAM,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE,EAAE,EAAE,WAAW,EAAE,CAAC;IAC/C,OAAO,IAAI,KAAK,aAAa,IAAI,EAAE,KAAK,aAAa,CAAC;AACxD,CAAC,6GAYuB,OAGvB;IACC,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;QACnB,OAAO;IACT,CAAC;IACD,IAAI,CAAC,uBAAA,IAAI,0GAAuC,MAA3C,IAAI,EAAwC,OAAO,CAAC,EAAE,CAAC;QAC1D,OAAO;IACT,CAAC;IACD,MAAM,UAAU,GAAG,OAAO,EAAE,OAAO,CAAC;IACpC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO;IACT,CAAC;IACD,MAAM,WAAW,GAAG,UAAU,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,EAAa,CAAC;IACpE,MAAM,SAAS,GAAG,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,CAAC,WAAW,CAAC,CAAC,CAAC;IAC7D,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACpE,GAAG,CAAC,oDAAoD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC,6GASC,OAA0E;IAE1E,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;QACnB,OAAO;IACT,CAAC;IACD,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAU,CAAC;IAC5C,KAAK,MAAM,IAAI,IAAI,OAAO,IAAI,EAAE,EAAE,CAAC;QACjC,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC;YACnB,SAAS;QACX,CAAC;QACD,IAAI,uBAAA,IAAI,0GAAuC,MAA3C,IAAI,EAAwC,IAAI,CAAC,EAAE,CAAC;YACtD,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IACD,MAAM,YAAY,GAAG,CAAC,GAAG,iBAAiB,CAAC,CAAC,GAAG,CAC7C,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,EAAa,CAChE,CAAC;IACF,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO;IACT,CAAC;IACD,MAAM,SAAS,GAAG,uBAAA,IAAI,0FAAuB,MAA3B,IAAI,EAAwB,YAAY,CAAC,CAAC;IAC5D,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACpE,GAAG,CAAC,8DAA8D,EAAE;gBAClE,KAAK;aACN,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC,2GASC,QAAmB;IAEnB,MAAM,SAAS,GAAqD,EAAE,CAAC;IACvE,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC;IACnC,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9D,KAAK,MAAM,EACT,OAAO,EACP,eAAe,GAChB,IAAI,YAAY,CAAC,2BAA2B,EAAE,CAAC;YAC9C,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;gBACtC,IAAI,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC1B,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;gBACvC,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;IAQC,MAAM,SAAS,GAAqD,EAAE,CAAC;IACvE,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9D,KAAK,MAAM,EACT,OAAO,EACP,eAAe,GAChB,IAAI,YAAY,CAAC,2BAA2B,EAAE,CAAC;YAC9C,KAAK,MAAM,OAAO,IAAI,eAAe,EAAE,CAAC;gBACtC,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YACvC,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAgBD;;;;;GAKG;AACH,KAAK,wEACH,SAA2D;IAE3D,MAAM,UAAU,GAAyC,EAAE,CAAC;IAC5D,MAAM,aAAa,GAGf,EAAE,CAAC;IAEP,KAAK,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,SAAS,EAAE,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,eAAe,GAAG,yBAAyB,CAAC,UAAU,CAAC,CAAC;YAC9D,IAAI,CAAC,eAAe,EAAE,CAAC;gBACrB,SAAS;YACX,CAAC;YAED,MAAM,KAAK,GAA8B;gBACvC,OAAO,EAAE,UAAU;gBACnB,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,cAAc,EAAE,OAAO,CAAC,OAAc;aACvC,CAAC;YAEF,MAAM,MAAM,GACV,MAAM,uBAAA,IAAI,qDAAsB,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAC7D,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;YAExD,UAAU,CAAC,OAAO,CAAC,GAAG,mBAAmB,CAAC;YAC1C,MAAM,QAAQ,GAAG,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3C,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC,GAAG;gBAC1B,GAAG,QAAQ;gBACX,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE;aACrC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,uDAAuD,EAAE;gBAC3D,OAAO;gBACP,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,KAAK;aACN,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1C,MAAM,QAAQ,GAAiB,EAAE,UAAU,EAAE,aAAa,EAAE,CAAC;QAC7D,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;YAC9D,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;gBAC9D,GAAG,CAAC,0DAA0D,EAAE;oBAC9D,KAAK;iBACN,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;AACH,CAAC;IAQC,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,uBAAA,IAAI,0CAAW,CAAC,IAAI,CAChC,sCAAsC,CACvC,CAAC;QACF,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EACF,KAAK,EAAE,iBAAiB,IAAI,EAAE,CAC/B,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,iDAAiD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAClE,uBAAA,IAAI,yGAAsC,MAA1C,IAAI,EAAuC,EAAE,CAAC,CAAC;IACjD,CAAC;AACH,CAAC,yIAUC,iBAA0D;IAE1D,IAAI,CAAC,uBAAA,IAAI,wCAAS,EAAE,CAAC;QACnB,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAC9C,IAAI,CAAC,kBAAkB,CAAC,EAAE,EAAE,CAAC,aAAa,EAAE,EAAE,CAC5C,uBAAA,IAAI,sDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;QACF,OAAO;IACT,CAAC;IAED,MAAM,YAAY,GAAc,EAAE,CAAC;IACnC,MAAM,SAAS,GAAG,iBAAiB,CAAC,MAAM,CAAC;IAC3C,IAAI,SAAS,EAAE,CAAC;QACd,KAAK,MAAM,KAAK,IAAI,uBAAA,IAAI,kDAAmB,EAAE,CAAC;YAC5C,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1B,SAAS;YACX,CAAC;YACD,MAAM,EAAE,SAAS,EAAE,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;YAC9C,MAAM,UAAU,GAAG,WAAW,CAAC,QAAQ,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC;YACxD,IAAI,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC1B,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAC9C,IAAI,CAAC,kBAAkB,CAAC,YAAY,EAAE,CAAC,aAAa,EAAE,EAAE,CACtD,uBAAA,IAAI,sDAAuB,MAA3B,IAAI,EAAwB,IAAI,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,QAAQ,CAAC,CACrE,CAAC;AACJ,CAAC,uFASY,UAAe;IAC1B,MAAM,MAAM,GAAG,uBAAA,IAAI,8CAAe,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACnD,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,CAAC;QACH,MAAM,YAAY,GAAG,uBAAA,IAAI,0CAAW,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAExE,MAAM,EAAE,8BAA8B,EAAE,GAAG,YAAY,CAAC;QACxD,IAAI,CAAC,8BAA8B,EAAE,CAAC;YACpC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,WAAW,GAAG,8BAA8B,CAAC,UAAU,CAAC,CAAC;QAC/D,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,6EAA6E;QAC7E,MAAM,EAAE,YAAY,EAAE,uBAAuB,EAAE,GAAG,WAAW,CAAC;QAC9D,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/C,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,KAAK,GACT,OAAO,uBAAuB,KAAK,QAAQ;YAC3C,uBAAuB,IAAI,CAAC;YAC5B,uBAAuB,GAAG,YAAY,CAAC,MAAM;YAC3C,CAAC,CAAC,uBAAuB;YACzB,CAAC,CAAC,CAAC,CAAC;QACR,MAAM,eAAe,GAAG,YAAY,CAAC,KAAK,CAEzC,CAAC;QACF,MAAM,eAAe,GAAG,eAAe,EAAE,eAAe,CAAC;QACzD,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,aAAa,GAAG,uBAAA,IAAI,0CAAW,CAAC,IAAI,CACxC,wCAAwC,EACxC,eAAe,CAChB,CAAC;QAEF,IAAI,CAAC,aAAa,EAAE,QAAQ,EAAE,CAAC;YAC7B,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC1D,uBAAA,IAAI,8CAAe,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC9C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,kCAAkC,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/D,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC,mHAS0B,MAAgC;IACzD,MAAM,eAAe,GAAG,yBAAyB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAClE,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO;IACT,CAAC;IACD,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IACpD,MAAM,WAAW,GAAG,UAAU,cAAc,EAAa,CAAC;IAC1D,MAAM,OAAO,GAAG,aAAa,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC;IAE5D,MAAM,QAAQ,GAAiB;QAC7B,UAAU,EAAE;YACV,CAAC,OAAO,CAAC,EAAE,mBAAmB;SAC/B;QACD,aAAa,EAAE;YACb,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE;gBAClB,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE;aAC7C;SACF;KACF,CAAC;IAEF,GAAG,CAAC,uBAAuB,EAAE;QAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,OAAO,EAAE,WAAW;QACpB,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,MAAM;KAC9B,CAAC,CAAC;IAEH,KAAK,MAAM,YAAY,IAAI,uBAAA,IAAI,oDAAqB,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9D,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YAC9D,GAAG,CAAC,wCAAwC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC;AACH,CAAC","sourcesContent":["import { toChecksumAddress } from '@ethereumjs/util';\nimport { Web3Provider } from '@ethersproject/providers';\nimport type { InternalAccount } from '@metamask/keyring-internal-api';\nimport type { NetworkEnablementControllerState } from '@metamask/network-enablement-controller';\nimport {\n isStrictHexString,\n isCaipChainId,\n numberToHex,\n parseCaipChainId,\n} from '@metamask/utils';\nimport type { Hex } from '@metamask/utils';\n\nimport type {\n DataSourceState,\n SubscriptionRequest,\n} from './AbstractDataSource';\nimport { AbstractDataSource } from './AbstractDataSource';\nimport type {\n StakedBalancePollingInput,\n StakedBalanceFetchResult,\n} from './evm-rpc-services';\nimport {\n StakedBalanceFetcher,\n getStakingContractAddress,\n getSupportedStakingChainIds,\n} from './evm-rpc-services';\nimport type { AssetsControllerMessenger } from '../AssetsController';\nimport { projectLogger, createModuleLogger } from '../logger';\nimport type {\n AccountId,\n ChainId,\n Caip19AssetId,\n AssetBalance,\n AssetMetadata,\n DataRequest,\n DataResponse,\n Middleware,\n} from '../types';\n\nconst CONTROLLER_NAME = 'StakedBalanceDataSource';\nconst DEFAULT_POLL_INTERVAL = 180_000; // 3 minutes\n\n/** Metadata for staked ETH (same symbol and decimals as native ETH). */\nconst STAKED_ETH_METADATA: AssetMetadata = {\n type: 'erc20',\n name: 'staked ethereum',\n symbol: 'ETH',\n decimals: 18,\n};\n\nconst log = createModuleLogger(projectLogger, CONTROLLER_NAME);\n\n/** Optional configuration for StakedBalanceDataSource. */\nexport type StakedBalanceDataSourceConfig = {\n /** Whether staked balance fetching is enabled (default: true). */\n enabled?: boolean;\n /** Polling interval in ms (default: 180s / 3 min). */\n pollInterval?: number;\n};\n\nexport type StakedBalanceDataSourceOptions = StakedBalanceDataSourceConfig & {\n /** The AssetsController messenger (for accessing NetworkController). */\n messenger: AssetsControllerMessenger;\n /** Called when active chains are updated. */\n onActiveChainsUpdated: (\n dataSourceName: string,\n chains: ChainId[],\n previousChains: ChainId[],\n ) => void;\n};\n\n/** Per-account supported chains (same shape as in DataRequest). */\ntype AccountWithSupportedChains = {\n account: InternalAccount;\n supportedChains: ChainId[];\n};\n\n/**\n * Subscription data stored for each active subscription.\n * Stores accountsWithSupportedChains so refresh paths use the same per-account\n * scope as normal subscription setup (avoids querying unsupported chains/accounts).\n */\ntype SubscriptionData = {\n /** Polling tokens from StakedBalanceFetcher. */\n pollingTokens: string[];\n /** Chain IDs being polled (union of all account chains). */\n chains: ChainId[];\n /** Accounts being polled. */\n accounts: InternalAccount[];\n /** Per-account supported chains; used by refreshStakedBalance and transaction handlers. */\n accountsWithSupportedChains: AccountWithSupportedChains[];\n /** Callback to report asset updates. */\n onAssetsUpdate: (response: DataResponse) => void | Promise<void>;\n};\n\n/**\n * Convert CAIP chain ID or hex chain ID to hex chain ID.\n *\n * @param chainId - CAIP chain ID or hex chain ID.\n * @returns Hex chain ID.\n */\nfunction caipChainIdToHex(chainId: string): Hex {\n if (isStrictHexString(chainId)) {\n return chainId;\n }\n\n if (isCaipChainId(chainId)) {\n const ref = parseCaipChainId(chainId).reference;\n return numberToHex(parseInt(ref, 10));\n }\n\n throw new Error('caipChainIdToHex - Failed to provide CAIP-2 or Hex chainId');\n}\n\n/**\n * Build the CAIP-19 asset ID for staked balance (same format as ERC20).\n * Uses the staking contract address (checksummed) so it is consistent with\n * other token assets.\n * Example: \"eip155:1/erc20:0x4fef9d741011476750a243ac70b9789a63dd47df\"\n *\n * @param chainId - CAIP-2 chain ID (e.g. \"eip155:1\").\n * @param contractAddress - Staking contract address (hex).\n * @returns The staked asset CAIP-19 ID with checksummed address.\n */\nfunction stakedAssetId(\n chainId: ChainId,\n contractAddress: string,\n): Caip19AssetId {\n const checksummed = toChecksumAddress(contractAddress);\n return `${chainId}/erc20:${checksummed}` as Caip19AssetId;\n}\n\n/**\n * Data source for fetching staked ETH balances via on-chain staking contracts.\n *\n * Delegates to {@link StakedBalanceFetcher} for the actual RPC calls\n * (getShares + convertToAssets on ERC-4626-style staking contracts).\n * Reports balances as CAIP-19 asset IDs using the ERC20 format with the\n * staking contract address (e.g. \"eip155:1/erc20:0x4fef9d741011476750a243ac70b9789a63dd47df\").\n *\n * Only supports chains with known staking contracts (mainnet, Hoodi).\n * Polling is managed by StakedBalanceFetcher via startPolling/stopPollingByPollingToken.\n */\nexport class StakedBalanceDataSource extends AbstractDataSource<\n typeof CONTROLLER_NAME,\n DataSourceState\n> {\n readonly #messenger: AssetsControllerMessenger;\n\n readonly #onActiveChainsUpdated: (\n dataSourceName: string,\n chains: ChainId[],\n previousChains: ChainId[],\n ) => void;\n\n readonly #pollInterval: number;\n\n readonly #enabled: boolean;\n\n /** The StakedBalanceFetcher that handles polling and RPC calls. */\n readonly #stakedBalanceFetcher: StakedBalanceFetcher;\n\n /** Active subscriptions by ID. */\n readonly #activeSubscriptions: Map<string, SubscriptionData> = new Map();\n\n /** Cache of Web3Provider instances by hex chain ID. */\n readonly #providerCache: Map<Hex, Web3Provider> = new Map();\n\n /** CAIP-2 chain IDs that have known staking contracts. */\n readonly #supportedChainIds: ChainId[];\n\n constructor(options: StakedBalanceDataSourceOptions) {\n super(CONTROLLER_NAME, { activeChains: [] });\n this.#messenger = options.messenger;\n this.#onActiveChainsUpdated = options.onActiveChainsUpdated;\n this.#pollInterval = options.pollInterval ?? DEFAULT_POLL_INTERVAL;\n this.#enabled = options.enabled !== false;\n this.#supportedChainIds = getSupportedStakingChainIds() as ChainId[];\n\n log('Initializing StakedBalanceDataSource', {\n enabled: this.#enabled,\n pollInterval: this.#pollInterval,\n });\n\n // Create StakedBalanceFetcher with provider getter\n this.#stakedBalanceFetcher = new StakedBalanceFetcher({\n pollingInterval: this.#pollInterval,\n getNetworkProvider: (hexChainId): Web3Provider | undefined =>\n this.#getProvider(hexChainId),\n });\n\n // Wire the callback so polling results flow back to subscriptions\n this.#stakedBalanceFetcher.setOnStakedBalanceUpdate(\n this.#handleStakedBalanceUpdate.bind(this),\n );\n\n this.#messenger.subscribe(\n 'TransactionController:transactionConfirmed',\n this.#onTransactionConfirmed.bind(this),\n );\n\n this.#messenger.subscribe(\n 'TransactionController:incomingTransactionsReceived',\n this.#onIncomingTransactions.bind(this),\n );\n\n this.#messenger.subscribe(\n 'NetworkController:stateChange',\n this.#onNetworkStateChange.bind(this),\n );\n\n this.#messenger.subscribe(\n 'NetworkEnablementController:stateChange',\n this.#onNetworkEnablementControllerStateChange.bind(this),\n );\n\n this.#initializeActiveChains();\n }\n\n /**\n * When NetworkController state changes (e.g. RPC endpoints or network clients\n * reconfigured), clear the provider cache so subsequent fetches use fresh\n * providers.\n */\n #onNetworkStateChange(): void {\n this.#providerCache.clear();\n log('Provider cache cleared after network state change');\n }\n\n /**\n * When NetworkEnablementController state changes (user enables/disables\n * networks), recompute active chains so we only fetch for enabled staking chains.\n *\n * @param state - The new NetworkEnablementController state.\n */\n #onNetworkEnablementControllerStateChange(\n state: NetworkEnablementControllerState,\n ): void {\n const { enabledNetworkMap } = state ?? {};\n if (!enabledNetworkMap) {\n return;\n }\n this.#initializeActiveChainsFromEnabledMap(enabledNetworkMap);\n }\n\n /**\n * Returns true if the transaction involves the staking contract (from or to)\n * for the payload's chain, so we only refresh staked balance when relevant.\n *\n * @param payload - Transaction payload.\n * @param payload.chainId - Hex chain ID.\n * @param payload.txParams - Optional transaction params.\n * @param payload.txParams.from - Sender address.\n * @param payload.txParams.to - Recipient address.\n * @returns True if txParams.from or txParams.to matches the staking contract address.\n */\n #isTransactionInvolvingStakingContract(payload: {\n chainId?: string;\n txParams?: { from?: string; to?: string };\n }): boolean {\n const hexChainId = payload?.chainId;\n if (!hexChainId) {\n return false;\n }\n const contractAddress = getStakingContractAddress(hexChainId);\n if (!contractAddress) {\n return false;\n }\n const contractLower = contractAddress.toLowerCase();\n const from = payload.txParams?.from?.toLowerCase();\n const to = payload.txParams?.to?.toLowerCase();\n return from === contractLower || to === contractLower;\n }\n\n /**\n * When a transaction is confirmed, refresh staked balance only if the\n * transaction is from or to the staking contract for that chain.\n *\n * @param payload - From TransactionController:transactionConfirmed.\n * @param payload.chainId - Hex chain ID of the transaction.\n * @param payload.txParams - Optional transaction params.\n * @param payload.txParams.from - Sender address.\n * @param payload.txParams.to - Recipient address.\n */\n #onTransactionConfirmed(payload: {\n chainId?: string;\n txParams?: { from?: string; to?: string };\n }): void {\n if (!this.#enabled) {\n return;\n }\n if (!this.#isTransactionInvolvingStakingContract(payload)) {\n return;\n }\n const hexChainId = payload?.chainId;\n if (!hexChainId) {\n return;\n }\n const caipChainId = `eip155:${parseInt(hexChainId, 16)}` as ChainId;\n const toRefresh = this.#getToRefreshForChains([caipChainId]);\n if (toRefresh.length > 0) {\n this.#refreshStakedBalanceAfterTransaction(toRefresh).catch((error) => {\n log('Failed to refresh staked balance after transaction', { error });\n });\n }\n }\n\n /**\n * When incoming transactions are received, refresh staked balance only for\n * chains where at least one transaction is from or to the staking contract.\n *\n * @param payload - From TransactionController:incomingTransactionsReceived (array of { chainId?, txParams? }).\n */\n #onIncomingTransactions(\n payload: { chainId?: string; txParams?: { from?: string; to?: string } }[],\n ): void {\n if (!this.#enabled) {\n return;\n }\n const chainIdsToRefresh = new Set<string>();\n for (const item of payload ?? []) {\n if (!item?.chainId) {\n continue;\n }\n if (this.#isTransactionInvolvingStakingContract(item)) {\n chainIdsToRefresh.add(item.chainId);\n }\n }\n const caipChainIds = [...chainIdsToRefresh].map(\n (hexChainId) => `eip155:${parseInt(hexChainId, 16)}` as ChainId,\n );\n if (caipChainIds.length === 0) {\n return;\n }\n const toRefresh = this.#getToRefreshForChains(caipChainIds);\n if (toRefresh.length > 0) {\n this.#refreshStakedBalanceAfterTransaction(toRefresh).catch((error) => {\n log('Failed to refresh staked balance after incoming transactions', {\n error,\n });\n });\n }\n }\n\n /**\n * Build toRefresh list for subscribed (account, chainId) pairs for the given chains.\n *\n * @param chainIds - CAIP-2 chain IDs to target.\n * @returns Pairs of account and chainId to refresh.\n */\n #getToRefreshForChains(\n chainIds: ChainId[],\n ): { account: InternalAccount; chainId: ChainId }[] {\n const toRefresh: { account: InternalAccount; chainId: ChainId }[] = [];\n const chainSet = new Set(chainIds);\n for (const subscription of this.#activeSubscriptions.values()) {\n for (const {\n account,\n supportedChains,\n } of subscription.accountsWithSupportedChains) {\n for (const chainId of supportedChains) {\n if (chainSet.has(chainId)) {\n toRefresh.push({ account, chainId });\n }\n }\n }\n }\n return toRefresh;\n }\n\n /**\n * Build toRefresh list for all subscribed (account, chainId) pairs.\n *\n * @returns Pairs of account and chainId to refresh.\n */\n #getToRefreshAll(): { account: InternalAccount; chainId: ChainId }[] {\n const toRefresh: { account: InternalAccount; chainId: ChainId }[] = [];\n for (const subscription of this.#activeSubscriptions.values()) {\n for (const {\n account,\n supportedChains,\n } of subscription.accountsWithSupportedChains) {\n for (const chainId of supportedChains) {\n toRefresh.push({ account, chainId });\n }\n }\n }\n return toRefresh;\n }\n\n /**\n * Refresh staked balance for all currently subscribed accounts and chains, then\n * push updates to the controller. Can be called from UI or after transaction events.\n */\n async refreshStakedBalance(): Promise<void> {\n if (!this.#enabled) {\n return;\n }\n const toRefresh = this.#getToRefreshAll();\n if (toRefresh.length > 0) {\n await this.#refreshStakedBalanceAfterTransaction(toRefresh);\n }\n }\n\n /**\n * Fetch staked balance for the given account/chain pairs and push a single\n * DataResponse to all active subscriptions.\n *\n * @param toRefresh - List of { account, chainId } to refresh.\n */\n async #refreshStakedBalanceAfterTransaction(\n toRefresh: { account: InternalAccount; chainId: ChainId }[],\n ): Promise<void> {\n const assetsInfo: Record<Caip19AssetId, AssetMetadata> = {};\n const assetsBalance: Record<\n AccountId,\n Record<Caip19AssetId, AssetBalance>\n > = {};\n\n for (const { account, chainId } of toRefresh) {\n try {\n const hexChainId = caipChainIdToHex(chainId);\n const contractAddress = getStakingContractAddress(hexChainId);\n if (!contractAddress) {\n continue;\n }\n\n const input: StakedBalancePollingInput = {\n chainId: hexChainId,\n accountId: account.id,\n accountAddress: account.address as Hex,\n };\n\n const result =\n await this.#stakedBalanceFetcher.fetchStakedBalance(input);\n const assetId = stakedAssetId(chainId, contractAddress);\n\n assetsInfo[assetId] = STAKED_ETH_METADATA;\n const existing = assetsBalance[account.id];\n assetsBalance[account.id] = {\n ...existing,\n [assetId]: { amount: result.amount },\n };\n } catch (error) {\n log('Failed to fetch staked balance in transaction refresh', {\n chainId,\n accountId: account.id,\n error,\n });\n }\n }\n\n if (Object.keys(assetsBalance).length > 0) {\n const response: DataResponse = { assetsInfo, assetsBalance };\n for (const subscription of this.#activeSubscriptions.values()) {\n subscription.onAssetsUpdate(response)?.catch((error: unknown) => {\n log('Failed to report staked balance update after transaction', {\n error,\n });\n });\n }\n }\n }\n\n /**\n * Set active chains from NetworkEnablementController state.\n * Only staking-supported chains that are enabled in the network enablement map\n * are active (e.g. if mainnet is not selected we do not fetch).\n */\n #initializeActiveChains(): void {\n try {\n const state = this.#messenger.call(\n 'NetworkEnablementController:getState',\n );\n this.#initializeActiveChainsFromEnabledMap(\n state?.enabledNetworkMap ?? {},\n );\n } catch (error) {\n log('Failed to get NetworkEnablementController state', { error });\n this.#initializeActiveChainsFromEnabledMap({});\n }\n }\n\n /**\n * Compute active chains as the intersection of supported staking chains and\n * enabled networks, then update state. Uses the same EIP-155 storage key\n * convention as NetworkEnablementController (hex for eip155).\n *\n * @param enabledNetworkMap - The enabled network map from NetworkEnablementController.\n */\n #initializeActiveChainsFromEnabledMap(\n enabledNetworkMap: Record<string, Record<string, boolean>>,\n ): void {\n if (!this.#enabled) {\n const previous = [...this.state.activeChains];\n this.updateActiveChains([], (updatedChains) =>\n this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n );\n return;\n }\n\n const activeChains: ChainId[] = [];\n const eip155Map = enabledNetworkMap.eip155;\n if (eip155Map) {\n for (const caip2 of this.#supportedChainIds) {\n if (!isCaipChainId(caip2)) {\n continue;\n }\n const { reference } = parseCaipChainId(caip2);\n const storageKey = numberToHex(parseInt(reference, 10));\n if (eip155Map[storageKey]) {\n activeChains.push(caip2);\n }\n }\n }\n\n const previous = [...this.state.activeChains];\n this.updateActiveChains(activeChains, (updatedChains) =>\n this.#onActiveChainsUpdated(this.getName(), updatedChains, previous),\n );\n }\n\n /**\n * Get or create a Web3Provider for the given hex chain ID.\n * Uses the same messenger-cast pattern as RpcDataSource.\n *\n * @param hexChainId - The hex chain ID.\n * @returns Web3Provider instance, or undefined if not available.\n */\n #getProvider(hexChainId: Hex): Web3Provider | undefined {\n const cached = this.#providerCache.get(hexChainId);\n if (cached) {\n return cached;\n }\n\n try {\n const networkState = this.#messenger.call('NetworkController:getState');\n\n const { networkConfigurationsByChainId } = networkState;\n if (!networkConfigurationsByChainId) {\n return undefined;\n }\n\n const chainConfig = networkConfigurationsByChainId[hexChainId];\n if (!chainConfig) {\n return undefined;\n }\n\n // Use the network's configured default RPC endpoint (same as RpcDataSource).\n const { rpcEndpoints, defaultRpcEndpointIndex } = chainConfig;\n if (!rpcEndpoints || rpcEndpoints.length === 0) {\n return undefined;\n }\n\n const index =\n typeof defaultRpcEndpointIndex === 'number' &&\n defaultRpcEndpointIndex >= 0 &&\n defaultRpcEndpointIndex < rpcEndpoints.length\n ? defaultRpcEndpointIndex\n : 0;\n const defaultEndpoint = rpcEndpoints[index] as {\n networkClientId?: string;\n };\n const networkClientId = defaultEndpoint?.networkClientId;\n if (!networkClientId) {\n return undefined;\n }\n\n const networkClient = this.#messenger.call(\n 'NetworkController:getNetworkClientById',\n networkClientId,\n );\n\n if (!networkClient?.provider) {\n return undefined;\n }\n\n const provider = new Web3Provider(networkClient.provider);\n this.#providerCache.set(hexChainId, provider);\n return provider;\n } catch (error) {\n log('Failed to get provider for chain', { hexChainId, error });\n return undefined;\n }\n }\n\n /**\n * Handle a staked balance update from StakedBalanceFetcher.\n * Converts the result into a DataResponse and forwards it to all active\n * subscriptions.\n *\n * @param result - The staked balance fetch result.\n */\n #handleStakedBalanceUpdate(result: StakedBalanceFetchResult): void {\n const contractAddress = getStakingContractAddress(result.chainId);\n if (!contractAddress) {\n return;\n }\n const chainIdDecimal = parseInt(result.chainId, 16);\n const caipChainId = `eip155:${chainIdDecimal}` as ChainId;\n const assetId = stakedAssetId(caipChainId, contractAddress);\n\n const response: DataResponse = {\n assetsInfo: {\n [assetId]: STAKED_ETH_METADATA,\n },\n assetsBalance: {\n [result.accountId]: {\n [assetId]: { amount: result.balance.amount },\n },\n },\n };\n\n log('Staked balance update', {\n accountId: result.accountId,\n chainId: caipChainId,\n amount: result.balance.amount,\n });\n\n for (const subscription of this.#activeSubscriptions.values()) {\n subscription.onAssetsUpdate(response)?.catch((error: unknown) => {\n log('Failed to report staked balance update', { error });\n });\n }\n }\n\n /**\n * Fetch staked balances for all accounts on supported chains.\n *\n * @param request - The data request with accounts and chains.\n * @returns DataResponse with staked balance data.\n */\n async fetch(request: DataRequest): Promise<DataResponse> {\n if (!this.#enabled) {\n return {};\n }\n const response: DataResponse = {};\n const activeChainsSet = new Set(this.state.activeChains);\n\n const chainsToFetch = request.chainIds.filter((chainId) =>\n activeChainsSet.has(chainId),\n );\n\n if (chainsToFetch.length === 0) {\n return response;\n }\n\n const balances: Record<AccountId, Record<Caip19AssetId, AssetBalance>> = {};\n\n for (const {\n account,\n supportedChains: accountChains,\n } of request.accountsWithSupportedChains) {\n const chains = chainsToFetch.filter((chain) =>\n accountChains.includes(chain),\n );\n\n for (const chainId of chains) {\n try {\n const hexChainId = caipChainIdToHex(chainId);\n const contractAddress = getStakingContractAddress(hexChainId);\n if (!contractAddress) {\n continue;\n }\n\n const input: StakedBalancePollingInput = {\n chainId: hexChainId,\n accountId: account.id,\n accountAddress: account.address as Hex,\n };\n\n const result =\n await this.#stakedBalanceFetcher.fetchStakedBalance(input);\n\n // Include zero amounts so merged updates clear prior non-zero state.\n balances[account.id] ??= {};\n const assetId = stakedAssetId(chainId, contractAddress);\n balances[account.id][assetId] = { amount: result.amount };\n } catch (error) {\n log('Failed to fetch staked balance', {\n chainId,\n accountId: account.id,\n error,\n });\n }\n }\n }\n\n if (Object.keys(balances).length > 0) {\n response.assetsBalance = balances;\n // Add metadata for each staked asset ID present in balances\n const assetIds = new Set<Caip19AssetId>();\n for (const accountBalances of Object.values(balances)) {\n for (const assetId of Object.keys(accountBalances)) {\n assetIds.add(assetId as Caip19AssetId);\n }\n }\n response.assetsInfo = {};\n for (const assetId of assetIds) {\n response.assetsInfo[assetId] = STAKED_ETH_METADATA;\n }\n }\n\n return response;\n }\n\n /**\n * Assets middleware for the fetch pipeline.\n * Fetches staked balances and merges them into the response, then passes\n * all chains to the next middleware (staked balance doesn't claim chains).\n *\n * @returns The middleware function for the assets pipeline.\n */\n get assetsMiddleware(): Middleware {\n return async (context, next) => {\n if (!this.#enabled) {\n return next(context);\n }\n const { request } = context;\n\n if (!request.dataTypes.includes('balance')) {\n return next(context);\n }\n\n if (request.chainIds.length === 0) {\n return next(context);\n }\n\n try {\n const fetchResponse = await this.fetch(request);\n\n if (fetchResponse.assetsInfo) {\n context.response.assetsInfo ??= {};\n Object.assign(context.response.assetsInfo, fetchResponse.assetsInfo);\n }\n if (fetchResponse.assetsBalance) {\n context.response.assetsBalance ??= {};\n for (const [accountId, accountBalances] of Object.entries(\n fetchResponse.assetsBalance,\n )) {\n context.response.assetsBalance[accountId] = {\n ...context.response.assetsBalance[accountId],\n ...accountBalances,\n };\n }\n }\n } catch (error) {\n log('Middleware fetch failed', { error });\n }\n\n // Pass all chains through (staked balance doesn't claim chains)\n return next(context);\n };\n }\n\n /**\n * Subscribe to staked balance updates with polling.\n * Starts polling via StakedBalanceFetcher for each account/chain combination.\n *\n * @param subscriptionRequest - The subscription request details.\n */\n async subscribe(subscriptionRequest: SubscriptionRequest): Promise<void> {\n const { request, subscriptionId, isUpdate } = subscriptionRequest;\n\n const activeChainsSet = new Set(this.state.activeChains);\n const chainsToSubscribe = request.chainIds.filter((chainId) =>\n activeChainsSet.has(chainId),\n );\n\n log('Subscribe requested', {\n subscriptionId,\n isUpdate,\n chainsToSubscribe,\n });\n\n if (chainsToSubscribe.length === 0) {\n log('No staking chains to subscribe');\n return;\n }\n\n // Handle subscription update - restart polling for new chains\n if (isUpdate) {\n const existing = this.#activeSubscriptions.get(subscriptionId);\n if (existing) {\n log('Updating existing subscription - restarting polling', {\n subscriptionId,\n });\n }\n }\n\n // Clean up existing subscription (stops old polling)\n await this.unsubscribe(subscriptionId);\n\n // Build subscription data first so it is available when the first poll runs\n const accountsWithSupportedChains: AccountWithSupportedChains[] =\n request.accountsWithSupportedChains\n .map(({ account, supportedChains }) => ({\n account,\n supportedChains: chainsToSubscribe.filter((chain) =>\n supportedChains.includes(chain),\n ),\n }))\n .filter(({ supportedChains }) => supportedChains.length > 0);\n\n const accounts = accountsWithSupportedChains.map((entry) => entry.account);\n const pollingTokens: string[] = [];\n\n // Store subscription before startPolling so first poll (setTimeout 0) has the callback\n this.#activeSubscriptions.set(subscriptionId, {\n pollingTokens,\n chains: chainsToSubscribe,\n accounts,\n accountsWithSupportedChains,\n onAssetsUpdate: subscriptionRequest.onAssetsUpdate,\n });\n\n this.activeSubscriptions.set(subscriptionId, {\n cleanup: () => {\n for (const token of pollingTokens) {\n this.#stakedBalanceFetcher.stopPollingByPollingToken(token);\n }\n this.#activeSubscriptions.delete(subscriptionId);\n },\n chains: chainsToSubscribe,\n request,\n onAssetsUpdate: subscriptionRequest.onAssetsUpdate,\n });\n\n // Start polling for each account/chain (first poll runs on next tick)\n for (const {\n account,\n supportedChains: accountChains,\n } of request.accountsWithSupportedChains) {\n const chainsForAccount = chainsToSubscribe.filter((chain) =>\n accountChains.includes(chain),\n );\n\n for (const chainId of chainsForAccount) {\n const hexChainId = caipChainIdToHex(chainId);\n\n const input: StakedBalancePollingInput = {\n chainId: hexChainId,\n accountId: account.id,\n accountAddress: account.address as Hex,\n };\n\n const pollingToken = this.#stakedBalanceFetcher.startPolling(input);\n pollingTokens.push(pollingToken);\n }\n }\n\n // Immediate initial fetch so state is updated without waiting for first poll\n try {\n const initialRequest: DataRequest = {\n accountsWithSupportedChains,\n chainIds: chainsToSubscribe,\n dataTypes: ['balance'],\n };\n const initialResponse = await this.fetch(initialRequest);\n if (\n initialResponse.assetsBalance &&\n Object.keys(initialResponse.assetsBalance).length > 0\n ) {\n subscriptionRequest\n .onAssetsUpdate?.(initialResponse)\n ?.catch((error) => {\n log('Initial staked balance update failed', { error });\n });\n }\n } catch (error) {\n log('Initial staked balance fetch failed', { error });\n }\n\n log('Subscription SUCCESS', {\n subscriptionId,\n chains: chainsToSubscribe,\n pollingCount: pollingTokens.length,\n });\n }\n\n /**\n * Unsubscribe from staked balance updates and stop polling.\n *\n * @param subscriptionId - The subscription ID to unsubscribe.\n */\n async unsubscribe(subscriptionId: string): Promise<void> {\n const subscription = this.#activeSubscriptions.get(subscriptionId);\n if (subscription) {\n for (const token of subscription.pollingTokens) {\n this.#stakedBalanceFetcher.stopPollingByPollingToken(token);\n }\n this.#activeSubscriptions.delete(subscriptionId);\n }\n\n await super.unsubscribe(subscriptionId);\n }\n\n /**\n * Destroy the data source and clean up all resources.\n */\n destroy(): void {\n for (const subscription of this.#activeSubscriptions.values()) {\n for (const token of subscription.pollingTokens) {\n this.#stakedBalanceFetcher.stopPollingByPollingToken(token);\n }\n }\n this.#activeSubscriptions.clear();\n this.#providerCache.clear();\n super.destroy();\n }\n}\n"]}
|