extensions rework pt1

2026-07-13 22:20:47 +04:00
parent 46487a4794
commit 0e7986996f
30 changed files with 1476 additions and 160 deletions
@@ -0,0 +1,47 @@
+using System.Collections.Immutable;
+using System.Linq;
+using System.Text;
+using Just.Railway.SourceGen.Executors.ResultExtensions.MainExtensions;
+using Microsoft.CodeAnalysis;
+
+namespace Just.Railway.SourceGen.Executors.ResultExtensions.ValueTaskExtensions;
+
+internal class ResultValueTaskAppendExecutor : ResultAppendExecutor
+{
+    protected override string NamespaceName => "Just.Railway.ValueTaskExtensions";
+    protected override string PartialClassName => "ResultValueTaskExtensions";
+
+    protected override void GenerateHelperMethods(StringBuilder sb)
+    {
+        sb.AppendLine("""
+        private static IEnumerable<string> GetBottom(ResultState r1, ResultState r2, string firstArg = "result", string secondArg = "next")
+        {
+            if (r1 == ResultState.Bottom)
+                yield return firstArg;
+            if (r2 == ResultState.Bottom)
+                yield return secondArg;
+        }
+        """);
+    }
+
+    protected override void GenerateMethodsForArgCount(StringBuilder sb, int argCount)
+    {
+        var templateArgNames = Enumerable.Range(1, argCount)
+            .Select(i => $"T{i}")
+            .ToImmutableArray();
+
+        string resultTypeDef = GenerateResultTypeDef(templateArgNames);
+        string resultValueExpansion = GenerateResultValueExpansion(templateArgNames);
+
+        sb.AppendLine($"#region {resultTypeDef}");
+
+        GenerateAsyncMethods("ValueTask", "ValueTask", sb, templateArgNames, resultTypeDef, resultValueExpansion);
+
+        if (argCount < Constants.MaxResultTupleSize)
+        {
+            GenerateExpandedAsyncMethods("ValueTask", "ValueTask", sb, templateArgNames, resultTypeDef, resultValueExpansion);
+        }
+
+        sb.AppendLine("#endregion");
+    }
+}
@@ -0,0 +1,29 @@
+using System.Collections.Immutable;
+using System.Linq;
+using System.Text;
+using Just.Railway.SourceGen.Executors.ResultExtensions.MainExtensions;
+using Microsoft.CodeAnalysis;
+
+namespace Just.Railway.SourceGen.Executors.ResultExtensions.ValueTaskExtensions;
+
+internal class ResultValueTaskBindExecutor : ResultBindExecutor
+{
+    protected override string NamespaceName => "Just.Railway.ValueTaskExtensions";
+    protected override string PartialClassName => "ResultValueTaskExtensions";
+
+    protected override void GenerateMethodsForArgCount(StringBuilder sb, int argCount)
+    {
+        var templateArgNames = Enumerable.Range(1, argCount)
+            .Select(i => $"T{i}")
+            .ToImmutableArray();
+
+        string resultTypeDef = GenerateResultTypeDef(templateArgNames);
+        string resultValueExpansion = GenerateResultValueExpansion(templateArgNames);
+
+        sb.AppendLine($"#region {resultTypeDef}");
+
+        GenerateAsyncMethods("ValueTask", "ValueTask", sb, templateArgNames, resultTypeDef, resultValueExpansion);
+
+        sb.AppendLine("#endregion");
+    }
+}
@@ -0,0 +1,35 @@
+using System.Collections.Immutable;
+using System.Linq;
+using System.Text;
+using Just.Railway.SourceGen.Executors.ResultExtensions.MainExtensions;
+using Microsoft.CodeAnalysis;
+
+namespace Just.Railway.SourceGen.Executors.ResultExtensions.ValueTaskExtensions;
+
+internal sealed class ResultValueTaskExtendExecutor : ResultExtendExecutor
+{
+    protected override string NamespaceName => "Just.Railway.ValueTaskExtensions";
+    protected override string PartialClassName => "ResultValueTaskExtensions";
+
+    protected override void GenerateMethodsForArgCount(StringBuilder sb, int argCount)
+    {
+        if (argCount == 0 || argCount == Constants.MaxResultTupleSize)
+        {
+            return;
+        }
+
+        var templateArgNames = Enumerable.Range(1, argCount)
+            .Select(i => $"T{i}")
+            .ToImmutableArray();
+
+        var expandedTemplateArgNames = templateArgNames.Add("R");
+        string resultTypeDef = GenerateResultTypeDef(templateArgNames);
+        string resultValueExpansion = GenerateResultValueExpansion(templateArgNames);
+
+        sb.AppendLine($"#region {resultTypeDef}");
+
+        GenerateAsyncMethods("ValueTask", "ValueTask", sb, templateArgNames, resultTypeDef, resultValueExpansion);
+
+        sb.AppendLine("#endregion");
+    }
+}
@@ -0,0 +1,28 @@
+using System.Collections.Immutable;
+using System.Linq;
+using System.Text;
+using Microsoft.CodeAnalysis;
+
+namespace Just.Railway.SourceGen.Executors.ResultExtensions.ValueTaskExtensions;
+
+internal sealed class ResultValueTaskMapExecutor : ResultMapExecutor
+{
+    protected override string NamespaceName => "Just.Railway.ValueTaskExtensions";
+    protected override string PartialClassName => "ResultValueTaskExtensions";
+
+    protected override void GenerateMethodsForArgCount(StringBuilder sb, int argCount)
+    {
+        var templateArgNames = Enumerable.Range(1, argCount)
+            .Select(i => $"T{i}")
+            .ToImmutableArray();
+
+        string resultTypeDef = GenerateResultTypeDef(templateArgNames);
+        string resultValueExpansion = GenerateResultValueExpansion(templateArgNames);
+
+        sb.AppendLine($"#region {resultTypeDef}");
+
+        GenerateAsyncMethods("ValueTask", "ValueTask", sb, templateArgNames, resultTypeDef, resultValueExpansion);
+
+        sb.AppendLine("#endregion");
+    }
+}
@@ -0,0 +1,29 @@
+using System.Collections.Immutable;
+using System.Linq;
+using System.Text;
+using Just.Railway.SourceGen.Executors.ResultExtensions.MainExtensions;
+using Microsoft.CodeAnalysis;
+
+namespace Just.Railway.SourceGen.Executors.ResultExtensions.ValueTaskExtensions;
+
+internal class ResultValueTaskMatchExecutor : ResultMatchExecutor
+{
+    protected override string NamespaceName => "Just.Railway.ValueTaskExtensions";
+    protected override string PartialClassName => "ResultValueTaskExtensions";
+
+    protected override void GenerateMethodsForArgCount(StringBuilder sb, int argCount)
+    {
+        var templateArgNames = Enumerable.Range(1, argCount)
+            .Select(i => $"T{i}")
+            .ToImmutableArray();
+
+        string resultTypeDef = GenerateResultTypeDef(templateArgNames);
+        string resultValueExpansion = GenerateResultValueExpansion(templateArgNames);
+
+        sb.AppendLine($"#region {resultTypeDef}");
+
+        GenerateAsyncMethods("ValueTask", "ValueTask", sb, templateArgNames, resultTypeDef, resultValueExpansion);
+
+        sb.AppendLine("#endregion");
+    }
+}
@@ -0,0 +1,29 @@
+using System.Collections.Immutable;
+using System.Linq;
+using System.Text;
+using Just.Railway.SourceGen.Executors.ResultExtensions.MainExtensions;
+using Microsoft.CodeAnalysis;
+
+namespace Just.Railway.SourceGen.Executors.ResultExtensions.ValueTaskExtensions;
+
+internal class ResultValueTaskTapExecutor : ResultTapExecutor
+{
+    protected override string NamespaceName => "Just.Railway.ValueTaskExtensions";
+    protected override string PartialClassName => "ResultValueTaskExtensions";
+
+    protected override void GenerateMethodsForArgCount(StringBuilder sb, int argCount)
+    {
+        var templateArgNames = Enumerable.Range(1, argCount)
+            .Select(i => $"T{i}")
+            .ToImmutableArray();
+
+        string resultTypeDef = GenerateResultTypeDef(templateArgNames);
+        string resultValueExpansion = GenerateResultValueExpansion(templateArgNames);
+
+        sb.AppendLine($"#region {resultTypeDef}");
+
+        GenerateAsyncMethods("ValueTask", "ValueTask", sb, templateArgNames, resultTypeDef, resultValueExpansion);
+
+        sb.AppendLine("#endregion");
+    }
+}
@@ -0,0 +1,30 @@
+using System.Collections.Immutable;
+using System.Linq;
+using System.Text;
+using Just.Railway.SourceGen.Executors.ResultExtensions.MainExtensions;
+
+namespace Just.Railway.SourceGen.Executors.ResultExtensions.ValueTaskExtensions;
+
+internal class ResultValueTaskTryRecoverExecutor : ResultTryRecoverExecutor
+{
+    protected override string NamespaceName => "Just.Railway.ValueTaskExtensions";
+    protected override string PartialClassName => "ResultValueTaskExtensions";
+
+
+    protected override void GenerateMethodsForArgCount(StringBuilder sb, int argCount)
+    {
+        if (argCount > 1) return;
+
+        var templateArgNames = Enumerable.Repeat("T", argCount)
+            .ToImmutableArray();
+
+        string methodTemplateDecl = GenerateTemplateDecl(templateArgNames);
+        string resultTypeDef = GenerateResultTypeDef(templateArgNames);
+
+        sb.AppendLine($"#region {resultTypeDef}");
+
+        GenerateAsyncMethods("ValueTask", "ValueTask", sb, templateArgNames, resultTypeDef, methodTemplateDecl);
+
+        sb.AppendLine("#endregion");
+    }
+}