/*
 * 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 azfile.
    /// </summary>
    public sealed class AzfileServiceConfig : IServiceConfig
    {
        /// <summary>
        /// The account key for azfile.
        /// </summary>
        public string? AccountKey { get; init; }
        /// <summary>
        /// The account name for azfile.
        /// </summary>
        public string? AccountName { get; init; }
        /// <summary>
        /// The endpoint for azfile.
        /// </summary>
        public string? Endpoint { get; init; }
        /// <summary>
        /// The root path for azfile.
        /// </summary>
        public string? Root { get; init; }
        /// <summary>
        /// The sas token for azfile.
        /// </summary>
        public string? SasToken { get; init; }
        /// <summary>
        /// The share name for azfile.
        /// </summary>
        public string? ShareName { get; init; }

        public string Scheme => "azfile";

        public IReadOnlyDictionary<string, string> ToOptions()
        {
            var map = new Dictionary<string, string>();
            if (AccountKey is not null)
            {
                map["account_key"] = Utilities.ToOptionString(AccountKey);
            }
            if (AccountName is not null)
            {
                map["account_name"] = Utilities.ToOptionString(AccountName);
            }
            if (Endpoint is not null)
            {
                map["endpoint"] = Utilities.ToOptionString(Endpoint);
            }
            if (Root is not null)
            {
                map["root"] = Utilities.ToOptionString(Root);
            }
            if (SasToken is not null)
            {
                map["sas_token"] = Utilities.ToOptionString(SasToken);
            }
            if (ShareName is not null)
            {
                map["share_name"] = Utilities.ToOptionString(ShareName);
            }
            return map;
        }
    }

}
