From e5ea2a2af88667efd99a64ec4188f4e6d1077113 Mon Sep 17 00:00:00 2001
From: Tomasz Godzik <tgodzik@virtuslab.com>
Date: Wed, 9 Apr 2025 20:21:41 +0200
Subject: [PATCH] improvement: Support using directives in worksheets

---
 .../tools/pc/completions/ScalaCliCompletions.scala   | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/presentation-compiler/src/main/dotty/tools/pc/completions/ScalaCliCompletions.scala b/presentation-compiler/src/main/dotty/tools/pc/completions/ScalaCliCompletions.scala
index e2a0a033ee6b..8df727b14155 100644
--- a/presentation-compiler/src/main/dotty/tools/pc/completions/ScalaCliCompletions.scala
+++ b/presentation-compiler/src/main/dotty/tools/pc/completions/ScalaCliCompletions.scala
@@ -13,14 +13,20 @@ class ScalaCliCompletions(
 ):
   def unapply(path: List[Tree]) =
     def scalaCliDep = CoursierComplete.isScalaCliDep(
-      pos.lineContent.take(pos.column).stripPrefix("/*<script>*/")
+      pos.lineContent.take(pos.column).stripPrefix("/*<script>*/").dropWhile(c => c == ' ' || c == '\t')
     )
+
+    lazy val supportsUsing =
+      val filename = pos.source.file.path
+      filename.endsWith(".sc.scala") ||
+      filename.endsWith(".worksheet.sc")
+
     path match
       case Nil | (_: PackageDef) :: _ => scalaCliDep
       // generated script file will end with .sc.scala
-      case (_: TypeDef) :: (_: PackageDef) :: Nil if pos.source.file.path.endsWith(".sc.scala") =>
+      case (_: TypeDef) :: (_: PackageDef) :: Nil if supportsUsing =>
         scalaCliDep
-      case (_: Template) :: (_: TypeDef) :: Nil if pos.source.file.path.endsWith(".sc.scala") =>
+      case (_: Template) :: (_: TypeDef) :: Nil if supportsUsing =>
         scalaCliDep
       case head :: next => None