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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2024 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
#if swift(>=6)
public import SwiftSyntax
#else
import SwiftSyntax
#endif
public struct ConvertStoredPropertyToComputed: SyntaxRefactoringProvider {
public static func refactor(syntax: VariableDeclSyntax, in context: ()) -> VariableDeclSyntax? {
guard syntax.bindings.count == 1, let binding = syntax.bindings.first, let initializer = binding.initializer else {
return nil
}
var codeBlockSyntax: CodeBlockItemListSyntax
if let functionExpression = initializer.value.as(FunctionCallExprSyntax.self),
let closureExpression = functionExpression.calledExpression.as(ClosureExprSyntax.self)
{
guard functionExpression.arguments.isEmpty else { return nil }
codeBlockSyntax = closureExpression.statements
codeBlockSyntax.leadingTrivia =
closureExpression.leftBrace.leadingTrivia + closureExpression.leftBrace.trailingTrivia
+ codeBlockSyntax.leadingTrivia
codeBlockSyntax.trailingTrivia +=
closureExpression.trailingTrivia + closureExpression.rightBrace.leadingTrivia
+ closureExpression.rightBrace.trailingTrivia + functionExpression.trailingTrivia
} else {
var body = CodeBlockItemListSyntax([
CodeBlockItemSyntax(
item: .expr(initializer.value)
)
])
body.leadingTrivia = initializer.equal.trailingTrivia + body.leadingTrivia
body.trailingTrivia += .space
codeBlockSyntax = body
}
let newBinding =
binding
.with(\.initializer, nil)
.with(
\.accessorBlock,
AccessorBlockSyntax(
accessors: .getter(codeBlockSyntax)
)
)
let newBindingSpecifier =
syntax.bindingSpecifier
.with(\.tokenKind, .keyword(.var))
return
syntax
.with(\.bindingSpecifier, newBindingSpecifier)
.with(\.bindings, PatternBindingListSyntax([newBinding]))
}
}
|
