APIClientScope

Author: dankinsoid

Sources/SwiftAPIClient/APIClientType.swift

@@ -0,0 +1,17 @@
+import Foundation
+
+public protocol APIClientScope {
+
+	/// The API client used in this scope.
+	var client: APIClient { get set }
+}
+
+extension APIClientScope {
+
+	/// Returns a copy of the API client scope with the client transformed by the given closure.
+	public func mapClient(_ transform: (APIClient) -> APIClient) -> Self {
+		var copy = self
+		copy.client = transform(copy.client)
+		return copy
+	}
+}

Sources/SwiftAPIClient/Macros.swift

@@ -282,13 +282,15 @@ public macro PATCH(_ path: String...) = #externalMacro(module: "SwiftAPIClientMa
 @attached(peer, names: arbitrary)
 @attached(memberAttribute)
 @attached(member, names: arbitrary)
+@attached(extension, conformances: APIClientScope)
 public macro Path(_ path: String...) = #externalMacro(module: "SwiftAPIClientMacros", type: "SwiftAPIClientPathMacro")
 
 /// Macro that generates an API client struct.
 /// The struct can contain functions with `Call`, `GET`, `POST`, `PUT`, `PATCH`, and `DELETE` macros, as well as structs with the `Path` macro.
 /// If you specify a custom init, you must initialize the `client` property.
 @attached(memberAttribute)
 @attached(member, names: arbitrary)
+@attached(extension, conformances: APIClientScope)
 public macro API() = #externalMacro(module: "SwiftAPIClientMacros", type: "SwiftAPIClientPathMacro")
 
 @propertyWrapper

Sources/SwiftAPIClientMacros/SwiftAPIClientMacros.swift

@@ -142,7 +142,7 @@ public struct SwiftAPIClientCallMacro: PeerMacro {
 	}
 }
 
-public struct SwiftAPIClientPathMacro: MemberMacro, MemberAttributeMacro, PeerMacro {
+public struct SwiftAPIClientPathMacro: MemberMacro, MemberAttributeMacro, PeerMacro, ExtensionMacro {
 
 	public static func expansion(
 		of node: AttributeSyntax,
@@ -157,6 +157,18 @@ public struct SwiftAPIClientPathMacro: MemberMacro, MemberAttributeMacro, PeerMa
 		return ["@available(*, unavailable)", "@APICallFakeBuilder"]
 	}
 
+	public static func expansion(
+		of node: AttributeSyntax,
+		attachedTo declaration: some DeclGroupSyntax,
+		providingExtensionsOf type: some TypeSyntaxProtocol,
+		conformingTo protocols: [TypeSyntax],
+		in context: some MacroExpansionContext
+	) throws -> [ExtensionDeclSyntax] {
+		guard declaration.is(StructDeclSyntax.self) || declaration.is(EnumDeclSyntax.self) else { return [] }
+		let scopeExtension = try ExtensionDeclSyntax("extension \(type.trimmed): APIClientScope {}")
+		return [scopeExtension]
+	}
+
 	public static func expansion(
 		of node: AttributeSyntax,
 		providingPeersOf declaration: some DeclSyntaxProtocol,

Tests/SwiftAPIClientMacrosTests/APIMacroTests.swift

@@ -29,6 +29,9 @@ final class APIMacroTests: XCTestCase {
 			      self.client = client
 			  }
 			}
+			
+			extension Pets: APIClientScope {
+			}
 			""",
 			macros: macros,
 			indentationWidth: .spaces(2)
@@ -57,6 +60,9 @@ final class APIMacroTests: XCTestCase {
 
 			  public var client: APIClient
 			}
+			
+			extension Pets: APIClientScope {
+			}
 			""",
 			macros: macros,
 			indentationWidth: .spaces(2)

Tests/SwiftAPIClientMacrosTests/PathMacroTests.swift

@@ -34,6 +34,9 @@ final class PathMacroTests: XCTestCase {
 			var pets: Pets  {
 			    Pets (client: client.path("pets"))
 			}
+			
+			extension Pets: APIClientScope {
+			}
 			""",
 			macros: macros,
 			indentationWidth: .spaces(2)
@@ -65,6 +68,9 @@ final class PathMacroTests: XCTestCase {
 			var pets: Pets  {
 			    Pets (client: client.path("some", "long", "path"))
 			}
+			
+			extension Pets: APIClientScope {
+			}
 			""",
 			macros: macros,
 			indentationWidth: .spaces(2)
@@ -96,6 +102,9 @@ final class PathMacroTests: XCTestCase {
 			func pets(_ long: String, id: UUID) -> Pets  {
 			    Pets (client: client.path("some", "\\(long)", "path", "\\(id)"))
 			}
+			
+			extension Pets: APIClientScope {
+			}
 			""",
 			macros: macros,
 			indentationWidth: .spaces(2)
@@ -132,6 +141,9 @@ final class PathMacroTests: XCTestCase {
 			var pets: Pets  {
 			    Pets (client: client.path("pets"))
 			}
+			
+			extension Pets: APIClientScope {
+			}
 			""",
 			macros: macros,
 			indentationWidth: .spaces(2)