Adds DBTools.cs (Create and Insert Table)

TShockAPI/DB/DBTools.cs

@@ -0,0 +1,206 @@
+using System;
+using System.Collections.Generic;
+using System.Data;
+using System.Linq;
+using System.Text;
+using TShockAPI.DB;
+
+namespace TShockAPI.DB
+{
+    public class DBTools
+    {
+        public static IDbConnection database;
+
+        public static void CreateTable(string name, bool IfNotExists, List<Column> columns)
+        {
+            //Build up Creation string :)
+            StringBuilder sb = new StringBuilder();
+            sb.Append("CREATE TABLE "); 
+
+            if (IfNotExists)
+                sb.Append("IF NOT EXISTS ");
+
+            if (TShock.Config.StorageType.ToLower() == "sqlite")
+                sb.Append("'" + name + "' (");
+            else if (TShock.Config.StorageType.ToLower() == "mysql")
+                sb.Append(name + " (");
+
+            int count = 0;
+
+            foreach (Column column in columns)
+            {
+                count++;
+                if (column.Type.ToLower() == "int")
+                {
+                    if (TShock.Config.StorageType.ToLower() == "sqlite")
+                        sb.Append(column.Name + " NUMERIC ");
+                    else if (TShock.Config.StorageType.ToLower() == "mysql")
+                        sb.Append(column.Name + " INT(255) ");
+                }
+                else if (column.Type.ToLower() == "string")
+                {
+                    if (TShock.Config.StorageType.ToLower() == "sqlite")
+                        sb.Append(column.Name + " TEXT ");
+                    else if (TShock.Config.StorageType.ToLower() == "mysql")
+                        sb.Append(column.Name + " VARCHAR(255) ");
+                }
+                if (column.Unique)
+                {                    
+                    if (columns.Count != count)
+                    {
+                        sb.Append("UNIQUE, ");
+                    }
+                    else
+                        sb.Append("UNIQUE) ");
+                }
+                if (columns.Count == count)
+                    sb.Append(")");
+            }
+
+            using (var com = database.CreateCommand())
+            {
+                com.CommandText = sb.ToString();
+                com.ExecuteNonQuery();
+            }
+        }
+
+        public static void InsertTable(string tablename, bool Ignore, List<ColumnData> Values, List<ColumnData> WhereAndStatements)
+        {
+            StringBuilder sb = new StringBuilder();
+            sb.Append("INSERT ");
+
+            if (Ignore)
+            {
+                if (TShock.Config.StorageType.ToLower() == "sqlite")
+                    sb.Append("OR IGNORE ");
+                else if (TShock.Config.StorageType.ToLower() == "mysql")
+                    sb.Append("IGNORE ");
+            }
+
+            if (TShock.Config.StorageType.ToLower() == "sqlite")
+                sb.Append("INTO '" + tablename + "' (");
+            else if (TShock.Config.StorageType.ToLower() == "mysql")
+                sb.Append("INTO " + tablename + " ");
+
+            using (var com = database.CreateCommand())
+            {
+                //Values
+                if (TShock.Config.StorageType.ToLower() == "sqlite")
+                {
+                    int count = 0;
+
+                    foreach (ColumnData columnname in Values)
+                    {
+                        count++;
+                        if (Values.Count != count)
+                            sb.Append(columnname.Name + ", ");
+                        else
+                            sb.Append(columnname.Name + ") ");
+                    }
+
+                    sb.Append("VALUES (");
+                    count = 0;
+
+                    foreach (ColumnData columnname in Values)
+                    {
+                        count++;
+                        if (Values.Count != count)
+                        {
+                            sb.Append("@" + columnname.Name + ", ");
+                            com.AddParameter("@" + columnname.Name.ToLower(), columnname.Value);
+                        }
+                        else
+                        {
+                            sb.Append("@" + columnname.Name + ") ");
+                            com.AddParameter("@" + columnname.Name.ToLower(), columnname.Value);
+                        }
+                    }
+                }
+                else if (TShock.Config.StorageType.ToLower() == "mysql")
+                {
+                    sb.Append("SET ");
+                    int count = 0;
+
+                    foreach (ColumnData columnname in Values)
+                    {
+                        count++;
+                        if (Values.Count != count)
+                        {
+                            sb.Append("@" + columnname.Name + "=" + columnname.Value + ", ");
+                            com.AddParameter("@" + columnname.Name.ToLower(), columnname.Value);
+                        }
+                        else
+                        {
+                            sb.Append("@" + columnname.Name + "=" + columnname.Value + ") ");
+                            com.AddParameter("@" + columnname.Name.ToLower(), columnname.Value);
+                        }
+                    }
+                }
+
+                //Where Statement (if any)
+                if (WhereAndStatements.Count > 0)
+                {
+                    sb.Append("WHERE ");
+                    int count = 0;
+
+                    foreach (ColumnData columnname in WhereAndStatements)
+                    {
+                        count++;
+                        if (Values.Count != count)
+                        {
+                            sb.Append("@" + columnname.Name + "=" + columnname.Value + "-where" + " AND ");
+                            com.AddParameter("@" + columnname.Name.ToLower() + "-where", columnname.Value);
+                        }
+                        else
+                        {
+                            sb.Append("@" + columnname.Name + "=" + columnname.Value + "-where" + ";");
+                            com.AddParameter("@" + columnname.Name.ToLower() + "-where", columnname.Value);
+                        }
+                    }
+                }
+            }
+        }
+    }
+
+    public class Column
+    {
+        public string Name { get; set; }
+        public string Type { get; set; }
+        public bool Unique { get; set; }
+        public string Parameters { get; set; }
+
+        public Column(string name, bool unique, string type, string parameters)
+        {
+            Name = name;
+            Type = type;
+            Unique = unique;
+            Parameters = parameters;
+        }
+
+        public Column()
+        {
+            Name = string.Empty;
+            Type = string.Empty;
+            Unique = false;
+            Parameters = string.Empty;
+        }
+    }
+
+    public class ColumnData
+    {
+        public string Name { get; set; }
+        public object Value { get; set; }
+
+        public ColumnData(string name, object value)
+        {
+            Name = name;
+            Value = value;
+        }
+
+        public ColumnData()
+        {
+            Name = string.Empty;
+            Value = string.Empty;
+        }
+    }
+}

TShockAPI/DB/SqlTable.cs

@@ -1,6 +0,0 @@
-namespace TShockAPI.DB
-{
-    class SqlTable
-    {
-    }
-}

TShockAPI/TShock.cs

@@ -155,6 +155,7 @@ namespace TShockAPI
                 throw new Exception("Invalid storage type");
             }
 
+            DBTools.database = DB;
             Backups = new BackupManager(Path.Combine(SavePath, "backups"));
             Backups.KeepFor = Config.BackupKeepFor;
             Backups.Interval = Config.BackupInterval;

TShockAPI/TShockAPI.csproj

@@ -93,10 +93,10 @@
   <ItemGroup>
     <Compile Include="BackupManager.cs" />
     <Compile Include="DB\BanManager.cs" />
+    <Compile Include="DB\DBTools.cs" />
     <Compile Include="DB\ItemManager.cs" />
     <Compile Include="DB\DbExt.cs" />
     <Compile Include="DB\GroupManager.cs" />
-    <Compile Include="DB\SqlTable.cs" />
     <Compile Include="DB\UserManager.cs" />
     <Compile Include="IPackable.cs" />
     <Compile Include="Commands.cs" />
@@ -171,7 +171,7 @@
   </PropertyGroup>
   <ProjectExtensions>
     <VisualStudio>
-      <UserProperties BuildVersion_UpdateAssemblyVersion="True" BuildVersion_UpdateFileVersion="True" BuildVersion_BuildAction="Both" BuildVersion_BuildVersioningStyle="None.None.None.MonthAndDayStamp" BuildVersion_StartDate="2011/6/17" BuildVersion_IncrementBeforeBuild="False" />
+      <UserProperties BuildVersion_IncrementBeforeBuild="False" BuildVersion_StartDate="2011/6/17" BuildVersion_BuildVersioningStyle="None.None.None.MonthAndDayStamp" BuildVersion_BuildAction="Both" BuildVersion_UpdateFileVersion="True" BuildVersion_UpdateAssemblyVersion="True" />
     </VisualStudio>
   </ProjectExtensions>
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it.