WireMock.Net-wiremock/src/WireMock.Net/Matchers/SimMetricsMatcher.cs

using System.Linq;
using JetBrains.Annotations;
using SimMetrics.Net;
using SimMetrics.Net.API;
using SimMetrics.Net.Metric;
using WireMock.Validation;

namespace WireMock.Matchers
{
    /// <summary>
    /// SimMetricsMatcher
    /// </summary>
    /// <seealso cref="IStringMatcher" />
    public class SimMetricsMatcher : IStringMatcher
    {
        private readonly string[] _patterns;
        private readonly SimMetricType _simMetricType;

        /// <inheritdoc cref="IMatcher.MatchBehaviour"/>
        public MatchBehaviour MatchBehaviour { get; }

        /// <summary>
        /// Initializes a new instance of the <see cref="SimMetricsMatcher"/> class.
        /// </summary>
        /// <param name="pattern">The pattern.</param>
        /// <param name="simMetricType">The SimMetric Type</param>
        public SimMetricsMatcher([NotNull] string pattern, SimMetricType simMetricType = SimMetricType.Levenstein) : this(new[] { pattern }, simMetricType)
        {
        }

        /// <summary>
        /// Initializes a new instance of the <see cref="SimMetricsMatcher"/> class.
        /// </summary>
        /// <param name="matchBehaviour">The match behaviour.</param>
        /// <param name="pattern">The pattern.</param>
        /// <param name="simMetricType">The SimMetric Type</param>
        public SimMetricsMatcher(MatchBehaviour matchBehaviour, [NotNull] string pattern, SimMetricType simMetricType = SimMetricType.Levenstein) : this(matchBehaviour, new[] { pattern }, simMetricType)
        {
        }

        /// <summary>
        /// Initializes a new instance of the <see cref="SimMetricsMatcher"/> class.
        /// </summary>
        /// <param name="patterns">The patterns.</param>
        /// <param name="simMetricType">The SimMetric Type</param>
        public SimMetricsMatcher([NotNull] string[] patterns, SimMetricType simMetricType = SimMetricType.Levenstein) : this(MatchBehaviour.AcceptOnMatch, patterns, simMetricType)
        {
        }

        /// <summary>
        /// Initializes a new instance of the <see cref="SimMetricsMatcher"/> class.
        /// </summary>
        /// <param name="matchBehaviour">The match behaviour.</param>
        /// <param name="patterns">The patterns.</param>
        /// <param name="simMetricType">The SimMetric Type</param>
        public SimMetricsMatcher(MatchBehaviour matchBehaviour, [NotNull] string[] patterns, SimMetricType simMetricType = SimMetricType.Levenstein)
        {
            Check.NotNullOrEmpty(patterns, nameof(patterns));

            MatchBehaviour = matchBehaviour;
            _patterns = patterns;
            _simMetricType = simMetricType;
        }

        /// <inheritdoc cref="IStringMatcher.IsMatch"/>
        public double IsMatch(string input)
        {
            IStringMetric m = GetStringMetricType();

            return MatchBehaviourHelper.Convert(MatchBehaviour, MatchScores.ToScore(_patterns.Select(p => m.GetSimilarity(p, input))));
        }

        private IStringMetric GetStringMetricType()
        {
            switch (_simMetricType)
            {
                case SimMetricType.BlockDistance:
                    return new BlockDistance();
                case SimMetricType.ChapmanLengthDeviation:
                    return new ChapmanLengthDeviation();
                case SimMetricType.CosineSimilarity:
                    return new CosineSimilarity();
                case SimMetricType.DiceSimilarity:
                    return new DiceSimilarity();
                case SimMetricType.EuclideanDistance:
                    return new EuclideanDistance();
                case SimMetricType.JaccardSimilarity:
                    return new JaccardSimilarity();
                case SimMetricType.Jaro:
                    return new Jaro();
                case SimMetricType.JaroWinkler:
                    return new JaroWinkler();
                case SimMetricType.MatchingCoefficient:
                    return new MatchingCoefficient();
                case SimMetricType.MongeElkan:
                    return new MongeElkan();
                case SimMetricType.NeedlemanWunch:
                    return new NeedlemanWunch();
                case SimMetricType.OverlapCoefficient:
                    return new OverlapCoefficient();
                case SimMetricType.QGramsDistance:
                    return new QGramsDistance();
                case SimMetricType.SmithWaterman:
                    return new SmithWaterman();
                case SimMetricType.SmithWatermanGotoh:
                    return new SmithWatermanGotoh();
                case SimMetricType.SmithWatermanGotohWindowedAffine:
                    return new SmithWatermanGotohWindowedAffine();
                case SimMetricType.ChapmanMeanLength:
                    return new ChapmanMeanLength();
                default:
                    return new Levenstein();
            }
        }

        /// <inheritdoc cref="IStringMatcher.GetPatterns"/>
        public string[] GetPatterns()
        {
            return _patterns;
        }

        /// <inheritdoc cref="IMatcher.Name"/>
        public string Name => $"SimMetricsMatcher.{_simMetricType}";
    }
}