private void BindUsages(ICollection <MethodInvocation> usages, IDictionary <IMethod, ITypeParameter> map, IProgressIndicator pi) { if (usages.Count == 0) { return; } pi.Start(usages.Count); foreach (MethodInvocation usage in usages) { if (usage.IsValid()) { ITypeParameter typeParameter; if (map.TryGetValue(usage.Method, out typeParameter)) { // one more call to language specific code... Exec[usage.Reference.GetTreeNode().Language].BindUsage(usage, typeParameter); } else { Exec[usage.Reference.GetTreeNode().Language].BindUsage(usage, null); } } pi.Advance(1); } }
public override void Rename(IRenameRefactoring executer, IProgressIndicator pi, bool hasConflictsWithDeclarations, IRefactoringDriver driver) { // Rename the "declaration" var declaredElement = myPointer.FindDeclaredElement(); var originalTreeNode = declaredElement?.GetTreeNode(); if (originalTreeNode == null) { return; } var originalRange = originalTreeNode.GetDocumentRange(); var factory = JavaScriptElementFactory.GetInstance(originalTreeNode); var literalExpression = factory.CreateExpression("\"$0\"", NewName); var newExpression = originalTreeNode.ReplaceBy(literalExpression); RemoveFromTextualOccurrences(executer, originalRange); var references = executer.Workflow.GetElementReferences(declaredElement); if (!Enumerable.Any(references)) { return; } pi.Start(references.Count); // Create a new declared element (other implementations don't appear to cache this, either) var element = new AsmDefNameDeclaredElement(declaredElement.GetJsServices(), NewName, declaredElement.SourceFile, newExpression.GetTreeStartOffset().Offset); // Rename/bind the references foreach (var pair in LanguageUtil.SortReferences(references.Where(r => r.IsValid()))) { foreach (var sortedReference in LanguageUtil.GetSortedReferences(pair.Value)) { InterruptableActivityCookie.CheckAndThrow(pi); var referenceRange = sortedReference.GetDocumentRange(); if (sortedReference.IsValid()) { sortedReference.BindTo(element); } RemoveFromTextualOccurrences(executer, referenceRange); pi.Advance(); } } element.GetPsiServices().Caches.Update(); myNewPointer = element.CreateElementPointer(); }
private FileSystemPath Deobfuscate(IObfuscatedFile obfuscatedFile, IProgressIndicator progressIndicator) { try { progressIndicator.Start(5); progressIndicator.TaskName = "Deobfuscating..."; progressIndicator.CurrentItemText = string.Format("Saving to {0}", obfuscatedFile.NewFilename); progressIndicator.Advance(1); obfuscatedFile.deobfuscateBegin(); progressIndicator.Advance(1); obfuscatedFile.deobfuscate(); progressIndicator.Advance(1); obfuscatedFile.deobfuscateEnd(); // turn all flags on RenamerFlags flags = RenamerFlags.DontCreateNewParamDefs | RenamerFlags.DontRenameDelegateFields | RenamerFlags.RenameEvents | RenamerFlags.RenameFields | RenamerFlags.RenameGenericParams | RenamerFlags.RenameMethodArgs | RenamerFlags.RenameMethods | RenamerFlags.RenameNamespaces | RenamerFlags.RenameProperties | RenamerFlags.RenameTypes | RenamerFlags.RestoreEvents | RenamerFlags.RestoreEventsFromNames | RenamerFlags.RestoreProperties | RenamerFlags.RestorePropertiesFromNames; var renamer = new Renamer(obfuscatedFile.DeobfuscatorContext, new[] { obfuscatedFile }, flags); progressIndicator.Advance(1); renamer.rename(); progressIndicator.Advance(1); obfuscatedFile.save(); } finally { progressIndicator.Stop(); } return new FileSystemPath(obfuscatedFile.NewFilename); }
private FileSystemPath Deobfuscate(IObfuscatedFile obfuscatedFile, IProgressIndicator progressIndicator) { try { progressIndicator.Start(5); progressIndicator.TaskName = "Deobfuscating..."; progressIndicator.CurrentItemText = string.Format("Saving to {0}", obfuscatedFile.NewFilename); progressIndicator.Advance(1); obfuscatedFile.deobfuscateBegin(); progressIndicator.Advance(1); obfuscatedFile.deobfuscate(); progressIndicator.Advance(1); obfuscatedFile.deobfuscateEnd(); // turn all flags on RenamerFlags flags = RenamerFlags.DontCreateNewParamDefs | RenamerFlags.DontRenameDelegateFields | RenamerFlags.RenameEvents | RenamerFlags.RenameFields | RenamerFlags.RenameGenericParams | RenamerFlags.RenameMethodArgs | RenamerFlags.RenameMethods | RenamerFlags.RenameNamespaces | RenamerFlags.RenameProperties | RenamerFlags.RenameTypes | RenamerFlags.RestoreEvents | RenamerFlags.RestoreEventsFromNames | RenamerFlags.RestoreProperties | RenamerFlags.RestorePropertiesFromNames; var renamer = new Renamer(obfuscatedFile.DeobfuscatorContext, new[] { obfuscatedFile }, flags); progressIndicator.Advance(1); renamer.rename(); progressIndicator.Advance(1); obfuscatedFile.save(); } finally { progressIndicator.Stop(); } return(new FileSystemPath(obfuscatedFile.NewFilename)); }
private IEnumerable <MethodInvocation> PreProcessMethodUsages(ICollection <IReference> references, IProgressIndicator pi) { if (references.Count == 0) { yield break; } pi.Start(references.Count); foreach (IReference reference in references) { MethodInvocation usage = Exec[reference.GetTreeNode().Language].ProcessUsage(reference); if (usage != null) { yield return(usage); } pi.Advance(1); } }
private void ProcessParameterUsages(ICollection <IReference> references, IProgressIndicator pi) { // can not start progress indicator with zero count. if (references.Count == 0) { return; } pi.Start(references.Count); foreach (IReference reference in references) { // reference can be invalid if previous changes affected it's element. It is unlikely to be occured... if (reference.IsValid()) { // process reference with language specific implementation... Exec[reference.GetTreeNode().Language].ProcessParameterReference(reference); } pi.Advance(1); } }
public override ICollection<IOccurence> Search(IProgressIndicator progressIndicator) { // Create and configure utility class, which will perform search var searcher = new StringSearcher(mySearchString, myCaseSensitive); // Fetch all projects for solution, start progress ICollection<IProject> projects = Solution.GetAllProjects(); progressIndicator.Start(projects.Count); var items = new List<IOccurence>(); // visit each project and collect occurences var visitor = new ProjectTextSearcher(searcher, items, mySearchFlags, myDocumentManager, mySolution); foreach (IProject project in projects) { progressIndicator.CurrentItemText = string.Format("Scanning project '{0}'", project.Name); project.Accept(visitor); progressIndicator.Advance(1); } return items; }
public override ICollection <IOccurence> Search(IProgressIndicator progressIndicator) { // Create and configure utility class, which will perform search var searcher = new StringSearcher(mySearchString, myCaseSensitive); // Fetch all projects for solution, start progress ICollection <IProject> projects = Solution.GetAllProjects(); progressIndicator.Start(projects.Count); var items = new List <IOccurence>(); // visit each project and collect occurences var visitor = new ProjectTextSearcher(searcher, items, mySearchFlags, myDocumentManager, mySolution); foreach (IProject project in projects) { progressIndicator.CurrentItemText = string.Format("Scanning project '{0}'", project.Name); project.Accept(visitor); progressIndicator.Advance(1); } return(items); }
public override void Rename(RenameRefactoring executer, IProgressIndicator pi, bool hasConflictsWithDeclarations, IRefactoringDriver driver) { BuildDeclarations(); //Logger.Assert(myDeclarations.Count > 0, "myDeclarations.Count > 0"); IDeclaredElement declaredElement = myOriginalElementPointer.FindDeclaredElement(); if (declaredElement == null) { return; } IPsiServices psiServices = declaredElement.GetPsiServices(); IList<IReference> primaryReferences = executer.Workflow.GetElementReferences(PrimaryDeclaredElement); List<Pair<IDeclaredElement, IList<IReference>>> secondaryElementWithReferences = SecondaryDeclaredElements.Select(x => Pair.Of(x, executer.Workflow.GetElementReferences(x))).ToList(); pi.Start(myDeclarations.Count + primaryReferences.Count); foreach (IDeclaration declaration in myDeclarations) { InterruptableActivityCookie.CheckAndThrow(pi); declaration.SetName(myNewName); pi.Advance(); } psiServices.PsiManager.UpdateCaches(); IDeclaredElement newDeclaredElement = null; if (myDeclarations.Count > 0) { newDeclaredElement = myDeclarations[0].DeclaredElement; } Assertion.Assert(newDeclaredElement != null, "The condition (newDeclaredElement != null) is false."); myNewElementPointer = newDeclaredElement.CreateElementPointer(); Assertion.Assert(newDeclaredElement.IsValid(), "myNewDeclaredElement.IsValid()"); myNewReferences.Clear(); OneToSetMap<PsiLanguageType, IReference> references = LanguageUtil.SortReferences(primaryReferences.Where(x => x.IsValid())); IList<IReference> referencesToRename = new List<IReference>(); foreach (var pair in references) { List<IReference> sortedReferences = LanguageUtil.GetSortedReferences(pair.Value); foreach (IReference reference in sortedReferences) { IReference oldReferenceForConflict = reference; InterruptableActivityCookie.CheckAndThrow(pi); if (reference.IsValid()) // reference may invalidate during additional reference processing { RenameHelperBase rename = executer.Workflow.LanguageSpecific[reference.GetTreeNode().Language]; IReference reference1 = rename.TransformProjectedInitializer(reference); DeclaredElementInstance subst = GetSubst(newDeclaredElement, executer); IReference newReference; if (subst != null) { if (subst.Substitution.Domain.IsEmpty()) { newReference = reference1.BindTo(subst.Element); } else { newReference = reference1.BindTo(subst.Element, subst.Substitution); } } else { newReference = reference1.BindTo(newDeclaredElement); } if (!(newReference is IImplicitReference)) { IDeclaredElement element = newReference.Resolve().DeclaredElement; if (!hasConflictsWithDeclarations && !myDoNotShowBindingConflicts && (element == null || !element.Equals(newDeclaredElement)) && !rename.IsAlias(newDeclaredElement)) { driver.AddLateConflict(() => new Conflict(newReference.GetTreeNode().GetSolution(), "Usage {0} can not be updated correctly.", ConflictSeverity.Error, oldReferenceForConflict), "not bound"); } referencesToRename.Add(newReference); } myNewReferences.Insert(0, newReference); rename.AdditionalReferenceProcessing(newDeclaredElement, newReference, myNewReferences); } pi.Advance(); } } foreach (var pair in secondaryElementWithReferences) { IDeclaredElement element = pair.First; IList<IReference> secondaryReferences = pair.Second; foreach (IReference reference in secondaryReferences) { InterruptableActivityCookie.CheckAndThrow(pi); if (reference.IsValid()) { reference.BindTo(element); } } } }
public override void Rename(IRenameRefactoring executer, IProgressIndicator pi, bool hasConflictsWithDeclarations, IRefactoringDriver driver) { BuildDeclarations(); //Logger.Assert(myDeclarations.Count > 0, "myDeclarations.Count > 0"); IDeclaredElement declaredElement = myOriginalElementPointer.FindDeclaredElement(); if (declaredElement == null) { return; } IPsiServices psiServices = declaredElement.GetPsiServices(); IList <IReference> primaryReferences = executer.Workflow.GetElementReferences(PrimaryDeclaredElement); List <Pair <IDeclaredElement, IList <IReference> > > secondaryElementWithReferences = SecondaryDeclaredElements.Select(x => Pair.Of(x, executer.Workflow.GetElementReferences(x))).ToList(); pi.Start(myDeclarations.Count + primaryReferences.Count); foreach (IDeclaration declaration in myDeclarations) { InterruptableActivityCookie.CheckAndThrow(pi); declaration.SetName(myNewName); pi.Advance(); } psiServices.Caches.Update(); IDeclaredElement newDeclaredElement; if (myDeclarations.Count > 0) { newDeclaredElement = myDeclarations[0].DeclaredElement; } else { if (myElement is RoleDeclaredElement) { newDeclaredElement = myElement; ((RoleDeclaredElement)myElement).ChangeName = true; ((RoleDeclaredElement)myElement).NewName = NewName; ((RoleDeclaredElement)newDeclaredElement).SetName(NewName); } else { newDeclaredElement = null; } } Assertion.Assert(newDeclaredElement != null, "The condition (newDeclaredElement != null) is false."); myNewElementPointer = newDeclaredElement.CreateElementPointer(); Assertion.Assert(newDeclaredElement.IsValid(), "myNewDeclaredElement.IsValid()"); myNewReferences.Clear(); OneToSetMap <PsiLanguageType, IReference> references = LanguageUtil.SortReferences(primaryReferences.Where(x => x.IsValid())); IList <IReference> referencesToRename = new List <IReference>(); foreach (var pair in references) { List <IReference> sortedReferences = pair.Value.ToList();//LanguageUtil.GetSortedReferences(pair.Value); foreach (IReference reference in sortedReferences) { IReference oldReferenceForConflict = reference; InterruptableActivityCookie.CheckAndThrow(pi); if (reference.IsValid()) // reference may invalidate during additional reference processing { RenameHelperBase rename = executer.Workflow.LanguageSpecific[reference.GetTreeNode().Language]; IReference reference1 = rename.TransformProjectedInitializer(reference); DeclaredElementInstance subst = GetSubst(newDeclaredElement, executer); IReference newReference; if (subst != null) { if (subst.Substitution.Domain.IsEmpty()) { newReference = reference1.BindTo(subst.Element); } else { newReference = reference1.BindTo(subst.Element, subst.Substitution); } } else { newReference = reference1.BindTo(newDeclaredElement); } if (!(newReference is IImplicitReference)) { IDeclaredElement element = newReference.Resolve().DeclaredElement; if (!hasConflictsWithDeclarations && !myDoNotShowBindingConflicts && (element == null || !element.Equals(newDeclaredElement)) && !rename.IsAlias(newDeclaredElement)) { driver.AddLateConflict(() => new Conflict(newReference.GetTreeNode().GetSolution(), "Usage {0} can not be updated correctly.", ConflictSeverity.Error, oldReferenceForConflict), "not bound"); } referencesToRename.Add(newReference); } myNewReferences.Insert(0, newReference); rename.AdditionalReferenceProcessing(newDeclaredElement, newReference, myNewReferences); } pi.Advance(); } } foreach (var pair in secondaryElementWithReferences) { IDeclaredElement element = pair.First; IList <IReference> secondaryReferences = pair.Second; foreach (IReference reference in secondaryReferences) { InterruptableActivityCookie.CheckAndThrow(pi); if (reference.IsValid()) { reference.BindTo(element); } } } if (myElement is RoleDeclaredElement) { ((RoleDeclaredElement)myElement).ChangeName = false; ((RoleDeclaredElement)myElement).SetName(NewName); foreach (IReference reference in referencesToRename) { ((PsiRoleReference)reference).SetName(NewName); reference.CurrentResolveResult = null; ((PsiFile)((RoleDeclaredElement)myElement).File).ClearTables(); } } }