git.defcon.no Git - YAVote/blob - src/no/defcon/yavote/Votemanager.java

   1 package no.defcon.yavote;
   2
   3 import java.util.ArrayList;
   4 import java.util.List;
   5
   6 import org.bukkit.World;
   7 import org.bukkit.entity.Player;
   8 import org.bukkit.plugin.java.JavaPlugin;
   9
  10 public class Votemanager {
  11         private JavaPlugin plugin;
  12         private boolean voteRunning;
  13         private String voteType;
  14         private int expireTask;
  15         private List<String> yesVoters;
  16         private List<String> noVoters;
  17
  18
  19         public Votemanager(JavaPlugin plugin) {
  20                 super();
  21                 this.plugin = plugin;
  22                 this.voteRunning = false;
  23         }
  24
  25         public boolean cancelVote( )
  26         {
  27                 if ( voteRunning )
  28                 {
  29                         plugin.getServer().broadcastMessage("Currently running vote is being canceled.");
  30                         plugin.getServer().getScheduler().cancelTask(expireTask);
  31                         clearState();
  32                         return true;
  33                 }
  34                 return false;
  35         }
  36
  37         public boolean startVote( String type, Player p )
  38         {
  39                 if ( this.voteRunning )
  40                         return false;
  41
  42                 if ( ! (type.equalsIgnoreCase("sun") ||
  43                                 type.equalsIgnoreCase("rain") ||
  44                                 type.equalsIgnoreCase("storm") ||
  45                                 type.equalsIgnoreCase("day") ||
  46                                 type.equalsIgnoreCase("night")  ))
  47                 {
  48                         plugin.getLogger().info("Tried to start a vote of invalid type. Code error!");
  49                         return false;
  50                 }
  51
  52                 voteType = type;
  53                 voteRunning = true;
  54
  55                 if ( yesVoters == null )
  56                                 yesVoters = new ArrayList<String>();
  57                 else
  58                         yesVoters.clear();
  59
  60                 if ( noVoters == null )
  61                         noVoters = new ArrayList<String>();
  62                 else
  63                         noVoters.clear();
  64
  65                 yesVoters.add(p.getName());
  66
  67                 if ( ((float) yesVoters.size() / (float) plugin.getServer().getOnlinePlayers().length ) >= getRequired() )
  68                 {
  69                         applyVote( voteType );
  70                         return true;
  71                 }
  72
  73                 expireTask = plugin.getServer().getScheduler().scheduleSyncDelayedTask(this.plugin, new Runnable(){
  74                         @Override
  75                         public void run()
  76                         {
  77                                 if ( voteRunning )
  78                                 {
  79                                         plugin.getServer().broadcastMessage("Time expired for voting on " + ((voteType != null) ? voteType : " .. uhm *blush*"));
  80                                         clearState();
  81                                 }
  82                                 else
  83                                 {
  84                                         plugin.getLogger().info("Timer expired but vote was not running :S");
  85                                         clearState();
  86                                 }
  87                         }
  88                 }, plugin.getConfig().getLong("vote.timeoutSeconds")*20L );
  89
  90                 plugin.getServer().broadcastMessage("A vote has started for " +  voteType.toLowerCase() );
  91                 return true;
  92         }
  93
  94         public boolean isVoteRunning()
  95         {
  96                 return voteRunning;
  97         }
  98
  99         public String getVoteType ()
 100         {
 101                 return voteType;
 102         }
 103
 104         public void removeVote(Player player )
 105         {
 106                 if ( ! voteRunning || (voteType == null ) )
 107                         return;
 108
 109                 if ( yesVoters.contains(player.getName()) )
 110                         yesVoters.remove(player.getName());
 111
 112                 if ( noVoters.contains(player.getName()) )
 113                         noVoters.remove(player.getName());
 114
 115                 if ( yesVoters.size() + noVoters.size() == 0 )
 116                         cancelVote();
 117
 118                 else if ( ((float) yesVoters.size() / (float) (plugin.getServer().getOnlinePlayers().length - 1) ) >= getRequired() )
 119                         applyVote( voteType );
 120         }
 121
 122         public boolean addVote(Player player, boolean yes)
 123         {
 124                 if ( ! voteRunning || (voteType == null ) )
 125                         return false;
 126
 127                 if( yesVoters.contains(player.getName()) || noVoters.contains(player.getName()) )
 128                 {
 129                         player.sendMessage("You have already cast your vote");
 130                         return true;
 131                 }
 132                 if (yes == true) yesVoters.add(player.getName());
 133                 else noVoters.add(player.getName());
 134
 135                 if ( ( yes == true) && ( ((float) yesVoters.size() / (float) plugin.getServer().getOnlinePlayers().length ) >= getRequired() ) )
 136                 {
 137                         applyVote( voteType );
 138                         return true;
 139                 }
 140                 else
 141                 {
 142                         if ( yesVoters.size() + noVoters.size() >= plugin.getServer().getOnlinePlayers().length )
 143                         {
 144                                 plugin.getServer().broadcastMessage("Vote failed for " + voteType);
 145                                 plugin.getServer().getScheduler().cancelTask(expireTask);
 146                                 clearState();
 147                                 return true;
 148                         }
 149                         player.sendMessage("Vote counted. Thank you.");
 150                         return true;
 151                 }
 152         }
 153
 154         private void applyVote(String type)
 155         {
 156                 plugin.getServer().getScheduler().cancelTask(expireTask);
 157                 if ( ! voteRunning || (voteType == null ) )
 158                 {
 159                         plugin.getLogger().info("No vote running, but I was told to apply one. ERRR");
 160                         return;
 161                 }
 162
 163                 List<World> worlds = plugin.getServer().getWorlds();
 164                 for ( World w : worlds)
 165                 {
 166                         if ( voteType.equalsIgnoreCase("sun"))
 167                                 w.setStorm(false);
 168                         else if ( voteType.equalsIgnoreCase("rain"))
 169                                 w.setStorm(true);
 170                         else if ( voteType.equalsIgnoreCase("storm"))
 171                                 w.setThundering(true);
 172                         else  if ( voteType.equalsIgnoreCase("day"))
 173                                 w.setTime( plugin.getConfig().getInt("vote.time.dayStart") );
 174                         else  if ( voteType.equalsIgnoreCase("night"))
 175                                 w.setTime( plugin.getConfig().getInt("vote.time.nightStart") );
 176                 }
 177                 plugin.getServer().broadcastMessage("Vote succeeded for " + ((voteType != null) ? voteType : " .. uhm *blush*"));
 178                 clearState();
 179         }
 180
 181         private float getRequired( )
 182         {
 183                 float req = 0.0f;
 184                 if ( voteType.equalsIgnoreCase("sun") || voteType.equalsIgnoreCase("rain") || voteType.equalsIgnoreCase("storm") )
 185                         req = (float)plugin.getConfig().getInt("vote.weather.requiredPercent") / 100.0F;
 186                 else if ( voteType.equalsIgnoreCase("day") || voteType.equalsIgnoreCase("night") )
 187                         req = (float)plugin.getConfig().getInt("vote.time.requiredPercent") / 100.0F;
 188
 189                 return req;
 190         }
 191
 192         private void clearState()
 193         {
 194                 voteRunning = false;
 195                 voteType = null;
 196                 yesVoters.clear();
 197                 noVoters.clear();
 198         }
 199
 200 }