swift-syntax/Sources/SwiftRefactor/ConvertZeroParameterFunctionToComputedProperty.swift at 36955883043ae083191076884f2962b0d2811c2f · swiftlang/swift-syntax · GitHub

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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
//===----------------------------------------------------------------------===//
//
// 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 compiler(>=6)
public import SwiftSyntax
import SwiftBasicFormat
import SwiftSyntaxBuilder
#else
import SwiftSyntax
import SwiftBasicFormat
import SwiftSyntaxBuilder
#endif

public struct ConvertZeroParameterFunctionToComputedProperty: SyntaxRefactoringProvider {
  public static func refactor(syntax: FunctionDeclSyntax, in context: ()) throws -> VariableDeclSyntax {
    guard syntax.signature.parameterClause.parameters.isEmpty,
      let body = syntax.body
    else { throw RefactoringNotApplicableError("not a zero parameter function") }

    let variableName = PatternSyntax(
      IdentifierPatternSyntax(
        identifier: syntax.name
      )
    )

    let triviaFromParameters =
      (syntax.signature.parameterClause.leftParen.trivia + syntax.signature.parameterClause.rightParen.trivia)
      .droppingTrailingWhitespace

    var variableType: TypeAnnotationSyntax?

    if let returnClause = syntax.signature.returnClause {
      variableType = TypeAnnotationSyntax(
        colon: .colonToken(
          leadingTrivia: triviaFromParameters + returnClause.arrow.leadingTrivia,
          trailingTrivia: returnClause.arrow.trailingTrivia
        ),
        type: returnClause.type
      )
    } else {
      variableType = TypeAnnotationSyntax(
        colon: .colonToken(
          leadingTrivia: triviaFromParameters,
          trailingTrivia: .space
        ),
        type: TypeSyntax("Void").with(\.trailingTrivia, .space)
      )
    }

    let accessorEffectSpecifiers: AccessorEffectSpecifiersSyntax?
    if let fnEffectSpecifiers = syntax.signature.effectSpecifiers {
      accessorEffectSpecifiers = AccessorEffectSpecifiersSyntax(
        asyncSpecifier: fnEffectSpecifiers.asyncSpecifier,
        throwsClause: fnEffectSpecifiers.throwsClause
      )
    } else {
      accessorEffectSpecifiers = nil
    }

    let indentation = BasicFormat.inferIndentation(of: syntax) ?? .spaces(2)

    let accessorBlock: AccessorBlockSyntax

    if let accessorEffectSpecifiers {
      let indentedStatements = body.statements.indented(by: indentation)
      let getterBody = CodeBlockSyntax(
        leftBrace: body.leftBrace,
        statements: indentedStatements,
        rightBrace: .rightBraceToken(leadingTrivia: .newline + indentation)
      )

      let getAccessor = AccessorDeclSyntax(
        accessorSpecifier: .keyword(.get, trailingTrivia: .space),
        effectSpecifiers: accessorEffectSpecifiers,
        body: getterBody
      ).with(\.leadingTrivia, indentation)

      accessorBlock = AccessorBlockSyntax(
        leftBrace: .leftBraceToken(trailingTrivia: .newline),
        accessors: .accessors(AccessorDeclListSyntax([getAccessor])),
        rightBrace: .rightBraceToken(leadingTrivia: .newline)
      )
    } else {
      accessorBlock = AccessorBlockSyntax(
        leftBrace: body.leftBrace,
        accessors: .getter(body.statements),
        rightBrace: body.rightBrace
      )
    }

    let bindingSpecifier = syntax.funcKeyword.detached.with(\.tokenKind, .keyword(.var))

    let patternBinding = PatternBindingSyntax(
      pattern: variableName,
      typeAnnotation: variableType,
      accessorBlock: accessorBlock
    )

    return VariableDeclSyntax(
      attributes: syntax.attributes,
      modifiers: syntax.modifiers,
      bindingSpecifier: bindingSpecifier,
      bindings: PatternBindingListSyntax([patternBinding])
    )
  }
}