118 lines
3.3 KiB
C#
118 lines
3.3 KiB
C#
using Newtonsoft.Json;
|
|
using VoidCat.Model;
|
|
using VoidCat.Services.Abstractions;
|
|
|
|
namespace VoidCat.Services.Files;
|
|
|
|
public class LocalDiskFileMetadataStore : IFileMetadataStore
|
|
{
|
|
private const string MetadataDir = "metadata-v3";
|
|
private readonly ILogger<LocalDiskFileMetadataStore> _logger;
|
|
private readonly VoidSettings _settings;
|
|
|
|
public LocalDiskFileMetadataStore(VoidSettings settings, ILogger<LocalDiskFileMetadataStore> logger)
|
|
{
|
|
_settings = settings;
|
|
_logger = logger;
|
|
|
|
var metaPath = Path.Combine(_settings.DataDirectory, MetadataDir);
|
|
if (!Directory.Exists(metaPath))
|
|
{
|
|
Directory.CreateDirectory(metaPath);
|
|
}
|
|
}
|
|
|
|
public ValueTask<TMeta?> Get<TMeta>(Guid id) where TMeta : VoidFileMeta
|
|
{
|
|
return GetMeta<TMeta>(id);
|
|
}
|
|
|
|
public async ValueTask<IReadOnlyList<TMeta>> Get<TMeta>(Guid[] ids) where TMeta : VoidFileMeta
|
|
{
|
|
var ret = new List<TMeta>();
|
|
foreach (var id in ids)
|
|
{
|
|
var r = await GetMeta<TMeta>(id);
|
|
if (r != null)
|
|
{
|
|
ret.Add(r);
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
public async ValueTask Update<TMeta>(Guid id, TMeta meta) where TMeta : VoidFileMeta
|
|
{
|
|
var oldMeta = await Get<SecretVoidFileMeta>(id);
|
|
if (oldMeta == default) return;
|
|
|
|
oldMeta.Description = meta.Description ?? oldMeta.Description;
|
|
oldMeta.Name = meta.Name ?? oldMeta.Name;
|
|
oldMeta.MimeType = meta.MimeType ?? oldMeta.MimeType;
|
|
|
|
await Set(id, oldMeta);
|
|
}
|
|
|
|
public async ValueTask<IFileMetadataStore.StoreStats> Stats()
|
|
{
|
|
var count = 0;
|
|
var size = 0UL;
|
|
foreach (var metaFile in Directory.EnumerateFiles(Path.Join(_settings.DataDirectory, MetadataDir), "*.json"))
|
|
{
|
|
try
|
|
{
|
|
var json = await File.ReadAllTextAsync(metaFile);
|
|
var meta = JsonConvert.DeserializeObject<VoidFileMeta>(json);
|
|
|
|
if (meta != null)
|
|
{
|
|
count++;
|
|
size += meta.Size;
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Failed to load metadata file: {File}", metaFile);
|
|
}
|
|
}
|
|
|
|
return new(count, size);
|
|
}
|
|
|
|
public ValueTask<VoidFileMeta?> Get(Guid id)
|
|
{
|
|
return GetMeta<VoidFileMeta>(id);
|
|
}
|
|
|
|
public async ValueTask Set(Guid id, SecretVoidFileMeta meta)
|
|
{
|
|
var path = MapMeta(id);
|
|
var json = JsonConvert.SerializeObject(meta);
|
|
await File.WriteAllTextAsync(path, json);
|
|
}
|
|
|
|
public ValueTask Delete(Guid id)
|
|
{
|
|
var path = MapMeta(id);
|
|
if (File.Exists(path))
|
|
{
|
|
_logger.LogInformation("Deleting metadata file {Path}", path);
|
|
File.Delete(path);
|
|
}
|
|
|
|
return ValueTask.CompletedTask;
|
|
}
|
|
|
|
private async ValueTask<TMeta?> GetMeta<TMeta>(Guid id)
|
|
{
|
|
var path = MapMeta(id);
|
|
if (!File.Exists(path)) return default;
|
|
|
|
var json = await File.ReadAllTextAsync(path);
|
|
return JsonConvert.DeserializeObject<TMeta>(json);
|
|
}
|
|
|
|
private string MapMeta(Guid id) =>
|
|
Path.ChangeExtension(Path.Join(_settings.DataDirectory, MetadataDir, id.ToString()), ".json");
|
|
} |