SabreTools/SabreTools.Helper/Tools/DatabaseTools.cs

using Mono.Data.Sqlite;
using System;
using System.Collections.Generic;

using SabreTools.Helper.Data;

#if MONO
using System.IO;
#else
using Alphaleonis.Win32.Filesystem;
#endif

namespace SabreTools.Helper.Tools
{
	/// <summary>
	/// All general database operations
	/// </summary>
	public static class DatabaseTools
	{
		/// <summary>
		/// Add a header to the database
		/// </summary>
		/// <param name="header">String representing the header bytes</param>
		/// <param name="SHA1">SHA-1 of the deheadered file</param>
		/// <param name="type">Name of the source skipper file</param>
		/// <param name="logger">Logger object for console and file output</param>
		public static void AddHeaderToDatabase(string header, string SHA1, string source, Logger logger)
		{
			bool exists = false;

			// Ensure the database exists
			EnsureDatabase(Constants.HeadererDbSchema, Constants.HeadererFileName, Constants.HeadererConnectionString);

			// Open the database connection
			SqliteConnection dbc = new SqliteConnection(Constants.HeadererConnectionString);
			dbc.Open();

			string query = @"SELECT * FROM data WHERE sha1='" + SHA1 + "' AND header='" + header + "'";
			SqliteCommand slc = new SqliteCommand(query, dbc);
			SqliteDataReader sldr = slc.ExecuteReader();
			exists = sldr.HasRows;

			if (!exists)
			{
				query = @"INSERT INTO data (sha1, header, type) VALUES ('" +
				SHA1 + "', " +
				"'" + header + "', " +
				"'" + source + "')";
				slc = new SqliteCommand(query, dbc);
				logger.Verbose("Result of inserting header: " + slc.ExecuteNonQuery());
			}

			// Dispose of database objects
			slc.Dispose();
			sldr.Dispose();
			dbc.Dispose();
		}

		/// <summary>
		/// Ensure that the databse exists and has the proper schema
		/// </summary>
		/// <param name="type">Schema type to use</param>
		/// <param name="db">Name of the databse</param>
		/// <param name="connectionString">Connection string for SQLite</param>
		public static void EnsureDatabase(string type, string db, string connectionString)
		{
			// Set the type to lowercase
			type = type.ToLowerInvariant();

			// Make sure the file exists
			if (!File.Exists(db))
			{
				SqliteConnection.CreateFile(db);
			}

			// Open the database connection
			SqliteConnection dbc = new SqliteConnection(connectionString);
			dbc.Open();

			// Make sure the database has the correct schema
			try
			{
				if (type == "rombasharp")
				{
					string query = @"
CREATE TABLE IF NOT EXISTS crc (
	'crc'	TEXT		NOT NULL,
	PRIMARY KEY (crc)
)";
					SqliteCommand slc = new SqliteCommand(query, dbc);
					slc.ExecuteNonQuery();

					query = @"
CREATE TABLE IF NOT EXISTS md5 (
	'md5'	TEXT		NOT NULL,
	PRIMARY KEY (md5)
)";
					slc = new SqliteCommand(query, dbc);
					slc.ExecuteNonQuery();

					query = @"
CREATE TABLE IF NOT EXISTS sha1 (
	'sha1'	TEXT		NOT NULL,
	'depot'	TEXT,
	PRIMARY KEY (sha1)
)";
					slc = new SqliteCommand(query, dbc);
					slc.ExecuteNonQuery();

					query = @"
CREATE TABLE IF NOT EXISTS crcsha1 (
	'crc'	TEXT		NOT NULL,
	'sha1'	TEXT		NOT NULL,
	PRIMARY KEY (crc, sha1)
)";
					slc = new SqliteCommand(query, dbc);
					slc.ExecuteNonQuery();

					query = @"
CREATE TABLE IF NOT EXISTS md5sha1 (
	'md5'	TEXT		NOT NULL,
	'sha1'	TEXT		NOT NULL,
	PRIMARY KEY (md5, sha1)
)";
					slc = new SqliteCommand(query, dbc);
					slc.ExecuteNonQuery();

					query = @"
CREATE TABLE IF NOT EXISTS dat (
	'hash'	TEXT		NOT NULL,
	PRIMARY KEY (hash)
)";
					slc = new SqliteCommand(query, dbc);
					slc.ExecuteNonQuery();
					slc.Dispose();
				}
				else if (type == "headerer")
				{
					string query = @"
CREATE TABLE IF NOT EXISTS data (
	'sha1'		TEXT		NOT NULL,
	'header'	TEXT		NOT NULL,
	'type'		TEXT		NOT NULL,
	PRIMARY KEY (sha1, header, type)
)";
					SqliteCommand slc = new SqliteCommand(query, dbc);
					slc.ExecuteNonQuery();
					slc.Dispose();
				}
			}
			catch (Exception ex)
			{
				Console.WriteLine(ex);
			}
			finally
			{
				dbc.Dispose();
			}
		}

		/// <summary>
		/// Retrieve headers from the database
		/// </summary>
		/// <param name="SHA1">SHA-1 of the deheadered file</param>
		/// <param name="logger">Logger object for console and file output</param>
		/// <returns>List of strings representing the headers to add</returns>
		public static List<string> RetrieveHeadersFromDatabase(string SHA1, Logger logger)
		{
			// Ensure the database exists
			EnsureDatabase(Constants.HeadererDbSchema, Constants.HeadererFileName, Constants.HeadererConnectionString);

			// Open the database connection
			SqliteConnection dbc = new SqliteConnection(Constants.HeadererConnectionString);
			dbc.Open();

			// Create the output list of headers
			List<string> headers = new List<string>();

			string query = @"SELECT header, type FROM data WHERE sha1='" + SHA1 + "'";
			SqliteCommand slc = new SqliteCommand(query, dbc);
			SqliteDataReader sldr = slc.ExecuteReader();

			if (sldr.HasRows)
			{
				while (sldr.Read())
				{
					logger.Verbose("Found match with rom type " + sldr.GetString(1));
					headers.Add(sldr.GetString(0));
				}
			}
			else
			{
				logger.Warning("No matching header could be found!");
			}

			// Dispose of database objects
			slc.Dispose();
			sldr.Dispose();
			dbc.Dispose();

			return headers;
		}
	}
}