155 lines
5.3 KiB
C#
155 lines
5.3 KiB
C#
/*
|
|
Copyright (C) 2011-2012 de4dot@gmail.com
|
|
|
|
This file is part of de4dot.
|
|
|
|
de4dot 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.
|
|
|
|
de4dot 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 de4dot. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
using Mono.Cecil;
|
|
|
|
namespace de4dot.code.renamer {
|
|
class VariableNameState {
|
|
ExistingNames existingVariableNames;
|
|
ExistingNames existingMethodNames;
|
|
ExistingNames existingPropertyNames;
|
|
ExistingNames existingEventNames;
|
|
TypeNames variableNameCreator; // For fields and method args
|
|
TypeNames propertyNameCreator;
|
|
NameCreator eventNameCreator;
|
|
NameCreator genericPropertyNameCreator;
|
|
public NameCreator staticMethodNameCreator;
|
|
public NameCreator instanceMethodNameCreator;
|
|
|
|
public static VariableNameState create() {
|
|
var vns = new VariableNameState();
|
|
vns.existingVariableNames = new ExistingNames();
|
|
vns.existingMethodNames = new ExistingNames();
|
|
vns.existingPropertyNames = new ExistingNames();
|
|
vns.existingEventNames = new ExistingNames();
|
|
vns.variableNameCreator = new VariableNameCreator();
|
|
vns.propertyNameCreator = new PropertyNameCreator();
|
|
vns.eventNameCreator = new NameCreator("Event_");
|
|
vns.genericPropertyNameCreator = new NameCreator("Prop_");
|
|
vns.staticMethodNameCreator = new NameCreator("smethod_");
|
|
vns.instanceMethodNameCreator = new NameCreator("method_");
|
|
return vns;
|
|
}
|
|
|
|
VariableNameState() {
|
|
}
|
|
|
|
// Cloning only params will speed up the method param renaming code
|
|
public VariableNameState cloneParamsOnly() {
|
|
var vns = new VariableNameState();
|
|
vns.existingVariableNames = new ExistingNames();
|
|
vns.variableNameCreator = new VariableNameCreator();
|
|
vns.existingVariableNames.merge(existingVariableNames);
|
|
vns.variableNameCreator.merge(variableNameCreator);
|
|
return vns;
|
|
}
|
|
|
|
public VariableNameState merge(VariableNameState other) {
|
|
existingVariableNames.merge(other.existingVariableNames);
|
|
existingMethodNames.merge(other.existingMethodNames);
|
|
existingPropertyNames.merge(other.existingPropertyNames);
|
|
existingEventNames.merge(other.existingEventNames);
|
|
variableNameCreator.merge(other.variableNameCreator);
|
|
propertyNameCreator.merge(other.propertyNameCreator);
|
|
eventNameCreator.merge(other.eventNameCreator);
|
|
genericPropertyNameCreator.merge(other.genericPropertyNameCreator);
|
|
staticMethodNameCreator.merge(other.staticMethodNameCreator);
|
|
instanceMethodNameCreator.merge(other.instanceMethodNameCreator);
|
|
return this;
|
|
}
|
|
|
|
public void mergeMethods(VariableNameState other) {
|
|
existingMethodNames.merge(other.existingMethodNames);
|
|
}
|
|
|
|
public void mergeProperties(VariableNameState other) {
|
|
existingPropertyNames.merge(other.existingPropertyNames);
|
|
}
|
|
|
|
public void mergeEvents(VariableNameState other) {
|
|
existingEventNames.merge(other.existingEventNames);
|
|
}
|
|
|
|
public string getNewPropertyName(PropertyDefinition propertyDefinition) {
|
|
var propType = propertyDefinition.PropertyType;
|
|
string newName;
|
|
if (propType is GenericParameter)
|
|
newName = existingPropertyNames.getName(propertyDefinition.Name, genericPropertyNameCreator);
|
|
else
|
|
newName = existingPropertyNames.getName(propertyDefinition.Name, () => propertyNameCreator.create(propType));
|
|
addPropertyName(newName);
|
|
return newName;
|
|
}
|
|
|
|
public string getNewEventName(EventDefinition eventDefinition) {
|
|
string newName = eventNameCreator.create();
|
|
addEventName(newName);
|
|
return newName;
|
|
}
|
|
|
|
public void addFieldName(string fieldName) {
|
|
existingVariableNames.add(fieldName);
|
|
}
|
|
|
|
public void addParamName(string paramName) {
|
|
existingVariableNames.add(paramName);
|
|
}
|
|
|
|
public void addMethodName(string methodName) {
|
|
existingMethodNames.add(methodName);
|
|
}
|
|
|
|
public void addPropertyName(string propName) {
|
|
existingPropertyNames.add(propName);
|
|
}
|
|
|
|
public void addEventName(string eventName) {
|
|
existingEventNames.add(eventName);
|
|
}
|
|
|
|
public bool isMethodNameUsed(string methodName) {
|
|
return existingMethodNames.exists(methodName);
|
|
}
|
|
|
|
public bool isPropertyNameUsed(string propName) {
|
|
return existingPropertyNames.exists(propName);
|
|
}
|
|
|
|
public bool isEventNameUsed(string eventName) {
|
|
return existingEventNames.exists(eventName);
|
|
}
|
|
|
|
public string getNewFieldName(FieldDefinition field) {
|
|
return existingVariableNames.getName(field.Name, () => variableNameCreator.create(field.FieldType));
|
|
}
|
|
|
|
public string getNewFieldName(string oldName, INameCreator nameCreator) {
|
|
return existingVariableNames.getName(oldName, () => nameCreator.create());
|
|
}
|
|
|
|
public string getNewParamName(string oldName, ParameterDefinition param) {
|
|
return existingVariableNames.getName(oldName, () => variableNameCreator.create(param.ParameterType));
|
|
}
|
|
|
|
public string getNewMethodName(string oldName, INameCreator nameCreator) {
|
|
return existingMethodNames.getName(oldName, nameCreator);
|
|
}
|
|
}
|
|
}
|