UnkeyedEncodingContainer.swift

import Foundation

extension _CBOREncoder {
    final class UnkeyedContainer {
        private var storage: [CBOREncodingContainer] = []

        var count: Int {
            return storage.count
        }

        var codingPath: [CodingKey]

        var nestedCodingPath: [CodingKey] {
            return self.codingPath + [AnyCodingKey(intValue: self.count)]
        }

        var userInfo: [CodingUserInfoKey: Any]

        let options: CodableCBOREncoder._Options

        init(codingPath: [CodingKey], userInfo: [CodingUserInfoKey : Any], options: CodableCBOREncoder._Options) {
            self.codingPath = codingPath
            self.userInfo = userInfo
            self.options = options
        }
    }
}

extension _CBOREncoder.UnkeyedContainer: UnkeyedEncodingContainer {
    func encodeNil() throws {
        var container = self.nestedSingleValueContainer()
        try container.encodeNil()
    }

    func encode<T: Encodable>(_ value: T) throws {
        var container = self.nestedSingleValueContainer()
        try container.encode(value)
    }

    private func nestedSingleValueContainer() -> SingleValueEncodingContainer {
        let container = _CBOREncoder.SingleValueContainer(codingPath: self.nestedCodingPath, userInfo: self.userInfo, options: self.options)
        self.storage.append(container)

        return container
    }

    func nestedContainer<NestedKey: CodingKey>(keyedBy keyType: NestedKey.Type) -> KeyedEncodingContainer<NestedKey> {
        let container = _CBOREncoder.KeyedContainer<NestedKey>(codingPath: self.nestedCodingPath, userInfo: self.userInfo, options: self.options)
        self.storage.append(container)

        return KeyedEncodingContainer(container)
    }

    func nestedUnkeyedContainer() -> UnkeyedEncodingContainer {
        let container = _CBOREncoder.UnkeyedContainer(codingPath: self.nestedCodingPath, userInfo: self.userInfo, options: self.options)
        self.storage.append(container)

        return container
    }

    func superEncoder() -> Encoder {
        let encoder = _CBOREncoder(options: self.options)
        encoder.codingPath = self.nestedCodingPath
        encoder.userInfo = self.userInfo
        self.storage.append(encoder)
        return encoder
    }
}

extension _CBOREncoder.UnkeyedContainer: CBOREncodingContainer {
    var data: Data {
        // TODO: Check that this works for all sizes of array
        var data = storage.count.encode()
        data[0] = data[0] | 0b100_00000
        for container in storage {
            data.append(contentsOf: container.data)
        }
        return Data(data)
    }
}