/*
 * 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 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;
        }
    }

}
