fix: proxy versions, KlerosCoreSnapshotProxy

Author: jaybuidl

contracts/src/arbitration/KlerosCore.sol

@@ -8,7 +8,7 @@ import "./KlerosCoreBase.sol";
 /// @notice KlerosCore implementation with new StakeController architecture for testing environments
 contract KlerosCore is KlerosCoreBase {
     /// @notice Version of the implementation contract
-    string public constant override version = "0.0.1";
+    string public constant override version = "0.10.0";
 
     // ************************************* //
     // *            Constructor            * //

contracts/src/arbitration/KlerosCoreBase.sol

@@ -1071,7 +1071,7 @@ abstract contract KlerosCoreBase is IArbitratorV2, Initializable, UUPSProxiable
         emit DisputeKitEnabled(_courtID, _disputeKitID, _enable);
     }
 
-    /// @dev If called only once then set _onError to Revert, otherwise for batch staking set it to Return
+    /// @dev It may revert if the stake change in invalid.
     /// @param _account The account to set the stake for.
     /// @param _courtID The ID of the court to set the stake for.
     /// @param _newStake The new stake.

contracts/src/arbitration/KlerosCoreNeo.sol

@@ -3,12 +3,11 @@
 pragma solidity 0.8.24;
 
 import "./KlerosCoreBase.sol";
-import "@openzeppelin/contracts/token/ERC721/IERC721.sol";
 
 /// @title KlerosCoreNeo
 /// @notice KlerosCore with whitelisted arbitrables
 contract KlerosCoreNeo is KlerosCoreBase {
-    string public constant override version = "0.0.1";
+    string public constant override version = "0.10.0";
 
     // ************************************* //
     // *             Storage               * //

contracts/src/arbitration/SortitionSumTree.sol

@@ -13,7 +13,7 @@ import "../libraries/Constants.sol";
 /// @notice Responsible for sortition operations
 /// @dev Contains only tree management and drawing logic, no phase management or token operations
 contract SortitionSumTree is ISortitionSumTree, Initializable, UUPSProxiable {
-    string public constant override version = "2.0.0";
+    string public constant override version = "0.9.0";
 
     // ************************************* //
     // *         Enums / Structs           * //

contracts/src/arbitration/StakeController.sol

@@ -16,7 +16,7 @@ import "../libraries/Constants.sol";
 /// @notice Responsible for coordinating between Vault and SortitionSumTree
 /// @dev Manages phases, delayed stakes, and coordination logic
 contract StakeController is IStakeController, Initializable, UUPSProxiable {
-    string public constant override version = "1.0.0";
+    string public constant override version = "0.9.0";
 
     // ************************************* //
     // *         Enums / Structs           * //

contracts/src/arbitration/Vault.sol

@@ -8,7 +8,7 @@ import {VaultBase, IERC20} from "./VaultBase.sol";
 /// @notice PNK Vault for handling deposits, withdrawals, locks, and penalties
 /// @dev Follows the same pattern as KlerosCore for upgradeable contracts
 contract Vault is VaultBase {
-    string public constant override version = "1.0.0";
+    string public constant override version = "0.1.0";
 
     // ************************************* //
     // *            Constructor            * //

contracts/src/arbitration/VaultNeo.sol

@@ -12,7 +12,7 @@ import "@openzeppelin/contracts/token/ERC721/IERC721.sol";
 contract VaultNeo is VaultBase {
     using SafeERC20 for IERC20;
 
-    string public constant override version = "1.0.0";
+    string public constant override version = "0.1.0";
 
     // ************************************* //
     // *             Storage               * //

contracts/src/arbitration/view/KlerosCoreSnapshotProxy.sol

@@ -2,10 +2,10 @@
 
 pragma solidity 0.8.24;
 
-import {ISortitionModule} from "../interfaces/ISortitionModule.sol";
+import {IStakeController} from "../interfaces/IStakeController.sol";
 
 interface IKlerosCore {
-    function sortitionModule() external view returns (ISortitionModule);
+    function stakeController() external view returns (IStakeController);
 }
 
 /// @title KlerosCoreSnapshotProxy
@@ -67,6 +67,6 @@ contract KlerosCoreSnapshotProxy {
     /// @param _account The address to query.
     /// @return totalStaked Total amount staked in V2 by the address.
     function balanceOf(address _account) external view returns (uint256 totalStaked) {
-        (totalStaked, , , ) = core.sortitionModule().getJurorBalance(_account, 0);
+        (, , , totalStaked, , ) = core.stakeController().getJurorBalance(_account, 0);
     }
 }