234 lines
11 KiB
C#
234 lines
11 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Collections.Immutable;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using Microsoft.CodeAnalysis;
|
|
|
|
namespace Just.Railway.SourceGen;
|
|
|
|
public sealed class EnsureExtensionsExecutor : IGeneratorExecutor
|
|
{
|
|
public void Execute(SourceProductionContext context, Compilation source)
|
|
{
|
|
var methods = GenerateMethods();
|
|
var code = $$"""
|
|
#nullable enable
|
|
using System;
|
|
using System.Linq;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics.Contracts;
|
|
using System.CodeDom.Compiler;
|
|
|
|
namespace Just.Railway;
|
|
|
|
public static partial class Ensure
|
|
{
|
|
{{methods}}
|
|
}
|
|
""";
|
|
|
|
context.AddSource("Ensure.Extensions.g.cs", code);
|
|
}
|
|
|
|
private string GenerateMethods()
|
|
{
|
|
List<(string ErrorParameterDecl, string ErrorValueExpr)> errorGenerationDefinitions =
|
|
[
|
|
("Error error = default!", "error"),
|
|
("ErrorFactory errorFactory", "errorFactory(ensure.ValueExpression)")
|
|
];
|
|
|
|
var sb = new StringBuilder();
|
|
|
|
sb.AppendLine("#region Satisfies");
|
|
errorGenerationDefinitions.ForEach(def => GenerateSatisfiesExtensions(sb, def.ErrorParameterDecl, def.ErrorValueExpr));
|
|
sb.AppendLine("#endregion");
|
|
|
|
sb.AppendLine("#region NotNull");
|
|
errorGenerationDefinitions.ForEach(def => GenerateNotNullExtensions(sb, def.ErrorParameterDecl, def.ErrorValueExpr));
|
|
sb.AppendLine("#endregion");
|
|
|
|
sb.AppendLine("#region NotEmpty");
|
|
errorGenerationDefinitions.ForEach(def => GenerateNotEmptyExtensions(sb, def.ErrorParameterDecl, def.ErrorValueExpr));
|
|
sb.AppendLine("#endregion");
|
|
|
|
sb.AppendLine("#region NotWhitespace");
|
|
errorGenerationDefinitions.ForEach(def => GenerateNotWhitespaceExtensions(sb, def.ErrorParameterDecl, def.ErrorValueExpr));
|
|
sb.AppendLine("#endregion");
|
|
|
|
return sb.ToString();
|
|
}
|
|
|
|
private void GenerateNotWhitespaceExtensions(StringBuilder sb, string errorParameterDecl, string errorValueExpr)
|
|
{
|
|
string defaultErrorExpr = "?? Error.New(DefaultErrorType, $\"Value {{{ensure.ValueExpression}}} is empty or consists exclusively of white-space characters.\")";
|
|
|
|
sb.AppendLine($$"""
|
|
[PureAttribute]
|
|
[GeneratedCodeAttribute("{{nameof(EnsureExtensionsExecutor)}}", "1.0.0.0")]
|
|
public static Ensure<string> NotWhitespace(this in Ensure<string> ensure, {{errorParameterDecl}})
|
|
{
|
|
return ensure.State switch
|
|
{
|
|
ResultState.Success => string.IsNullOrWhiteSpace(ensure.Value)
|
|
? new({{errorValueExpr}} {{defaultErrorExpr}}, ensure.ValueExpression)
|
|
: new(ensure.Value!, ensure.ValueExpression),
|
|
ResultState.Error => new(ensure.Error!, ensure.ValueExpression),
|
|
_ => throw new EnsureNotInitializedException(nameof(ensure))
|
|
};
|
|
}
|
|
""");
|
|
}
|
|
|
|
private void GenerateNotEmptyExtensions(StringBuilder sb, string errorParameterDecl, string errorValueExpr)
|
|
{
|
|
string defaultErrorExpr = "?? Error.New(DefaultErrorType, $\"Value {{{ensure.ValueExpression}}} is empty.\")";
|
|
List<(string TemplateDef, string CollectionType, string NotEmptyTest)> typeOverloads =
|
|
[
|
|
("<T>", "IEnumerable<T>", "ensure.Value?.Any() == true"),
|
|
("<T>", "ICollection<T>", "ensure.Value?.Count > 0"),
|
|
("<T>", "IReadOnlyCollection<T>", "ensure.Value?.Count > 0"),
|
|
("<T>", "IList<T>", "ensure.Value?.Count > 0"),
|
|
("<T>", "IReadOnlyList<T>", "ensure.Value?.Count > 0"),
|
|
("<T>", "ISet<T>", "ensure.Value?.Count > 0"),
|
|
("<T>", "IReadOnlySet<T>", "ensure.Value?.Count > 0"),
|
|
("<TKey,TValue>", "IDictionary<TKey,TValue>", "ensure.Value?.Count > 0"),
|
|
("<TKey,TValue>", "IReadOnlyDictionary<TKey,TValue>", "ensure.Value?.Count > 0"),
|
|
("<T>", "T[]", "ensure.Value?.Length > 0"),
|
|
("<T>", "List<T>", "ensure.Value?.Count > 0"),
|
|
("<T>", "Queue<T>", "ensure.Value?.Count > 0"),
|
|
("<T>", "HashSet<T>", "ensure.Value?.Count > 0"),
|
|
("", "string", "!string.IsNullOrEmpty(ensure.Value)"),
|
|
];
|
|
|
|
typeOverloads.ForEach(def => sb.AppendLine($$"""
|
|
[PureAttribute]
|
|
[GeneratedCodeAttribute("{{nameof(EnsureExtensionsExecutor)}}", "1.0.0.0")]
|
|
public static Ensure<{{def.CollectionType}}> NotEmpty{{def.TemplateDef}}(this in Ensure<{{def.CollectionType}}> ensure, {{errorParameterDecl}})
|
|
{
|
|
return ensure.State switch
|
|
{
|
|
ResultState.Success => {{def.NotEmptyTest}}
|
|
? new(ensure.Value, ensure.ValueExpression)
|
|
: new({{errorValueExpr}} {{defaultErrorExpr}}, ensure.ValueExpression),
|
|
ResultState.Error => new(ensure.Error!, ensure.ValueExpression),
|
|
_ => throw new EnsureNotInitializedException(nameof(ensure))
|
|
};
|
|
}
|
|
"""));
|
|
}
|
|
|
|
private void GenerateNotNullExtensions(StringBuilder sb, string errorParameterDecl, string errorValueExpr)
|
|
{
|
|
string defaultErrorExpr = "?? Error.New(DefaultErrorType, $\"Value {{{ensure.ValueExpression}}} is null.\")";
|
|
|
|
sb.AppendLine($$"""
|
|
[PureAttribute]
|
|
[GeneratedCodeAttribute("{{nameof(EnsureExtensionsExecutor)}}", "1.0.0.0")]
|
|
public static Ensure<T> NotNull<T>(this in Ensure<T?> ensure, {{errorParameterDecl}})
|
|
where T : struct
|
|
{
|
|
return ensure.State switch
|
|
{
|
|
ResultState.Success => ensure.Value.HasValue
|
|
? new(ensure.Value.Value, ensure.ValueExpression)
|
|
: new({{errorValueExpr}} {{defaultErrorExpr}}, ensure.ValueExpression),
|
|
ResultState.Error => new(ensure.Error!, ensure.ValueExpression),
|
|
_ => throw new EnsureNotInitializedException(nameof(ensure))
|
|
};
|
|
}
|
|
""");
|
|
sb.AppendLine($$"""
|
|
[PureAttribute]
|
|
[GeneratedCodeAttribute("{{nameof(EnsureExtensionsExecutor)}}", "1.0.0.0")]
|
|
public static Ensure<T> NotNull<T>(this in Ensure<T?> ensure, {{errorParameterDecl}})
|
|
where T : notnull
|
|
{
|
|
return ensure.State switch
|
|
{
|
|
ResultState.Success => ensure.Value is not null
|
|
? new(ensure.Value!, ensure.ValueExpression)
|
|
: new({{errorValueExpr}} {{defaultErrorExpr}}, ensure.ValueExpression),
|
|
ResultState.Error => new(ensure.Error!, ensure.ValueExpression),
|
|
_ => throw new EnsureNotInitializedException(nameof(ensure))
|
|
};
|
|
}
|
|
""");
|
|
}
|
|
|
|
private void GenerateSatisfiesExtensions(StringBuilder sb, string errorParameterDecl, string errorValueExpr)
|
|
{
|
|
string defaultErrorExpr = "?? Error.New(DefaultErrorType, $\"Value {{{ensure.ValueExpression}}} does not satisfy the requirement.\")";
|
|
sb.AppendLine($$"""
|
|
[PureAttribute]
|
|
[GeneratedCodeAttribute("{{nameof(EnsureExtensionsExecutor)}}", "1.0.0.0")]
|
|
public static Ensure<T> Satisfies<T>(this in Ensure<T> ensure, Func<T, bool> requirement, {{errorParameterDecl}})
|
|
{
|
|
return ensure.State switch
|
|
{
|
|
ResultState.Success => requirement(ensure.Value)
|
|
? new(ensure.Value, ensure.ValueExpression)
|
|
: new({{errorValueExpr}} {{defaultErrorExpr}}, ensure.ValueExpression),
|
|
ResultState.Error => new(ensure.Error!, ensure.ValueExpression),
|
|
_ => throw new EnsureNotInitializedException(nameof(ensure))
|
|
};
|
|
}
|
|
""");
|
|
|
|
GenerateSatisfiesAsyncExtensions(sb, "Task", errorParameterDecl, errorValueExpr, defaultErrorExpr);
|
|
GenerateSatisfiesAsyncExtensions(sb, "ValueTask", errorParameterDecl, errorValueExpr, defaultErrorExpr);
|
|
}
|
|
|
|
private void GenerateSatisfiesAsyncExtensions(StringBuilder sb, string taskType, string errorParameterDecl, string errorValueExpr, string defaultErrorExpr)
|
|
{
|
|
sb.AppendLine($$"""
|
|
[PureAttribute]
|
|
[GeneratedCodeAttribute("{{nameof(EnsureExtensionsExecutor)}}", "1.0.0.0")]
|
|
public static async {{taskType}}<Ensure<T>> Satisfies<T>(this {{taskType}}<Ensure<T>> ensureTask, Func<T, bool> requirement, {{errorParameterDecl}})
|
|
{
|
|
var ensure = await ensureTask.ConfigureAwait(false);
|
|
return ensure.State switch
|
|
{
|
|
ResultState.Success => requirement(ensure.Value)
|
|
? new(ensure.Value, ensure.ValueExpression)
|
|
: new({{errorValueExpr}} {{defaultErrorExpr}}, ensure.ValueExpression),
|
|
ResultState.Error => new(ensure.Error!, ensure.ValueExpression),
|
|
_ => throw new EnsureNotInitializedException(nameof(ensureTask))
|
|
};
|
|
}
|
|
""");
|
|
sb.AppendLine($$"""
|
|
[PureAttribute]
|
|
[GeneratedCodeAttribute("{{nameof(EnsureExtensionsExecutor)}}", "1.0.0.0")]
|
|
public static async {{taskType}}<Ensure<T>> Satisfies<T>(this Ensure<T> ensure, Func<T, {{taskType}}<bool>> requirement, {{errorParameterDecl}})
|
|
{
|
|
return ensure.State switch
|
|
{
|
|
ResultState.Success => await requirement(ensure.Value).ConfigureAwait(false)
|
|
? new(ensure.Value, ensure.ValueExpression)
|
|
: new({{errorValueExpr}} {{defaultErrorExpr}}, ensure.ValueExpression),
|
|
ResultState.Error => new(ensure.Error!, ensure.ValueExpression),
|
|
_ => throw new EnsureNotInitializedException(nameof(ensure))
|
|
};
|
|
}
|
|
""");
|
|
sb.AppendLine($$"""
|
|
[PureAttribute]
|
|
[GeneratedCodeAttribute("{{nameof(EnsureExtensionsExecutor)}}", "1.0.0.0")]
|
|
public static async {{taskType}}<Ensure<T>> Satisfies<T>(this {{taskType}}<Ensure<T>> ensureTask, Func<T, {{taskType}}<bool>> requirement, {{errorParameterDecl}})
|
|
{
|
|
var ensure = await ensureTask.ConfigureAwait(false);
|
|
return ensure.State switch
|
|
{
|
|
ResultState.Success => await requirement(ensure.Value).ConfigureAwait(false)
|
|
? new(ensure.Value, ensure.ValueExpression)
|
|
: new({{errorValueExpr}} {{defaultErrorExpr}}, ensure.ValueExpression),
|
|
ResultState.Error => new(ensure.Error!, ensure.ValueExpression),
|
|
_ => throw new EnsureNotInitializedException(nameof(ensureTask))
|
|
};
|
|
}
|
|
""");
|
|
}
|
|
}
|