Use macro to generate authorization property (#6)

pixel-foundry · Feb 17, 2024 · 9b8c570 · 9b8c570
1 parent fe96c02
commit 9b8c570
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 4 deletions.
diff --git a/Sources/EndpointBuilder/EndpointMacro.swift b/Sources/EndpointBuilder/EndpointMacro.swift
@@ -6,5 +6,5 @@ import RoutingKit
 /// Generates a `path` string for the given path components,
 /// along with a helper type for initializing any path parameters.
 @attached(extension, conformances: Endpoint)
-@attached(member, names: named(PathParameters), named(pathParameters), named(path))
+@attached(member, names: named(authorization), named(PathParameters), named(pathParameters), named(path))
 public macro Endpoint() = #externalMacro(module: "EndpointBuilderMacros", type: "EndpointMacro")
diff --git a/Sources/EndpointBuilderMacros/EndpointMacro.swift b/Sources/EndpointBuilderMacros/EndpointMacro.swift
@@ -40,6 +40,15 @@ extension EndpointMacro: MemberMacro {
 		providingMembersOf declaration: D,
 		in context: C
 	) throws -> [DeclSyntax] {
+		let authorizationDefinition = [
+			VariableDeclSyntax(
+				modifiers: declaration.modifiers,
+				Keyword.var,
+				name: PatternSyntax(stringLiteral: "authorization"),
+				type: TypeAnnotationSyntax(type: TypeSyntax(stringLiteral: "Authorization?"))
+			).as(DeclSyntax.self)
+		].compactMap { $0 }
+
 		let (pathComponents, pathVariableSyntax) = try endpointPathParameters(declaration: declaration)
 		let pathParameters = pathComponents.filter { pathComponent in
 			guard case .parameter = pathComponent else {
@@ -57,7 +66,7 @@ extension EndpointMacro: MemberMacro {
 			return try endpointPathParametersStructDefinition(declaration: declaration, pathParameters)
 		}()
 
-		return pathDefinition + structDefinition
+		return authorizationDefinition + pathDefinition + structDefinition
 	}
 
 	@usableFromInline

diff --git a/Tests/EndpointBuilderMacrosTests/EndpointBuilderMacrosTests.swift b/Tests/EndpointBuilderMacrosTests/EndpointBuilderMacrosTests.swift
@@ -20,7 +20,6 @@ final class EndpointBuilderMacrosTests: XCTestCase {
 				public static let path: [RoutingKit.PathComponent] = ["user", ":id"]
 				public static let httpMethod = HTTPRequest.Method.get
 				public static let responseType = User.self
-				public let authorization: Authorization?
 			}
 			"""
 		} expansion: {
@@ -29,7 +28,8 @@ final class EndpointBuilderMacrosTests: XCTestCase {
 				public static let path: [RoutingKit.PathComponent] = ["user", ":id"]
 				public static let httpMethod = HTTPRequest.Method.get
 				public static let responseType = User.self
-				public let authorization: Authorization?
+
+				public var authorization: Authorization?
 
 				public var path: String {
 					"/" + ["user", pathParameters.id].joined(separator: "/")
@@ -68,6 +68,8 @@ final class EndpointBuilderMacrosTests: XCTestCase {
 				public static let httpMethod = HTTPRequest.Method.get
 				public static let responseType = User.self
 
+				public var authorization: Authorization?
+
 				public var path: String {
 					"/" + ["team", pathParameters.teamId, "user", pathParameters.userId].joined(separator: "/")
 				}