@airgap/icp 0.13.15-beta.5 → 0.13.15-beta.6

package/v1/protocol/ICPProtocol.js

@@ -1,4 +1,27 @@
 "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 __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 __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) {
@@ -21,9 +44,11 @@ const assert_1 = require("@airgap/coinlib-core/utils/assert");
 const array_1 = require("@airgap/coinlib-core/utils/array");
 const crypto_1 = require("@airgap/crypto");
 const module_kit_1 = require("@airgap/module-kit");
-const module_1 = require("../module");
+const governance_1 = require("../types/governance");
 const transaction_1 = require("../types/transaction");
 const account_1 = require("../utils/account");
+const Cbor = __importStar(require("../utils/cbor"));
+const IDL = __importStar(require("../utils/idl"));
 const principal_1 = require("../utils/principal");
 const ICPGovernance_1 = require("./ICPGovernance");
 const ICPImplementation_1 = require("./ICPImplementation");
@@ -75,6 +100,10 @@ class ICPProtocolImpl {
             algorithm: 'secp256k1'
         };
         this.options = createICPProtocolOptions(options.network);
+        this.canisterId = {
+            ledger: this.options.network.ledgerCanisterId,
+            governance: this.options.network.governanceCanisterId
+        };
     }
     getMetadata() {
         return __awaiter(this, void 0, void 0, function* () {
@@ -105,21 +134,19 @@ class ICPProtocolImpl {
                 case transaction_1.ICPActionType.TRANSFER:
                     return (0, ICPImplementation_1.getDetailsFromSignedTransactionTransfer)(encoded, publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.GET_NEURON_INFO:
-                    return (0, ICPGovernance_1.getDetailsFromSignedTransactionGetNeuronInfo)(publicKey.value, this.options.network);
+                    return (0, ICPGovernance_1.getDetailsFromSignedGetNeuronInfo)(publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.TRANSFER_TO_SUBACCOUNT:
                     return (0, ICPGovernance_1.getDetailsFromSignedTransferToSubaccount)(encoded, publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.CLAIM_GOVERNANCE:
                     return (0, ICPGovernance_1.getDetailsFromTransactionClaimGovernance)(publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.FOLLOW_NEURON:
                     return (0, ICPGovernance_1.getDetailsFromSignedFollowNeuron)(publicKey.value, this.options.network);
-                case transaction_1.ICPActionType.INCREASE_STAKE:
-                    throw new Error('TODO: implement');
                 case transaction_1.ICPActionType.DISBURSE:
-                    return (0, ICPGovernance_1.getDetailsFromSignedTransactionRemoveStake)(publicKey.value, this.options.network);
+                    return (0, ICPGovernance_1.getDetailsFromSignedDisburse)(publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.INCREASE_DISSOLVING_DELAY:
                     return (0, ICPGovernance_1.getDetailsFromSignedTransactionIncreaseDelay)(publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.START_DISSOLVING:
-                    return (0, ICPGovernance_1.getDetailsFromSignedTransactionStartDissolving)(publicKey.value, this.options.network);
+                    return (0, ICPGovernance_1.getDetailsFromSignedStartDissolving)(publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.STOP_DISSOLVING:
                     return (0, ICPGovernance_1.getDetailsFromSignedTransactionStopDissolving)(publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.RESTAKE_MATURITY:
@@ -137,19 +164,17 @@ class ICPProtocolImpl {
                 case transaction_1.ICPActionType.TRANSFER:
                     return (0, ICPImplementation_1.getDetailsFromUnsignedTransactionTransfer)(encoded, publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.GET_NEURON_INFO:
-                    return (0, ICPGovernance_1.getDetailsFromUnsignedTransactionGetNeuronInfo)(publicKey.value, this.options.network);
+                    return (0, ICPGovernance_1.getDetailsFromUnsignedGetNeuronInfo)(publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.TRANSFER_TO_SUBACCOUNT:
                     return (0, ICPGovernance_1.getDetailsFromUnsignedTransferToSubaccount)(encoded, publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.CLAIM_GOVERNANCE:
                     return (0, ICPGovernance_1.getDetailsFromUnsignedClaimGovernance)(publicKey.value, this.options.network);
-                case transaction_1.ICPActionType.INCREASE_STAKE:
-                    throw new Error('TODO: implement');
                 case transaction_1.ICPActionType.DISBURSE:
-                    return (0, ICPGovernance_1.getDetailsFromUnsignedTransactionRemoveStake)(publicKey.value, this.options.network);
+                    return (0, ICPGovernance_1.getDetailsFromUnsignedDisburse)(publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.INCREASE_DISSOLVING_DELAY:
                     return (0, ICPGovernance_1.getDetailsFromUnsignedTransactionIncreaseDelay)(publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.START_DISSOLVING:
-                    return (0, ICPGovernance_1.getDetailsFromUnsignedTransactionStartDissolving)(publicKey.value, this.options.network);
+                    return (0, ICPGovernance_1.getDetailsFromUnsignedStartDissolving)(publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.STOP_DISSOLVING:
                     return (0, ICPGovernance_1.getDetailsFromUnsignedTransactionStopDissolving)(publicKey.value, this.options.network);
                 case transaction_1.ICPActionType.FOLLOW_NEURON:
@@ -193,7 +218,7 @@ class ICPProtocolImpl {
                     case transaction_1.ICPActionType.GET_NEURON_INFO:
                         return {
                             actionType,
-                            encoded: yield (0, ICPGovernance_1.signTransactionGetNeuronInfo)(encoded, secretKey.value, this.options.network.governanceCanisterId)
+                            encoded: yield (0, ICPGovernance_1.signGetNeuronInfo)(encoded, secretKey.value, this.options.network.governanceCanisterId)
                         };
                     case transaction_1.ICPActionType.TRANSFER_TO_SUBACCOUNT:
                         return {
@@ -203,39 +228,37 @@ class ICPProtocolImpl {
                     case transaction_1.ICPActionType.CLAIM_GOVERNANCE:
                         return {
                             actionType,
-                            encoded: yield (0, ICPGovernance_1.signClaimGovernance)(encoded, secretKey.value, this.options.network.ledgerCanisterId)
+                            encoded: yield (0, ICPGovernance_1.signClaimGovernance)(encoded, secretKey.value, this.options.network.governanceCanisterId)
                         };
-                    case transaction_1.ICPActionType.INCREASE_STAKE:
-                        throw new Error('TODO: implement');
                     case transaction_1.ICPActionType.DISBURSE:
                         return {
                             actionType,
-                            encoded: yield (0, ICPGovernance_1.signTransactionRemoveStake)(encoded, secretKey.value, this.options.network.ledgerCanisterId)
+                            encoded: yield (0, ICPGovernance_1.signDisburse)(encoded, secretKey.value, this.options.network.governanceCanisterId)
                         };
                     case transaction_1.ICPActionType.INCREASE_DISSOLVING_DELAY:
                         return {
                             actionType,
-                            encoded: yield (0, ICPGovernance_1.signTransactionIncreaseDelay)(encoded, secretKey.value, this.options.network.ledgerCanisterId)
+                            encoded: yield (0, ICPGovernance_1.signTransactionIncreaseDelay)(encoded, secretKey.value, this.options.network.governanceCanisterId)
                         };
                     case transaction_1.ICPActionType.START_DISSOLVING:
                         return {
                             actionType,
-                            encoded: yield (0, ICPGovernance_1.signTransactionStartDissolving)(encoded, secretKey.value, this.options.network.ledgerCanisterId)
+                            encoded: yield (0, ICPGovernance_1.signStartDissolving)(encoded, secretKey.value, this.options.network.governanceCanisterId)
                         };
                     case transaction_1.ICPActionType.STOP_DISSOLVING:
                         return {
                             actionType,
-                            encoded: yield (0, ICPGovernance_1.signTransactionStopDissolving)(encoded, secretKey.value, this.options.network.ledgerCanisterId)
+                            encoded: yield (0, ICPGovernance_1.signTransactionStopDissolving)(encoded, secretKey.value, this.options.network.governanceCanisterId)
                         };
                     case transaction_1.ICPActionType.FOLLOW_NEURON:
                         return {
                             actionType,
-                            encoded: yield (0, ICPGovernance_1.signFollowNeuron)(encoded, secretKey.value, this.options.network.ledgerCanisterId)
+                            encoded: yield (0, ICPGovernance_1.signFollowNeuron)(encoded, secretKey.value, this.options.network.governanceCanisterId)
                         };
                     case transaction_1.ICPActionType.RESTAKE_MATURITY:
                         return {
                             actionType,
-                            encoded: yield (0, ICPGovernance_1.signTransactionRestakeMaturity)(encoded, secretKey.value, this.options.network.ledgerCanisterId)
+                            encoded: yield (0, ICPGovernance_1.signTransactionRestakeMaturity)(encoded, secretKey.value, this.options.network.governanceCanisterId)
                         };
                     default:
                         (0, coinlib_core_1.assertNever)(actionType);
@@ -296,7 +319,16 @@ class ICPProtocolImpl {
     getBalanceOfPublicKey(publicKey) {
         return __awaiter(this, void 0, void 0, function* () {
             const address = (0, ICPImplementation_1.getAddressFromPublicKey)(publicKey.value);
-            return this.getBalanceOfAddress(address);
+            const mainBalance = yield this.getBalanceOfAddress(address);
+            const neuronBalance = yield this.getNeuronBalance(publicKey);
+            const totalBalance = (0, module_kit_1.newAmount)(mainBalance.total)
+                .blockchain(this.units)
+                .toBigNumber()
+                .plus((0, module_kit_1.newAmount)(neuronBalance.total).blockchain(this.units).toBigNumber());
+            return {
+                total: (0, module_kit_1.newAmount)(totalBalance, 'blockchain'),
+                transferable: mainBalance.total
+            };
         });
     }
     getBalanceOfAddress(address) {
@@ -317,7 +349,7 @@ class ICPProtocolImpl {
             }
             else {
                 const feeEstimation = yield this.getTransactionFeeWithPublicKey(publicKey, []);
-                fee = new bignumber_1.default((0, module_kit_1.newAmount)(feeEstimation.medium).value);
+                fee = new bignumber_1.default((0, module_kit_1.newAmount)(feeEstimation).value);
                 if (fee.gte(balanceBn)) {
                     fee = new bignumber_1.default(0);
                 }
@@ -339,7 +371,7 @@ class ICPProtocolImpl {
             //   medium: newAmount(feeDefault.toString(10), 'ICP'),
             //   high: newAmount(feeDefault.toString(10), 'ICP')
             // }
-            return this.feeDefaults;
+            return this.feeDefaults.medium;
         });
     }
     prepareTransactionWithPublicKey(publicKey, details, configuration) {
@@ -355,15 +387,12 @@ class ICPProtocolImpl {
                 throw new errors_1.BalanceError(coinlib_core_1.Domain.ICP, 'not enough balance');
             }
             const amountBn = BigInt((0, module_kit_1.newAmount)(details[0].amount).blockchain(this.units).value);
-            // Transaction object
-            const transaction = {
-                from: (0, ICPImplementation_1.getAddressFromPublicKey)(publicKey.value),
+            // Create unsigned
+            const unsignedTransaction = (0, ICPImplementation_1.createUnsignedTransaction)({
                 to: details[0].to,
                 amount: amountBn,
                 fee: feeBn
-            };
-            // Create unsigned
-            const unsignedTransaction = (0, ICPImplementation_1.createUnsignedTransaction)(transaction);
+            });
             return (0, module_kit_1.newUnsignedTransaction)({
                 transactions: [
                     {
@@ -379,10 +408,87 @@ class ICPProtocolImpl {
     // Search for : Transfer ICP from the caller to the destination accountIdentifier. Returns the index of the block containing the tx if it was successful.
     broadcastTransaction(transaction) {
         return __awaiter(this, void 0, void 0, function* () {
-            const txHashes = yield Promise.all(transaction.transactions.map(({ actionType, encoded }) => (0, ICPImplementation_1.broadcastTransaction)(encoded, this.options.network.rpcUrl, this.options.network.governanceCanisterId, transaction_1.ICP_REQUEST_TYPE_PER_ACTION[actionType])));
-            return txHashes[0];
+            for (const { actionType, encoded } of transaction.transactions) {
+                const requestType = transaction_1.ICP_REQUEST_TYPE_PER_ACTION[actionType];
+                if (requestType !== 'call')
+                    continue;
+                yield (0, ICPImplementation_1.broadcastTransaction)(encoded, this.options.network.rpcUrl, this.canisterId[transaction_1.ICP_CANISTER_TYPE_PER_ACTION[actionType]], transaction_1.ICP_REQUEST_TYPE_PER_ACTION[actionType]);
+            }
+            return '';
         });
     }
+    sendQuery(transaction) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return Promise.all(transaction.transactions
+                .filter(({ actionType }) => transaction_1.ICP_REQUEST_TYPE_PER_ACTION[actionType] === 'query')
+                .map(({ actionType, encoded }) => __awaiter(this, void 0, void 0, function* () {
+                const response = yield (0, ICPImplementation_1.broadcastTransaction)(encoded, this.options.network.rpcUrl, this.canisterId[transaction_1.ICP_CANISTER_TYPE_PER_ACTION[actionType]], transaction_1.ICP_REQUEST_TYPE_PER_ACTION[actionType]);
+                const queryResponse = Cbor.decode(response);
+                if (queryResponse.status == "rejected" /* QueryResponseStatus.Rejected */) {
+                    throw new errors_1.NetworkError(coinlib_core_1.Domain.ICP, {}, `Error (${queryResponse.reject_code}) ${queryResponse.reject_message}`);
+                }
+                return this.decodeQueryResponse(actionType, queryResponse.reply.arg);
+            })));
+        });
+    }
+    decodeQueryResponse(actionType, response) {
+        switch (actionType) {
+            case transaction_1.ICPActionType.GET_NEURON_INFO:
+                const NeuronId = IDL.Record({ id: IDL.Nat64 });
+                const BallotInfo = IDL.Record({
+                    vote: IDL.Int32,
+                    proposal_id: IDL.Opt(NeuronId)
+                });
+                const DissolveState = IDL.Variant({
+                    DissolveDelaySeconds: IDL.Nat64,
+                    WhenDissolvedTimestampSeconds: IDL.Nat64
+                });
+                const Followees = IDL.Record({ followees: IDL.Vec(NeuronId) });
+                const NeuronStakeTransfer = IDL.Record({
+                    to_subaccount: IDL.Vec(IDL.Nat8),
+                    neuron_stake_e8s: IDL.Nat64,
+                    from: IDL.Opt(IDL.Principal),
+                    memo: IDL.Nat64,
+                    from_subaccount: IDL.Vec(IDL.Nat8),
+                    transfer_timestamp: IDL.Nat64,
+                    block_height: IDL.Nat64
+                });
+                const KnownNeuronData = IDL.Record({
+                    name: IDL.Text,
+                    description: IDL.Opt(IDL.Text)
+                });
+                const Neuron = IDL.Record({
+                    id: IDL.Opt(NeuronId),
+                    staked_maturity_e8s_equivalent: IDL.Opt(IDL.Nat64),
+                    controller: IDL.Opt(IDL.Principal),
+                    recent_ballots: IDL.Vec(BallotInfo),
+                    kyc_verified: IDL.Bool,
+                    not_for_profit: IDL.Bool,
+                    maturity_e8s_equivalent: IDL.Nat64,
+                    cached_neuron_stake_e8s: IDL.Nat64,
+                    created_timestamp_seconds: IDL.Nat64,
+                    auto_stake_maturity: IDL.Opt(IDL.Bool),
+                    aging_since_timestamp_seconds: IDL.Nat64,
+                    hot_keys: IDL.Vec(IDL.Principal),
+                    account: IDL.Vec(IDL.Nat8),
+                    joined_community_fund_timestamp_seconds: IDL.Opt(IDL.Nat64),
+                    dissolve_state: IDL.Opt(DissolveState),
+                    followees: IDL.Vec(IDL.Tuple(IDL.Int32, Followees)),
+                    neuron_fees_e8s: IDL.Nat64,
+                    transfer: IDL.Opt(NeuronStakeTransfer),
+                    known_neuron_data: IDL.Opt(KnownNeuronData),
+                    spawn_at_timestamp_seconds: IDL.Opt(IDL.Nat64)
+                });
+                const GovernanceError = IDL.Record({
+                    error_message: IDL.Text,
+                    error_type: IDL.Int32
+                });
+                const Result = IDL.Variant({ Ok: Neuron, Err: GovernanceError });
+                return IDL.decode([Result], response)[0];
+            default:
+                throw new errors_1.UnsupportedError(coinlib_core_1.Domain.ICP, `Decode response for action type ${actionType} is not supported`);
+        }
+    }
     // DELEGATION
     getDefaultDelegatee() {
         var _a, _b, _c;
@@ -406,19 +512,22 @@ class ICPProtocolImpl {
         });
     }
     getDelegateeDetails(address, data) {
-        var _a, _b;
+        var _a, _b, _c, _d;
         return __awaiter(this, void 0, void 0, function* () {
             const partialNeuron = yield (0, ICPImplementation_1.getNeuronInfo)(address, this.options.network.rpcUrl, this.options.network.governanceCanisterId);
+            const neuronFolowees = (data === null || data === void 0 ? void 0 : data.neuron) ? this.getNeuronFollowees(data.neuron) : undefined;
             return {
                 name: ((_b = (_a = partialNeuron === null || partialNeuron === void 0 ? void 0 : partialNeuron.known_neuron_data) === null || _a === void 0 ? void 0 : _a[0]) === null || _b === void 0 ? void 0 : _b.name) || '',
-                address: address
+                address: address,
+                status: neuronFolowees ? (neuronFolowees.includes(address) ? 'followed' : 'not-followed') : 'unknown',
+                votingPower: (_d = (_c = partialNeuron === null || partialNeuron === void 0 ? void 0 : partialNeuron.voting_power) === null || _c === void 0 ? void 0 : _c.toString()) !== null && _d !== void 0 ? _d : '0'
             };
         });
     }
     isPublicKeyDelegating(publicKey, data) {
         var _a, _b;
         return __awaiter(this, void 0, void 0, function* () {
-            const partialNeuron = yield this.getNeuron(publicKey.value);
+            const partialNeuron = yield this.getNeuron(publicKey);
             if (partialNeuron === undefined) {
                 return false;
             }
@@ -444,17 +553,26 @@ class ICPProtocolImpl {
         });
     }
     getDelegatorDetailsFromPublicKey(publicKey, data) {
+        var _a, _b, _c, _d;
         return __awaiter(this, void 0, void 0, function* () {
             const address = yield this.getAddressFromPublicKey(publicKey);
             const availableBalance = yield this.getBalanceOfAddress(address);
             const followees = (data === null || data === void 0 ? void 0 : data.neuron) ? this.getNeuronFollowees(data.neuron) : [];
-            const availableActions = yield this.getAvailableDelegatorActions(publicKey.value, data === null || data === void 0 ? void 0 : data.neuron);
+            const availableActions = yield this.getAvailableDelegatorActions(publicKey, data === null || data === void 0 ? void 0 : data.neuron);
+            const neuronBalance = yield this.getNeuronBalance(publicKey);
+            const partialNeuron = yield this.getNeuron(publicKey);
             const totalBalance = (0, module_kit_1.newAmount)(availableBalance.total).blockchain(this.units);
+            const subaccountBalance = (0, module_kit_1.newAmount)(neuronBalance.total).blockchain(this.units);
             return {
                 address,
                 balance: totalBalance.value,
                 delegatees: followees,
-                availableActions: availableActions
+                availableActions: availableActions,
+                subaccountBalance: subaccountBalance.value,
+                stake: (_a = partialNeuron === null || partialNeuron === void 0 ? void 0 : partialNeuron.stake_e8s) === null || _a === void 0 ? void 0 : _a.toString(),
+                votingPower: (_b = partialNeuron === null || partialNeuron === void 0 ? void 0 : partialNeuron.voting_power) === null || _b === void 0 ? void 0 : _b.toString(),
+                age: (_c = partialNeuron === null || partialNeuron === void 0 ? void 0 : partialNeuron.age_seconds) === null || _c === void 0 ? void 0 : _c.toString(),
+                dissolveDelay: (_d = partialNeuron === null || partialNeuron === void 0 ? void 0 : partialNeuron.dissolve_delay_seconds) === null || _d === void 0 ? void 0 : _d.toString()
             };
         });
     }
@@ -469,7 +587,7 @@ class ICPProtocolImpl {
                 return Promise.reject('Multiple validators for a single delegation are not supported.');
             }
             const validator = delegatees[0];
-            const results = yield Promise.all([this.getDelegatorDetailsFromPublicKey(publicKey), this.getDelegateeDetails(validator)]);
+            const results = yield Promise.all([this.getDelegatorDetailsFromPublicKey(publicKey, data), this.getDelegateeDetails(validator, data)]);
             const delegatorDetails = results[0];
             const validatorDetails = results[1];
             return {
@@ -483,58 +601,73 @@ class ICPProtocolImpl {
             throw new errors_1.UnsupportedError(coinlib_core_1.Domain.ICP, '`getDelegationDetailsFromAddress` is not supported');
         });
     }
-    getAvailableDelegatorActions(publicKey, neuron) {
+    getAvailableDelegatorActions(publicKey, fullNeuron) {
         return __awaiter(this, void 0, void 0, function* () {
             const actions = [];
-            const hasNeuron = yield this.getNeuronBalance(publicKey).then((balance) => new bignumber_1.default(balance.total.value).gt(0));
-            const claimedGovernance = yield this.getNeuron(publicKey).then((neuron) => neuron !== undefined);
-            const isFollowing = neuron ? this.getNeuronFollowees(neuron).length > 0 : false;
-            const isDissolving = neuron ? this.isNeuronDissolving(neuron) : false;
-            const canDisburse = neuron ? this.canNeuronDisburse(neuron) : false;
-            if (!hasNeuron) {
+            const neuronBalance = yield this.getNeuronBalance(publicKey);
+            const partialNeuron = yield this.getNeuron(publicKey);
+            const isStaking = new bignumber_1.default(neuronBalance.total.value).gt(0);
+            const hasNeuron = partialNeuron !== undefined;
+            const isNeuronUpToDate = hasNeuron && partialNeuron.stake_e8s.toString() === (0, module_kit_1.newAmount)(neuronBalance.total).blockchain(this.units).value;
+            const isFollowing = fullNeuron ? this.getNeuronFollowees(fullNeuron).length > 0 : false;
+            const isDissolving = fullNeuron ? this.isNeuronDissolving(fullNeuron) : false;
+            const canDisburse = partialNeuron ? this.canNeuronDisburse(partialNeuron) : false;
+            if (!isStaking && !hasNeuron) {
                 actions.push({
-                    type: module_1.ICPStakingActionType.CREATE_NEURON,
+                    type: governance_1.ICPStakingActionType.CREATE_NEURON,
                     args: ['followee', 'amount', 'delay']
                 });
             }
-            if (hasNeuron && !claimedGovernance) {
+            if (isStaking && !hasNeuron) {
                 actions.push({
-                    type: module_1.ICPStakingActionType.CLAIM_GOVERNANCE,
+                    type: governance_1.ICPStakingActionType.CLAIM_GOVERNANCE,
                     args: ['followee', 'amount', 'delay']
                 });
             }
-            if (claimedGovernance && !isFollowing && neuron !== undefined) {
+            if (!isNeuronUpToDate) {
+                actions.push({
+                    type: governance_1.ICPStakingActionType.REFRESH_NEURON,
+                    args: ['amount']
+                });
+            }
+            if (isNeuronUpToDate && fullNeuron === undefined) {
+                actions.push({
+                    type: governance_1.ICPStakingActionType.INCREASE_STAKE,
+                    args: ['amount']
+                });
+            }
+            if (isNeuronUpToDate && !isFollowing && fullNeuron !== undefined) {
                 actions.push({
-                    type: module_1.ICPStakingActionType.FOLLOW_NEURON,
+                    type: governance_1.ICPStakingActionType.FOLLOW_NEURON,
                     args: ['followee', 'amount', 'delay']
                 });
             }
-            if (isFollowing) {
+            if (isFollowing && fullNeuron !== undefined) {
                 actions.push({
-                    type: module_1.ICPStakingActionType.CHANGE_FOLLOWEE,
+                    type: governance_1.ICPStakingActionType.CHANGE_FOLLOWEE,
                     args: ['followee', 'amount', 'delay']
                 });
             }
-            if (claimedGovernance) {
+            if (hasNeuron) {
                 actions.push({
-                    type: module_1.ICPStakingActionType.GET_STAKING_DETAILS
+                    type: governance_1.ICPStakingActionType.GET_STAKING_DETAILS
                 });
             }
-            if (claimedGovernance && !isDissolving) {
+            if (isStaking && hasNeuron && !isDissolving && fullNeuron !== undefined) {
                 actions.push({
-                    type: module_1.ICPStakingActionType.INCREASE_DISSOLVING_DELAY
+                    type: governance_1.ICPStakingActionType.INCREASE_DISSOLVING_DELAY
                 }, {
-                    type: module_1.ICPStakingActionType.START_DISSOLVING
+                    type: governance_1.ICPStakingActionType.START_DISSOLVING
                 });
             }
             if (isDissolving) {
                 actions.push({
-                    type: module_1.ICPStakingActionType.STOP_DISSOLVING
+                    type: governance_1.ICPStakingActionType.STOP_DISSOLVING
                 });
             }
             if (canDisburse) {
                 actions.push({
-                    type: module_1.ICPStakingActionType.DISBURSE
+                    type: governance_1.ICPStakingActionType.DISBURSE
                 });
             }
             return actions;
@@ -543,47 +676,121 @@ class ICPProtocolImpl {
     prepareDelegatorActionFromPublicKey(publicKey, type, data) {
         return __awaiter(this, void 0, void 0, function* () {
             switch (type) {
-                case module_1.ICPStakingActionType.CREATE_NEURON:
-                    (0, assert_1.assertFields)(`${type} action`, data, 'followee', 'amount', 'delay');
-                    return [
-                        (0, module_kit_1.newUnsignedTransaction)({
-                            transactions: yield Promise.all([
-                                (0, ICPGovernance_1.prepareTransferToSubaccount)(publicKey.value, this.options.network.ledgerCanisterId, BigInt(data.amount), BigInt((0, module_kit_1.newAmount)(this.feeDefaults.medium).blockchain(this.units).value)),
-                                (0, ICPGovernance_1.prepareClaimGovernance)(publicKey.value)
-                            ])
-                        })
-                    ];
+                case governance_1.ICPStakingActionType.CREATE_NEURON:
+                case governance_1.ICPStakingActionType.INCREASE_STAKE:
+                    (0, assert_1.assertFields)(`${type} action`, data, 'amount');
+                    return this.prepareTransferToNeuron(publicKey, data.amount);
+                case governance_1.ICPStakingActionType.CLAIM_GOVERNANCE:
+                case governance_1.ICPStakingActionType.REFRESH_NEURON:
+                    (0, assert_1.assertFields)(`${type} action`, data, 'amount');
+                    return this.prepareClaimOrRefreshNeuron(publicKey, data.amount);
+                case governance_1.ICPStakingActionType.GET_STAKING_DETAILS:
+                    return this.prepareGetStakingDetails();
+                case governance_1.ICPStakingActionType.START_DISSOLVING:
+                    return this.prepareStartDissolving(publicKey);
+                case governance_1.ICPStakingActionType.DISBURSE:
+                    return this.prepareDisburse(publicKey);
                 default:
                     return Promise.reject(`Delegator action type ${type} is not supported.`);
             }
         });
     }
-    getKnownNeurons() {
+    prepareTransferToNeuron(publicKey, amount) {
         return __awaiter(this, void 0, void 0, function* () {
-            return (0, ICPImplementation_1.listKnownNeurons)(this.options.network.rpcUrl, this.options.network.governanceCanisterId);
+            const amountBigInt = BigInt(amount);
+            const feeBigInt = BigInt((0, module_kit_1.newAmount)(this.feeDefaults.medium).blockchain(this.units).value);
+            return [
+                (0, module_kit_1.newUnsignedTransaction)({
+                    transactions: yield Promise.all([
+                        (0, ICPGovernance_1.prepareTransferToSubaccount)(publicKey.value, this.options.network.governanceCanisterId, amountBigInt, feeBigInt),
+                        (0, ICPGovernance_1.prepareClaimOrRefreshNeuron)(publicKey.value)
+                    ])
+                })
+            ];
+        });
+    }
+    prepareClaimOrRefreshNeuron(publicKey, amount) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const amountBigInt = BigInt(amount);
+            const feeBigInt = BigInt((0, module_kit_1.newAmount)(this.feeDefaults.medium).blockchain(this.units).value);
+            return [
+                (0, module_kit_1.newUnsignedTransaction)({
+                    transactions: (yield Promise.all([
+                        amountBigInt > 0
+                            ? (0, ICPGovernance_1.prepareTransferToSubaccount)(publicKey.value, this.options.network.governanceCanisterId, amountBigInt, feeBigInt)
+                            : Promise.resolve(undefined),
+                        (0, ICPGovernance_1.prepareClaimOrRefreshNeuron)(publicKey.value)
+                    ])).filter((tx) => tx !== undefined)
+                })
+            ];
+        });
+    }
+    prepareGetStakingDetails() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return [
+                (0, module_kit_1.newUnsignedTransaction)({
+                    transactions: [yield (0, ICPGovernance_1.prepareGetNeuronInfo)()]
+                })
+            ];
         });
     }
-    getNeuronVotingPower(neuronId) {
+    prepareStartDissolving(publicKey) {
         return __awaiter(this, void 0, void 0, function* () {
-            const neuron = yield (0, ICPImplementation_1.getNeuronInfo)(neuronId, this.options.network.rpcUrl, this.options.network.governanceCanisterId);
-            if (neuron === undefined) {
-                return '0';
-            }
-            return new bignumber_1.default(neuron.voting_power.toString()).shiftedBy(-this.units.ICP.decimals).toString();
+            return [
+                (0, module_kit_1.newUnsignedTransaction)({
+                    transactions: [yield (0, ICPGovernance_1.prepareStartDissolving)(publicKey.value)]
+                })
+            ];
+        });
+    }
+    prepareDisburse(publicKey) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return [
+                (0, module_kit_1.newUnsignedTransaction)({
+                    transactions: [yield (0, ICPGovernance_1.prepareDisburse)(publicKey.value)]
+                })
+            ];
+        });
+    }
+    getMaxStakingAmount(publicKey) {
+        var _a, _b;
+        return __awaiter(this, void 0, void 0, function* () {
+            const balance = yield this.getBalanceOfPublicKey(publicKey);
+            const { subAccount } = (0, ICPImplementation_1.getFixedSubaccountFromPublicKey)(publicKey.value);
+            const accountIdentifier = account_1.AccountIdentifier.fromPrincipal({
+                principal: principal_1.Principal.from(this.options.network.ledgerCanisterId),
+                subAccount: subAccount
+            });
+            const fee = yield this.getTransactionFeeWithPublicKey(publicKey, [
+                {
+                    to: accountIdentifier.toHex(),
+                    amount: (_a = balance.transferable) !== null && _a !== void 0 ? _a : balance.total
+                }
+            ]);
+            const maxAmount = (0, module_kit_1.newAmount)((_b = balance.transferable) !== null && _b !== void 0 ? _b : balance.total)
+                .blockchain(this.units)
+                .toBigNumber()
+                .minus((0, module_kit_1.newAmount)(fee).blockchain(this.units).value);
+            return (0, module_kit_1.newAmount)(maxAmount, 'blockchain');
+        });
+    }
+    getKnownNeurons() {
+        return __awaiter(this, void 0, void 0, function* () {
+            return (0, ICPImplementation_1.listKnownNeurons)(this.options.network.rpcUrl, this.options.network.governanceCanisterId);
         });
     }
     getNeuron(publicKey) {
         return __awaiter(this, void 0, void 0, function* () {
-            const neuron = yield (0, ICPImplementation_1.getNeuronInfoBySubAccount)(publicKey, this.options.network.rpcUrl, this.options.network.governanceCanisterId);
+            const neuron = yield (0, ICPImplementation_1.getNeuronInfoBySubAccount)(publicKey.value, this.options.network.rpcUrl, this.options.network.governanceCanisterId);
             return neuron;
         });
     }
     getNeuronBalance(publicKey) {
         return __awaiter(this, void 0, void 0, function* () {
             // Create subaccount from publicKey
-            const { subAccount, nonce: _ } = (0, ICPImplementation_1.getFixedSubaccountFromPublicKey)(publicKey);
+            const { subAccount, nonce: _ } = (0, ICPImplementation_1.getFixedSubaccountFromPublicKey)(publicKey.value);
             const accountIdentifier = account_1.AccountIdentifier.fromPrincipal({
-                principal: principal_1.Principal.from(this.options.network.ledgerCanisterId),
+                principal: principal_1.Principal.from(this.options.network.governanceCanisterId),
                 subAccount: subAccount
             });
             return this.getBalanceOfAddress(accountIdentifier.toHex());
@@ -599,10 +806,7 @@ class ICPProtocolImpl {
             dissolveState.WhenDissolvedTimestampSeconds > BigInt(0));
     }
     canNeuronDisburse(neuron) {
-        const dissolveState = neuron.dissolve_state[0];
-        return (dissolveState !== undefined &&
-            'WhenDissolvedTimestampSeconds' in dissolveState &&
-            dissolveState.WhenDissolvedTimestampSeconds === BigInt(0));
+        return neuron.dissolve_delay_seconds === BigInt(0) && neuron.stake_e8s > BigInt(0);
     }
 }
 exports.ICPProtocolImpl = ICPProtocolImpl;