/*
 * 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 dropbox.
    /// </summary>
    public sealed class DropboxServiceConfig : IServiceConfig
    {
        /// <summary>
        /// access token for dropbox.
        /// </summary>
        public string? AccessToken { get; init; }
        /// <summary>
        /// client_id for dropbox.
        /// </summary>
        public string? ClientId { get; init; }
        /// <summary>
        /// client_secret for dropbox.
        /// </summary>
        public string? ClientSecret { get; init; }
        /// <summary>
        /// refresh_token for dropbox.
        /// </summary>
        public string? RefreshToken { get; init; }
        /// <summary>
        /// root path for dropbox.
        /// </summary>
        public string? Root { get; init; }

        public string Scheme => "dropbox";

        public IReadOnlyDictionary<string, string> ToOptions()
        {
            var map = new Dictionary<string, string>();
            if (AccessToken is not null)
            {
                map["access_token"] = Utilities.ToOptionString(AccessToken);
            }
            if (ClientId is not null)
            {
                map["client_id"] = Utilities.ToOptionString(ClientId);
            }
            if (ClientSecret is not null)
            {
                map["client_secret"] = Utilities.ToOptionString(ClientSecret);
            }
            if (RefreshToken is not null)
            {
                map["refresh_token"] = Utilities.ToOptionString(RefreshToken);
            }
            if (Root is not null)
            {
                map["root"] = Utilities.ToOptionString(Root);
            }
            return map;
        }
    }

}
