@@ -45,7 +45,7 @@ class ShortenedTypePrinter(
4545 isTextEdit : Boolean = false ,
4646 renameConfigMap : Map [Symbol , String ] = Map .empty
4747)(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
4949 private val defaultWidth = 1000
5050
5151 private val methodFlags =
@@ -81,7 +81,8 @@ class ShortenedTypePrinter(
8181 * Returns a list of TextEdits (auto-imports) of the symbols
8282 */
8383 def imports (autoImportsGen : AutoImportsGenerator ): List [TextEdit ] =
84- missingImports.toList
84+ missingImports
85+ .toList
8586 .filterNot(selector => selector.sym.isRoot)
8687 .sortBy(_.sym.effectiveName)
8788 .flatMap(selector => autoImportsGen.renderImports(List (selector)))
@@ -112,7 +113,9 @@ class ShortenedTypePrinter(
112113 def ownersAfterRename (owner : Symbol ): List [Symbol ] =
113114 prefix.ownersIterator.takeWhile(_ != owner).toList
114115
115- prefix.ownersIterator.flatMap { owner =>
116+ val prefixIterator = if isTextEdit then prefix.ownersIterator else Iterator (prefix)
117+
118+ prefixIterator.flatMap { owner =>
116119 val prefixAfterRename = ownersAfterRename(owner)
117120 val currentRenamesSearchResult =
118121 indexedCtx.rename(owner).map(Found (owner, _, prefixAfterRename))
0 commit comments