Innovenergy_trunk/csharp/App/Backend/Database/Db.cs

234 lines
7.6 KiB
C#
Raw Normal View History

2023-03-21 10:40:23 +00:00
using System.Reactive.Concurrency;
2023-03-15 13:38:06 +00:00
using System.Reactive.Linq;
using InnovEnergy.App.Backend.DataTypes;
2023-03-16 09:34:47 +00:00
using InnovEnergy.App.Backend.DataTypes.Methods;
2023-03-15 13:38:06 +00:00
using InnovEnergy.App.Backend.Relations;
2023-10-16 09:27:19 +00:00
using InnovEnergy.Lib.S3Utils;
using InnovEnergy.Lib.S3Utils.DataTypes;
2023-10-26 12:09:38 +00:00
using InnovEnergy.Lib.Utils;
2023-02-16 12:57:06 +00:00
using SQLite;
using SQLiteConnection = SQLite.SQLiteConnection;
2023-02-16 12:57:06 +00:00
2023-03-15 13:38:06 +00:00
2023-03-08 12:20:33 +00:00
namespace InnovEnergy.App.Backend.Database;
2023-02-16 12:57:06 +00:00
2023-11-20 16:29:45 +00:00
public static partial class Db
2023-02-16 12:57:06 +00:00
{
private static SQLiteConnection Connection { get; } = InitConnection();
2023-03-09 15:32:11 +00:00
private static SQLiteConnection InitConnection()
{
var latestDb = new DirectoryInfo("DbBackups")
2023-11-20 16:29:45 +00:00
.GetFiles()
.OrderBy(f => f.LastWriteTime)
.Last().Name;
//This is the file connection from the DbBackups folder
var fileConnection = new SQLiteConnection("DbBackups/" + latestDb);
2023-11-20 16:29:45 +00:00
//Create a table if it does not exist
fileConnection.CreateTable<User>();
fileConnection.CreateTable<Installation>();
fileConnection.CreateTable<Folder>();
fileConnection.CreateTable<FolderAccess>();
fileConnection.CreateTable<InstallationAccess>();
fileConnection.CreateTable<Session>();
fileConnection.CreateTable<OrderNumber2Installation>();
fileConnection.CreateTable<Error>();
2023-11-20 16:29:45 +00:00
fileConnection.CreateTable<Warning>();
return CopyDbToMemory(fileConnection);
}
2023-09-15 11:34:28 +00:00
private static SQLiteConnection CopyDbToMemory(SQLiteConnection fileConnection)
{
var memoryConnection = new SQLiteConnection(":memory:");
2023-11-20 16:29:45 +00:00
//Create a table if it does not exist in main memory
memoryConnection.CreateTable<User>();
memoryConnection.CreateTable<Installation>();
memoryConnection.CreateTable<Folder>();
memoryConnection.CreateTable<FolderAccess>();
memoryConnection.CreateTable<InstallationAccess>();
memoryConnection.CreateTable<Session>();
memoryConnection.CreateTable<OrderNumber2Installation>();
memoryConnection.CreateTable<Error>();
2023-11-20 16:29:45 +00:00
memoryConnection.CreateTable<Warning>();
//Copy all the existing tables from the disk to main memory
fileConnection.Table<Session>().ForEach(memoryConnection.Insert);
fileConnection.Table<Folder>().ForEach(memoryConnection.Insert);
fileConnection.Table<Installation>().ForEach(memoryConnection.Insert);
fileConnection.Table<User>().ForEach(memoryConnection.Insert);
fileConnection.Table<FolderAccess>().ForEach(memoryConnection.Insert);
fileConnection.Table<InstallationAccess>().ForEach(memoryConnection.Insert);
fileConnection.Table<OrderNumber2Installation>().ForEach(memoryConnection.Insert);
fileConnection.Table<Error>().ForEach(memoryConnection.Insert);
2023-11-20 16:29:45 +00:00
fileConnection.Table<Warning>().ForEach(memoryConnection.Insert);
return memoryConnection;
}
public static void BackupDatabase()
{
var filename = "db-" + DateTimeOffset.UtcNow.ToUnixTimeSeconds() + ".sqlite";
2023-09-15 11:34:28 +00:00
Connection.Backup("DbBackups/" + filename);
}
2023-11-20 16:29:45 +00:00
public static TableQuery<Session> Sessions => Connection.Table<Session>();
public static TableQuery<Folder> Folders => Connection.Table<Folder>();
public static TableQuery<Installation> Installations => Connection.Table<Installation>();
public static TableQuery<User> Users => Connection.Table<User>();
public static TableQuery<FolderAccess> FolderAccess => Connection.Table<FolderAccess>();
public static TableQuery<InstallationAccess> InstallationAccess => Connection.Table<InstallationAccess>();
public static TableQuery<OrderNumber2Installation> OrderNumber2Installation => Connection.Table<OrderNumber2Installation>();
public static TableQuery<Error> Errors => Connection.Table<Error>();
public static TableQuery<Warning> Warnings => Connection.Table<Warning>();
2023-11-20 16:29:45 +00:00
public static void Init()
{
// used to force static constructor
//Since this class is static, we call Init method from the Program.cs to initialize all the fields of the class
2023-03-23 12:28:55 +00:00
}
2023-11-20 16:29:45 +00:00
//This is the constructor of the class
2023-02-16 12:57:06 +00:00
static Db()
{
2023-03-15 13:38:06 +00:00
Connection.RunInTransaction(() =>
2023-02-16 12:57:06 +00:00
{
2023-03-15 13:38:06 +00:00
Connection.CreateTable<User>();
Connection.CreateTable<Installation>();
Connection.CreateTable<Folder>();
Connection.CreateTable<FolderAccess>();
Connection.CreateTable<InstallationAccess>();
2023-03-15 13:38:06 +00:00
Connection.CreateTable<Session>();
Connection.CreateTable<OrderNumber2Installation>();
Connection.CreateTable<Error>();
2023-11-20 16:29:45 +00:00
Connection.CreateTable<Warning>();
2023-02-16 12:57:06 +00:00
});
2023-11-20 16:29:45 +00:00
UpdateKeys().SupressAwaitWarning();
CleanupSessions().SupressAwaitWarning();
}
private static async Task CleanupSessions()
{
while (true)
{
try
{
DeleteStaleSessions();
}
catch(Exception e)
{
Console.WriteLine("An error has occured when cleaning stale sessions, exception is:\n"+e);
}
2023-03-09 15:32:11 +00:00
2023-11-20 16:29:45 +00:00
await Task.Delay(TimeSpan.FromHours(0.5));
}
2023-02-16 12:57:06 +00:00
}
2023-03-09 15:32:11 +00:00
2023-11-20 16:29:45 +00:00
private static async Task UpdateKeys()
{
while (true)
{
try
{
await UpdateS3Urls();
}
catch(Exception e)
{
Console.WriteLine("An error has occured when updating S3 keys, exception is:\n"+e);
}
await Task.Delay(TimeSpan.FromHours(24));
}
}
2023-03-15 13:38:06 +00:00
private static Boolean RunTransaction(Func<Boolean> func)
2023-02-16 12:57:06 +00:00
{
2023-03-15 13:38:06 +00:00
var savepoint = Connection.SaveTransactionPoint();
var success = false;
2023-11-20 16:29:45 +00:00
2023-02-16 12:57:06 +00:00
try
{
2023-03-15 13:38:06 +00:00
success = func();
2023-02-16 12:57:06 +00:00
}
2023-03-15 13:38:06 +00:00
finally
2023-02-16 12:57:06 +00:00
{
2023-03-15 13:38:06 +00:00
if (success)
Connection.Release(savepoint);
else
Connection.RollbackTo(savepoint);
2023-02-16 12:57:06 +00:00
}
2023-03-09 15:32:11 +00:00
2023-03-15 13:38:06 +00:00
return success;
2023-02-16 12:57:06 +00:00
}
2023-11-20 16:29:45 +00:00
2023-03-09 11:50:21 +00:00
2023-03-15 13:38:06 +00:00
private static void DeleteStaleSessions()
2023-03-09 11:50:21 +00:00
{
2023-11-20 16:29:45 +00:00
var deadline = DateTime.Now.AddDays((-1) * Session.MaxAge.Days);
2023-03-15 13:38:06 +00:00
Sessions.Delete(s => s.LastSeen < deadline);
2023-03-09 11:50:21 +00:00
}
2023-07-06 08:57:57 +00:00
private static async Task UpdateS3Urls()
{
2023-10-16 09:27:19 +00:00
var regions = Installations
2023-11-20 16:29:45 +00:00
.Select(i => i.S3Region)
.Distinct()
.ToList();
2023-10-16 09:27:19 +00:00
const String provider = "exo.io";
2023-11-20 16:29:45 +00:00
2023-10-16 09:27:19 +00:00
foreach (var region in regions)
2023-07-06 08:57:57 +00:00
{
2023-11-20 16:29:45 +00:00
var s3Region = new S3Region($"https://{region}.{provider}", ExoCmd.S3Credentials!);
2023-10-26 12:09:38 +00:00
var bucketList = await s3Region.ListAllBuckets();
2023-11-20 16:29:45 +00:00
var installations = from bucket in bucketList.Buckets
from installation in Installations
where installation.BucketName() == bucket.BucketName
select installation;
2023-10-26 12:09:38 +00:00
foreach (var installation in installations)
2023-10-16 09:27:19 +00:00
{
2023-10-26 12:09:38 +00:00
await installation.RenewS3Credentials();
2023-10-16 09:27:19 +00:00
}
2023-07-06 08:57:57 +00:00
}
}
2023-09-15 12:23:22 +00:00
2023-10-26 12:09:38 +00:00
public static async Task<Boolean> SendPasswordResetEmail(User user, String sessionToken)
2023-09-15 12:23:22 +00:00
{
2023-10-26 12:09:38 +00:00
try
{
await user.SendPasswordResetEmail(sessionToken);
return true;
}
2023-11-20 16:29:45 +00:00
catch
2023-10-26 12:09:38 +00:00
{
return false;
}
2023-09-15 12:23:22 +00:00
}
2023-11-20 16:29:45 +00:00
2023-10-26 12:09:38 +00:00
public static async Task<Boolean> SendNewUserEmail(User user)
2023-10-23 14:19:03 +00:00
{
2023-10-26 12:09:38 +00:00
try
{
await user.SendNewUserWelcomeMessage();
return true;
}
2023-11-20 16:29:45 +00:00
catch
2023-10-26 12:09:38 +00:00
{
return false;
}
2023-10-23 14:19:03 +00:00
}
2023-09-15 12:23:22 +00:00
public static Boolean DeleteUserPassword(User user)
{
user.Password = "";
user.MustResetPassword = true;
return Update(user);
}
2023-03-09 15:32:11 +00:00
}