-
Notifications
You must be signed in to change notification settings - Fork 234
/
Copy pathCodeEnumWriter.cs
38 lines (35 loc) · 1.55 KB
/
CodeEnumWriter.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
using System;
using System.Linq;
using Kiota.Builder.CodeDOM;
using Kiota.Builder.Extensions;
namespace Kiota.Builder.Writers.CSharp;
public class CodeEnumWriter : BaseElementWriter<CodeEnum, CSharpConventionService>
{
public CodeEnumWriter(CSharpConventionService conventionService) : base(conventionService) { }
public override void WriteCodeElement(CodeEnum codeElement, LanguageWriter writer)
{
ArgumentNullException.ThrowIfNull(codeElement);
ArgumentNullException.ThrowIfNull(writer);
if (!codeElement.Options.Any())
return;
var codeNamespace = codeElement.Parent as CodeNamespace;
if (codeNamespace != null)
writer.StartBlock($"namespace {codeNamespace.Name} {{");
if (codeElement.Flags)
writer.WriteLine("[Flags]");
conventions.WriteShortDescription(codeElement.Documentation.Description, writer);
writer.WriteLine($"public enum {codeElement.Name.ToFirstCharacterUpperCase()} {{");
writer.IncreaseIndent();
var idx = 0;
foreach (var option in codeElement.Options)
{
conventions.WriteShortDescription(option.Documentation.Description, writer);
writer.WriteLine($"{option.Name.ToFirstCharacterUpperCase()}{(codeElement.Flags ? " = " + GetEnumFlag(idx) : string.Empty)},");
idx++;
}
if (codeNamespace != null)
writer.CloseBlock();
}
private static readonly Func<int, string> GetEnumFlag = static idx =>
(idx == 0 ? 1 : Math.Pow(2, idx)).ToString();
}