Extract function

Author: Alexander Schmutz

Sources/StructBuilderMacro/Enum/Helper/GetFirstEnumCaseName.swift

@@ -0,0 +1,15 @@
+//
+//  GetFirstEnumCaseName.swift
+//
+//
+//  Created by Alexander Schmutz on 09.02.24.
+//
+
+import SwiftSyntax
+
+func getFirstEnumCaseName(from enumDecl: EnumDeclSyntax) throws -> TokenSyntax {
+    guard let firstCaseName = enumDecl.memberBlock.members.first?.decl.as(EnumCaseDeclSyntax.self)?.elements.first?.name else {
+        throw "Missing enum case"
+    }
+    return firstCaseName
+}

Sources/StructBuilderMacro/Enum/MakeEnumBuilder.swift

@@ -9,20 +9,13 @@ import Foundation
 import SwiftSyntax
 
 func makeEnumBuilder(enumDecl: EnumDeclSyntax) throws -> StructDeclSyntax {
-    guard let firstCaseName = enumDecl.memberBlock.members.first?.decl.as(EnumCaseDeclSyntax.self)?.elements.first?.name else {
-        throw "Missing enum case"
-    }
-
-    let structIdentifier = getStructBuilderName(from: enumDecl.name)
-    let valueVariable = TokenSyntax(stringLiteral: "value")
-    let returningType = TypeSyntax(stringLiteral: enumDecl.name.text)
     let enumMember = EnumMember(
-        identifier: valueVariable,
-        type: returningType,
-        value: firstCaseName
+        identifier: TokenSyntax(stringLiteral: "value"),
+        type: TypeSyntax(stringLiteral: enumDecl.name.text),
+        value: try getFirstEnumCaseName(from: enumDecl)
     )
-
-    return StructDeclSyntax(name: structIdentifier) {
+    
+    return StructDeclSyntax(name: getStructBuilderName(from: enumDecl.name)) {
         MemberBlockItemListSyntax {
             MemberBlockItemSyntax(decl: makeVariableDeclWithValue(enumMember: enumMember))
             MemberBlockItemSyntax(

Sources/StructBuilderMacro/Struct/MakeStructBuilder.swift

@@ -10,9 +10,8 @@ import SwiftSyntax
 
 func makeStructBuilder(structDecl: StructDeclSyntax) -> StructDeclSyntax {
     let structMembers = extractMembersFrom(structDecl.memberBlock.members)
-    let structIdentifier = getStructBuilderName(from: structDecl.name)
 
-    return StructDeclSyntax(name: structIdentifier) {
+    return StructDeclSyntax(name: getStructBuilderName(from: structDecl.name)) {
         MemberBlockItemListSyntax {
             for structMember in structMembers {
                 MemberBlockItemSyntax(decl: makeVariableDecl(structMember: structMember))