claunia/SabreTools
1
0
Fork 0
mirror of https://github.com/claunia/SabreTools.git synced 2025-12-16 19:14:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Consolodate DB inits

Browse Source
This commit is contained in:
Matt Nadareski
2016-04-06 00:26:13 -07:00
parent ad8d026134
commit 97ee13681b
2 changed files with 41 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
Deheader/Headerer.cs
View File

@@ -6,6 +6,8 @@ using System.Linq;
using System.Security.Cryptography; using System.Security.Cryptography;
using System.Text.RegularExpressions; using System.Text.RegularExpressions;
using SabreTools.Helper;
namespace SabreTools namespace SabreTools
{ {
/// <summary> /// <summary>
@@ -40,7 +42,7 @@ Options:
types.Add("snes", 512); types.Add("snes", 512);
// Ensure that the header database is set up // Ensure that the header database is set up
EnsureDatabase(_dbName, _connectionString); DBTools.EnsureDatabase(_dbName, _connectionString);
if (args.Length == 0 || args.Length > 2) if (args.Length == 0 || args.Length > 2)
{ {
@@ -275,45 +277,5 @@ Options:
} }
} }
} }
/// <summary>
/// Ensure that the databse exists and has the proper schema
/// </summary>
/// <param name="db">Name of the databse</param>
/// <param name="connectionString">Connection string for SQLite</param>
public static void EnsureDatabase(string db, string connectionString)
{
// Make sure the file exists
if (!File.Exists(db))
{
SQLiteConnection.CreateFile(db);
}
// Connect to the file
SQLiteConnection dbc = new SQLiteConnection(connectionString);
dbc.Open();
try
{
// Make sure the database has the correct schema
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();
}
catch (Exception ex)
{
Console.WriteLine(ex);
}
finally
{
// Close and return the database connection
dbc.Close();
}
}
} }
} }

57
SabreHelper/DBTools.cs
View File

@@ -22,13 +22,19 @@ namespace SabreTools.Helper
SQLiteConnection.CreateFile(db); SQLiteConnection.CreateFile(db);
} }
//Get "type" from the filename
string type = Path.GetFileNameWithoutExtension(db);
// Connect to the file // Connect to the file
SQLiteConnection dbc = new SQLiteConnection(connectionString); SQLiteConnection dbc = new SQLiteConnection(connectionString);
dbc.Open(); dbc.Open();
// Make sure the database has the correct schema
try try
{ {
// Make sure the database has the correct schema if (type == "DATabase")
string query = @" {
string query = @"
CREATE TABLE IF NOT EXISTS checksums ( CREATE TABLE IF NOT EXISTS checksums (
'file' INTEGER NOT NULL, 'file' INTEGER NOT NULL,
'size' INTEGER NOT NULL DEFAULT -1, 'size' INTEGER NOT NULL DEFAULT -1,
@@ -37,10 +43,10 @@ CREATE TABLE IF NOT EXISTS checksums (
'sha1' TEXT NOT NULL, 'sha1' TEXT NOT NULL,
PRIMARY KEY (file, size, crc, md5, sha1) PRIMARY KEY (file, size, crc, md5, sha1)
)"; )";
SQLiteCommand slc = new SQLiteCommand(query, dbc); SQLiteCommand slc = new SQLiteCommand(query, dbc);
slc.ExecuteNonQuery(); slc.ExecuteNonQuery();
query = @" query = @"
CREATE TABLE IF NOT EXISTS files ( CREATE TABLE IF NOT EXISTS files (
'id' INTEGER PRIMARY KEY NOT NULL, 'id' INTEGER PRIMARY KEY NOT NULL,
'setid' INTEGER NOT NULL, 'setid' INTEGER NOT NULL,
@@ -48,10 +54,10 @@ CREATE TABLE IF NOT EXISTS files (
'type' TEXT NOT NULL DEFAULT 'rom', 'type' TEXT NOT NULL DEFAULT 'rom',
'lastupdated' TEXT NOT NULL 'lastupdated' TEXT NOT NULL
)"; )";
slc = new SQLiteCommand(query, dbc); slc = new SQLiteCommand(query, dbc);
slc.ExecuteNonQuery(); slc.ExecuteNonQuery();
query = @" query = @"
CREATE TABLE IF NOT EXISTS games ( CREATE TABLE IF NOT EXISTS games (
'id' INTEGER PRIMARY KEY NOT NULL, 'id' INTEGER PRIMARY KEY NOT NULL,
'system' INTEGER NOT NULL, 'system' INTEGER NOT NULL,
@@ -59,34 +65,47 @@ CREATE TABLE IF NOT EXISTS games (
'parent' INTEGER NOT NULL DEFAULT '0', 'parent' INTEGER NOT NULL DEFAULT '0',
'source' INTEGER NOT NULL DEFAULT '0' 'source' INTEGER NOT NULL DEFAULT '0'
)"; )";
slc = new SQLiteCommand(query, dbc); slc = new SQLiteCommand(query, dbc);
slc.ExecuteNonQuery(); slc.ExecuteNonQuery();
query = @" query = @"
CREATE TABLE IF NOT EXISTS parent ( CREATE TABLE IF NOT EXISTS parent (
'id' INTEGER PRIMARY KEY NOT NULL, 'id' INTEGER PRIMARY KEY NOT NULL,
'name' TEXT NOT NULL 'name' TEXT NOT NULL
)"; )";
slc = new SQLiteCommand(query, dbc); slc = new SQLiteCommand(query, dbc);
slc.ExecuteNonQuery(); slc.ExecuteNonQuery();
query = @" query = @"
CREATE TABLE IF NOT EXISTS sources ( CREATE TABLE IF NOT EXISTS sources (
'id' INTEGER PRIMARY KEY NOT NULL, 'id' INTEGER PRIMARY KEY NOT NULL,
'name' TEXT NOT NULL UNIQUE, 'name' TEXT NOT NULL UNIQUE,
'url' TEXT NOT NULL 'url' TEXT NOT NULL
)"; )";
slc = new SQLiteCommand(query, dbc); slc = new SQLiteCommand(query, dbc);
slc.ExecuteNonQuery(); slc.ExecuteNonQuery();
query = @" query = @"
CREATE TABLE IF NOT EXISTS systems ( CREATE TABLE IF NOT EXISTS systems (
'id' INTEGER PRIMARY KEY NOT NULL, 'id' INTEGER PRIMARY KEY NOT NULL,
'manufacturer' TEXT NOT NULL, 'manufacturer' TEXT NOT NULL,
'system' TEXT NOT NULL 'system' TEXT NOT NULL
)"; )";
slc = new SQLiteCommand(query, dbc); slc = new SQLiteCommand(query, dbc);
slc.ExecuteNonQuery(); slc.ExecuteNonQuery();
}
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();
}
} }
catch (Exception ex) catch (Exception ex)
{ {