Cosmos/Users/Orvid/PlugViewer/MainForm.cs

586 lines
20 KiB
C#

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Text;
using System.Windows.Forms;
using System.Reflection;
using System.IO;
using PlugViewer.TreeViewNodes;
namespace PlugViewer
{
public partial class MainForm : Form
{
public MainForm()
{
InitializeComponent();
treeView1.TreeViewNodeSorter = new TreeViewSorter();
}
private void AddAssemblyToView(Assembly a)
{
if (a != null)
{
PlugTemplateDumper.Dump(a);
try
{
TreeNode nd = new AssemblyTreeNode(a);
treeView1.Nodes.Add(nd);
foreach (Module md in a.GetModules())
{
LoadModule(nd, md);
}
treeView1.Sort();
}
catch {}
}
}
#region LoadModule
private void LoadModule(TreeNode parent, Module md)
{
ModuleTreeNode tn = new ModuleTreeNode(md);
parent.Nodes.Add(tn);
foreach (Type t in md.GetTypes())
{
if (!tn.Namespaces.ContainsKey((t.Namespace == null ? "-" : t.Namespace)))
{
TreeNode tnd = new NamespaceTreeNode((t.Namespace == null ? "-" : t.Namespace));
tn.Nodes.Add(tnd);
tn.Namespaces.Add((t.Namespace == null ? "-" : t.Namespace), tnd);
}
LoadType(tn.Namespaces[(t.Namespace == null ? "-" : t.Namespace)], t);
}
}
#endregion
#region LoadType
private void LoadType(TreeNode parentNode, Type type)
{
if (!type.IsEnum && !type.IsInterface && !type.IsValueType && type.IsClass)
{
TreeNode node;
if (type.IsPublic)
{
node = new ClassTreeNode(type, ClassType.Class, Access.Public);
parentNode.Nodes.Add(node);
}
else if (type.IsNestedPrivate)
{
node = new ClassTreeNode(type, ClassType.Class, Access.Private);
parentNode.Nodes.Add(node);
}
else if (type.IsNestedFamORAssem)
{
node = new ClassTreeNode(type, ClassType.Class, Access.Internal);
parentNode.Nodes.Add(node);
}
else
{
// It must be Protected.
node = new ClassTreeNode(type, ClassType.Class, Access.Protected);
parentNode.Nodes.Add(node);
}
//TreeNode node = (treeView1.Nodes.Find(type.FullName, true))[0];
foreach (Type ntd in type.GetNestedTypes())
{
LoadType(node, ntd);
}
foreach (MethodInfo md in type.GetMethods())
{
LoadMethod(node, md);
}
foreach (Type ntd in type.GetInterfaces())
{
LoadImplementedInterface(node, ntd);
}
foreach (EventInfo ed in type.GetEvents())
{
LoadEvent(node, ed);
}
foreach (PropertyInfo pd in type.GetProperties())
{
LoadProperty(node, pd);
}
foreach (FieldInfo fd in type.GetFields())
{
LoadField(node, fd);
}
}
else if (type.IsEnum)
{
LoadEnum(parentNode, type);
}
else if (type.IsInterface)
{
LoadInterface(parentNode, type.GetElementType());
}
else if (type.IsValueType)
{
LoadStruct(parentNode, type);
}
else
{
throw new Exception();
}
}
#endregion
/// <summary>
/// Checks if the method is part of an event or property definition.
/// </summary>
/// <param name="method"></param>
/// <returns></returns>
private bool ShouldLoadMethod(MethodInfo method)
{
string s;
if (method.Name.Length <= 4) // Too short to be one of things being tested
{
return true;
}
else if (method.Name.Length < 7) // only check add, get, and set
{
s = method.Name.Substring(0,4);
if (s == "add_" || s == "get_" || s == "set_")
{
return false;
}
return true;
}
else // Check add, remove, get, and set
{
s = method.Name.Substring(0, 4);
if (s == "add_" || s == "get_" || s == "set_")
{
return false;
}
else if (method.Name.Substring(0, 7) == "remove_")
{
return false;
}
return true;
}
}
private void LoadMethod(TreeNode parentNode, MethodInfo method)
{
string dispkey = NameBuilder.BuildMethodDisplayName(method);
TreeNode tn;
// Check that the method isn't part of a property or event definition.
if (ShouldLoadMethod(method))
{
if (method.IsVirtual)
{
if (method.GetMethodBody() != null) // The method is an override
{
if (method.IsPublic)
{
tn = new MethodTreeNode(method, MethodType.OverrideMethod, Access.Public);
parentNode.Nodes.Add(tn);
}
else if (method.IsPrivate)
{
tn = new MethodTreeNode(method, MethodType.OverrideMethod, Access.Private);
parentNode.Nodes.Add(tn);
}
else if (method.IsFamilyOrAssembly)
{
tn = new MethodTreeNode(method, MethodType.OverrideMethod, Access.Internal);
parentNode.Nodes.Add(tn);
}
else
{
// It must be protected.
tn = new MethodTreeNode(method, MethodType.OverrideMethod, Access.Protected);
parentNode.Nodes.Add(tn);
}
}
else // The method is virtual, but not an override.
{
if (method.IsPublic)
{
tn = new MethodTreeNode(method, MethodType.VirtualMethod, Access.Public);
parentNode.Nodes.Add(tn);
}
else if (method.IsPrivate)
{
tn = new MethodTreeNode(method, MethodType.VirtualMethod, Access.Private);
parentNode.Nodes.Add(tn);
}
else if (method.IsFamilyOrAssembly)
{
tn = new MethodTreeNode(method, MethodType.VirtualMethod, Access.Internal);
parentNode.Nodes.Add(tn);
}
else
{
// It must be protected.
tn = new MethodTreeNode(method, MethodType.VirtualMethod, Access.Protected);
parentNode.Nodes.Add(tn);
}
}
}
else // The method is a regular method
{
if (method.IsPublic)
{
tn = new MethodTreeNode(method, MethodType.BasicMethod, Access.Public);
parentNode.Nodes.Add(tn);
}
else if (method.IsPrivate)
{
tn = new MethodTreeNode(method, MethodType.BasicMethod, Access.Private);
parentNode.Nodes.Add(tn);
}
else if (method.IsFamilyOrAssembly)
{
tn = new MethodTreeNode(method, MethodType.BasicMethod, Access.Internal);
parentNode.Nodes.Add(tn);
}
else
{
// It must be protected.
tn = new MethodTreeNode(method, MethodType.BasicMethod, Access.Protected);
parentNode.Nodes.Add(tn);
}
}
}
}
#region LoadImplementedInterface
private void LoadImplementedInterface(TreeNode parentNode, Type intface)
{
TreeNode node;
if (intface.IsPublic)
{
node = new ClassTreeNode(intface, ClassType.ImplementedInterface, Access.Public);
parentNode.Nodes.Add(node);
}
else if (intface.IsNestedPrivate)
{
node = new ClassTreeNode(intface, ClassType.ImplementedInterface, Access.Private);
parentNode.Nodes.Add(node);
}
else if (intface.IsNestedFamORAssem)
{
node = new ClassTreeNode(intface, ClassType.ImplementedInterface, Access.Internal);
parentNode.Nodes.Add(node);
}
else
{
// It must be protected.
node = new ClassTreeNode(intface, ClassType.ImplementedInterface, Access.Protected);
parentNode.Nodes.Add(node);
}
foreach (Type ntd in intface.GetNestedTypes())
{
LoadType(node, ntd);
}
foreach (MethodInfo md in intface.GetMethods())
{
LoadMethod(node, md);
}
foreach (Type ntd in intface.GetInterfaces())
{
LoadImplementedInterface(node, ntd);
}
foreach (EventInfo ed in intface.GetEvents())
{
LoadEvent(node, ed);
}
foreach (PropertyInfo pd in intface.GetProperties())
{
LoadProperty(node, pd);
}
foreach (FieldInfo fd in intface.GetFields())
{
LoadField(node, fd);
}
}
#endregion
#region LoadInterface
private void LoadInterface(TreeNode parentNode, Type intface)
{
TreeNode node;
if (intface.IsPublic)
{
node = new ClassTreeNode(intface, ClassType.Interface, Access.Public);
parentNode.Nodes.Add(node);
}
else if (intface.IsNestedPrivate)
{
node = new ClassTreeNode(intface, ClassType.Interface, Access.Private);
parentNode.Nodes.Add(node);
}
else if (intface.IsNestedFamORAssem)
{
node = new ClassTreeNode(intface, ClassType.Interface, Access.Internal);
parentNode.Nodes.Add(node);
}
else
{
// It must be protected.
node = new ClassTreeNode(intface, ClassType.Interface, Access.Protected);
parentNode.Nodes.Add(node);
}
foreach (Type ntd in intface.GetNestedTypes())
{
LoadType(node, ntd);
}
foreach (MethodInfo md in intface.GetMethods())
{
LoadMethod(node, md);
}
foreach (Type ntd in intface.GetInterfaces())
{
LoadImplementedInterface(node, ntd);
}
foreach (EventInfo ed in intface.GetEvents())
{
LoadEvent(node, ed);
}
foreach (PropertyInfo pd in intface.GetProperties())
{
LoadProperty(node, pd);
}
foreach (FieldInfo fd in intface.GetFields())
{
LoadField(node, fd);
}
}
#endregion
#region LoadEvent
private void LoadEvent(TreeNode parentNode, EventInfo evnt)
{
TreeNode n = new EventTreeNode(evnt);
parentNode.Nodes.Add(n);
}
#endregion
#region LoadProperty
private void LoadProperty(TreeNode parentNode, PropertyInfo ptd)
{
TreeNode n;
if (ptd.CanWrite)
{
n = new PropertyTreeNode(ptd, true);
parentNode.Nodes.Add(n);
}
else
{
n = new PropertyTreeNode(ptd, false);
parentNode.Nodes.Add(n);
}
}
#endregion
//private void LoadConstant(TreeNode parentNode, PropertyInfo ptd)
//{
// //TreeNode tr = new FieldTreeNode(fd, Access.Protected, false);
// //parentNode.Nodes.Add(tr);
// parentNode.Nodes.Add(ptd.FullName, ptd.Name, Constants.ConstantIcon, Constants.ConstantIcon);
//}
#region LoadEnum
private void LoadEnum(TreeNode parentNode, Type td)
{
TreeNode enode;
TreeNode tr;
if (td.IsPublic)
{
enode = new ClassTreeNode(td, ClassType.Enum, Access.Public);
parentNode.Nodes.Add(enode);
}
else if (td.IsNestedPrivate)
{
enode = new ClassTreeNode(td, ClassType.Enum, Access.Private);
parentNode.Nodes.Add(enode);
}
else if (td.IsNestedFamORAssem)
{
enode = new ClassTreeNode(td, ClassType.Enum, Access.Internal);
parentNode.Nodes.Add(enode);
}
else
{
// It must be protected.
enode = new ClassTreeNode(td, ClassType.Enum, Access.Protected);
parentNode.Nodes.Add(enode);
}
// Now we get to load it's values.
foreach (FieldInfo fd in td.GetFields())
{
// We don't want to show "value__" because it's automatically added
// by the compiler.
if (fd.Name != "value__")
{
tr = new FieldTreeNode(fd, Access.Public, true);
enode.Nodes.Add(tr);
}
}
}
#endregion
#region LoadStruct
private void LoadStruct(TreeNode parentNode, Type td)
{
TreeNode node;
if (td.IsPublic)
{
node = new ClassTreeNode(td, ClassType.Struct, Access.Public);
parentNode.Nodes.Add(node);
}
else if (td.IsNestedPrivate)
{
node = new ClassTreeNode(td, ClassType.Struct, Access.Private);
parentNode.Nodes.Add(node);
}
else if (td.IsNestedFamORAssem)
{
node = new ClassTreeNode(td, ClassType.Struct, Access.Internal);
parentNode.Nodes.Add(node);
}
else
{
// It must be protected.
node = new ClassTreeNode(td, ClassType.Struct, Access.Protected);
parentNode.Nodes.Add(node);
}
foreach (Type ntd in td.GetNestedTypes())
{
LoadType(node, ntd);
}
foreach (MethodInfo md in td.GetMethods())
{
LoadMethod(node, md);
}
foreach (Type ntd in td.GetInterfaces())
{
LoadImplementedInterface(node, ntd);
}
foreach (EventInfo ed in td.GetEvents())
{
LoadEvent(node, ed);
}
foreach (PropertyInfo pd in td.GetProperties())
{
LoadProperty(node, pd);
}
foreach (FieldInfo fd in td.GetFields())
{
LoadField(node, fd);
}
}
#endregion
#region LoadField
private void LoadField(TreeNode parentNode, FieldInfo fd)
{
TreeNode tr;
if (fd.IsLiteral)
{
tr = new FieldTreeNode(fd, Access.Public, true);
parentNode.Nodes.Add(tr);
}
else if (fd.IsPublic)
{
tr = new FieldTreeNode(fd, Access.Public, false);
parentNode.Nodes.Add(tr);
}
else if (fd.IsPrivate)
{
tr = new FieldTreeNode(fd, Access.Private, false);
parentNode.Nodes.Add(tr);
}
else if (fd.IsFamilyOrAssembly)
{
tr = new FieldTreeNode(fd, Access.Internal, false);
parentNode.Nodes.Add(tr);
}
else
{
// It must be protected.
tr = new FieldTreeNode(fd, Access.Protected, false);
parentNode.Nodes.Add(tr);
}
}
#endregion
private void OpenDll(string loc)
{
Assembly asmb = null;
try
{
asmb = Assembly.LoadFrom(loc);
}
catch { }
AddAssemblyToView(asmb);
}
private void openToolStripMenuItem_Click(object sender, EventArgs e)
{
if (openFileDialog1.ShowDialog() == System.Windows.Forms.DialogResult.OK)
{
foreach (string str in openFileDialog1.FileNames)
{
OpenDll(str);
}
TestRunner.RunTests();
}
}
private void treeView1_NodeMouseClick(object sender, TreeNodeMouseClickEventArgs e)
{
OTreeNode n = (OTreeNode)e.Node;
n.ShowNodeInfo(Rtb);
bool hErr = false;
bool hWarn = false;
int ErrSelStart = 0;
int WarnSelStart = 0;
if (n.Errors.Count > 0)
{
ErrSelStart = Rtb.Text.Length - 1;
foreach (Errors.BaseError er in n.Errors)
{
Rtb.Text += er.Name + ": " + er.Description + "\r\n";
}
hErr = true;
}
if (n.Warnings.Count > 0)
{
WarnSelStart = Rtb.Text.Length - 1;
foreach (Warnings.BaseWarning b in n.Warnings)
{
Rtb.Text += b.Name + ": " + b.Description + "\r\n";
}
hWarn = true;
}
if (hErr)
{
Rtb.SelectionStart = ErrSelStart;
Rtb.SelectionLength = Rtb.Text.Length - Rtb.SelectionStart;
Rtb.SelectionColor = System.Drawing.Color.FromArgb(0xff, 0x00, 0x00);
Rtb.SelectionLength = 0;
}
if (hWarn)
{
Rtb.SelectionStart = WarnSelStart;
Rtb.SelectionLength = Rtb.Text.Length - Rtb.SelectionStart;
Rtb.SelectionColor = System.Drawing.Color.FromArgb(0xff, 0x99, 0x00);
Rtb.SelectionLength = 0;
}
}
}
}