refactor: Tweaks after reviews

scalameta · Dec 15, 2023 · e643cf5 · e643cf5
1 parent 4ed074b
commit e643cf5
Show file tree

Hide file tree

Showing 5 changed files with 22 additions and 18 deletions.
diff --git a/mtags/src/main/scala-3/scala/meta/internal/pc/CompilerSearchVisitor.scala b/mtags/src/main/scala-3/scala/meta/internal/pc/CompilerSearchVisitor.scala
@@ -21,13 +21,15 @@ class CompilerSearchVisitor(
 
   val logger: Logger = Logger.getLogger(classOf[CompilerSearchVisitor].getName)
 
+  private def isAccessibleImplicitClass(sym: Symbol) =
+    val owner = sym.maybeOwner
+    owner != NoSymbol && owner.isClass &&
+    owner.is(Flags.Implicit) &&
+    owner.isStatic && owner.isPublic
+
   private def isAccessible(sym: Symbol): Boolean = try
-    sym != NoSymbol && sym.isPublic && sym.isStatic || {
-      val owner = sym.maybeOwner
-      owner != NoSymbol && owner.isClass &&
-      owner.is(Flags.Implicit) &&
-      owner.isStatic && owner.isPublic
-    }
+    sym != NoSymbol && sym.isPublic && sym.isStatic ||
+      isAccessibleImplicitClass(sym)
   catch
     case err: AssertionError =>
       logger.log(Level.WARNING, err.getMessage())

diff --git a/mtags/src/main/scala-3/scala/meta/internal/pc/SemanticdbSymbols.scala b/mtags/src/main/scala-3/scala/meta/internal/pc/SemanticdbSymbols.scala
@@ -51,15 +51,11 @@ object SemanticdbSymbols:
                   /**
                    * Looks like decl doesn't work for:
                    *  package a:
-                   *   implicit class A (i: Int):
+                   *   implicit class <<A>> (i: Int):
                    *      def inc = i + 1
                    */
                   else if typeSym == NoSymbol then
-                    val searched = typeName(value)
-                    owner.info.allMembers
-                      .find(_.name == searched)
-                      .map(_.symbol)
-                      .toList
+                    owner.info.member(typeName(value)).symbol :: Nil
                   else typeSym :: Nil
                   end if
                 case Descriptor.Term(value) =>

diff --git a/mtags/src/main/scala-3/scala/meta/internal/pc/completions/Completions.scala b/mtags/src/main/scala-3/scala/meta/internal/pc/completions/Completions.scala
@@ -599,9 +599,14 @@ class Completions(
             qualType.widenDealias <:< sym.extensionParam.info.widenDealias
 
           def isImplicitClass(owner: Symbol) =
+
             val constructorParam =
-              owner.info.allMembers
-                .find(_.symbol.isAllOf(Flags.PrivateParamAccessor))
+              owner.info
+                .membersBasedOnFlags(
+                  Flags.ParamAccessor,
+                  Flags.EmptyFlags,
+                )
+                .headOption
                 .map(_.info)
             owner.isClass && owner.is(Flags.Implicit) &&
             constructorParam.exists(p =>

diff --git a/mtags/src/main/scala/scala/meta/internal/mtags/ScalaToplevelMtags.scala b/mtags/src/main/scala/scala/meta/internal/mtags/ScalaToplevelMtags.scala
@@ -222,7 +222,8 @@ class ScalaToplevelMtags(
           loop(
             indent,
             isAfterNewline = false,
-            if (needsToGenerateFileClass) currRegion.withTermOwner(owner) else currRegion,
+            if (needsToGenerateFileClass) currRegion.withTermOwner(owner)
+            else currRegion,
             template
           )
         // also covers extension methods because of `def` inside

diff --git a/tests/cross/src/test/scala/tests/pc/CompletionExtensionMethodSuite.scala b/tests/cross/src/test/scala/tests/pc/CompletionExtensionMethodSuite.scala
@@ -125,7 +125,7 @@ class CompletionExtensionMethodSuite extends BaseCompletionSuite {
        |def main = "foo".iden@@
        |""".stripMargin,
     """|identity: String (implicit)
-       |""".stripMargin // incr won't be available
+       |""".stripMargin // identity2 won't be available
 
   )
 
@@ -266,7 +266,7 @@ class CompletionExtensionMethodSuite extends BaseCompletionSuite {
     """|package example
        |
        |object enrichments:
-       |  implicit class A (num: Int):
+       |  implicit class A (val num: Int):
        |    def plus(other: Int): Int = num + other
        |
        |def main = 100.pl@@
@@ -276,7 +276,7 @@ class CompletionExtensionMethodSuite extends BaseCompletionSuite {
        |import example.enrichments.A
        |
        |object enrichments:
-       |  implicit class A (num: Int):
+       |  implicit class A (val num: Int):
        |    def plus(other: Int): Int = num + other
        |
        |def main = 100.plus($0)