swift-protobuf/Sources/SwiftProtobuf/Google_Protobuf_Struct+Extensions.swift

// Sources/SwiftProtobuf/Google_Protobuf_Struct+Extensions.swift - Struct extensions
//
// Copyright (c) 2014 - 2017 Apple Inc. and the project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See LICENSE.txt for license information:
// https://github.com/apple/swift-protobuf/blob/main/LICENSE.txt
//
// -----------------------------------------------------------------------------
///
/// Struct is a well-known message type that can be used to parse or encode
/// arbitrary JSON objects without a predefined schema.
///
// -----------------------------------------------------------------------------

extension Google_Protobuf_Struct: ExpressibleByDictionaryLiteral {
    public typealias Key = String
    public typealias Value = Google_Protobuf_Value

    /// Creates a new `Google_Protobuf_Struct` from a dictionary of string keys to
    /// values of type `Google_Protobuf_Value`.
    public init(dictionaryLiteral: (String, Google_Protobuf_Value)...) {
        self.init()
        for (k, v) in dictionaryLiteral {
            fields[k] = v
        }
    }
}

extension Google_Protobuf_Struct: _CustomJSONCodable {
    internal func encodedJSONString(options: JSONEncodingOptions) throws -> String {
        var jsonEncoder = JSONEncoder()
        jsonEncoder.startObject()
        var mapVisitor = JSONMapEncodingVisitor(encoder: jsonEncoder, options: options)
        for (k, v) in fields {
            try mapVisitor.visitSingularStringField(value: k, fieldNumber: 1)
            try mapVisitor.visitSingularMessageField(value: v, fieldNumber: 2)
        }
        mapVisitor.encoder.endObject()
        return mapVisitor.encoder.stringResult
    }

    internal mutating func decodeJSON(from decoder: inout JSONDecoder) throws {
        try decoder.scanner.skipRequiredObjectStart()
        if decoder.scanner.skipOptionalObjectEnd() {
            return
        }
        while true {
            let key = try decoder.scanner.nextQuotedString()
            try decoder.scanner.skipRequiredColon()
            var value = Google_Protobuf_Value()
            try value.decodeJSON(from: &decoder)
            fields[key] = value
            if decoder.scanner.skipOptionalObjectEnd() {
                return
            }
            try decoder.scanner.skipRequiredComma()
        }
    }
}

extension Google_Protobuf_Struct {
    /// Creates a new `Google_Protobuf_Struct` from a dictionary of string keys to
    /// values of type `Google_Protobuf_Value`.
    ///
    /// - Parameter fields: The dictionary from field names to
    ///   `Google_Protobuf_Value` messages that should be used to create the
    ///   `Struct`.
    public init(fields: [String: Google_Protobuf_Value]) {
        self.init()
        self.fields = fields
    }

    /// Accesses the `Google_Protobuf_Value` with the given key for reading and
    /// writing.
    ///
    /// This key-based subscript returns the `Value` for the given key if the key
    /// is found in the `Struct`, or nil if the key is not found. If you assign
    /// nil as the `Value` for the given key, the `Struct` removes that key and
    /// its associated `Value`.
    public subscript(key: String) -> Google_Protobuf_Value? {
        get { fields[key] }
        set(newValue) { fields[key] = newValue }
    }
}