Add KubernetesAPIResource protocol

swiftkube · Oct 4, 2020 · 176e954 · 176e954
1 parent c44b04a
commit 176e954
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/Sources/SwiftkubeModelGen/Schema.swift b/Sources/SwiftkubeModelGen/Schema.swift
@@ -376,6 +376,7 @@ struct Resource: Decodable {
 	var requiresCodableExtension: Bool
 	var hasMetadata: Bool
 	var listResource: Bool
+	var isAPIResource: Bool
 
 	enum CodingKeys: String, CodingKey {
 		case type
@@ -396,6 +397,7 @@ struct Resource: Decodable {
 			self.requiresCodableExtension = false
 			self.hasMetadata = false
 			self.listResource = false
+			self.isAPIResource = false
 			return
 		}
 
@@ -410,6 +412,7 @@ struct Resource: Decodable {
 		self.requiresCodableExtension = false
 		self.hasMetadata = false
 		self.listResource = false
+		self.isAPIResource = false
 
 		guard container.allKeys.contains(.properties) else {
 			return
@@ -440,6 +443,10 @@ struct Resource: Decodable {
 		self.requiresCodableExtension = properties.contains { $0.type.requiresCodableExtension }
 		self.hasMetadata = properties.contains { $0.type.isMetadata && $0.isOptional }
 		self.listResource = properties.contains(where: { $0.name == "items" }) && gvk?.kind.hasSuffix("List") ?? false
+		self.isAPIResource =
+			!self.listResource &&
+			properties.contains(where: { $0.name == "apiVersion" && $0.isContant }) &&
+			properties.contains(where: { $0.name == "kind" && $0.isContant })
 	}
 }
 

diff --git a/Sources/SwiftkubeModelGen/Stencil+Extensions.swift b/Sources/SwiftkubeModelGen/Stencil+Extensions.swift
@@ -40,6 +40,9 @@ extension Extension {
 			if schema.hasMetadata {
 				protocols.append("ResourceWithMetadata")
 			}
+			if schema.isAPIResource {
+				protocols.append("KubernetesAPIResource")
+			}
 
 			return protocols.joined(separator: ", ")
 		}