Skip to content

feat: Feat/crosschain baln #472

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
TRSSMG wants to merge 6 commits into
base: main
Choose a base branch
from
Open

feat: Feat/crosschain baln #472

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
826c64e
NetworkAddress implementation on boosted baln
sagars Dec 11, 2024
33513d2
x methods added
sagars Dec 12, 2024
cd99282
baln crosschain functionalities added with exiting unit tests all pass
sagars Dec 16, 2024
891785b
missing crosschain methods added for Dividends
sagars Dec 17, 2024
02eb26c
recursion correction
sagars Dec 17, 2024
53d27f0
getVotesOfUserV2
sagars Dec 18, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions core-contracts/Bribing/build.gradle
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ dependencies {
implementation Dependencies.javaeeScorex
implementation Dependencies.minimalJson
implementation project(':score-lib')
implementation 'xyz.venture23:xcall-lib:2.1.0'

testImplementation Dependencies.javaeeUnitTest
testImplementation Dependencies.javaeeTokens
Expand Down
67 changes: 50 additions & 17 deletions core-contracts/Bribing/src/main/java/network/balanced/score/core/bribing/BribingImpl.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,10 @@

package network.balanced.score.core.bribing;

import network.balanced.score.lib.utils.Versions;
import foundation.icon.xcall.NetworkAddress;
import network.balanced.score.lib.interfaces.BribingXCall;
import network.balanced.score.lib.interfaces.GovernanceXCall;
import network.balanced.score.lib.utils.*;
import score.*;
import score.annotation.External;

Expand All @@ -29,7 +32,9 @@
import com.eclipsesource.json.JsonArray;
import com.eclipsesource.json.JsonObject;
import com.eclipsesource.json.JsonValue;
import score.annotation.Optional;

import static network.balanced.score.lib.utils.BalancedAddressManager.getXCall;
import static network.balanced.score.lib.utils.Check.*;
import static network.balanced.score.lib.utils.Constants.MICRO_SECONDS_IN_A_DAY;
import static network.balanced.score.lib.utils.Constants.EXA;
Expand All @@ -51,7 +56,9 @@ public class BribingImpl implements Bribing {
//Source->bribeToken->period
public static final BranchDB<String, DictDB<Address, BigInteger>> activePeriod = Context.newBranchDB("activePeriod", BigInteger.class);
//userAddress->Source->bribeToken->timeOfLastClaim
public static final BranchDB<Address, BranchDB<String, DictDB<Address, BigInteger>>> lastUserClaim = Context.newBranchDB("lastUserClaim", BigInteger.class);
//public static final BranchDB<Address, BranchDB<String, DictDB<Address, BigInteger>>> lastUserClaim = Context.newBranchDB("lastUserClaim", BigInteger.class);
public static final NetworkAddressBranchedStringBranchDictDB<String, Address, BigInteger> lastUserClaim = new NetworkAddressBranchedStringBranchDictDB<>("lastUserClaim", BigInteger.class);


public static final BranchDB<String, ArrayDB<Address>> bribesPerSource = Context.newBranchDB("bribesPerSource", Address.class);
public static final BranchDB<Address, ArrayDB<String>> sourcesPerBribe = Context.newBranchDB("sourcesPerBribe", String.class);
Expand All @@ -61,7 +68,9 @@ public class BribingImpl implements Bribing {
private final VarDB<String> currentVersion = Context.newVarDB("version", String.class);
private final VarDB<BigInteger> migrationPeriod = Context.newVarDB("migration_period", BigInteger.class);

private class SourceStatus {
public static String NATIVE_NID;

private static class SourceStatus {
BigInteger period;
BigInteger bribesPerToken;
}
Expand All @@ -78,6 +87,7 @@ public BribingImpl(Address rewards) {
Context.revert("Can't Update same version of code");
}
this.currentVersion.set(Versions.BRIBING);
NATIVE_NID = Context.call(String.class, getXCall(), "getNetworkId");
}

@External(readonly=true)
Expand Down Expand Up @@ -148,7 +158,14 @@ public BigInteger getFutureBribe(String source, Address bribeToken, BigInteger p

@External(readonly=true)
public BigInteger claimable(Address user, String source, Address bribeToken) {
return getBribesAmount(user, source, bribeToken, true);
String networkAddress = new NetworkAddress(NATIVE_NID, user).toString();
return getBribesAmount(networkAddress, source, bribeToken, true);
}

@External(readonly=true)
public BigInteger xClaimable(String user, String source, Address bribeToken) {
String networkAddress = NetworkAddress.valueOf(user, NATIVE_NID).toString();
return getBribesAmount(networkAddress, source, bribeToken, true);
}

@External
Expand All @@ -159,13 +176,30 @@ public void updatePeriod(String source, Address bribeToken) {
@External
public void claimBribe(String source, Address bribeToken) {
Address user = Context.getCaller();
BigInteger amount = getBribesAmount(user, source, bribeToken, false);
Context.require(amount.compareTo(BigInteger.ZERO) > 0, user.toString() + " has no bribe in " + bribeToken.toString() + " to claim for source: " + source);
String networkAddress = new NetworkAddress(NATIVE_NID, user).toString();
BigInteger amount = getBribesAmount(networkAddress, source, bribeToken, false);
Context.require(amount.compareTo(BigInteger.ZERO) > 0, user + " has no bribe in " + bribeToken.toString() + " to claim for source: " + source);
BigInteger prevClaims = claimsPerSource.at(source).getOrDefault(bribeToken, BigInteger.ZERO);
claimsPerSource.at(source).set(bribeToken, prevClaims.add(amount));
Context.call(bribeToken, "transfer", user, amount, new byte[0]);
}

@External
public void handleCallMessage(String _from, byte[] _data, @Optional String[] _protocols) {
Check.checkStatus();
only(getXCall());
XCallUtils.verifyXCallProtocols(_from, _protocols);
BribingXCall.process(this, _from, _data);
}

public void xClaimTo(String from, String source, Address bribeToken) {
BigInteger amount = getBribesAmount(from, source, bribeToken, false);
Context.require(amount.compareTo(BigInteger.ZERO) > 0, from + " has no bribe in " + bribeToken.toString() + " to claim for source: " + source);
BigInteger prevClaims = claimsPerSource.at(source).getOrDefault(bribeToken, BigInteger.ZERO);
claimsPerSource.at(source).set(bribeToken, prevClaims.add(amount));
TokenTransfer.transfer(bribeToken, from, amount);
}

@External
public void tokenFallback(Address _from, BigInteger _value, byte[] _data) {
Address token = Context.getCaller();
Expand Down Expand Up @@ -219,27 +253,26 @@ private void scheduledBribes(String source, Address bribeToken, BigInteger total
add(source, bribeToken);
}

private BigInteger getBribesAmount(Address user, String source, Address bribeToken, boolean readonly) {
private BigInteger getBribesAmount(String user, String source, Address bribeToken, boolean readonly) {
SourceStatus status = updateSource(source, bribeToken, readonly);
DictDB<Address, BigInteger> lastUserClaim = BribingImpl.lastUserClaim.at(user).at(source);
if (lastUserClaim.getOrDefault(bribeToken, BigInteger.ZERO).compareTo(status.period) >= 0) {
NetworkAddress userNetworkAddress = NetworkAddress.valueOf(user, NATIVE_NID);
BigInteger lastUserClaimAmount = BribingImpl.lastUserClaim.getOrDefault(userNetworkAddress, source, bribeToken, BigInteger.ZERO);
if (lastUserClaimAmount.compareTo(status.period) >= 0) {
return BigInteger.ZERO;
}

if (!readonly) {
lastUserClaim.set(bribeToken, status.period);
lastUserClaim.set(userNetworkAddress, source, bribeToken, status.period);
}

BigInteger lastVote = Context.call(BigInteger.class, rewards.get(), "getLastUserVote", user, source);
BigInteger lastVote = Context.call(BigInteger.class, rewards.get(), "getLastUserVoteV2", user, source);

if (lastVote.compareTo(status.period) >= 0) {
return BigInteger.ZERO;
}

BigInteger bias = calculateUserBias(user, source);
BigInteger amount = bias.multiply(status.bribesPerToken).divide(EXA);

return amount;
return bias.multiply(status.bribesPerToken).divide(EXA);
}

private SourceStatus updateSource(String source, Address bribeToken, boolean readOnly) {
Expand Down Expand Up @@ -298,12 +331,12 @@ public BigInteger getSourceBias(String source, BigInteger period) {
}


public Map<String, BigInteger> getUserSlope(Address user, String source) {
Map<String, BigInteger> userSlope = (Map<String, BigInteger>)Context.call(rewards.get(), "getUserSlope", user, source);
public Map<String, BigInteger> getUserSlope(String user, String source) {
Map<String, BigInteger> userSlope = (Map<String, BigInteger>)Context.call(rewards.get(), "getUserSlopeV2", user, source);
return userSlope;
}

public BigInteger calculateUserBias(Address user, String source) {
public BigInteger calculateUserBias(String user, String source) {
BigInteger period = getCurrentPeriod();
Map<String, BigInteger> userSlope = getUserSlope(user, source);
BigInteger end = userSlope.get("end");
Expand Down
33 changes: 17 additions & 16 deletions ...-contracts/Bribing/src/test/java/network/balanced/score/core/bribing/BribingImplTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@
import java.math.BigInteger;
import java.util.Map;

import foundation.icon.xcall.NetworkAddress;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.function.Executable;
Expand Down Expand Up @@ -55,9 +56,9 @@ public void addReward() {
// Act
addBribe(source, amount);

when(rewards.mock.getLastUserVote(user.getAddress(), source)).thenReturn(startPeriod.add(BigInteger.ONE));
when(rewards.mock.getLastUserVoteV2(new NetworkAddress(NATIVE_NID, user.getAddress()).toString(), source)).thenReturn(startPeriod.add(BigInteger.ONE));

doReturn(userWeight).when(bribingSpy).calculateUserBias(user.getAddress(), source);
doReturn(userWeight).when(bribingSpy).calculateUserBias(new NetworkAddress(NATIVE_NID, user.getAddress()).toString(), source);

assertEquals(BigInteger.ZERO, bribing.call("claimable", user.getAddress(), source, bribeToken.getAddress()));

Expand Down Expand Up @@ -103,8 +104,8 @@ public void claimBribe() {
addBribe(source, amount);

// Assert
when(rewards.mock.getLastUserVote(user.getAddress(), source)).thenReturn(startPeriod.subtract(BigInteger.ONE));
doReturn(userWeight).when(bribingSpy).calculateUserBias(user.getAddress(), source);
when(rewards.mock.getLastUserVoteV2(new NetworkAddress(NATIVE_NID, user.getAddress()).toString(), source)).thenReturn(startPeriod.subtract(BigInteger.ONE));
doReturn(userWeight).when(bribingSpy).calculateUserBias(new NetworkAddress(NATIVE_NID, user.getAddress()).toString(), source);

sm.getBlock().increase(WEEK);
doReturn(totalWeight).when(bribingSpy).getSourceBias(source, getPeriod());
Expand All @@ -131,8 +132,8 @@ public void claimBribe_carryUnclaimed() {
addBribe(source, amount);

// Assert
when(rewards.mock.getLastUserVote(user.getAddress(), source)).thenReturn(startPeriod.add(BigInteger.ONE));
doReturn(userWeight).when(bribingSpy).calculateUserBias(user.getAddress(), source);
when(rewards.mock.getLastUserVoteV2(new NetworkAddress(NATIVE_NID, user.getAddress()).toString(), source)).thenReturn(startPeriod.add(BigInteger.ONE));
doReturn(userWeight).when(bribingSpy).calculateUserBias(new NetworkAddress(NATIVE_NID, user.getAddress()).toString(), source);

sm.getBlock().increase(WEEK);
doReturn(totalWeight).when(bribingSpy).getSourceBias(source, getPeriod());
Expand Down Expand Up @@ -168,8 +169,8 @@ public void claimBribe_twice() {
// Act
addBribe(source, amount);

when(rewards.mock.getLastUserVote(user.getAddress(), source)).thenReturn(startPeriod.subtract(BigInteger.ONE));
doReturn(userWeight).when(bribingSpy).calculateUserBias(user.getAddress(), source);
when(rewards.mock.getLastUserVoteV2(new NetworkAddress(NATIVE_NID, user.getAddress()).toString(), source)).thenReturn(startPeriod.subtract(BigInteger.ONE));
doReturn(userWeight).when(bribingSpy).calculateUserBias(new NetworkAddress(NATIVE_NID, user.getAddress()).toString(), source);

sm.getBlock().increase(WEEK);
doReturn(totalWeight).when(bribingSpy).getSourceBias(source, getPeriod());;
Expand Down Expand Up @@ -210,10 +211,10 @@ public void scheduledBribes() {
scheduledBribes(source, total, amounts);

// Assert
when(rewards.mock.getLastUserVote(user1.getAddress(), source)).thenReturn(startPeriod.subtract(BigInteger.ONE));
doReturn(userWeight).when(bribingSpy).calculateUserBias(user1.getAddress(), source);
when(rewards.mock.getLastUserVote(user2.getAddress(), source)).thenReturn(startPeriod.subtract(BigInteger.ONE));
doReturn(userWeight).when(bribingSpy).calculateUserBias(user2.getAddress(), source);
when(rewards.mock.getLastUserVoteV2(new NetworkAddress(NATIVE_NID, user1.getAddress()).toString(), source)).thenReturn(startPeriod.subtract(BigInteger.ONE));
doReturn(userWeight).when(bribingSpy).calculateUserBias(new NetworkAddress(NATIVE_NID, user1.getAddress()).toString(), source);
when(rewards.mock.getLastUserVoteV2(new NetworkAddress(NATIVE_NID, user2.getAddress()).toString(), source)).thenReturn(startPeriod.subtract(BigInteger.ONE));
doReturn(userWeight).when(bribingSpy).calculateUserBias(new NetworkAddress(NATIVE_NID, user2.getAddress()).toString(), source);

for (int i = 0; i < 4; i++) {
sm.getBlock().increase(WEEK);
Expand Down Expand Up @@ -286,10 +287,10 @@ public void liveDataTest() {
"slope", new BigInteger("2c9bb4a", 16)
);

doReturn(user1Slope).when(bribingSpy).getUserSlope(user1.getAddress(), source);
doReturn(user2Slope).when(bribingSpy).getUserSlope(user2.getAddress(), source);
when(rewards.mock.getLastUserVote(user1.getAddress(), source)).thenReturn(startPeriod.subtract(BigInteger.ONE));
when(rewards.mock.getLastUserVote(user2.getAddress(), source)).thenReturn(startPeriod.subtract(BigInteger.ONE));
doReturn(user1Slope).when(bribingSpy).getUserSlope(new NetworkAddress(NATIVE_NID, user1.getAddress()).toString(), source);
doReturn(user2Slope).when(bribingSpy).getUserSlope(new NetworkAddress(NATIVE_NID, user2.getAddress()).toString(), source);
when(rewards.mock.getLastUserVoteV2(new NetworkAddress(NATIVE_NID, user1.getAddress()).toString(), source)).thenReturn(startPeriod.subtract(BigInteger.ONE));
when(rewards.mock.getLastUserVoteV2(new NetworkAddress(NATIVE_NID, user2.getAddress()).toString(), source)).thenReturn(startPeriod.subtract(BigInteger.ONE));

// Act
doReturn(period.subtract(WEEK_IN_MS)).when(bribingSpy).getBlockTime();
Expand Down
16 changes: 15 additions & 1 deletion ...tracts/Bribing/src/test/java/network/balanced/score/core/bribing/BribingImplTestBase.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,12 @@
import network.balanced.score.lib.interfaces.tokens.*;
import network.balanced.score.lib.structs.BalancedAddresses;
import network.balanced.score.lib.test.UnitTest;
import network.balanced.score.lib.test.integration.Balanced;
import network.balanced.score.lib.test.mock.MockBalanced;
import network.balanced.score.lib.test.mock.MockContract;
import network.balanced.score.lib.utils.BalancedAddressManager;
import network.balanced.score.lib.utils.Names;
import score.Context;

import java.math.BigInteger;
import java.util.Map;
Expand All @@ -51,13 +56,22 @@ class BribingImplTestBase extends UnitTest {

protected MockContract<Rewards> rewards;
protected MockContract<IRC2> bribeToken;
protected MockContract<XCall> xCall;
protected MockContract<Governance> governance;
private MockBalanced balanced;

protected Score bribing;
protected BribingImpl bribingSpy;

public final String NATIVE_NID = "0x1.ICON";

protected void setupBase() throws Exception {
rewards = new MockContract<>(RewardsScoreInterface.class, sm, owner);
balanced = new MockBalanced(sm, owner);
rewards = balanced.rewards;
governance = balanced.governance;
bribeToken = new MockContract<>(IRC2ScoreInterface.class, sm, owner);
xCall = balanced.xCall;
when(xCall.mock.getNetworkId()).thenReturn(NATIVE_NID);
bribing = sm.deploy(owner, BribingImpl.class, rewards.getAddress());

bribingSpy = (BribingImpl) spy(bribing.getInstance());
Expand Down
1 change: 0 additions & 1 deletion core-contracts/Dex/build.gradle
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,6 @@ dependencies {
testImplementation Dependencies.mockitoCore
testImplementation Dependencies.mockitoInline


testImplementation project(':test-lib')
testImplementation Dependencies.junitJupiter
testRuntimeOnly Dependencies.junitJupiterEngine
Expand Down
1 change: 1 addition & 0 deletions core-contracts/Dex/src/main/java/network/balanced/score/core/dex/DexImpl.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -144,6 +144,7 @@ public void xTokenFallback(String _from, BigInteger _value, byte[] _data) {
require(_value.compareTo(BigInteger.ZERO) > 0, TAG + ": Invalid token transfer value");

if (method.equals("_deposit")) {

JsonObject params = json.get("params").asObject();
String to = _from;
if (params.get("address") != null) {
Expand Down
1 change: 1 addition & 0 deletions core-contracts/Dividends/build.gradle
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ dependencies {
implementation Dependencies.javaeeScorex
implementation Dependencies.minimalJson
implementation project(':score-lib')
implementation 'xyz.venture23:xcall-lib:2.1.0'

testImplementation Dependencies.javaeeUnitTest
testImplementation Dependencies.javaeeTokens
Expand Down
Loading