Skip to content

Commit ae09ed0

Browse files
committed
Add consistent reporting and a kick method removing icon addresses
1 parent d96c95e commit ae09ed0

3 files changed

Lines changed: 20 additions & 6 deletions

File tree

core-contracts/StakedLP/src/main/java/network/balanced/score/core/stakedlp/StakedLPImpl.java

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,11 @@ private void unstake(BigInteger id, NetworkAddress user, BigInteger value) {
176176

177177
Unstake(user.toString(), id, value);
178178

179-
Context.call(rewards.get(), "updateBalanceAndSupply", poolName, newTotal, user.toString(), newBalance);
179+
if (user.net().equals(NATIVE_NID)) {
180+
Context.call(rewards.get(), "updateBalanceAndSupply", poolName, newTotal, user.account(), newBalance);
181+
} else {
182+
Context.call(rewards.get(), "updateBalanceAndSupply", poolName, newTotal, user.toString(), newBalance);
183+
}
180184

181185
try {
182186
Context.call(dex.get(), "hubTransfer", user.toString(), value, id, new byte[0]);
@@ -209,11 +213,21 @@ public void addDataSource(BigInteger id, String name) {
209213
dataSourceNames.set(id, name);
210214
}
211215

216+
@External
217+
public void kick(String _name, Address _owner) {
218+
BigInteger poolId = dataSourceIds.get(_name);
219+
BigInteger totalSupply = totalStaked(poolId);
220+
BigInteger balance = balanceOf(_owner, poolId);
221+
222+
Context.call(rewards.get(), "updateBalanceAndSupply", _name, totalSupply, _owner.toString(), balance);
223+
Context.call(rewards.get(), "updateBalanceAndSupply", _name, totalSupply, new NetworkAddress(NATIVE_NID, _owner).toString(), 0);
224+
}
225+
212226
@External(readonly = true)
213227
public Map<String, BigInteger> getBalanceAndSupply(String _name, String _owner) {
214228
BigInteger poolId = dataSourceIds.get(_name);
215229
BigInteger totalSupply = totalStaked(poolId);
216-
BigInteger balance = balanceOf(Address.fromString(_owner), poolId);
230+
BigInteger balance = xBalanceOf(_owner, poolId);
217231

218232
Map<String, BigInteger> rewardsData = new HashMap<>();
219233
rewardsData.put("_balance", balance);

core-contracts/StakedLP/src/test/java/network/balanced/score/core/stakedlp/StakedLPTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -318,7 +318,7 @@ void testUnstake() {
318318
assertEquals(totalStakedBalanceBeforeUnstake, stakedLpScore.call("totalStaked",
319319
BigInteger.ONE));
320320
verify(dex.mock).hubTransfer(NetworkAddress.valueOf(alice.getAddress().toString(), NATIVE_NID).toString(), aliceUnstakeAmount, BigInteger.ONE, new byte[0]);
321-
verify(rewards.mock).updateBalanceAndSupply(poolOneName, totalStakedBalanceBeforeUnstake, NetworkAddress.valueOf(alice.getAddress().toString(), NATIVE_NID).toString(),
321+
verify(rewards.mock).updateBalanceAndSupply(poolOneName, totalStakedBalanceBeforeUnstake, alice.getAddress().toString(),
322322
aliceStakedBalance);
323323

324324
// Adjust the values after first unstake
@@ -389,6 +389,6 @@ void testStakeUnstake_unnamedPool() {
389389
balanceAndSupply = (Map<String, BigInteger>) stakedLpScore.call("getBalanceAndSupply", name, alice.getAddress().toString());
390390
assertEquals(BigInteger.valueOf(8), balanceAndSupply.get("_balance"));
391391
assertEquals(BigInteger.valueOf(8), balanceAndSupply.get("_totalSupply"));
392-
verify(rewards.mock).updateBalanceAndSupply(name, balanceAndSupply.get("_totalSupply"), NetworkAddress.valueOf(alice.getAddress().toString(), NATIVE_NID).toString(), balanceAndSupply.get("_balance"));
392+
verify(rewards.mock).updateBalanceAndSupply(name, balanceAndSupply.get("_totalSupply"), alice.getAddress().toString(), balanceAndSupply.get("_balance"));
393393
}
394394
}

score-lib/src/main/java/network/balanced/score/lib/utils/Versions.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,15 +26,15 @@ public class Versions {
2626
public final static String WORKERTOKEN = "v1.0.0";
2727
public final static String BNUSD = "v1.1.1";
2828
public final static String FEEHANDLER = "v1.0.1";
29-
public final static String REWARDS = "v1.2.4";
29+
public final static String REWARDS = "v1.2.5";
3030
public final static String STABILITY = "v1.1.1";
3131
public final static String BALANCEDORACLE = "v1.2.0";
3232
public final static String DAOFUND = "v1.1.4";
3333
public final static String DEX = "v1.1.8";
3434
public final static String GOVERNANCE = "v1.0.2";
3535
public final static String REBALANCING = "v1.0.0";
3636
public final static String ROUTER = "v1.2.4";
37-
public final static String STAKEDLP = "v1.0.9";
37+
public final static String STAKEDLP = "v1.1.0";
3838
public final static String BOOSTED_BALN = "v1.1.0";
3939
public final static String BRIBING = "v1.0.1";
4040
public final static String BALANCED_OTC = "v1.0.0";

0 commit comments

Comments
 (0)