chore: getter for remaining milestones

icon-community · Jan 25, 2024 · 9859573 · 9859573
1 parent 75024a1
commit 9859573
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/CPSCore/src/main/java/community/icon/cps/score/cpscore/CPSCore.java b/CPSCore/src/main/java/community/icon/cps/score/cpscore/CPSCore.java
@@ -2562,6 +2562,32 @@ public Map<String, Object> getProposalsHistory(@Optional int startIndex) {
         return Map.of(DATA, proposalHistory, COUNT, size);
     }
 
+    @Override
+    @External(readonly = true)
+    public List<Integer> getRemainingMilestones(String ipfsHash){
+        String ipfsHashPrefix = proposalPrefix(ipfsHash);
+        ArrayDB<Integer> milestoneIDs = milestoneIds.at(ipfsHashPrefix);
+
+        List<Integer> milestoneIdList = new ArrayList<>();
+        for (int i = 0; i < milestoneIDs.size(); i++) {
+            int milestoneId = milestoneIDs.get(i);
+            String milestonePrefix = mileStonePrefix(ipfsHash, milestoneId);
+            int proposalTotalPeriod = proposalPeriod.at(ipfsHashPrefix).getOrDefault(0);
+            int completionPeriod = MilestoneDb.completionPeriod.at(milestonePrefix).getOrDefault(0);
+
+            int computedCompletionPeriod = proposalTotalPeriod + completionPeriod;
+
+            int currentPeriod = getPeriodCount();
+            if (computedCompletionPeriod > currentPeriod) {
+                int status = MilestoneDb.status.at(milestonePrefix).getOrDefault(0);
+                if (status != MILESTONE_REPORT_APPROVED ){
+                    milestoneIdList.add(milestoneId);
+                }
+            }
+        }
+        return milestoneIdList;
+    }
+
     //    =====================================TEMPORARY MIGRATIONS METHODS===============================================
 
     @External

diff --git a/score-lib/src/main/java/community/icon/cps/score/lib/interfaces/CPSCoreInterface.java b/score-lib/src/main/java/community/icon/cps/score/lib/interfaces/CPSCoreInterface.java
@@ -265,6 +265,9 @@ public static class MilestoneVoteAttributes {
     @External(readonly = true)
     Map<String, Object> getProposalsHistory(@Optional int startIndex);
 
+    @External(readonly = true)
+    List<Integer> getRemainingMilestones(String ipfsHash);
+
     //    EventLogs
     @EventLog(indexed = 1)
     void ProposalSubmitted(Address _sender_address, String note);