Premium system.

trunk/java/com/l2jserver/gameserver/instancemanager/PremiumManager.java

@@ -0,0 +1,138 @@
+/*
+ * Copyright (C) 2004-2015 L2J Server
+ * 
+ * This file is part of L2J Server.
+ * 
+ * L2J Server is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * 
+ * L2J Server is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * General Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+package com.l2jserver.gameserver.instancemanager;
+
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.Calendar;
+
+import com.l2jserver.Config;
+import com.l2jserver.L2DatabaseFactory;
+import com.l2jserver.gameserver.model.L2World;
+import com.l2jserver.gameserver.model.actor.instance.L2PcInstance;
+
+/**
+ * @author Mobius, RobíkBobík
+ */
+public class PremiumManager
+{
+	private long endDate = 0;
+	
+	public long getPremiumEndDate(String accountName)
+	{
+		try (Connection con = L2DatabaseFactory.getInstance().getConnection())
+		{
+			PreparedStatement statement = con.prepareStatement("SELECT premium_service,enddate FROM account_premium WHERE account_name=?");
+			statement.setString(1, accountName);
+			ResultSet rset = statement.executeQuery();
+			while (rset.next())
+			{
+				if (Config.PREMIUM_SYSTEM_ENABLED)
+				{
+					endDate = rset.getLong("enddate");
+					if (endDate <= System.currentTimeMillis())
+					{
+						endDate = 0;
+						removePremiumStatus(accountName);
+					}
+				}
+			}
+			statement.close();
+		}
+		catch (Exception e)
+		{
+		}
+		
+		return endDate;
+	}
+	
+	public void updatePremiumData(int months, String accountName)
+	{
+		long remainingTime = getPremiumEndDate(accountName);
+		if (remainingTime > 0)
+		{
+			remainingTime -= System.currentTimeMillis();
+		}
+		
+		try (Connection con = L2DatabaseFactory.getInstance().getConnection())
+		{
+			Calendar endDate = Calendar.getInstance();
+			endDate.setTimeInMillis(System.currentTimeMillis() + remainingTime);
+			endDate.set(Calendar.SECOND, 0);
+			endDate.add(Calendar.MONTH, months);
+			
+			PreparedStatement statement = con.prepareStatement("UPDATE account_premium SET premium_service=?,enddate=? WHERE account_name=?");
+			statement.setInt(1, 1);
+			statement.setLong(2, endDate.getTimeInMillis());
+			statement.setString(3, accountName);
+			statement.execute();
+			statement.close();
+		}
+		catch (SQLException e)
+		{
+		}
+		
+		for (L2PcInstance player : L2World.getInstance().getPlayers())
+		{
+			if (player.getAccountNamePlayer().equalsIgnoreCase(accountName))
+			{
+				player.setPremiumStatus(getPremiumEndDate(accountName) > 0 ? true : false);
+			}
+		}
+	}
+	
+	public void removePremiumStatus(String accountName)
+	{
+		// TODO: Add check if account exists. XD
+		try (Connection con = L2DatabaseFactory.getInstance().getConnection())
+		{
+			PreparedStatement statement = con.prepareStatement("INSERT INTO account_premium (account_name,premium_service,enddate) values(?,?,?) ON DUPLICATE KEY UPDATE premium_service = ?, enddate = ?");
+			statement.setString(1, accountName);
+			statement.setInt(2, 0);
+			statement.setLong(3, 0);
+			statement.setInt(4, 0);
+			statement.setLong(5, 0);
+			statement.execute();
+			statement.close();
+		}
+		catch (SQLException e)
+		{
+		}
+		
+		for (L2PcInstance player : L2World.getInstance().getPlayers())
+		{
+			if (player.getAccountNamePlayer().equalsIgnoreCase(accountName))
+			{
+				player.setPremiumStatus(false);
+			}
+		}
+	}
+	
+	public static final PremiumManager getInstance()
+	{
+		return SingletonHolder._instance;
+	}
+	
+	private static class SingletonHolder
+	{
+		protected static final PremiumManager _instance = new PremiumManager();
+	}
+}