//
// Copyright © 2023 Nevis Security AG. All rights reserved.
//

import NevisMobileAuthentication

struct TypedRetryPolicy: Decodable {
	// MARK: Properties

	let wrapped: RetryPolicy

	// MARK: Decodable

	init(from decoder: Decoder) throws {
		let container = try decoder.container(keyedBy: TypedCodingKeys.self)
		let type = try container.decode(String.self, forKey: .type)
		switch type {
		case RetryPolicy.CodingKeys.Case.noRetry.stringValue:
			self.wrapped = .noRetry
		case RetryPolicy.CodingKeys.Case.constant.stringValue:
			let dataContainer = try container.nestedContainer(keyedBy: RetryPolicy.CodingKeys.Constant.self,
			                                                  forKey: .data)
			let maxRetries = try dataContainer.decode(Int.self, forKey: .maxRetries)
			let delayInSeconds = try dataContainer.decode(TimeInterval.self, forKey: .delayInSeconds)
			self.wrapped = .constant(maxRetries: maxRetries,
			                         delayInSeconds: delayInSeconds)
		case RetryPolicy.CodingKeys.Case.exponential.stringValue:
			let dataContainer = try container.nestedContainer(keyedBy: RetryPolicy.CodingKeys.Exponential.self,
			                                                  forKey: .data)
			let maxRetries = try dataContainer.decode(Int.self, forKey: .maxRetries)
			let initialDelayInSeconds = try dataContainer.decode(TimeInterval.self, forKey: .initialDelayInSeconds)
			let multiplier = try dataContainer.decode(Double.self, forKey: .multiplier)
			let maximumDelayInSeconds = try dataContainer.decode(TimeInterval.self, forKey: .maximumDelayInSeconds)
			self.wrapped = .exponential(maxRetries: maxRetries,
			                            initialDelayInSeconds: initialDelayInSeconds,
			                            multiplier: multiplier,
			                            maximumDelayInSeconds: maximumDelayInSeconds)
		default:
			throw DecodingError.dataCorrupted(
				DecodingError.Context(
					codingPath: container.codingPath,
					debugDescription: "Failed to decode retry policy!"
				)
			)
		}
	}
}

extension RetryPolicy {
	enum CodingKeys {
		enum Case: CodingKey {
			case noRetry
			case constant
			case exponential
		}

		enum Constant: CodingKey {
			case maxRetries
			case delayInSeconds
		}

		enum Exponential: CodingKey {
			case maxRetries
			case initialDelayInSeconds
			case multiplier
			case maximumDelayInSeconds
		}
	}

	var caseName: String {
		Mirror(reflecting: self).children.first?.label ?? String(describing: self)
	}
}