63 lines
2.4 KiB
C#
63 lines
2.4 KiB
C#
|
/*
|
||
|
|
Copyright (c) 2015 Ki
|
||
|
|
|
||
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||
|
|
of this software and associated documentation files (the "Software"), to deal
|
||
|
|
in the Software without restriction, including without limitation the rights
|
||
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||
|
|
copies of the Software, and to permit persons to whom the Software is
|
||
|
|
furnished to do so, subject to the following conditions:
|
||
|
|
|
||
|
|
The above copyright notice and this permission notice shall be included in
|
||
|
|
all copies or substantial portions of the Software.
|
||
|
|
|
||
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||
|
|
THE SOFTWARE.
|
||
|
|
*/
|
||
|
|
|
||
|
|
using System.Linq;
|
||
|
|
using System.Xml.Linq;
|
||
|
|
using dnlib.DotNet;
|
||
|
|
using dnSpy.BamlDecompiler.Xaml;
|
||
|
|
|
||
|
|
namespace dnSpy.BamlDecompiler.Rewrite {
|
||
|
|
internal class XClassRewritePass : IRewritePass {
|
||
|
|
public void Run(XamlContext ctx, XDocument document) {
|
||
|
|
foreach (var elem in document.Elements(ctx.GetPseudoName("Document")).Elements())
|
||
|
|
RewriteClass(ctx, elem);
|
||
|
|
}
|
||
|
|
|
||
|
|
void RewriteClass(XamlContext ctx, XElement elem) {
|
||
|
|
var type = elem.Annotation<XamlType>();
|
||
|
|
if (type is null || type.ResolvedType is null)
|
||
|
|
return;
|
||
|
|
|
||
|
|
var typeDef = type.ResolvedType.ResolveTypeDef();
|
||
|
|
var comparer = new AssemblyNameComparer(AssemblyNameComparerFlags.All);
|
||
|
|
if (typeDef is null || !comparer.Equals(typeDef.DefinitionAssembly, ctx.Module.Assembly))
|
||
|
|
return;
|
||
|
|
|
||
|
|
var newType = typeDef.BaseType;
|
||
|
|
var xamlType = new XamlType(newType.DefinitionAssembly, newType.ReflectionNamespace, newType.Name);
|
||
|
|
xamlType.ResolveNamespace(elem, ctx);
|
||
|
|
|
||
|
|
elem.Name = xamlType.ToXName(ctx);
|
||
|
|
|
||
|
|
var attrName = ctx.GetXamlNsName("Class", elem);
|
||
|
|
|
||
|
|
var attrs = elem.Attributes().ToList();
|
||
|
|
if (typeDef.IsNotPublic) {
|
||
|
|
var classModifierName = ctx.GetXamlNsName("ClassModifier", elem);
|
||
|
|
attrs.Insert(0, new XAttribute(classModifierName, ctx.BamlDecompilerOptions.InternalClassModifier));
|
||
|
|
}
|
||
|
|
attrs.Insert(0, new XAttribute(attrName, type.ResolvedType.ReflectionFullName));
|
||
|
|
elem.ReplaceAttributes(attrs);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|