private static void GenerateGetKeywordKind(KindList kinds, SourceWriter writer) { var optimized = new OptimizedSwitch(); foreach (var keyword in kinds.Keywords.OrderBy(kind => kind.Field.Name)) { optimized.AddClause(keyword.TokenInfo !.Value.Text !, writer => { writer.Write("return SyntaxKind."); writer.Write(keyword.Field.Name); writer.WriteLine(';'); }); } optimized.DefaultBodyWriter = writer => writer.WriteLine("return SyntaxKind.IdentifierToken;"); writeHeader(writer); using (writer.CurlyIndenter("public static SyntaxKind GetKeywordKind(String text)")) { optimized.Generate(writer, "text", false); } writer.WriteLine(); writeHeader(writer); using (writer.CurlyIndenter("public static SyntaxKind GetKeywordKind(ReadOnlySpan<char> span)")) { optimized.Generate(writer, "span", true); }
private static void GenerateGetBinaryOperatorPrecedence(KindList kinds, SourceWriter writer) { writer.WriteLine("/// <summary>"); writer.WriteLine("/// Returns the precedence for a given binary operator. Returns 0 if kind is not a binary operator."); writer.WriteLine("/// </summary>"); writer.WriteLine("/// <param name=\"kind\"></param>"); writer.WriteLine("/// <returns>"); writer.WriteLine("/// A positive number indicating the binary operator precedence or 0 if the kind is not a binary operator."); writer.WriteLine("/// </returns>"); using (writer.CurlyIndenter("public static int GetBinaryOperatorPrecedence(SyntaxKind kind)")) using (writer.CurlyIndenter("switch(kind)")) { var groups = kinds.BinaryOperators.GroupBy(kind => kind.BinaryOperatorInfo !.Value.Precedence); foreach (var group in groups.OrderByDescending(g => g.Key)) { foreach (var kind in group.OrderByDescending(info => info.Field.Name)) { writer.WriteLine($"case SyntaxKind.{kind.Field.Name}:"); } using (new Indenter(writer)) writer.WriteLine($"return {group.Key};"); writer.WriteLineNoTabs(""); } using (writer.Indenter("default:")) writer.WriteLine("return 0;"); } }
private static void GenerateGetBinaryExpression(KindList kinds, SourceWriter writer) { writer.WriteLine("/// <summary>"); writer.WriteLine("/// Returns the expression kind for a given unary operator or None if not a unary operator."); writer.WriteLine("/// </summary>"); writer.WriteLine("/// <param name=\"kind\"></param>"); writer.WriteLine("/// <returns>"); writer.WriteLine("/// A positive number indicating the binary operator precedence or 0 if the kind is not a binary operator."); writer.WriteLine("/// </returns>"); using (writer.Indenter("public static Option<SyntaxKind> GetBinaryExpression(SyntaxKind kind) =>")) using (writer.CurlyIndenter("kind switch", ";")) { foreach (var binaryOperator in kinds.BinaryOperators) { writer.WriteLine($"SyntaxKind.{binaryOperator.Field.Name} => {binaryOperator.BinaryOperatorInfo!.Value.Expression.ToCSharpString()},"); } writer.WriteLine("_ => default,"); } }
private static void GenerateSyntaxFacts(GeneratorExecutionContext context, KindList kinds) { SourceText sourceText; using (var writer = new SourceWriter()) { writer.WriteLine("// <auto-generated />"); writer.WriteLine(); writer.WriteLine("using System;"); writer.WriteLine("using System.Collections.Generic;"); writer.WriteLine("using System.Collections.Immutable;"); writer.WriteLine("using System.Diagnostics.CodeAnalysis;"); writer.WriteLine("using Tsu;"); writer.WriteLine(); writer.WriteLine("#nullable enable"); writer.WriteLine(); using (writer.CurlyIndenter("namespace Loretta.CodeAnalysis.Lua")) using (writer.CurlyIndenter("public static partial class SyntaxFacts")) { GenerateMinMaxLength(kinds.Tokens.Concat(kinds.Keywords), writer, "Token"); GenerateMinMaxLength(kinds.Tokens, writer, "NonKeywordToken"); GenerateMinMaxLength(kinds.Keywords, writer, "Keyword"); GenerateMinMaxLength(kinds.UnaryOperators, writer, "UnaryOperator"); GenerateMinMaxLength(kinds.BinaryOperators, writer, "BinaryOperator"); writer.WriteLineNoTabs(""); GenerateGetUnaryOperatorPrecedence(kinds, writer); writer.WriteLineNoTabs(""); GenerateGetUnaryExpression(kinds, writer); writer.WriteLineNoTabs(""); GenerateGetBinaryOperatorPrecedence(kinds, writer); writer.WriteLineNoTabs(""); GenerateGetBinaryExpression(kinds, writer); writer.WriteLineNoTabs(""); GenerateGetKeywordKind(kinds, writer); writer.WriteLineNoTabs(""); GenerateGetUnaryOperatorKinds(kinds, writer); writer.WriteLineNoTabs(""); GenerateGetBinaryOperatorKinds(kinds, writer); writer.WriteLineNoTabs(""); GenerateGetText(kinds, writer); var properties = kinds.SelectMany(kind => kind.Properties.Select(kv => (kind, key: kv.Key, value: kv.Value))) .GroupBy(t => t.key, t => (t.kind, t.value)); foreach (var propertyGroup in properties) { var possibleTypes = propertyGroup.Select(t => t.value.Type) .Where(t => t is not null) .Distinct(SymbolEqualityComparer.Default) .ToImmutableArray(); string type; if (possibleTypes.Length > 1) { type = context.Compilation.GetSpecialType(SpecialType.System_Object) + "?"; } else { type = possibleTypes.Single() !.ToString(); } writer.WriteLineNoTabs(""); using (new CurlyIndenter(writer, $"public static partial Option<{type}> Get{propertyGroup.Key}(SyntaxKind kind)")) { var values = propertyGroup.GroupBy(t => t.value, t => t.kind); writer.WriteLine("return kind switch"); writer.WriteLine("{"); using (new Indenter(writer)) { foreach (var value in values) { writer.Write(string.Join(" or ", value.Select(k => $"SyntaxKind.{k.Field.Name}"))); writer.Write(" => "); writer.Write(value.Key.ToCSharpString()); writer.WriteLine(","); } writer.WriteLine("_ => default,"); } writer.WriteLine("};"); } } writer.WriteLineNoTabs(""); // Generate IsTrivia GenerateIsX(kinds, writer, "Trivia", kind => kind.IsTrivia); writer.WriteLineNoTabs(""); // Generate IsKeyword GenerateIsX(kinds, writer, "Keyword", kind => kind.TokenInfo?.IsKeyword is true); writer.WriteLineNoTabs(""); // Generate IsToken GenerateIsX(kinds, writer, "Token", kind => kind.TokenInfo is not null); // Generate Is(Unary|Binary)Operator GenerateIsX(kinds, writer, "OperatorToken", kind => kind.UnaryOperatorInfo is not null || kind.BinaryOperatorInfo is not null); GenerateIsX(kinds, writer, "UnaryOperatorToken", kind => kind.UnaryOperatorInfo is not null); GenerateIsX(kinds, writer, "BinaryOperatorToken", kind => kind.BinaryOperatorInfo is not null); writer.WriteLineNoTabs(""); // Extra Categories var extraCategories = kinds.SelectMany(kind => kind.ExtraCategories.Select(cat => (cat, kind))) .GroupBy(t => t.cat, t => t.kind); foreach (var group in extraCategories) { writer.WriteLineNoTabs(""); var groupKinds = new KindList(group.ToImmutableArray()); GenerateIsX(groupKinds, writer, group.Key, k => true); writer.WriteLineNoTabs(""); writer.WriteLine("/// <summary>"); writer.WriteLine($"/// Returns all <see cref=\"SyntaxKind\"/>s that are in the {group.Key} category."); writer.WriteLine("/// </summary>"); writer.WriteLine("/// <returns></returns>"); using (writer.CurlyIndenter($"public static IEnumerable<SyntaxKind> Get{group.Key}Kinds() => ImmutableArray.Create(new[]", ");")) { foreach (var kind in group) { writer.WriteLine($"SyntaxKind.{kind.Field.Name},"); } } } } sourceText = writer.GetText(); } context.AddSource("SyntaxFacts.g.cs", sourceText); }