Merge pull request #890 from wert23239/SQL

Added SQLite Calls to Lua

BizHawk.Client.Common/BizHawk.Client.Common.csproj

@@ -54,6 +54,10 @@
     </Reference>
     <Reference Include="System" />
     <Reference Include="System.Core" />
+    <Reference Include="System.Data.SQLite, Version=1.0.105.2, Culture=neutral, PublicKeyToken=db937bc2d44ff139, processorArchitecture=AMD64">
+      <SpecificVersion>False</SpecificVersion>
+      <HintPath>..\References\x64\SQLite\System.Data.SQLite.dll</HintPath>
+    </Reference>
     <Reference Include="System.Drawing" />
     <Reference Include="System.IO.Compression" />
     <Reference Include="System.Xml.Linq" />
@@ -149,6 +153,7 @@
     <Compile Include="lua\EmuLuaLibrary.Movie.cs" />
     <Compile Include="lua\EmuLuaLibrary.NES.cs" />
     <Compile Include="lua\EmuLuaLibrary.SNES.cs" />
+    <Compile Include="lua\EmuLuaLibrary.SQL.cs" />
     <Compile Include="lua\EmuLuaLibrary.String.cs" />
     <Compile Include="lua\EmuLuaLibrary.UserData.cs" />
     <Compile Include="lua\EnvironmentSandbox.cs" />

BizHawk.Client.Common/lua/EmuLuaLibrary.SQL.cs

@@ -0,0 +1,145 @@
+using System;
+using System.Collections;
+using System.ComponentModel;
+using System.Data.SQLite;
+using LuaInterface;
+using System.Collections.Generic;
+
+namespace BizHawk.Client.Common
+{
+	[Description("A library for performing standard bitwise operations.")]
+	public sealed class SQLLuaLibrary : LuaLibraryBase
+	{
+		public SQLLuaLibrary(Lua lua)
+			: base(lua) { }
+
+		public SQLLuaLibrary(Lua lua, Action<string> logOutputCallback)
+			: base(lua, logOutputCallback) { }
+
+		public override string Name => "SQL";
+
+		SQLiteConnection m_dbConnection;
+		string connectionString;
+
+		[LuaMethodAttributes("createdatabase","Creates a SQLite Database. Name should end with .db")]
+		public string CreateDatabase(string name)
+		{
+			try
+			{
+				SQLiteConnection.CreateFile(name);
+				return "Database Created Successfully";
+			}
+			catch (SQLiteException sqlEX)
+			{
+				return sqlEX.Message;
+			}
+
+		}
+
+
+		[LuaMethodAttributes("opendatabase", "Opens a SQLite database. Name should end with .db")]
+		public string OpenDatabase(string name)
+		{
+			try
+			{
+				SQLiteConnectionStringBuilder connBuilder = new SQLiteConnectionStringBuilder();
+				connBuilder.DataSource = name;
+				connBuilder.Version = 3; //SQLite version 
+				connBuilder.JournalMode = SQLiteJournalModeEnum.Wal;  //Allows for reads and writes to happen at the same time
+				connBuilder.DefaultIsolationLevel = System.Data.IsolationLevel.ReadCommitted;  //This only helps make the database lock left. May be pointless now
+				connBuilder.SyncMode = SynchronizationModes.Off; //This shortens the delay for do synchronous calls.
+				m_dbConnection = new SQLiteConnection(connBuilder.ToString());
+				connectionString = connBuilder.ToString();
+				m_dbConnection.Open();
+				m_dbConnection.Close();
+				return "Database Opened Successfully";
+			}
+			catch(SQLiteException sqlEX)
+			{
+				return sqlEX.Message;
+			}
+		}
+
+		[LuaMethodAttributes("writecommand", "Runs a SQLite write command which includes CREATE,INSERT, UPDATE. " +
+			"Ex: create TABLE rewards (ID integer  PRIMARY KEY, action VARCHAR(20)) ")]
+		public string WriteCommand(string query="")
+		{
+			if (query == "")
+			{
+				return "query is empty";
+			}
+			try
+			{
+				m_dbConnection.Open();
+				string sql = query;
+				SQLiteCommand command = new SQLiteCommand(sql, m_dbConnection);
+				command.ExecuteNonQuery();
+				m_dbConnection.Close();
+
+				return "Command ran successfully";
+
+			}
+			catch (NullReferenceException nullEX)
+			{
+				return "Database not open.";
+			}
+			catch(SQLiteException sqlEX)
+			{
+				m_dbConnection.Close();
+				return sqlEX.Message;
+			}
+		}
+
+		[LuaMethodAttributes("readcommand", "Run a SQLite read command which includes Select. Returns all rows into a LuaTable." +
+			"Ex: select * from rewards")]
+		public dynamic ReadCommand(string query="")
+		{
+			if (query=="")
+			{
+				return "query is empty";
+			}
+			try
+			{
+				var table = Lua.NewTable();
+				m_dbConnection.Open();
+				string sql = "PRAGMA read_uncommitted =1;"+query;
+				SQLiteCommand command = new SQLiteCommand(sql, m_dbConnection);
+				SQLiteDataReader reader = command.ExecuteReader();
+				bool rows=reader.HasRows;
+				long rowCount = 0;
+				var columns = new List<string>();
+				for (int i = 0; i < reader.FieldCount; ++i) //Add all column names into list
+				{
+					columns.Add(reader.GetName(i));
+				}
+				while (reader.Read())
+				{   
+					for (int i = 0; i < reader.FieldCount; ++i)
+					{
+						table[columns[i]+" "+rowCount.ToString()] = reader.GetValue(i);
+					}
+					rowCount += 1;
+				}
+				reader.Close();
+				m_dbConnection.Close();
+				if (rows==false)
+				{
+					return "No rows found";
+				}
+
+				return table;
+
+			}
+			catch (NullReferenceException)
+			{
+				return "Database not opened.";
+			}
+			catch (SQLiteException sqlEX)
+			{
+				m_dbConnection.Close();
+				return sqlEX.Message;
+			}
+		}
+
+	}
+}