/*
 * 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 b2.
    /// </summary>
    public sealed class B2ServiceConfig : IServiceConfig
    {
        /// <summary>
        /// applicationKey of this backend. - If application_key is set, we will take user's input first. - If not, we will try to load it from environment.
        /// </summary>
        public string? ApplicationKey { get; init; }
        /// <summary>
        /// keyID of this backend. - If application_key_id is set, we will take user's input first. - If not, we will try to load it from environment.
        /// </summary>
        public string? ApplicationKeyId { get; init; }
        /// <summary>
        /// bucket of this backend. required.
        /// </summary>
        public string? Bucket { get; init; }
        /// <summary>
        /// bucket id of this backend. required.
        /// </summary>
        public string? BucketId { get; init; }
        /// <summary>
        /// root of this backend. All operations will happen under this root.
        /// </summary>
        public string? Root { get; init; }

        public string Scheme => "b2";

        public IReadOnlyDictionary<string, string> ToOptions()
        {
            var map = new Dictionary<string, string>();
            if (ApplicationKey is not null)
            {
                map["application_key"] = Utilities.ToOptionString(ApplicationKey);
            }
            if (ApplicationKeyId is not null)
            {
                map["application_key_id"] = Utilities.ToOptionString(ApplicationKeyId);
            }
            if (Bucket is not null)
            {
                map["bucket"] = Utilities.ToOptionString(Bucket);
            }
            if (BucketId is not null)
            {
                map["bucket_id"] = Utilities.ToOptionString(BucketId);
            }
            if (Root is not null)
            {
                map["root"] = Utilities.ToOptionString(Root);
            }
            return map;
        }
    }

}
