mirror of https://github.com/icsharpcode/ILSpy.git
Jan Kučera
6 years ago
9 changed files with 246 additions and 108 deletions
@ -0,0 +1,125 @@
@@ -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 @@
@@ -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