highperfocused/void.cat
1
0
Fork 0
mirror of https://git.v0l.io/Kieran/void.cat.git synced 2025-09-26 02:36:23 +02:00
Code Issues Packages Projects Releases Wiki Activity

Compute digest if missing

Browse Source
This commit is contained in:
Kieran
2022-06-13 18:49:28 +01:00
parent 7155c995c0
commit 5a424e4d16
1 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
VoidCat/Services/Migrations/MigrateToPostgres.cs
View File

@@ -1,4 +1,5 @@
using Newtonsoft.Json;
using System.Security.Cryptography;
using Newtonsoft.Json;
using VoidCat.Model;
using VoidCat.Services.Abstractions;
using VoidCat.Services.Files;
@@ -17,9 +18,11 @@ public class MigrateToPostgres : IMigration
private readonly IPaywallStore _paywallStore;
private readonly IUserStore _userStore;
private readonly IUserUploadsStore _userUploads;
private readonly IFileStore _fileStore;
public MigrateToPostgres(VoidSettings settings, ILogger<MigrateToPostgres> logger, IFileMetadataStore fileMetadata,
ICache cache, IPaywallStore paywallStore, IUserStore userStore, IUserUploadsStore userUploads)
ICache cache, IPaywallStore paywallStore, IUserStore userStore, IUserUploadsStore userUploads,
IFileStore fileStore)
{
_logger = logger;
_settings = settings;
@@ -28,6 +31,7 @@ public class MigrateToPostgres : IMigration
_paywallStore = paywallStore;
_userStore = userStore;
_userUploads = userUploads;
_fileStore = fileStore;
}
/// <inheritdoc />
@@ -64,12 +68,21 @@ public class MigrateToPostgres : IMigration
{
try
{
if (string.IsNullOrEmpty(file.Digest))
{
var fs = await _fileStore.Open(new(file.Id, Enumerable.Empty<RangeRequest>()),
CancellationToken.None);
var hash = await SHA256.Create().ComputeHashAsync(fs);
file.Digest = hash.ToHex();
}
file.MimeType ??= "application/octet-stream";
await _fileMetadata.Set(file.Id, file);
if (file.Uploader.HasValue)
{
await _userUploads.AddFile(file.Uploader.Value, file.Id);
}
await localDiskMetaStore.Delete(file.Id);
_logger.LogInformation("Migrated file metadata for {File}", file.Id);
}
@@ -114,7 +127,7 @@ public class MigrateToPostgres : IMigration
{
var privateUser = await cacheUsers.Get<PrivateUser>(user.Id);
privateUser!.Password ??= privateUser.PasswordHash;
await _userStore.Set(privateUser!.Id, new InternalVoidUser()
{
Id = privateUser.Id,
@@ -144,6 +157,8 @@ public class MigrateToPostgres : IMigration
private record UploaderSecretVoidFileMeta : SecretVoidFileMeta
{
public new string? Digest { get; set; }
[JsonConverter(typeof(Base58GuidConverter))]
public Guid? Uploader { get; set; }
}