Convert between computed properties and zero-parameters functions

Author: antigluten

Sources/SwiftRefactor/ConvertComputedPropertyToZeroParameterFunction.swift

@@ -0,0 +1,101 @@
+//===----------------------------------------------------------------------===//
+//
+// 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 ConvertComputedPropertyToZeroParameterFunction: SyntaxRefactoringProvider {
+  public static func refactor(syntax: VariableDeclSyntax, in context: Void) -> FunctionDeclSyntax? {
+    guard syntax.bindings.count == 1,
+      let binding = syntax.bindings.first,
+      let identifierPattern = binding.pattern.as(IdentifierPatternSyntax.self)
+    else { return nil }
+
+    var statements: CodeBlockItemListSyntax
+
+    guard let typeAnnotation = binding.typeAnnotation,
+      let accessorBlock = binding.accessorBlock
+    else { return nil }
+
+    switch accessorBlock.accessors {
+    case .accessors(let accessors):
+      guard accessors.count == 1, let accessor = accessors.first,
+        accessor.accessorSpecifier.tokenKind == .keyword(.get), let codeBlock = accessor.body
+      else { return nil }
+      statements = codeBlock.statements
+      statements.leadingTrivia = accessor.leadingTrivia + codeBlock.leadingTrivia + statements.leadingTrivia
+      statements.trailingTrivia += codeBlock.trailingTrivia + accessor.trailingTrivia
+      statements.trailingTrivia = statements.trailingTrivia.droppingTrailingWhitespace
+    case .getter(let codeBlock):
+      statements = codeBlock
+    }
+
+    let returnType = typeAnnotation.type
+
+    var returnClause: ReturnClauseSyntax?
+    let triviaAfterSignature: Trivia
+
+    if !returnType.isVoid {
+      triviaAfterSignature = .space
+      returnClause = ReturnClauseSyntax(
+        arrow: .arrowToken(
+          leadingTrivia: typeAnnotation.colon.leadingTrivia,
+          trailingTrivia: typeAnnotation.colon.trailingTrivia
+        ),
+        type: returnType
+      )
+    } else {
+      triviaAfterSignature = typeAnnotation.colon.leadingTrivia + typeAnnotation.colon.trailingTrivia
+    }
+
+    let body = CodeBlockSyntax(
+      leftBrace: accessorBlock.leftBrace,
+      statements: statements,
+      rightBrace: accessorBlock.rightBrace
+    )
+
+    var parameterClause = FunctionParameterClauseSyntax(parameters: [])
+    parameterClause.trailingTrivia = identifierPattern.identifier.trailingTrivia + triviaAfterSignature
+
+    let functionSignature = FunctionSignatureSyntax(
+      parameterClause: parameterClause,
+      returnClause: returnClause
+    )
+
+    return FunctionDeclSyntax(
+      modifiers: syntax.modifiers,
+      funcKeyword: .keyword(.func, trailingTrivia: syntax.bindingSpecifier.trailingTrivia),
+      name: identifierPattern.identifier.with(\.trailingTrivia, []),
+      signature: functionSignature,
+      body: body
+    )
+  }
+}
+
+fileprivate extension TypeSyntax {
+  var isVoid: Bool {
+    switch self.as(TypeSyntaxEnum.self) {
+    case .identifierType(let identifierType) where identifierType.name.text == "Void": return true
+    case .tupleType(let tupleType) where tupleType.elements.isEmpty: return true
+    default: return false
+    }
+  }
+}
+
+fileprivate extension Trivia {
+  var droppingTrailingWhitespace: Trivia {
+    return Trivia(pieces: self.reversed().drop(while: \.isWhitespace).reversed())
+  }
+}

Sources/SwiftRefactor/ConvertZeroParameterFunctionToComputedProperty.swift

@@ -0,0 +1,82 @@
+//===----------------------------------------------------------------------===//
+//
+// 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 ConvertZeroParameterFunctionToComputedProperty: SyntaxRefactoringProvider {
+  public static func refactor(syntax: FunctionDeclSyntax, in context: ()) -> VariableDeclSyntax? {
+    guard syntax.signature.parameterClause.parameters.isEmpty,
+      let body = syntax.body
+    else { return nil }
+
+    let variableName = PatternSyntax(
+      IdentifierPatternSyntax(
+        leadingTrivia: syntax.funcKeyword.trailingTrivia,
+        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 accessorBlock = AccessorBlockSyntax(
+      leftBrace: body.leftBrace,
+      accessors: .getter(body.statements),
+      rightBrace: body.rightBrace
+    )
+
+    return VariableDeclSyntax(
+      modifiers: syntax.modifiers,
+      .var,
+      name: variableName,
+      type: variableType,
+      accessorBlock: accessorBlock
+    )
+  }
+}
+
+fileprivate extension TokenSyntax {
+  var trivia: Trivia {
+    return leadingTrivia + trailingTrivia
+  }
+}
+
+fileprivate extension Trivia {
+  var droppingTrailingWhitespace: Trivia {
+    return Trivia(pieces: self.reversed().drop(while: \.isWhitespace).reversed())
+  }
+}

Sources/SwiftSyntaxBuilder/ConvenienceInitializers.swift

@@ -403,7 +403,8 @@ extension VariableDeclSyntax {
     _ bindingSpecifier: Keyword,
     name: PatternSyntax,
     type: TypeAnnotationSyntax? = nil,
-    initializer: InitializerClauseSyntax? = nil
+    initializer: InitializerClauseSyntax? = nil,
+    accessorBlock: AccessorBlockSyntax? = nil
   ) {
     self.init(
       leadingTrivia: leadingTrivia,
@@ -414,7 +415,8 @@ extension VariableDeclSyntax {
       PatternBindingSyntax(
         pattern: name,
         typeAnnotation: type,
-        initializer: initializer
+        initializer: initializer,
+        accessorBlock: accessorBlock
       )
     }
   }