Merge pull request #3 from Lucien/feat/optional-security-requirements

Add support to optional Security Requirements and top-level inheritance yonaskolb#310

Author: arasan01_dev

Sources/SwagGenKit/CodeFormatter.swift

@@ -380,6 +380,7 @@ public class CodeFormatter {
         var context: Context = [:]
 
         context["name"] = securityRequirement.name
+        context["isRequired"] = securityRequirement.isRequired
         context["scopes"] = securityRequirement.scopes
         context["scope"] = securityRequirement.scopes.first
 

Sources/Swagger/Operation.swift

@@ -41,7 +41,11 @@ public struct Operation {
 
 extension Operation {
 
-    public init(path: String, method: Method, pathParameters: [PossibleReference<Parameter>], jsonDictionary: JSONDictionary) throws {
+    public init(path: String,
+                method: Method,
+                pathParameters: [PossibleReference<Parameter>],
+                jsonDictionary: JSONDictionary,
+                topLevelSecurityRequirements: [SecurityRequirement]?) throws {
         json = jsonDictionary
         self.path = path
         self.method = method
@@ -57,7 +61,10 @@ extension Operation {
 
         identifier = jsonDictionary.json(atKeyPath: "operationId")
         tags = (jsonDictionary.json(atKeyPath: "tags")) ?? []
-        securityRequirements = jsonDictionary.json(atKeyPath: "security")
+        securityRequirements = type(of: self).getSecurityRequirements(
+            from: jsonDictionary,
+            topLevelSecurityRequirements: topLevelSecurityRequirements
+        )
 
         let allResponses: [String: PossibleReference<Response>] = try jsonDictionary.json(atKeyPath: "responses")
         var mappedResponses: [OperationResponse] = []
@@ -88,4 +95,18 @@ extension Operation {
 
         deprecated = (jsonDictionary.json(atKeyPath: "deprecated")) ?? false
     }
+
+    private static func getSecurityRequirements(
+        from jsonDictionary: JSONDictionary,
+        topLevelSecurityRequirements: [SecurityRequirement]?
+    ) -> [SecurityRequirement]? {
+        // Even an empty list of operation security requirements
+        // can remove top-level security requirements.
+        guard let securityRequirements: [SecurityRequirement] =
+                jsonDictionary.json(atKeyPath: "security") else {
+            return topLevelSecurityRequirements
+        }
+
+        return securityRequirements.updatingRequiredFlag()
+    }
 }

Sources/Swagger/Path.swift

@@ -9,15 +9,22 @@ public struct Path {
 
 extension Path: NamedMappable {
 
-    public init(name: String, jsonDictionary: JSONDictionary) throws {
+    public init(name: String,
+                jsonDictionary: JSONDictionary,
+                topLevelSecurityRequirements: [SecurityRequirement]?) throws {
         path = name
         parameters = (jsonDictionary.json(atKeyPath: "parameters")) ?? []
 
         var mappedOperations: [Operation] = []
         for (key, value) in jsonDictionary {
             if let method = Operation.Method(rawValue: key) {
                 if let json = value as? [String: Any] {
-                    let operation = try Operation(path: path, method: method, pathParameters: parameters, jsonDictionary: json)
+                    let operation = try Operation(
+                        path: path,
+                        method: method,
+                        pathParameters: parameters,
+                        jsonDictionary: json,
+                        topLevelSecurityRequirements: topLevelSecurityRequirements)
                     mappedOperations.append(operation)
                 }
             }

Sources/Swagger/Security/SecurityRequirement.swift

@@ -1,12 +1,37 @@
 import Foundation
 import JSONUtilities
 
-public struct SecurityRequirement: JSONObjectConvertible {
+public struct SecurityRequirement: Equatable {
     public let name: String
+    public let isRequired: Bool
     public let scopes: [String]
 
+    public static let optionalMarker: SecurityRequirement = {
+        .init(name: "optional-marker", isRequired: false, scopes: [])
+    }()
+}
+
+extension SecurityRequirement: JSONObjectConvertible {
     public init(jsonDictionary: JSONDictionary) throws {
-        name = jsonDictionary.keys.first!
+        guard let firstKey = jsonDictionary.keys.first else {
+            self = .optionalMarker
+            return
+        }
+        name = firstKey
+        isRequired = true
         scopes = try jsonDictionary.json(atKeyPath: .key(name))
     }
 }
+
+extension Array where Element == SecurityRequirement {
+    func updatingRequiredFlag() -> [Element] {
+        // Check if there is an optional security requirement marker
+        guard (contains(where: { $0 == .optionalMarker })) else {
+            return self
+        }
+        // Remove the optional marker and set all security requirements as optional
+        return self
+            .drop { $0 == .optionalMarker }
+            .map { .init(name: $0.name, isRequired: false, scopes: $0.scopes) }
+    }
+}

Sources/Swagger/SwaggerSpec.swift

@@ -30,7 +30,9 @@ public enum TransferScheme: String {
 }
 
 public protocol NamedMappable {
-    init(name: String, jsonDictionary: JSONDictionary) throws
+    init(name: String,
+         jsonDictionary: JSONDictionary,
+         topLevelSecurityRequirements: [SecurityRequirement]?) throws
 }
 
 extension SwaggerSpec {
@@ -69,12 +71,20 @@ extension SwaggerSpec: JSONObjectConvertible {
 
     public init(jsonDictionary: JSONDictionary) throws {
 
-        func decodeNamed<T: NamedMappable>(jsonDictionary: JSONDictionary, key: String) throws -> [T] {
+        func decodeNamed<T: NamedMappable>(
+            jsonDictionary: JSONDictionary,
+            key: String,
+            topLevelSecurityRequirements: [SecurityRequirement]?
+        ) throws -> [T] {
             var values: [T] = []
             if let dictionary = jsonDictionary[key] as? [String: Any] {
                 for (key, value) in dictionary {
                     if let dictionary = value as? [String: Any] {
-                        let value = try T(name: key, jsonDictionary: dictionary)
+                        let value = try T(
+                            name: key,
+                            jsonDictionary: dictionary,
+                            topLevelSecurityRequirements: topLevelSecurityRequirements
+                        )
                         values.append(value)
                     }
                 }
@@ -91,13 +101,15 @@ extension SwaggerSpec: JSONObjectConvertible {
 
         info = try jsonDictionary.json(atKeyPath: "info")
         servers = jsonDictionary.json(atKeyPath: "servers") ?? []
-        securityRequirements = jsonDictionary.json(atKeyPath: "security")
+        securityRequirements = type(of: self).getSecurityRequirements(from: jsonDictionary)
         if jsonDictionary["components"] != nil {
             components = try jsonDictionary.json(atKeyPath: "components")
         } else {
             components = Components()
         }
-        paths = try decodeNamed(jsonDictionary: jsonDictionary, key: "paths")
+        paths = try decodeNamed(jsonDictionary: jsonDictionary,
+                                key: "paths",
+                                topLevelSecurityRequirements: securityRequirements)
         operations = paths.reduce([]) { $0 + $1.operations }
             .sorted(by: { (lhs, rhs) -> Bool in
                 if lhs.path == rhs.path {
@@ -110,4 +122,15 @@ extension SwaggerSpec: JSONObjectConvertible {
         try resolver.resolve()
         self = resolver.spec
     }
+
+    private static func getSecurityRequirements(
+        from jsonDictionary: JSONDictionary
+    ) -> [SecurityRequirement]? {
+        guard let securityRequirements: [SecurityRequirement] =
+                jsonDictionary.json(atKeyPath: "security") else {
+            return nil
+        }
+
+        return securityRequirements.updatingRequiredFlag()
+    }
 }

Specs/Petstore/generated/Swift/Sources/APIService.swift

@@ -37,10 +37,12 @@ extension APIService: CustomStringConvertible {
 
 public struct SecurityRequirement {
     public let type: String
+    public let isRequired: Bool
     public let scopes: [String]
 
-    public init(type: String, scopes: [String]) {
+    public init(type: String, isRequired: Bool, scopes: [String]) {
         self.type = type
+        self.isRequired = isRequired
         self.scopes = scopes
     }
 }

Specs/Petstore/generated/Swift/Sources/Requests/Pets/CreatePets.swift

@@ -10,7 +10,7 @@ extension Petstore.Pets {
     /** Create a pet */
     public enum CreatePets {
 
-        public static let service = APIService<Response>(id: "createPets", tag: "pets", method: "POST", path: "/pets", hasBody: false, securityRequirements: [SecurityRequirement(type: "petstore_auth", scopes: ["write:pets", "read:pets"])])
+        public static let service = APIService<Response>(id: "createPets", tag: "pets", method: "POST", path: "/pets", hasBody: false, securityRequirements: [SecurityRequirement(type: "petstore_auth", isRequired: true, scopes: ["write:pets", "read:pets"])])
 
         public final class Request: APIRequest<Response> {
 

Specs/Petstore/generated/Swift/Sources/Requests/Pets/ListPets.swift

@@ -10,7 +10,7 @@ extension Petstore.Pets {
     /** List all pets */
     public enum ListPets {
 
-        public static let service = APIService<Response>(id: "listPets", tag: "pets", method: "GET", path: "/pets", hasBody: false, securityRequirements: [])
+        public static let service = APIService<Response>(id: "listPets", tag: "pets", method: "GET", path: "/pets", hasBody: false, securityRequirements: [SecurityRequirement(type: "petstore_auth", isRequired: false, scopes: ["read:pets"])])
 
         public final class Request: APIRequest<Response> {
 

Specs/Petstore/generated/Swift/Sources/Requests/Pets/ShowPetById.swift

@@ -10,7 +10,7 @@ extension Petstore.Pets {
     /** Info for a specific pet */
     public enum ShowPetById {
 
-        public static let service = APIService<Response>(id: "showPetById", tag: "pets", method: "GET", path: "/pets/{petId}", hasBody: false, securityRequirements: [])
+        public static let service = APIService<Response>(id: "showPetById", tag: "pets", method: "GET", path: "/pets/{petId}", hasBody: false, securityRequirements: [SecurityRequirement(type: "petstore_auth", isRequired: true, scopes: ["read:pets"])])
 
         public final class Request: APIRequest<Response> {
 

Specs/Petstore/generated/Swift/Sources/Requests/Pets/UpdatePetWithForm.swift

@@ -10,7 +10,7 @@ extension Petstore.Pets {
     /** Updates a pet in the store with form data */
     public enum UpdatePetWithForm {
 
-        public static let service = APIService<Response>(id: "updatePetWithForm", tag: "pets", method: "POST", path: "/pets/{petId}", hasBody: true, securityRequirements: [SecurityRequirement(type: "petstore_auth", scopes: ["write:pets", "read:pets"])])
+        public static let service = APIService<Response>(id: "updatePetWithForm", tag: "pets", method: "POST", path: "/pets/{petId}", hasBody: true, securityRequirements: [])
 
         public final class Request: APIRequest<Response> {