TShock/TShockAPI/DB/ProjectileManager.cs

/*
TShock, a server mod for Terraria
Copyright (C) 2011-2019 Pryaxis & TShock Contributors

This program 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.

This program 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/>.
*/

using System;
using System.Collections.Generic;
using System.Data;
using System.Linq;
using MySql.Data.MySqlClient;
using TShockAPI.DB.Queries;
using TShockAPI.Hooks;

namespace TShockAPI.DB
{
	public class ProjectileManagager
	{
		private IDbConnection database;
		public List<ProjectileBan> ProjectileBans = new List<ProjectileBan>();

		public ProjectileManagager(IDbConnection db)
		{
			database = db;

			var table = new SqlTable("ProjectileBans",
				new SqlColumn("ProjectileID", MySqlDbType.Int32) {Primary = true},
				new SqlColumn("AllowedGroups", MySqlDbType.Text)
				);
			var creator = new SqlTableCreator(db,
				db.GetSqlType() == SqlType.Sqlite
					? (IQueryBuilder) new SqliteQueryCreator()
					: new MysqlQueryCreator());
			creator.EnsureTableStructure(table);
			UpdateBans();
		}

		public void UpdateBans()
		{
			ProjectileBans.Clear();

			using (var reader = database.QueryReader("SELECT * FROM ProjectileBans"))
			{
				while (reader != null && reader.Read())
				{
					ProjectileBan ban = new ProjectileBan((short) reader.Get<Int32>("ProjectileID"));
					ban.SetAllowedGroups(reader.Get<string>("AllowedGroups"));
					ProjectileBans.Add(ban);
				}
			}
		}

		public void AddNewBan(short id = 0)
		{
			try
			{
				database.Query("INSERT INTO ProjectileBans (ProjectileID, AllowedGroups) VALUES (@0, @1);",
					id, "");

				if (!ProjectileIsBanned(id, null))
					ProjectileBans.Add(new ProjectileBan(id));
			}
			catch (Exception ex)
			{
				TShock.Log.Error(ex.ToString());
			}
		}

		public void RemoveBan(short id)
		{
			if (!ProjectileIsBanned(id, null))
				return;
			try
			{
				database.Query("DELETE FROM ProjectileBans WHERE ProjectileId=@0;", id);
				ProjectileBans.Remove(new ProjectileBan(id));
			}
			catch (Exception ex)
			{
				TShock.Log.Error(ex.ToString());
			}
		}

		public bool ProjectileIsBanned(short id)
		{
			if (ProjectileBans.Contains(new ProjectileBan(id)))
			{
				return true;
			}
			return false;
		}

		public bool ProjectileIsBanned(short id, TSPlayer ply)
		{
			if (ProjectileBans.Contains(new ProjectileBan(id)))
			{
				ProjectileBan b = GetBanById(id);
				return !b.HasPermissionToCreateProjectile(ply);
			}
			return false;
		}

		public bool AllowGroup(short id, string name)
		{
			string groupsNew = "";
			ProjectileBan b = GetBanById(id);
			if (b != null)
			{
					try
				{
					groupsNew = String.Join(",", b.AllowedGroups);
					if (groupsNew.Length > 0)
						groupsNew += ",";
					groupsNew += name;
					b.SetAllowedGroups(groupsNew);

					int q = database.Query("UPDATE ProjectileBans SET AllowedGroups=@0 WHERE ProjectileId=@1", groupsNew,
						id);

					return q > 0;
				}
				catch (Exception ex)
				{
					TShock.Log.Error(ex.ToString());
				}
			}

			return false;
		}

		public bool RemoveGroup(short id, string group)
		{
			ProjectileBan b = GetBanById(id);
			if (b != null)
			{
				try
				{
					b.RemoveGroup(group);
					string groups = string.Join(",", b.AllowedGroups);
					int q = database.Query("UPDATE ProjectileBans SET AllowedGroups=@0 WHERE ProjectileId=@1", groups,
						id);

					if (q > 0)
						return true;
				}
				catch (Exception ex)
				{
					TShock.Log.Error(ex.ToString());
				}
			}
			return false;
		}

		public ProjectileBan GetBanById(short id)
		{
			foreach (ProjectileBan b in ProjectileBans)
			{
				if (b.ID == id)
				{
					return b;
				}
			}
			return null;
		}
	}

	public class ProjectileBan : IEquatable<ProjectileBan>
	{
		public short ID { get; set; }
		public List<string> AllowedGroups { get; set; }

		public ProjectileBan(short id)
			: this()
		{
			ID = id;
			AllowedGroups = new List<string>();
		}

		public ProjectileBan()
		{
			ID = 0;
			AllowedGroups = new List<string>();
		}

		public bool Equals(ProjectileBan other)
		{
			return ID == other.ID;
		}

		public bool HasPermissionToCreateProjectile(TSPlayer ply)
		{
			if (ply == null)
				return false;

			if (ply.HasPermission(Permissions.canusebannedprojectiles))
				return true;

			PermissionHookResult hookResult = PlayerHooks.OnPlayerProjbanPermission(ply, this);
			if (hookResult != PermissionHookResult.Unhandled)
				return hookResult == PermissionHookResult.Granted;

			var cur = ply.Group;
			var traversed = new List<Group>();
			while (cur != null)
			{
				if (AllowedGroups.Contains(cur.Name))
				{
					return true;
				}
				if (traversed.Contains(cur))
				{
					throw new InvalidOperationException(GetString($"Infinite group parenting ({cur.Name})"));
				}
				traversed.Add(cur);
				cur = cur.Parent;
			}
			return false;
			// could add in the other permissions in this class instead of a giant if switch.
		}

		public void SetAllowedGroups(String groups)
		{
			// prevent null pointer exceptions
			if (!string.IsNullOrEmpty(groups))
			{
				List<String> groupArr = groups.Split(',').ToList();

				for (int i = 0; i < groupArr.Count; i++)
				{
					groupArr[i] = groupArr[i].Trim();
					//Console.WriteLine(groupArr[i]);
				}
				AllowedGroups = groupArr;
			}
		}

		public bool RemoveGroup(string groupName)
		{
			return AllowedGroups.Remove(groupName);
		}

		public override string ToString()
		{
			return ID + (AllowedGroups.Count > 0 ? " (" + String.Join(",", AllowedGroups) + ")" : "");
		}
	}
}