WireMock.Net-wiremock/src/WireMock.Net.Shared/Settings/SimpleSettingsParser.cs

// Copyright © WireMock.Net

using System.Collections;
using WireMock.Extensions;
using WireMock.Util;

namespace WireMock.Settings;

// Based on http://blog.gauffin.org/2014/12/simple-command-line-parser/
internal class SimpleSettingsParser
{
    private const string Sigil = "--";
    private const string Prefix = $"{nameof(WireMockServerSettings)}__";
    private static readonly int PrefixLength = Prefix.Length;

    private IDictionary<string, string[]> Arguments { get; } = new Dictionary<string, string[]>(StringComparer.OrdinalIgnoreCase);

    public void Parse(string[] arguments, IDictionary? environment = null)
    {
        string currentName = string.Empty;

        var values = new List<string>();

        // Split a single argument on a space character to fix issue (e.g. Azure Service Fabric) when an argument is supplied like "--x abc" or '--x abc'
        foreach (string arg in arguments.SelectMany(arg => arg.Split(' ')))
        {
            if (arg.StartsWith(Sigil))
            {
                if (!string.IsNullOrEmpty(currentName))
                {
                    Arguments[currentName] = values.ToArray();
                }

                values.Clear();
                currentName = arg.Substring(Sigil.Length);
            }
            else if (string.IsNullOrEmpty(currentName))
            {
                Arguments[arg] = [];
            }
            else
            {
                values.Add(arg);
            }
        }

        if (!string.IsNullOrEmpty(currentName))
        {
            Arguments[currentName] = values.ToArray();
        }

        // Now also parse environment
        if (environment != null)
        {
            foreach (var key in environment.Keys.OfType<string>())
            {
                if (key.StartsWith(Prefix, StringComparison.OrdinalIgnoreCase) && environment.TryGetStringValue(key, out var value))
                {
                    Arguments[key.Substring(PrefixLength)] = value.Split(' ').ToArray();
                }
            }
        }
    }

    public bool Contains(string name)
    {
        return Arguments.ContainsKey(name);
    }

    public bool ContainsAny(params string[] names)
    {
        return names.Any(Arguments.ContainsKey);
    }

    public string[] GetValues(string name, string[] defaultValue)
    {
        return Contains(name) ? Arguments[name] : defaultValue;
    }

    public string[]? GetValues(string name)
    {
        return Contains(name) ? Arguments[name] : default;
    }

    public T GetValue<T>(string name, Func<string[], T> func, T defaultValue)
    {
        return Contains(name) ? func(Arguments[name]) : defaultValue;
    }

    public T? GetValue<T>(string name, Func<string[], T> func)
    {
        return Contains(name) ? func(Arguments[name]) : default;
    }

    public bool GetBoolValue(string name, bool defaultValue = false)
    {
        return GetValue(name, values =>
        {
            var value = values.FirstOrDefault();
            return !string.IsNullOrEmpty(value) ? bool.Parse(value) : defaultValue;
        }, defaultValue);
    }

    public bool GetBoolWithDefault(string key1, string key2, bool defaultValue)
    {
        if (Contains(key1))
        {
            return GetBoolValue(key1);
        }

        if (Contains(key2))
        {
            return GetBoolValue(key2);
        }

        return defaultValue;
    }

    public bool GetBoolSwitchValue(string name)
    {
        return Contains(name);
    }

    public int? GetIntValue(string name)
    {
        return GetValue<int?>(name, values =>
        {
            var value = values.FirstOrDefault();
            return !string.IsNullOrEmpty(value) ? int.Parse(value) : null;
        }, null);
    }

    public int GetIntValue(string name, int defaultValue)
    {
        return GetValue(name, values =>
        {
            var value = values.FirstOrDefault();
            return !string.IsNullOrEmpty(value) ? int.Parse(value) : defaultValue;
        }, defaultValue);
    }

    public TEnum? GetEnumValue<TEnum>(string name)
        where TEnum : struct
    {
        return GetValue(name, values =>
        {
            var value = values.FirstOrDefault();
            return Enum.TryParse<TEnum>(value, true, out var enumValue) ? enumValue : (TEnum?)null;
        });
    }

    public TEnum GetEnumValue<TEnum>(string name, TEnum defaultValue)
        where TEnum : struct
    {
        return GetValue(name, values =>
        {
            var value = values.FirstOrDefault();
            return Enum.TryParse<TEnum>(value, true, out var enumValue) ? enumValue : defaultValue;
        }, defaultValue);
    }

    public TEnum[] GetEnumValues<TEnum>(string name, TEnum[] defaultValues)
        where TEnum : struct
    {
        var values = GetValues(name);
        if (values == null)
        {
            return defaultValues;
        }

        var enums = new List<TEnum>();

        foreach (var value in values)
        {
            if (Enum.TryParse<TEnum>(value, true, out var enumValue))
            {
                enums.Add(enumValue);
            }
        }

        return enums.ToArray();
    }

    public string GetStringValue(string name, string defaultValue)
    {
        return GetValue(name, values => values.FirstOrDefault() ?? defaultValue, defaultValue);
    }

    public string? GetStringValue(string name)
    {
        return GetValue(name, values => values.FirstOrDefault());
    }

    public T? GetObjectValueFromJson<T>(string name)
    {
        var value = GetValue(name, values => values.FirstOrDefault());
        return string.IsNullOrWhiteSpace(value) ? default : JsonUtils.DeserializeObject<T>(value!);
    }
}