From bcbc17e45828c0628feae0ccfc589cb9c8e63f9e Mon Sep 17 00:00:00 2001
From: Albert Meltzer <7529386+kitbellew@users.noreply.github.com>
Date: Wed, 27 Mar 2024 06:36:38 -0700
Subject: [PATCH] FormatTokensRewrite: token index in Replacement

---
 .../org/scalafmt/rewrite/FormatTokensRewrite.scala     | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/scalafmt-core/shared/src/main/scala/org/scalafmt/rewrite/FormatTokensRewrite.scala b/scalafmt-core/shared/src/main/scala/org/scalafmt/rewrite/FormatTokensRewrite.scala
index b6eb481dc7..dbf030a043 100644
--- a/scalafmt-core/shared/src/main/scala/org/scalafmt/rewrite/FormatTokensRewrite.scala
+++ b/scalafmt-core/shared/src/main/scala/org/scalafmt/rewrite/FormatTokensRewrite.scala
@@ -304,7 +304,7 @@ object FormatTokensRewrite {
       claimed.get(ftIdx) match {
         case Some(x) =>
           val repl = tokens(x)
-          val ok = (repl eq null) || repl.ft.meta.idx == ftIdx
+          val ok = (repl eq null) || repl.idx == ftIdx
           if (ok) Some((x, repl)) else None
         case _ => None
       }
@@ -330,9 +330,8 @@ object FormatTokensRewrite {
                 val ok = orepl != null && (orepl.rule eq repl.rule) &&
                   orepl.isRemove
                 if (ok) {
-                  tokens(oldidx) =
-                    repl.copy(ft = repl.ft.withIdx(orepl.ft.meta.idx))
-                  Some(repl.copy(ft = orepl.ft.withIdx(repl.ft.meta.idx)))
+                  tokens(oldidx) = repl.copy(ft = repl.ft.withIdx(orepl.idx))
+                  Some(repl.copy(ft = orepl.ft.withIdx(repl.idx)))
                 } else None
               }
             case _ => None
@@ -376,7 +375,7 @@ object FormatTokensRewrite {
 
     private[rewrite] def isRemovedOnLeftOpt(x: FormatToken): Option[Boolean] = {
       val ftIdx = x.meta.idx - 1
-      claimedRule(ftIdx).filter(_.ft.meta.idx == ftIdx).map(_.isRemove)
+      claimedRule(ftIdx).filter(_.idx == ftIdx).map(_.isRemove)
     }
 
     private[rewrite] def isRemovedOnLeft(x: FormatToken, ok: Boolean): Boolean =
@@ -393,6 +392,7 @@ object FormatTokensRewrite {
       claim: Iterable[Int] = Nil
   ) {
     @inline def isRemove: Boolean = how eq ReplacementType.Remove
+    @inline def idx: Int = ft.meta.idx
   }
 
   private[rewrite] sealed trait ReplacementType