/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

// Generated from bindings/java ServiceConfig definitions.

using OpenDAL.ServiceConfig.Abstractions;

namespace OpenDAL.ServiceConfig
{
    /// <summary>
    /// Configuration for service obs.
    /// </summary>
    public sealed class ObsServiceConfig : IServiceConfig
    {
        /// <summary>
        /// Access key id for obs.
        /// </summary>
        public string? AccessKeyId { get; init; }
        /// <summary>
        /// Bucket for obs.
        /// </summary>
        public string? Bucket { get; init; }
        /// <summary>
        /// Is bucket versioning enabled for this bucket
        /// </summary>
        public bool? EnableVersioning { get; init; }
        /// <summary>
        /// Endpoint for obs.
        /// </summary>
        public string? Endpoint { get; init; }
        /// <summary>
        /// Root for obs.
        /// </summary>
        public string? Root { get; init; }
        /// <summary>
        /// Secret access key for obs.
        /// </summary>
        public string? SecretAccessKey { get; init; }

        public string Scheme => "obs";

        public IReadOnlyDictionary<string, string> ToOptions()
        {
            var map = new Dictionary<string, string>();
            if (AccessKeyId is not null)
            {
                map["access_key_id"] = Utilities.ToOptionString(AccessKeyId);
            }
            if (Bucket is not null)
            {
                map["bucket"] = Utilities.ToOptionString(Bucket);
            }
            if (EnableVersioning is not null)
            {
                map["enable_versioning"] = Utilities.ToOptionString(EnableVersioning);
            }
            if (Endpoint is not null)
            {
                map["endpoint"] = Utilities.ToOptionString(Endpoint);
            }
            if (Root is not null)
            {
                map["root"] = Utilities.ToOptionString(Root);
            }
            if (SecretAccessKey is not null)
            {
                map["secret_access_key"] = Utilities.ToOptionString(SecretAccessKey);
            }
            return map;
        }
    }

}
