ai-guiders/roslyn-mcp / ServiceLayer/GenerateConstructor.cs

/// <summary>По позиции на класс генерирует конструктор с параметрами по выбранным полям/свойствам и присвоениями. Опционально: member_names, insert_into_file.</summary>

27

public static async Task<string> GenerateConstructorFromMembersAsync(

28

string solutionOrProjectPath,

string filePath,

int line,

int column,

IReadOnlyList<string>? memberNames = null,

33

bool insertIntoFile = false,

34

CancellationToken cancellationToken = default)

35

{

36

if (!File.Exists(solutionOrProjectPath))

37

return $"Error: solution/project not found: {solutionOrProjectPath}";

38

if (!File.Exists(filePath))

39

return $"Error: file not found: {filePath}";

40

41

var targetPath = NormalizePath(filePath);

42

Solution? solution = null;

43

try

44

{

45

var workspace = MSBuildWorkspace.Create(RoslynMcpWorkspaceProperties.MsBuild);

46

solution = await WorkspaceOpen.OpenSolutionOrProjectAsync(workspace, solutionOrProjectPath, cancellationToken).ConfigureAwait(false);

47

48

if (solution is null)

49

return "Error: failed to open solution.";

50

51

var document = solution.Projects

52

.SelectMany(p => p.Documents)

53

.FirstOrDefault(d => string.Equals(NormalizePath(d.FilePath ?? ""), targetPath, StringComparison.OrdinalIgnoreCase));

54

if (document is null)

55

return $"Error: file not found in solution: {filePath}";

56

57

var root = await document.GetSyntaxRootAsync(cancellationToken).ConfigureAwait(false);

58

var semanticModel = await document.GetSemanticModelAsync(cancellationToken).ConfigureAwait(false);

59

if (root is null || semanticModel is null)

60

return "Error: could not get syntax/semantic model.";

61

62

var sourceText = await document.GetTextAsync(cancellationToken).ConfigureAwait(false);

63

var lines = sourceText.Lines;

64

if (line < 1 || line > lines.Count)

65

return $"Error: line {line} out of range (1..{lines.Count}).";

66

var lineInfo = lines[line - 1];

67

var columnIndex = column - 1;

68

if (columnIndex < 0)

69

return "Error: column must be >= 1.";

70

var position = lineInfo.Span.Length == 0

71

? lineInfo.Start

72

: lineInfo.Start + Math.Min(columnIndex, lineInfo.Span.Length);

73

74

var node = root.FindToken(position, findInsideTrivia: true).Parent;

75

INamedTypeSymbol? typeSymbol = null;

76

SyntaxNode? classDeclaration = null;

77

while (node != null)

78

{

79

cancellationToken.ThrowIfCancellationRequested();

80

var sym = semanticModel.GetDeclaredSymbol(node, cancellationToken);

81

if (sym is INamedTypeSymbol named && named.TypeKind == TypeKind.Class)

82

{

83

typeSymbol = named;

84

classDeclaration = node;

break;

}

node = node.Parent;

}

if (typeSymbol is null)

91

return $"Error: no class at {filePath}:{line}:{column}. Position the cursor on the class name or inside the class body.";

92

93

var style = EditorConfigStyle.GetOptionsForDirectory(Path.GetDirectoryName(document.FilePath) ?? "");

94

95

var memberSet = memberNames != null && memberNames.Count > 0

96

? new HashSet<string>(memberNames.Select(n => n.Trim()), StringComparer.OrdinalIgnoreCase)

97

: null;

98

99

var propertyNames = typeSymbol.GetMembers()

100

.OfType<IPropertySymbol>()

101

.Where(p => !p.IsIndexer)

102

.Select(p => p.Name)

103

.ToHashSet(StringComparer.OrdinalIgnoreCase);

104

105

var members = new List<(string type, string memberName, string paramName)>();

106

foreach (var member in typeSymbol.GetMembers())

107

{

108

if (member.IsImplicitlyDeclared)

109

continue;

110

if (memberSet != null && !memberSet.Contains(member.Name))

continue;

switch (member)

{

case IFieldSymbol field when field.IsConst == false && field.IsStatic == false:

116

if (IsBackingFieldForProperty(field.Name, propertyNames))

117

continue;

118

var p1 = ToCamelCase(field.Name);

119

var type1 = style.FormatTypeName(field.Type.ToDisplayString(TypeFormat));

120

members.Add((type1, field.Name, p1));

121

break;

122

case IPropertySymbol prop when prop.IsIndexer == false && !prop.IsReadOnly && prop.SetMethod != null:

123

var p2 = ToCamelCase(prop.Name);

124

var type2 = style.FormatTypeName(prop.Type.ToDisplayString(TypeFormat));

125

members.Add((type2, prop.Name, p2));

break;

}

}

if (members.Count == 0)

131

return "Error: no instance fields or writable properties to include. Add member_names or ensure the class has instance fields/properties with setter.";

132

133

var indent = "\t\t";

134

var ctorParams = string.Join(", ", members.Select(m => $"{m.type} {m.paramName}"));

135

var assignments = string.Join(Environment.NewLine + indent, members.Select(m => $"{m.memberName} = {m.paramName};"));

136

var ctorText = $"public {typeSymbol.Name}({ctorParams})" + Environment.NewLine + indent + "{" + Environment.NewLine + indent + assignments + Environment.NewLine + indent + "}";

137

138

if (insertIntoFile && classDeclaration != null)

139

{

140

var (closeBrace, indentBeforeBrace) = GetClassCloseBraceAndIndent(classDeclaration, root);

141

if (closeBrace != null)

142

{

143

var insertIndent = indentBeforeBrace ?? style.IndentString;

144

var toInsert = Environment.NewLine + insertIndent + ctorText.Replace(indent, insertIndent) + Environment.NewLine + insertIndent;

145

var text = root.GetText();

146

var change = new TextChange(new TextSpan(closeBrace.Value.Span.Start, 0), toInsert);

147

var newText = text.WithChanges(change);

148

var newSolution = solution.WithDocumentText(document.Id, newText);

149

var applied = newSolution.Workspace.TryApplyChanges(newSolution);

150

if (applied)

151

{

152

newSolution.Workspace.Dispose();

153

return $"# Constructor inserted into {filePath}\n# Parameters: {members.Count}\n\n{ctorText}";

}

}

}

solution?.Workspace.Dispose();

159

return $"# Generated constructor for {typeSymbol.Name}\n# Paste into the class body. (Use insert_into_file: true to insert automatically.)\n\n{ctorText}";

160

}

161

catch (InvalidOperationException ex) when (ex.Message.Contains("slnx") || ex.Message.Contains("Slnx"))

162

{

163

return "Error: .slnx format is not supported. Use .sln or .csproj.";

164

}

165

finally