themaster1970sf/TShock
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

BanManager now fully sql based

Browse source
This commit is contained in:
high 2011-07-01 17:08:42 -04:00
parent 6a960c2db9
commit f1c86a6216
2 changed files with 66 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

47
TShockAPI/BanManager.cs
View file

@ -27,14 +27,8 @@ namespace TShockAPI
{
public class BanManager
{
private DateTime LastLoad;
private IDbConnection database;
/// <summary>
/// IP - Name - Reason
/// </summary>
private List<Ban> Bans = new List<Ban>();
public BanManager(IDbConnection db)
{
database = db;
@ -48,6 +42,8 @@ namespace TShockAPI
}
public Ban GetBanByIp(string ip)
{
try
{
using (var com = database.CreateCommand())
{
@ -59,9 +55,13 @@ namespace TShockAPI
return new Ban((string)reader["IP"], (string)reader["Name"], (string)reader["Reason"]);
}
}
}
catch (SqliteExecutionException ex)
{
}
return null;
}
IDbDataParameter AddParameter(IDbCommand command, string name, object data)
static IDbDataParameter AddParameter(IDbCommand command, string name, object data)
{
var parm = command.CreateParameter();
parm.ParameterName = name;
@ -72,6 +72,8 @@ namespace TShockAPI
public Ban GetBanByName(string name, bool casesensitive = true)
{
try
{
using (var com = database.CreateCommand())
{
@ -86,10 +88,16 @@ namespace TShockAPI
return new Ban((string)reader["IP"], (string)reader["Name"], (string)reader["Reason"]);
}
}
}
catch (SqliteExecutionException ex)
{
}
return null;
}
public void AddBan(string ip, string name = "", string reason = "")
public bool AddBan(string ip, string name = "", string reason = "")
{
try
{
using (var com = database.CreateCommand())
{
@ -99,11 +107,30 @@ namespace TShockAPI
AddParameter(com, "@reason", reason);
com.ExecuteNonQuery();
}
return true;
}
catch (SqliteExecutionException ex)
{
}
return false;
}
public void RemoveBan(Ban ban)
public bool RemoveBan(string ip)
{
try
{
using (var com = database.CreateCommand())
{
com.CommandText = "DELETE FROM Bans WHERE IP=@ip";
AddParameter(com, "@ip", ip);
com.ExecuteNonQuery();
return true;
}
}
catch (SqliteExecutionException ex)
{
}
return false;
}
}

8
TShockAPI/Commands.cs
View file

@ -456,8 +456,10 @@ namespace TShockAPI
var ban = TShock.Bans.GetBanByName(plStr);
if (ban != null)
{
TShock.Bans.RemoveBan(ban);
if (TShock.Bans.RemoveBan(ban.IP))
args.Player.SendMessage(string.Format("Unbanned {0} ({1})!", ban.Name, ban.IP), Color.Red);
else
args.Player.SendMessage(string.Format("Failed to Unbanned {0} ({1})!", ban.Name, ban.IP), Color.Red);
}
else
{
@ -482,8 +484,10 @@ namespace TShockAPI
var ban = TShock.Bans.GetBanByIp(plStr);
if (ban != null)
{
TShock.Bans.RemoveBan(ban);
if (TShock.Bans.RemoveBan(ban.IP))
args.Player.SendMessage(string.Format("Unbanned {0} ({1})!", ban.Name, ban.IP), Color.Red);
else
args.Player.SendMessage(string.Format("Failed to Unbanned {0} ({1})!", ban.Name, ban.IP), Color.Red);
}
else
{