ProxyFactory ensure that entity exists in database

Author: usercode

src/DragonFly.Core/Modules/Assets/Models/Asset.cs

@@ -11,15 +11,6 @@ public class Asset : ContentBase<Asset>
 {
     public Asset()
     {
-        //Name = string.Empty;
-        //Alt = string.Empty;
-        //Description = string.Empty;
-        //Slug = string.Empty;
-        //MimeType = string.Empty;
-        //Hash = string.Empty;
-        //PreviewUrl = string.Empty;
-        //Size = 0;
-        //_metaddata = new AssetMetadatas();
     }
 
     public Asset(Guid id)

src/DragonFly.Generator/ProxyGenerator.cs

@@ -98,6 +98,11 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
                                 {
                                     bool ignoreProperty = ignoredProperties.Contains(propertySymbol.Name);
 
+                                    if (ignoreProperty)
+                                    {
+                                        continue;
+                                    }
+
                                     builder.AppendLine($"public override {propertySymbol.Type.ToDisplayString()} {propertySymbol.Name}");
                                     builder.AppendBlock(x =>
                                     {
@@ -106,13 +111,9 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
                                             builder.AppendLine($"get");
                                             builder.AppendBlock(x =>
                                             {
-                                                if (ignoreProperty == false)
-                                                {
-                                                    x.AppendLine($"{methodCaller}(\"{propertySymbol.GetMethod.Name}\"){methodCallerAsyncExtensions};");
-                                                }
+                                                x.AppendLine($"{methodCaller}(\"{propertySymbol.GetMethod.Name}\"){methodCallerAsyncExtensions};");                
                                                 x.AppendLine($"return _invocationTarget.{propertySymbol.Name};");
                                             });
-
                                         }
                                         if (propertySymbol.SetMethod != null)
                                         {
@@ -136,17 +137,19 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
                                 {
                                     bool ignoreMethod = ignoredMethods.Contains(methodSymbol.Name);
 
+                                    if (ignoreMethod)
+                                    {
+                                        continue;
+                                    }
+
                                     string parameterDefinition = $"{string.Join(", ", methodSymbol.Parameters.Select(x => $"{x.Type.ToDisplayString()} {x.Name}"))}";
 
                                     builder.AppendTabs();
                                     builder.Append($"public override {methodSymbol.ReturnType.OriginalDefinition.ToDisplayString()} {methodSymbol.Name}({parameterDefinition})");
                                     builder.AppendLineBreak();
                                     builder.AppendBlock(x =>
                                     {
-                                        if (ignoreMethod == false)
-                                        {
-                                            builder.AppendLine($"{methodCaller}(\"{methodSymbol.Name}\"){methodCallerAsyncExtensions};");
-                                        }
+                                        builder.AppendLine($"{methodCaller}(\"{methodSymbol.Name}\"){methodCallerAsyncExtensions};");
 
                                         builder.AppendTabs();
 

src/DragonFly.Generator/SourceBuilder/SourceBuilder.cs

@@ -67,5 +67,23 @@ public void AppendBlock(Action<SourceBuilder> action, bool command = false)
         }
     }
 
+    public void AppendTryCatch(Action<SourceBuilder> tryBlock, Action<SourceBuilder>? catchBlock = null, Action<SourceBuilder>? finallyBlock = null)
+    {
+        AppendLine("try");
+        AppendBlock(tryBlock);
+
+        if (catchBlock != null)
+        {
+            AppendLine("catch (Exception ex)");
+            AppendBlock(catchBlock);
+        }
+
+        if (finallyBlock != null)
+        {
+            AppendLine("finally");
+            AppendBlock(finallyBlock);
+        }
+    }
+
     public override string ToString() => Builder.ToString();
 }

src/DragonFly.MongoDB/Proxies/ProxyFactory.cs

@@ -2,23 +2,41 @@
 // https://github.com/usercode/DragonFly
 // MIT License
 
+using MongoDB.Driver;
+
 namespace DragonFly.MongoDB;
 
 internal static class ProxyFactory
 {
-    public static ContentItem CreateContent(string schema, Guid id)
+    public static ContentItem? CreateContent(string schema, Guid id)
     {
-        return new ContentItemProxy(CreateSchema(schema), id);
+        var collection = MongoStorage.Default.GetMongoCollection(schema);
+
+        if (collection.AsQueryable().Where(x => x.Id == id).Any())
+        {
+            return new ContentItemProxy(CreateSchema(schema), id);
+        }
+        else
+        {
+            return null;
+        }
     }
 
     public static ContentSchema CreateSchema(string schema)
     {
         return new ContentSchemaProxy(schema);
     }
 
-    public static Asset CreateAsset(Guid assetId)
+    public static Asset? CreateAsset(Guid assetId)
     {
-        return new AssetProxy(assetId);
+        if (MongoStorage.Default.Assets.AsQueryable().Where(x => x.Id == assetId).Any())
+        {
+            return new AssetProxy(assetId);
+        }
+        else
+        {
+            return null;
+        }
     }
 
     public static AssetFolder CreateAssetFolder(Guid assetFolderId)