@@ -45,7 +45,7 @@ class ShortenedTypePrinter(
45
45
isTextEdit : Boolean = false ,
46
46
renameConfigMap : Map [Symbol , String ] = Map .empty
47
47
)(using indexedCtx : IndexedContext , reportCtx : ReportContext ) extends RefinedPrinter (indexedCtx.ctx):
48
- private val missingImports : mutable.ListBuffer [ImportSel ] = mutable.ListBuffer .empty
48
+ private val missingImports : mutable.Set [ImportSel ] = mutable.Set .empty
49
49
private val defaultWidth = 1000
50
50
51
51
private val methodFlags =
@@ -81,7 +81,8 @@ class ShortenedTypePrinter(
81
81
* Returns a list of TextEdits (auto-imports) of the symbols
82
82
*/
83
83
def imports (autoImportsGen : AutoImportsGenerator ): List [TextEdit ] =
84
- missingImports.toList
84
+ missingImports
85
+ .toList
85
86
.filterNot(selector => selector.sym.isRoot)
86
87
.sortBy(_.sym.effectiveName)
87
88
.flatMap(selector => autoImportsGen.renderImports(List (selector)))
@@ -112,7 +113,9 @@ class ShortenedTypePrinter(
112
113
def ownersAfterRename (owner : Symbol ): List [Symbol ] =
113
114
prefix.ownersIterator.takeWhile(_ != owner).toList
114
115
115
- prefix.ownersIterator.flatMap { owner =>
116
+ val prefixIterator = if isTextEdit then prefix.ownersIterator else Iterator (prefix)
117
+
118
+ prefixIterator.flatMap { owner =>
116
119
val prefixAfterRename = ownersAfterRename(owner)
117
120
val currentRenamesSearchResult =
118
121
indexedCtx.rename(owner).map(Found (owner, _, prefixAfterRename))
0 commit comments