From f352f6fed004ffe81967a30fe9e7a5438f6c0c3a Mon Sep 17 00:00:00 2001
From: Ivan Smolin <ivan.smolin@touchin.ru>
Date: Fri, 4 Mar 2022 17:53:46 +0300
Subject: [PATCH] apiKey in header Plugin generation

---
 .../touchin/codegen/TINetworkingCodegen.java  |  4 +++
 .../handlebars/TINetworking/Models.mustache   | 36 -------------------
 .../handlebars/TINetworking/Plugins.mustache  | 26 ++++++++++++++
 3 files changed, 30 insertions(+), 36 deletions(-)
 delete mode 100644 src/main/resources/handlebars/TINetworking/Models.mustache
 create mode 100644 src/main/resources/handlebars/TINetworking/Plugins.mustache

diff --git a/src/main/java/ru/touchin/codegen/TINetworkingCodegen.java b/src/main/java/ru/touchin/codegen/TINetworkingCodegen.java
index 258f6c0..6a0021d 100644
--- a/src/main/java/ru/touchin/codegen/TINetworkingCodegen.java
+++ b/src/main/java/ru/touchin/codegen/TINetworkingCodegen.java
@@ -229,6 +229,10 @@ public class TINetworkingCodegen extends DefaultCodegenConfig {
                 sourceFolder,
                 "APIDateFormat.swift"));
 
+        supportingFiles.add(new SupportingFile("Plugins.mustache",
+                sourceFolder,
+                "Plugins.swift"));
+
         copyFistAllOfProperties = true;
     }
 
diff --git a/src/main/resources/handlebars/TINetworking/Models.mustache b/src/main/resources/handlebars/TINetworking/Models.mustache
deleted file mode 100644
index d7dfe63..0000000
--- a/src/main/resources/handlebars/TINetworking/Models.mustache
+++ /dev/null
@@ -1,36 +0,0 @@
-// Models.swift
-//
-// Generated by swagger-codegen
-// https://github.com/swagger-api/swagger-codegen
-//
-
-import Foundation
-
-protocol JSONEncodable {
-    func encodeToJSON() -> Any
-}
-
-public enum ErrorResponse : Error {
-    case error(Int, Data?, Error)
-}
-
-open class Response<T> {
-    public let statusCode: Int
-    public let header: [String: String]
-    public let body: T?
-
-    public init(statusCode: Int, header: [String: String], body: T?) {
-        self.statusCode = statusCode
-        self.header = header
-        self.body = body
-    }
-
-    public convenience init(response: HTTPURLResponse, body: T?) {
-        let rawHeader = response.allHeaderFields
-        var header = [String:String]()
-        for case let (key, value) as (String, String) in rawHeader {
-            header[key] = value
-        }
-        self.init(statusCode: response.statusCode, header: header, body: body)
-    }
-}
diff --git a/src/main/resources/handlebars/TINetworking/Plugins.mustache b/src/main/resources/handlebars/TINetworking/Plugins.mustache
new file mode 100644
index 0000000..6bf1d89
--- /dev/null
+++ b/src/main/resources/handlebars/TINetworking/Plugins.mustache
@@ -0,0 +1,26 @@
+import TIMoyaNetworking
+import Alamofire
+
+{{#authMethods}}
+{{#isApiKey}}
+{{#isKeyInHeader}}
+public struct {{name}}Plugin: AdditionalHeadersPlugin {
+    public typealias HeaderValueProvider = () -> String?
+
+    private let headerValueProvider: HeaderValueProvider
+
+    public init(headerValueProvider: @escaping HeaderValueProvider) {
+        self.headerValueProvider = headerValueProvider
+    }
+
+    public var additionalHeaders: HTTPHeaders {
+        guard let value = headerValueProvider() else {
+            return HTTPHeaders()
+        }
+
+        return HTTPHeaders(["{{{keyParamName}}}": value])
+    }
+}
+{{/isKeyInHeader}}
+{{/isApiKey}}
+{{/authMethods}}