"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    var desc = Object.getOwnPropertyDescriptor(m, k);
    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
      desc = { enumerable: true, get: function() { return m[k]; } };
    }
    Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
    return c > 3 && r && Object.defineProperty(target, key, r), r;
};
var __importStar = (this && this.__importStar) || function (mod) {
    if (mod && mod.__esModule) return mod;
    var result = {};
    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
    __setModuleDefault(result, mod);
    return result;
};
var __metadata = (this && this.__metadata) || function (k, v) {
    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
};
var __param = (this && this.__param) || function (paramIndex, decorator) {
    return function (target, key) { decorator(target, key, paramIndex); }
};
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.VotingRewardsContractPositionFetcher = void 0;
const common_1 = require("@nestjs/common");
const ethers_1 = require("ethers/lib/ethers");
const lodash_1 = __importStar(require("lodash"));
const app_toolkit_interface_1 = require("../../../app-toolkit/app-toolkit.interface");
const drill_balance_helper_1 = require("../../../app-toolkit/helpers/drill-balance.helper");
const position_interface_1 = require("../../../position/position.interface");
const custom_contract_position_template_position_fetcher_1 = require("../../../position/template/custom-contract-position.template.position-fetcher");
const velodrome_definitions_resolver_1 = require("../common/velodrome.definitions-resolver");
const contracts_1 = require("../contracts");
let VotingRewardsContractPositionFetcher = class VotingRewardsContractPositionFetcher extends custom_contract_position_template_position_fetcher_1.CustomContractPositionTemplatePositionFetcher {
    constructor(appToolkit, contractFactory, definitionsResolver) {
        super(appToolkit);
        this.appToolkit = appToolkit;
        this.contractFactory = contractFactory;
        this.definitionsResolver = definitionsResolver;
        this.veTokenAddress = '0x9c7305eb78a432ced5c4d14cac27e8ed569a2e26';
    }
    getTokenDefinitions({ contract }) {
        return __awaiter(this, void 0, void 0, function* () {
            const numRewards = Number(yield contract.rewardsListLength());
            const bribeTokens = yield Promise.all((0, lodash_1.range)(numRewards).map((n) => __awaiter(this, void 0, void 0, function* () { return yield contract.rewards(n); })));
            const baseTokens = yield this.appToolkit.getBaseTokenPrices(this.network);
            const tokenDefinitions = bribeTokens.map(token => {
                const tokenFound = baseTokens.find(p => p.address === token.toLowerCase());
                if (!tokenFound)
                    return null;
                return {
                    metaType: position_interface_1.MetaType.CLAIMABLE,
                    address: token,
                    network: this.network,
                };
            });
            return lodash_1.default.compact(tokenDefinitions);
        });
    }
    getTokenBalancesPerPosition() {
        throw new common_1.NotImplementedException();
    }
    getBalances(address) {
        return __awaiter(this, void 0, void 0, function* () {
            const multicall = this.appToolkit.getMulticall(this.network);
            const escrow = this.contractFactory.velodromeVe({ address: this.veTokenAddress, network: this.network });
            const mcEscrow = multicall.wrap(escrow);
            const veCount = Number(yield mcEscrow.balanceOf(address));
            const veTokenIds = yield Promise.all((0, lodash_1.range)(veCount).map((i) => __awaiter(this, void 0, void 0, function* () { return mcEscrow.tokenOfOwnerByIndex(address, i); })));
            if (veTokenIds.length === 0)
                return [];
            const contractPositions = yield this.appToolkit.getAppContractPositions({
                appId: this.appId,
                network: this.network,
                groupIds: [this.groupId],
            });
            const balances = yield Promise.all(contractPositions.map((contractPosition) => __awaiter(this, void 0, void 0, function* () {
                const bribeContract = multicall.wrap(this.getContract(contractPosition.address));
                const tokens = yield Promise.all(contractPosition.tokens.map((bribeToken) => __awaiter(this, void 0, void 0, function* () {
                    const balancesPerBribePromises = veTokenIds.map((id) => __awaiter(this, void 0, void 0, function* () { return bribeContract.earned(bribeToken.address, id); }));
                    const balancesPerBribe = yield Promise.all(balancesPerBribePromises);
                    const balancesPerBribeSum = balancesPerBribe.reduce((acc, v) => acc.add(v), ethers_1.BigNumber.from(0));
                    return (0, drill_balance_helper_1.drillBalance)(bribeToken, balancesPerBribeSum.toString());
                })));
                const balanceUSD = (0, lodash_1.sumBy)(tokens, t => t.balanceUSD);
                const balance = Object.assign(Object.assign({}, contractPosition), { tokens, balanceUSD });
                return balance;
            })));
            return balances;
        });
    }
    getRawBalances(address) {
        return __awaiter(this, void 0, void 0, function* () {
            const multicall = this.appToolkit.getMulticall(this.network);
            const escrow = this.contractFactory.velodromeVe({ address: this.veTokenAddress, network: this.network });
            const mcEscrow = multicall.wrap(escrow);
            const veCount = Number(yield mcEscrow.balanceOf(address));
            const veTokenIds = yield Promise.all((0, lodash_1.range)(veCount).map((i) => __awaiter(this, void 0, void 0, function* () { return mcEscrow.tokenOfOwnerByIndex(address, i); })));
            if (veTokenIds.length === 0)
                return [];
            const contractPositions = yield this.appToolkit.getAppContractPositions({
                appId: this.appId,
                network: this.network,
                groupIds: [this.groupId],
            });
            const balances = yield Promise.all(contractPositions.map((contractPosition) => __awaiter(this, void 0, void 0, function* () {
                const bribeContract = multicall.wrap(this.getContract(contractPosition.address));
                const balance = {
                    key: this.appToolkit.getPositionKey(contractPosition),
                    tokens: yield Promise.all(contractPosition.tokens.map((bribeToken) => __awaiter(this, void 0, void 0, function* () {
                        const balancesPerBribePromises = veTokenIds.map((id) => __awaiter(this, void 0, void 0, function* () { return bribeContract.earned(bribeToken.address, id); }));
                        const balancesPerBribe = yield Promise.all(balancesPerBribePromises);
                        const balancesPerBribeSum = balancesPerBribe.reduce((acc, v) => acc.add(v), ethers_1.BigNumber.from(0));
                        return { key: this.appToolkit.getPositionKey(bribeToken), balance: balancesPerBribeSum.toString() };
                    }))),
                };
                return balance;
            })));
            return balances;
        });
    }
};
VotingRewardsContractPositionFetcher = __decorate([
    __param(0, (0, common_1.Inject)(app_toolkit_interface_1.APP_TOOLKIT)),
    __param(1, (0, common_1.Inject)(contracts_1.VelodromeContractFactory)),
    __param(2, (0, common_1.Inject)(velodrome_definitions_resolver_1.VelodromeDefinitionsResolver)),
    __metadata("design:paramtypes", [Object, contracts_1.VelodromeContractFactory,
        velodrome_definitions_resolver_1.VelodromeDefinitionsResolver])
], VotingRewardsContractPositionFetcher);
exports.VotingRewardsContractPositionFetcher = VotingRewardsContractPositionFetcher;
//# sourceMappingURL=velodrome.voting-rewards.contract-position-fetcher.js.map