From ee8fe8e9d55f7adb180956d0b3a44e4193948a34 Mon Sep 17 00:00:00 2001
From: Hecate2 <2474101468@qq.com>
Date: Wed, 8 Mar 2023 10:16:02 +0800
Subject: [PATCH] filter candidates not in whitelist

---
 BurgerStrategist/Program.cs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/BurgerStrategist/Program.cs b/BurgerStrategist/Program.cs
index 8dd99aa..927040c 100644
--- a/BurgerStrategist/Program.cs
+++ b/BurgerStrategist/Program.cs
@@ -6,6 +6,7 @@
 using LibRPC;
 using LibWallet;
 using Neo;
+using Neo.SmartContract;
 using Neo.SmartContract.Native;
 using Neo.VM;
 
@@ -29,8 +30,8 @@ static void Main(string[] args)
 
             List<BigInteger> CANDIDATE_V = CANDIDATES.Zip(CANDIDATE_VOTES).Select(v => v.Second - AGENT_TO.Zip(AGENT_HOLD).FindByOrDefault(v.First)).ToList();
             List<byte[]> ELECTEDS = CANDIDATES.Zip(CANDIDATE_V).OrderBy(v => v.Second).TakeLast(21).Select(v => v.Item1).ToList();
-            List<byte[]> CNS = ELECTEDS.TakeLast(7).ToList();
-            List<byte[]> CMS = ELECTEDS.Take(14).ToList();
+            List<byte[]> CNS = ELECTEDS.TakeLast(7).ToList().FindAll(v => BNEO.MakeScript("candidate", v).Call().Single().GetBoolean());
+            List<byte[]> CMS = ELECTEDS.Take(14).ToList().FindAll(v => BNEO.MakeScript("candidate", v).Call().Single().GetBoolean());
             $"CANDIDATE_V: {String.Join(", ", CANDIDATE_V)}".Log();
             $"ELECTEDS: {String.Join(", ", ELECTEDS.Select(v => v.ToHexString()))}".Log();
             $"CNS: {String.Join(", ", CNS.Select(v => v.ToHexString()))}".Log();