mirror of https://github.com/icsharpcode/ILSpy.git
9 changed files with 246 additions and 108 deletions
@ -0,0 +1,125 @@ |
|||||||
|
using System; |
||||||
|
using System.Collections.Concurrent; |
||||||
|
using System.Text.RegularExpressions; |
||||||
|
using System.Threading; |
||||||
|
using System.Windows.Media; |
||||||
|
using ICSharpCode.Decompiler.Metadata; |
||||||
|
using ICSharpCode.Decompiler.TypeSystem; |
||||||
|
using ICSharpCode.ILSpy.TreeNodes; |
||||||
|
|
||||||
|
namespace ICSharpCode.ILSpy.Search |
||||||
|
{ |
||||||
|
abstract class AbstractEntitySearchStrategy : AbstractSearchStrategy |
||||||
|
{ |
||||||
|
protected readonly Language language; |
||||||
|
protected readonly ApiVisibility apiVisibility; |
||||||
|
|
||||||
|
protected AbstractEntitySearchStrategy(Language language, ApiVisibility apiVisibility, IProducerConsumerCollection<SearchResult> resultQueue, params string[] terms) |
||||||
|
: base(resultQueue, terms) |
||||||
|
{ |
||||||
|
this.language = language; |
||||||
|
this.apiVisibility = apiVisibility; |
||||||
|
} |
||||||
|
|
||||||
|
protected bool CheckVisibility(IEntity entity) |
||||||
|
{ |
||||||
|
if (apiVisibility == ApiVisibility.All) |
||||||
|
return true; |
||||||
|
|
||||||
|
do { |
||||||
|
if (apiVisibility == ApiVisibility.PublicOnly) { |
||||||
|
if (!(entity.Accessibility == Accessibility.Public || |
||||||
|
entity.Accessibility == Accessibility.Protected || |
||||||
|
entity.Accessibility == Accessibility.ProtectedOrInternal)) |
||||||
|
return false; |
||||||
|
} else if (apiVisibility == ApiVisibility.PublicAndInternal) { |
||||||
|
if (!language.ShowMember(entity)) |
||||||
|
return false; |
||||||
|
} |
||||||
|
entity = entity.DeclaringTypeDefinition; |
||||||
|
} |
||||||
|
while (entity != null); |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
protected void OnFoundResult(IEntity entity) |
||||||
|
{ |
||||||
|
var result = ResultFromEntity(entity); |
||||||
|
OnFoundResult(result); |
||||||
|
} |
||||||
|
|
||||||
|
SearchResult ResultFromEntity(IEntity item) |
||||||
|
{ |
||||||
|
var declaringType = item.DeclaringTypeDefinition; |
||||||
|
return new SearchResult { |
||||||
|
Reference = item, |
||||||
|
Fitness = CalculateFitness(item), |
||||||
|
Image = GetIcon(item), |
||||||
|
Name = GetLanguageSpecificName(item), |
||||||
|
LocationImage = declaringType != null ? TypeTreeNode.GetIcon(declaringType) : Images.Namespace, |
||||||
|
Location = declaringType != null ? language.TypeToString(declaringType, includeNamespace: true) : item.Namespace, |
||||||
|
AssemblyImage = Images.Assembly, |
||||||
|
Assembly = item.ParentModule.FullAssemblyName, |
||||||
|
ToolTip = item.ParentModule.PEFile?.FileName |
||||||
|
}; |
||||||
|
} |
||||||
|
|
||||||
|
float CalculateFitness(IEntity member) |
||||||
|
{ |
||||||
|
string text = member.Name; |
||||||
|
|
||||||
|
// Probably compiler generated types without meaningful names, show them last
|
||||||
|
if (text.StartsWith("<")) { |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
// Constructors always have the same name in IL:
|
||||||
|
// Use type name instead
|
||||||
|
if (text == ".cctor" || text == ".ctor") { |
||||||
|
text = member.DeclaringType.Name; |
||||||
|
} |
||||||
|
|
||||||
|
// Ignore generic arguments, it not possible to search based on them either
|
||||||
|
text = ReflectionHelper.SplitTypeParameterCountFromReflectionName(text); |
||||||
|
|
||||||
|
return 1.0f / text.Length; |
||||||
|
} |
||||||
|
|
||||||
|
string GetLanguageSpecificName(IEntity member) |
||||||
|
{ |
||||||
|
switch (member) { |
||||||
|
case ITypeDefinition t: |
||||||
|
return language.TypeToString(t, false); |
||||||
|
case IField f: |
||||||
|
return language.FieldToString(f, true, false, false); |
||||||
|
case IProperty p: |
||||||
|
return language.PropertyToString(p, true, false, false); |
||||||
|
case IMethod m: |
||||||
|
return language.MethodToString(m, true, false, false); |
||||||
|
case IEvent e: |
||||||
|
return language.EventToString(e, true, false, false); |
||||||
|
default: |
||||||
|
throw new NotSupportedException(member?.GetType() + " not supported!"); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
ImageSource GetIcon(IEntity member) |
||||||
|
{ |
||||||
|
switch (member) { |
||||||
|
case ITypeDefinition t: |
||||||
|
return TypeTreeNode.GetIcon(t); |
||||||
|
case IField f: |
||||||
|
return FieldTreeNode.GetIcon(f); |
||||||
|
case IProperty p: |
||||||
|
return PropertyTreeNode.GetIcon(p); |
||||||
|
case IMethod m: |
||||||
|
return MethodTreeNode.GetIcon(m); |
||||||
|
case IEvent e: |
||||||
|
return EventTreeNode.GetIcon(e); |
||||||
|
default: |
||||||
|
throw new NotSupportedException(member?.GetType() + " not supported!"); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,93 @@ |
|||||||
|
using System; |
||||||
|
using System.Collections.Concurrent; |
||||||
|
using System.Collections.Generic; |
||||||
|
using System.IO; |
||||||
|
using System.Reflection; |
||||||
|
using System.Threading; |
||||||
|
using System.Windows.Media; |
||||||
|
using System.Windows.Media.Imaging; |
||||||
|
using ICSharpCode.Decompiler.Metadata; |
||||||
|
using ICSharpCode.Decompiler.TypeSystem; |
||||||
|
using ICSharpCode.Decompiler.Util; |
||||||
|
using ICSharpCode.ILSpy.TreeNodes; |
||||||
|
using ICSharpCode.TreeView; |
||||||
|
|
||||||
|
namespace ICSharpCode.ILSpy.Search |
||||||
|
{ |
||||||
|
class ResourceSearchStrategy : AbstractSearchStrategy |
||||||
|
{ |
||||||
|
protected readonly bool searchInside; |
||||||
|
protected readonly ApiVisibility apiVisibility; |
||||||
|
|
||||||
|
public ResourceSearchStrategy(ApiVisibility apiVisibility, IProducerConsumerCollection<SearchResult> resultQueue, string term) |
||||||
|
: this(apiVisibility, resultQueue, new[] { term }) |
||||||
|
{ |
||||||
|
} |
||||||
|
|
||||||
|
public ResourceSearchStrategy(ApiVisibility apiVisibility, IProducerConsumerCollection<SearchResult> resultQueue, string[] terms) |
||||||
|
: base(resultQueue, terms) |
||||||
|
{ |
||||||
|
this.apiVisibility = apiVisibility; |
||||||
|
this.searchInside = true; |
||||||
|
} |
||||||
|
|
||||||
|
protected bool CheckVisibility(Resource resource) |
||||||
|
{ |
||||||
|
if (apiVisibility == ApiVisibility.All) |
||||||
|
return true; |
||||||
|
|
||||||
|
if (apiVisibility == ApiVisibility.PublicOnly && (resource.Attributes & ManifestResourceAttributes.VisibilityMask) == ManifestResourceAttributes.Private) |
||||||
|
return false; |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
public override void Search(PEFile module, CancellationToken cancellationToken) |
||||||
|
{ |
||||||
|
cancellationToken.ThrowIfCancellationRequested(); |
||||||
|
var resourcesNode = new ResourceListTreeNode(module); |
||||||
|
|
||||||
|
resourcesNode.EnsureLazyChildren(); |
||||||
|
foreach (var node in resourcesNode.Children) |
||||||
|
Search(module, null, resourcesNode, node, cancellationToken); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
void Search(PEFile module, object reference, SharpTreeNode parent, SharpTreeNode node, CancellationToken cancellationToken) |
||||||
|
{ |
||||||
|
cancellationToken.ThrowIfCancellationRequested(); |
||||||
|
|
||||||
|
if (node is ResourceTreeNode treeNode) { |
||||||
|
if (!CheckVisibility(treeNode.Resource)) |
||||||
|
return; |
||||||
|
reference = treeNode.Resource; |
||||||
|
} |
||||||
|
|
||||||
|
if (node.Text != null && IsMatch((string)node.Text)) |
||||||
|
OnFoundResult(module, reference, node, parent); |
||||||
|
|
||||||
|
if (!searchInside) |
||||||
|
return; |
||||||
|
|
||||||
|
node.EnsureLazyChildren(); |
||||||
|
foreach (var child in node.Children) |
||||||
|
Search(module, reference, node, child, cancellationToken); |
||||||
|
} |
||||||
|
|
||||||
|
void OnFoundResult(PEFile module, object reference, SharpTreeNode node, SharpTreeNode parent) |
||||||
|
{ |
||||||
|
var result = new SearchResult { |
||||||
|
Reference = reference, |
||||||
|
Fitness = 1f, |
||||||
|
Image = (ImageSource)node.Icon, |
||||||
|
Name = (string)node.Text, |
||||||
|
LocationImage = (ImageSource)parent.Icon, |
||||||
|
Location = (string)parent.Text, |
||||||
|
AssemblyImage = Images.Assembly, |
||||||
|
Assembly = module.Name, |
||||||
|
ToolTip = module.FileName, |
||||||
|
}; |
||||||
|
OnFoundResult(result); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue