Make Year and Month Codable

Author: koher

Sources/SwiftYMD/Month.swift

@@ -19,6 +19,22 @@ extension Month: Equatable, Comparable, Hashable {
     }
 }
 
+extension Month: Codable {
+    public init(from decoder: Decoder) throws {
+        let container = try decoder.singleValueContainer()
+        let value = try container.decode(String.self)
+        guard let month = Month.allCases.first(where: { $0.description == value }) else {
+            throw DecodingError.dataCorrupted(DecodingError.Context.init(codingPath: decoder.codingPath, debugDescription: "Illegal value: \(value)"))
+        }
+        self = month
+    }
+    
+    public func encode(to encoder: Encoder) throws {
+        var container = encoder.singleValueContainer()
+        try container.encode(description)
+    }
+}
+
 extension Month: ExpressibleByIntegerLiteral {
     public init(integerLiteral value: Int) {
         self.init(rawValue: value)!

Sources/SwiftYMD/Year.swift

@@ -39,6 +39,19 @@ extension Year: Strideable {
     }
 }
 
+extension Year: Codable {
+    public init(from decoder: Decoder) throws {
+        let container = try decoder.singleValueContainer()
+        let value = try container.decode(Int.self)
+        self.init(value)
+    }
+    
+    public func encode(to encoder: Encoder) throws {
+        var container = encoder.singleValueContainer()
+        try container.encode(rawValue)
+    }
+}
+
 extension Year: CustomStringConvertible {
     public var description: String {
         rawValue.description

Tests/SwiftYMDTests/MonthTests.swift

@@ -0,0 +1,22 @@
+import XCTest
+import SwiftYMD
+
+class MonthTests: XCTestCase {
+    func testCodable() {
+        do { // encoding
+            let month: Month = .december
+            let data: Data = try JSONEncoder().encode(month)
+            XCTAssertEqual(String(data: data, encoding: .utf8)!, "\"December\"")
+        } catch {
+            XCTFail("\(error)")
+        }
+        
+        do { // decoding
+            let data: Data = "\"December\"".data(using: .utf8)!
+            let month: Month = try JSONDecoder().decode(Month.self, from: data)
+            XCTAssertEqual(month, .december)
+        } catch {
+            XCTFail("\(error)")
+        }
+    }
+}

Tests/SwiftYMDTests/XCTestManifests.swift

@@ -10,6 +10,15 @@ extension FoundationTests {
     ]
 }
 
+extension MonthTests {
+    // DO NOT MODIFY: This is autogenerated, use:
+    //   `swift test --generate-linuxmain`
+    // to regenerate.
+    static let __allTests__MonthTests = [
+        ("testCodable", testCodable),
+    ]
+}
+
 extension SwiftYMDTests {
     // DO NOT MODIFY: This is autogenerated, use:
     //   `swift test --generate-linuxmain`
@@ -50,6 +59,7 @@ extension YearTests {
     //   `swift test --generate-linuxmain`
     // to regenerate.
     static let __allTests__YearTests = [
+        ("testCodable", testCodable),
         ("testIsLeapYear", testIsLeapYear),
         ("testNumberOfDays", testNumberOfDays),
         ("testOperators", testOperators),
@@ -59,6 +69,7 @@ extension YearTests {
 public func __allTests() -> [XCTestCaseEntry] {
     return [
         testCase(FoundationTests.__allTests__FoundationTests),
+        testCase(MonthTests.__allTests__MonthTests),
         testCase(SwiftYMDTests.__allTests__SwiftYMDTests),
         testCase(YMDTests.__allTests__YMDTests),
         testCase(YMTests.__allTests__YMTests),

Tests/SwiftYMDTests/YearTests.swift

@@ -54,6 +54,24 @@ class YearTests: XCTestCase {
         XCTAssertEqual((2105 as Year).numberOfDays, 365)
     }
 
+    func testCodable() {
+        do { // encoding
+            let year: Year = 1988
+            let data: Data = try JSONEncoder().encode(year)
+            XCTAssertEqual(String(data: data, encoding: .utf8)!, "1988")
+        } catch {
+            XCTFail("\(error)")
+        }
+        
+        do { // decoding
+            let data: Data = "1988".data(using: .utf8)!
+            let year: Year = try JSONDecoder().decode(Year.self, from: data)
+            XCTAssertEqual(year, 1988)
+        } catch {
+            XCTFail("\(error)")
+        }
+    }
+    
     func testOperators() {
         XCTAssertEqual((1988 as Year) + (31 as Year), (2019 as Year))
         do {