2021-09-20 18:20:01 +02:00

55 lines
1.7 KiB
C#

/*
Copyright (C) 2014-2019 de4dot@gmail.com
This file is part of dnSpy
dnSpy is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
dnSpy is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with dnSpy. If not, see <http://www.gnu.org/licenses/>.
*/
using System;
using System.Collections.Generic;
using System.Diagnostics;
using dnlib.DotNet;
using dnSpy.Contracts.Documents.TreeView;
namespace dnSpy.AsmEditor.Commands {
sealed class FieldNodeCreator {
public IEnumerable<DocumentTreeNodeData> OriginalNodes {
get { yield return ownerNode; }
}
readonly TypeNode ownerNode;
readonly FieldNode fieldNode;
public FieldNodeCreator(ModuleDocumentNode modNode, TypeNode ownerNode, FieldDef field) {
this.ownerNode = ownerNode;
fieldNode = modNode.Context.DocumentTreeView.Create(field);
}
public void Add() {
ownerNode.TreeNode.EnsureChildrenLoaded();
ownerNode.TypeDef.Fields.Add(fieldNode.FieldDef);
ownerNode.TreeNode.AddChild(fieldNode.TreeNode);
}
public void Remove() {
bool b = ownerNode.TreeNode.Children.Remove(fieldNode.TreeNode) &&
ownerNode.TypeDef.Fields.Remove(fieldNode.FieldDef);
Debug.Assert(b);
if (!b)
throw new InvalidOperationException();
}
}
}