@hashgraph/sdk
Version:
86 lines (80 loc) • 2.87 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.default = void 0;
var _TokenId = _interopRequireDefault(require("../token/TokenId.cjs"));
var _AccountId = _interopRequireDefault(require("../account/AccountId.cjs"));
var _TokenTransferAccountMap = _interopRequireDefault(require("./TokenTransferAccountMap.cjs"));
var _ObjectMap = _interopRequireDefault(require("../ObjectMap.cjs"));
function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
// SPDX-License-Identifier: Apache-2.0
/**
* @namespace proto
* @typedef {import("@hashgraph/proto").proto.ITokenTransferList} HieroProto.proto.ITokenTransferList
* @typedef {import("@hashgraph/proto").proto.IAccountAmount} HieroProto.proto.IAccountAmount
* @typedef {import("@hashgraph/proto").proto.ITokenID} HieroProto.proto.ITokenID
* @typedef {import("@hashgraph/proto").proto.IAccountID} HieroProto.proto.IAccountID
*/
/**
* @augments {ObjectMap<TokenId, TokenTransferAccountMap>}
*/
class TokenTransferMap extends _ObjectMap.default {
constructor() {
super(s => _TokenId.default.fromString(s));
}
/**
* @internal
* @param {TokenId} tokenId
* @param {AccountId} accountId
* @param {Long} amount
*/
__set(tokenId, accountId, amount) {
const token = tokenId.toString();
let _map = this._map.get(token);
if (_map == null) {
_map = new _TokenTransferAccountMap.default();
this._map.set(token, _map);
this.__map.set(tokenId, _map);
}
_map._set(accountId, amount);
}
/**
* @param {HieroProto.proto.ITokenTransferList[]} transfers
* @returns {TokenTransferMap}
*/
static _fromProtobuf(transfers) {
const tokenTransfersMap = new TokenTransferMap();
for (const transfer of transfers) {
const token = _TokenId.default._fromProtobuf(/** @type {HieroProto.proto.ITokenID} */transfer.token);
for (const aa of transfer.transfers != null ? transfer.transfers : []) {
const account = _AccountId.default._fromProtobuf(/** @type {HieroProto.proto.IAccountID} */aa.accountID);
tokenTransfersMap.__set(token, account, /** @type {Long} */aa.amount);
}
}
return tokenTransfersMap;
}
/**
* @returns {HieroProto.proto.ITokenTransferList[]}
*/
_toProtobuf() {
/** @type {HieroProto.proto.ITokenTransferList[]} */
const tokenTransferList = [];
for (const [tokenId, value] of this) {
/** @type {HieroProto.proto.IAccountAmount[]} */
const transfers = [];
for (const [accountId, amount] of value) {
transfers.push({
accountID: accountId._toProtobuf(),
amount: amount
});
}
tokenTransferList.push({
token: tokenId._toProtobuf(),
transfers: transfers
});
}
return tokenTransferList;
}
}
exports.default = TokenTransferMap;