FileDatabase refactor for normalization and consistency

2025-09-04 16:26:10 -04:00
parent e82366e47f
commit 6fefcbcfb5
24 changed files with 3069 additions and 186 deletions
@@ -0,0 +1,111 @@
+using DeepDrftContent.FileDatabase.Abstractions;
+using DeepDrftContent.FileDatabase.Models;
+using DeepDrftContent.FileDatabase.Utils;
+using IndexType = DeepDrftContent.FileDatabase.Services.IndexType;
+
+namespace DeepDrftContent.FileDatabase.Services;
+
+/// <summary>
+/// Factory service for creating and managing indexes
+/// </summary>
+public class IndexFactoryService : IIndexFactory, IIndexDataFactory
+{
+    private readonly Dictionary<IndexType, Func<string, IIndex>> _indexCreators;
+    private readonly Dictionary<IndexType, Func<object, IIndex>> _indexFromDataCreators;
+    private readonly Dictionary<IndexType, Func<IIndex, object>> _indexDataCreators;
+
+    public IndexFactoryService()
+    {
+        _indexCreators = new Dictionary<IndexType, Func<string, IIndex>>
+        {
+            { IndexType.Directory, rootPath => new DirectoryIndex(new DirectoryIndexData(Path.GetFileName(rootPath))) },
+            { IndexType.Vault, rootPath => new VaultIndex(new VaultIndexData(Path.GetFileName(rootPath))) }
+        };
+
+        _indexFromDataCreators = new Dictionary<IndexType, Func<object, IIndex>>
+        {
+            { IndexType.Directory, data => new DirectoryIndex((DirectoryIndexData)data) },
+            { IndexType.Vault, data => new VaultIndex((VaultIndexData)data) }
+        };
+
+        _indexDataCreators = new Dictionary<IndexType, Func<IIndex, object>>
+        {
+            { IndexType.Directory, index => DirectoryIndexData.FromIndex((DirectoryIndex)index) },
+            { IndexType.Vault, index => VaultIndexData.FromIndex((VaultIndex)index) }
+        };
+    }
+
+    public async Task<IIndex?> CreateIndexAsync(IndexType type, string rootPath)
+    {
+        if (!_indexCreators.TryGetValue(type, out var creator))
+        {
+            throw new ArgumentException($"Unknown index type: {type}");
+        }
+
+        var index = creator(rootPath);
+        
+        // Ensure directory exists and save the index
+        await FileUtils.MakeVaultDirectoryAsync(rootPath);
+        await SaveIndexAsync(rootPath, type, index);
+
+        return index;
+    }
+
+    public async Task<IIndex?> LoadIndexAsync(IndexType type, string rootPath)
+    {
+        if (!_indexFromDataCreators.TryGetValue(type, out var creator))
+        {
+            throw new ArgumentException($"Unknown index type: {type}");
+        }
+
+        var indexPath = Path.Combine(rootPath, "index");
+        
+        object indexData = type switch
+        {
+            IndexType.Directory => await FileUtils.FetchObjectAsync<DirectoryIndexData>(indexPath),
+            IndexType.Vault => await FileUtils.FetchObjectAsync<VaultIndexData>(indexPath),
+            _ => throw new ArgumentException($"Unknown index type: {type}")
+        };
+
+        return creator(indexData);
+    }
+
+    public async Task<IIndex?> LoadOrCreateIndexAsync(IndexType type, string rootPath)
+    {
+        try
+        {
+            return await LoadIndexAsync(type, rootPath);
+        }
+        catch
+        {
+            return await CreateIndexAsync(type, rootPath);
+        }
+    }
+
+    public object CreateIndexData(IndexType type, IIndex index)
+    {
+        if (!_indexDataCreators.TryGetValue(type, out var creator))
+        {
+            throw new ArgumentException($"Unknown index type: {type}");
+        }
+
+        return creator(index);
+    }
+
+    public IIndex CreateIndexFromData(IndexType type, object indexData)
+    {
+        if (!_indexFromDataCreators.TryGetValue(type, out var creator))
+        {
+            throw new ArgumentException($"Unknown index type: {type}");
+        }
+
+        return creator(indexData);
+    }
+
+    private async Task SaveIndexAsync(string rootPath, IndexType type, IIndex index)
+    {
+        var indexPath = Path.Combine(rootPath, "index");
+        var indexData = CreateIndexData(type, index);
+        await FileUtils.PutObjectAsync(indexPath, indexData);
+    }
+}