/*
 * 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.
 */

// DO NOT EDIT IT MANUALLY. This file is generated by opendal/dev/src/generate/dotnet.rs.

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>
        /// Deprecated: OBS versioning capability is not controlled by service config.
        /// </summary>
        [System.Obsolete("OBS versioning capability is not controlled by this option and this option is no longer needed.")]
        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);
            }
#pragma warning disable CS0618
            if (EnableVersioning is not null)
            {
                map["enable_versioning"] = Utilities.ToOptionString(EnableVersioning);
            }
#pragma warning restore CS0618
            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;
        }
    }

}
