TShock/TShockAPI/DB/GroupManager.cs

/*
TShock, a server mod for Terraria
Copyright (C) 2011 The TShock Team

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.IO;
using System.Linq;
using MySql.Data.MySqlClient;

namespace TShockAPI.DB
{
	public class GroupManager
	{
		private IDbConnection database;

		public List<Group> groups = new List<Group>();

		public GroupManager(IDbConnection db)
		{
			database = db;

			var table = new SqlTable("GroupList",
			                         new SqlColumn("GroupName", MySqlDbType.VarChar, 32) {Primary = true},
			                         new SqlColumn("Parent", MySqlDbType.VarChar, 32),
			                         new SqlColumn("Commands", MySqlDbType.Text),
			                         new SqlColumn("ChatColor", MySqlDbType.Text),
			                         new SqlColumn("Prefix", MySqlDbType.Text),
			                         new SqlColumn("Suffix", MySqlDbType.Text)
				);
			var creator = new SqlTableCreator(db,
			                                  db.GetSqlType() == SqlType.Sqlite
			                                  	? (IQueryBuilder) new SqliteQueryCreator()
			                                  	: new MysqlQueryCreator());
			creator.EnsureExists(table);

			//Add default groups
			AddGroup("guest", "canbuild,canregister,canlogin,canpartychat,cantalkinthird");
			AddGroup("default", "guest", "warp,canchangepassword");
			AddGroup("newadmin", "default", "kick,editspawn,reservedslot");
			AddGroup("admin", "newadmin",
			         "ban,unban,whitelist,causeevents,spawnboss,spawnmob,managewarp,time,tp,pvpfun,kill,logs,immunetokick,tphere");
			AddGroup("trustedadmin", "admin", "maintenance,cfg,butcher,item,heal,immunetoban,usebanneditem,manageusers");
			AddGroup("vip", "default", "reservedslot");
		}


		public bool GroupExists(string group)
		{
			if (group == "superadmin")
				return true;


			return groups.Any(g => g.Name.Equals(group));
		}

		/// <summary>
		/// Adds group with name and permissions if it does not exist.
		/// </summary>
		/// <param name="name">name of group</param>
		/// <param name="parentname">parent of group</param>
		/// <param name="permissions">permissions</param>
		public String AddGroup(String name, string parentname, String permissions, String chatcolor)
		{
			String message = "";
			if (GroupExists(name))
				return "Error: Group already exists.  Use /modGroup to change permissions.";

			var group = new Group(name, null, chatcolor);
			group.permissions.Add(permissions);
			if (!string.IsNullOrWhiteSpace(parentname))
			{
				var parent = groups.FirstOrDefault(gp => gp.Name == parentname);
				if (parent == null)
				{
					var error = "Invalid parent {0} for group {1}".SFormat(group.Name, parentname);
					Log.ConsoleError(error);
					return error;
				}
				group.Parent = parent;
			}

			string query = (TShock.Config.StorageType.ToLower() == "sqlite")
			               	? "INSERT OR IGNORE INTO GroupList (GroupName, Parent, Commands, ChatColor) VALUES (@0, @1, @2, @3);"
			               	: "INSERT IGNORE INTO GroupList SET GroupName=@0, Parent=@1, Commands=@2, ChatColor=@3";
			if (database.Query(query, name, parentname, permissions, chatcolor) == 1)
				message = "Group " + name + " has been created successfully.";

			groups.Add(group);

			return message;
		}

		public String AddGroup(String name, String permissions)
		{
			return AddGroup(name, "", permissions, "255,255,255");
		}

		public String AddGroup(String name, string parent, String permissions)
		{
			return AddGroup(name, parent, permissions, "255,255,255");
		}

		public String DeleteGroup(String name)
		{
			String message = "";
			if (!GroupExists(name))
				return "Error: Group doesn't exists.";

			if (database.Query("DELETE FROM GroupList WHERE GroupName=@0", name) == 1)
				message = "Group " + name + " has been deleted successfully.";
			groups.Remove(TShock.Utils.GetGroup(name));

			return message;
		}

		public String AddPermissions(String name, List<String> permissions)
		{
			String message = "";
			if (!GroupExists(name))
				return "Error: Group doesn't exists.";

			var group = TShock.Utils.GetGroup(name);
			//Add existing permissions (without duplicating)
			permissions.AddRange(group.permissions.Where(s => !permissions.Contains(s)));

			if (database.Query("UPDATE GroupList SET Commands=@0 WHERE GroupName=@1", String.Join(",", permissions), name) != 0)
			{
				message = "Group " + name + " has been modified successfully.";
				group.SetPermission(permissions);
			}
			return message;
		}

		public String DeletePermissions(String name, List<String> permissions)
		{
			String message = "";
			if (!GroupExists(name))
				return "Error: Group doesn't exists.";

			var group = TShock.Utils.GetGroup(name);

			//Only get permissions that exist in the group.
			var newperms = group.permissions.Except(permissions);

			if (database.Query("UPDATE GroupList SET Commands=@0 WHERE GroupName=@1", String.Join(",", newperms), name) != 0)
			{
				message = "Group " + name + " has been modified successfully.";
				group.SetPermission(newperms.ToList());
			}
			return message;
		}

		public void LoadPermisions()
		{
			//Create a temporary list so if there is an error it doesn't override the currently loaded groups with broken groups.
			var tempgroups = new List<Group>();
			tempgroups.Add(new SuperAdminGroup());

			if (groups == null || groups.Count < 2)
				groups = tempgroups;

			try
			{
				var groupsparents = new List<Tuple<Group, string>>();
				using (var reader = database.QueryReader("SELECT * FROM GroupList"))
				{
					while (reader.Read())
					{
						string groupname = reader.Get<String>("GroupName");
						var group = new Group(groupname);

						group.Prefix = reader.Get<String>("Prefix");
						group.Suffix = reader.Get<String>("Suffix");

						//Inherit Given commands
						String[] commands = reader.Get<String>("Commands").Split(',');
						foreach (var t in commands)
						{
							var str = t.Trim();
							if (str.StartsWith("!"))
							{
								group.NegatePermission(str.Substring(1));
							}
							else
							{
								group.AddPermission(str);
							}
						}
						String[] chatcolour = (reader.Get<String>("ChatColor") ?? "").Split(',');
						if (chatcolour.Length == 3)
						{
							byte.TryParse(chatcolour[0], out group.R);
							byte.TryParse(chatcolour[1], out group.G);
							byte.TryParse(chatcolour[2], out group.B);
						}

						groupsparents.Add(Tuple.Create(group, reader.Get<string>("Parent")));
					}
				}

				foreach (var t in groupsparents)
				{
					var group = t.Item1;
					var parentname = t.Item2;
					if (!string.IsNullOrWhiteSpace(parentname))
					{
						var parent = groupsparents.FirstOrDefault(gp => gp.Item1.Name == parentname);
						if (parent == null)
						{
							Log.ConsoleError("Invalid parent {0} for group {1}".SFormat(group.Name, parentname));
							return;
						}
						group.Parent = parent.Item1;
					}
					tempgroups.Add(group);
				}


				groups = tempgroups;
			}
			catch (Exception ex)
			{
				Log.Error(ex.ToString());
			}
		}
	}
}