[SabreTools.Library] Update folder name

2025-12-16 19:14:27 +00:00 · 2017-05-04 02:38:56 -07:00
parent 6b5a23247c
commit f72b06a561
111 changed files with 98 additions and 3 deletions
--- a/SabreTools.Library/Tools/DatabaseTools.cs
+++ b/SabreTools.Library/Tools/DatabaseTools.cs
@@ -0,0 +1,202 @@
+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>
+		public static void AddHeaderToDatabase(string header, string SHA1, string source)
+		{
+			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);
+				Globals.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>
+		/// <returns>List of strings representing the headers to add</returns>
+		public static List<string> RetrieveHeadersFromDatabase(string SHA1)
+		{
+			// 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())
+				{
+					Globals.Logger.Verbose("Found match with rom type " + sldr.GetString(1));
+					headers.Add(sldr.GetString(0));
+				}
+			}
+			else
+			{
+				Globals.Logger.Warning("No matching header could be found!");
+			}
+
+			// Dispose of database objects
+			slc.Dispose();
+			sldr.Dispose();
+			dbc.Dispose();
+
+			return headers;
+		}
+	}
+}