From 860a8285f577368ae326b6d639b440fe35c3b4d9 Mon Sep 17 00:00:00 2001
From: Albert Meltzer <7529386+kitbellew@users.noreply.github.com>
Date: Mon, 13 Mar 2023 15:50:20 +0100
Subject: [PATCH] Router: `extension` is probably also a keyword

---
 .../shared/src/main/scala/org/scalafmt/internal/Router.scala  | 2 +-
 scalafmt-tests/src/test/resources/scala3/Extension.stat       | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/scalafmt-core/shared/src/main/scala/org/scalafmt/internal/Router.scala b/scalafmt-core/shared/src/main/scala/org/scalafmt/internal/Router.scala
index 0beba3000b..583cfe76ca 100644
--- a/scalafmt-core/shared/src/main/scala/org/scalafmt/internal/Router.scala
+++ b/scalafmt-core/shared/src/main/scala/org/scalafmt/internal/Router.scala
@@ -680,7 +680,7 @@ class Router(formatOps: FormatOps) {
                 isSymbolicName(name.value) && name.parent.exists(isDefDef) =>
             Space
           case _: Defn.ExtensionGroup if left.is[soft.KwExtension] =>
-            Space
+            Space(style.spaces.afterKeywordBeforeParen)
           case _ => Space(left.is[T.Comment])
         }
         def baseNoSplit(implicit fileLine: FileLine) = Split(modification, 0)
diff --git a/scalafmt-tests/src/test/resources/scala3/Extension.stat b/scalafmt-tests/src/test/resources/scala3/Extension.stat
index 39c1578d14..33718ef79c 100644
--- a/scalafmt-tests/src/test/resources/scala3/Extension.stat
+++ b/scalafmt-tests/src/test/resources/scala3/Extension.stat
@@ -300,7 +300,7 @@ spaces.afterKeywordBeforeParen = false
 extension (s: String)
   def foo(): Unit = ???
 >>>
-extension (s: String) def foo(): Unit = ???
+extension(s: String) def foo(): Unit = ???
 <<< #3499 4
 spaces.afterKeywordBeforeParen = false
 newlines.beforeOpenParenDefnSite = unfold
@@ -308,4 +308,4 @@ newlines.beforeOpenParenDefnSite = unfold
 extension (s: String)
   def foo(): Unit = ???
 >>>
-extension (s: String) def foo(): Unit = ???
+extension(s: String) def foo(): Unit = ???