74 lines
2.3 KiB
C#
74 lines
2.3 KiB
C#
using AssemblyRemapper.Enums;
|
|
using AssemblyRemapper.Models;
|
|
using Mono.Cecil;
|
|
using Mono.Cecil.Rocks;
|
|
|
|
namespace AssemblyRemapper.Remapper.Search;
|
|
|
|
internal static class Methods
|
|
{
|
|
/// <summary>
|
|
/// returns a match on all types with the specified methods
|
|
/// </summary>
|
|
/// <param name="type"></param>
|
|
/// <param name="parms"></param>
|
|
/// <param name="score"></param>
|
|
/// <returns>Match if type contains any supplied methods</returns>
|
|
public static EMatchResult IncludeMethods(TypeDefinition type, SearchParams parms, ScoringModel score)
|
|
{
|
|
if (parms.IncludeMethods is null || parms.IncludeMethods.Count == 0) return EMatchResult.Disabled;
|
|
|
|
var matches = type.Methods
|
|
.Where(method => parms.IncludeMethods.Contains(method.Name))
|
|
.Count();
|
|
|
|
score.Score += matches;
|
|
|
|
return matches > 0
|
|
? EMatchResult.Match
|
|
: EMatchResult.NoMatch;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns a match on all types without methods
|
|
/// </summary>
|
|
/// <param name="type"></param>
|
|
/// <param name="parms"></param>
|
|
/// <param name="score"></param>
|
|
/// <returns>Match if type has no methods</returns>
|
|
public static EMatchResult ExcludeMethods(TypeDefinition type, SearchParams parms, ScoringModel score)
|
|
{
|
|
if (parms.ExcludeMethods is null || parms.ExcludeMethods.Count == 0) return EMatchResult.Disabled;
|
|
|
|
var matches = type.Methods
|
|
.Where(method => parms.ExcludeMethods.Contains(method.Name))
|
|
.Count();
|
|
|
|
score.Score += matches;
|
|
|
|
return matches > 0
|
|
? EMatchResult.NoMatch
|
|
: EMatchResult.Match;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns a match if the type has the provided number of methods
|
|
/// </summary>
|
|
/// <param name="type"></param>
|
|
/// <param name="parms"></param>
|
|
/// <param name="score"></param>
|
|
/// <returns></returns>
|
|
public static EMatchResult MatchMethodCount(TypeDefinition type, SearchParams parms, ScoringModel score)
|
|
{
|
|
if (parms.MethodCount is null) return EMatchResult.Disabled;
|
|
|
|
var numMethods = type.Methods.Count - type.GetConstructors().Count();
|
|
bool match = numMethods == parms.MethodCount;
|
|
|
|
if (match) { score.Score++; }
|
|
|
|
return match
|
|
? EMatchResult.Match
|
|
: EMatchResult.NoMatch;
|
|
}
|
|
} |