protoype for scala integration

Shashankss1205 · Shashankss1205 · commit 9d621bd6bb27 · 2026-01-03T19:18:10.000+05:30
diff --git a/src/codegraphcontext/tools/graph_builder.py b/src/codegraphcontext/tools/graph_builder.py
@@ -64,6 +64,9 @@ def __init__(self, language_name: str):
         elif self.language_name == 'kotlin':
             from .languages.kotlin import KotlinTreeSitterParser
             self.language_specific_parser = KotlinTreeSitterParser(self)
+        elif self.language_name == 'scala':
+            from .languages.scala import ScalaTreeSitterParser
+            self.language_specific_parser = ScalaTreeSitterParser(self)
 
 
 
@@ -105,6 +108,8 @@ def __init__(self, db_manager: DatabaseManager, job_manager: JobManager, loop: a
             '.cs': TreeSitterParser('c_sharp'),
             '.php': TreeSitterParser('php'),
             '.kt': TreeSitterParser('kotlin'),
+            '.scala': TreeSitterParser('scala'),
+            '.sc': TreeSitterParser('scala'),
         }
         self.create_schema()
 
@@ -213,6 +218,12 @@ def _pre_scan_for_imports(self, files: list[Path]) -> dict:
         if '.kt' in files_by_lang:
             from .languages import kotlin as kotlin_lang_module
             imports_map.update(kotlin_lang_module.pre_scan_kotlin(files_by_lang['.kt'], self.parsers['.kt']))
+        if '.scala' in files_by_lang:
+            from .languages import scala as scala_lang_module
+            imports_map.update(scala_lang_module.pre_scan_scala(files_by_lang['.scala'], self.parsers['.scala']))
+        if '.sc' in files_by_lang:
+            from .languages import scala as scala_lang_module
+            imports_map.update(scala_lang_module.pre_scan_scala(files_by_lang['.sc'], self.parsers['.sc']))
             
         return imports_map
 
diff --git a/src/codegraphcontext/tools/languages/scala.py b/src/codegraphcontext/tools/languages/scala.py
diff --git a/src/codegraphcontext/tools/query_tool_languages/scala_toolkit.py b/src/codegraphcontext/tools/query_tool_languages/scala_toolkit.py
@@ -0,0 +1,5 @@
+class ScalaToolkit:
+    """Template placeholder for future implementation."""
+
+    def get_cypher_query(query: str) -> str:
+        raise NotImplementedError("AdvancedLanguageQuery is not implemented yet.")
diff --git a/src/codegraphcontext/utils/tree_sitter_manager.py b/src/codegraphcontext/utils/tree_sitter_manager.py
@@ -46,6 +46,8 @@
     "rust": "rust",
     "kt": "kotlin",
     "kotlin": "kotlin",
+    "scala": "scala",
+    ".scala": "scala",
 }
 
 
diff --git a/tests/sample_project_scala/Animals.scala b/tests/sample_project_scala/Animals.scala
@@ -0,0 +1,23 @@
+package com.example.animals
+
+trait Animal {
+  val name: String
+  def speak(): String
+}
+
+trait Mammal extends Animal {
+  def hasFur: Boolean = true
+}
+
+trait CanRun {
+  def run(speed: Double): String = s"Running at $speed km/h"
+}
+
+class Dog(override val name: String) extends Mammal with CanRun {
+  override def speak(): String = "Woof!"
+}
+
+class Cat(override val name: String) extends Mammal {
+  override def speak(): String = "Meow"
+  override val hasFur: Boolean = true
+}
diff --git a/tests/sample_project_scala/Complex.scala b/tests/sample_project_scala/Complex.scala
@@ -0,0 +1,23 @@
+package com.example.complex
+
+class Outer {
+  private val secret = "I am outer"
+  
+  class Inner {
+    def reveal: String = secret
+  }
+  
+  object InnerObject {
+    def check: Boolean = true
+  }
+}
+
+object Outer {
+  private val staticSecret = "I am static"
+  
+  def getStaticSecret: String = staticSecret
+  
+  class NestedInObject {
+    def nothing: Unit = {}
+  }
+}
diff --git a/tests/sample_project_scala/Functional.scala b/tests/sample_project_scala/Functional.scala
@@ -0,0 +1,13 @@
+package com.example.functional
+
+object Functional {
+  def applyFunc(x: Int, f: Int => Int): Int = f(x)
+  
+  def curriedAdd(x: Int)(y: Int): Int = x + y
+  
+  val multiplier: Int => Int = (x: Int) => x * 2
+  
+  def filterList(list: List[Int], p: Int => Boolean): List[Int] = {
+    list.filter(p)
+  }
+}
diff --git a/tests/sample_project_scala/Geometry.scala b/tests/sample_project_scala/Geometry.scala
@@ -0,0 +1,21 @@
+package com.example.geometry
+
+abstract class Point(val x: Int, val y: Int) {
+  def distanceTo(other: Point): Double
+}
+
+class Point2D(x: Int, y: Int) extends Point(x, y) {
+  def distanceTo(other: Point): Double = {
+    val dx = x - other.x
+    val dy = y - other.y
+    Math.sqrt(dx * dx + dy * dy)
+  }
+}
+
+trait Resizable { 
+  this: Point => // Self-type annotation
+  
+  def resize(factor: Double): Unit = {
+    // resizing logic
+  }
+}
diff --git a/tests/sample_project_scala/Main.scala b/tests/sample_project_scala/Main.scala
@@ -0,0 +1,18 @@
+package com.example
+
+import com.example.shapes._
+import com.example.utils.Calculator
+
+object Main {
+  def main(args: Array[String]): Unit = {
+    println("Hello, Scala!")
+    
+    val calc = new Calculator()
+    val sum = calc.add(5, 10)
+    println(s"Sum: $sum")
+    
+    val circle = Circle(5.0)
+    val area = circle.area
+    println(s"Circle area: $area")
+  }
+}
diff --git a/tests/sample_project_scala/PackageObject.scala b/tests/sample_project_scala/PackageObject.scala
@@ -0,0 +1,8 @@
+package com.example
+
+package object constants {
+  val PI = 3.14159
+  val E = 2.71828
+  
+  def greet(name: String): String = s"Hello, $name"
+}
diff --git a/tests/sample_project_scala/Script.sc b/tests/sample_project_scala/Script.sc
@@ -0,0 +1,16 @@
+// This is a script file
+println("Starting script...")
+
+val x = 10
+val y = 20
+
+def sum(a: Int, b: Int): Int = a + b
+
+println(s"Result: ${sum(x, y)}")
+
+class ScriptHelper {
+  def help: String = "Helping"
+}
+
+val helper = new ScriptHelper()
+println(helper.help)
diff --git a/tests/sample_project_scala/Services.scala b/tests/sample_project_scala/Services.scala
@@ -0,0 +1,17 @@
+package com.example.services
+
+trait Service[T] {
+  def process(item: T): Boolean
+}
+
+class StringService extends Service[String] {
+  override def process(item: String): Boolean = item.nonEmpty
+}
+
+class IntService extends Service[Int] {
+  override def process(item: Int): Boolean = item > 0
+}
+
+object ServiceManager {
+  def runService[T](service: Service[T], item: T): Boolean = service.process(item)
+}
diff --git a/tests/sample_project_scala/Shapes.scala b/tests/sample_project_scala/Shapes.scala
@@ -0,0 +1,21 @@
+package com.example.shapes
+
+sealed trait Shape {
+  def area: Double
+  def perimeter: Double
+}
+
+case class Circle(radius: Double) extends Shape {
+  override def area: Double = Math.PI * radius * radius
+  override def perimeter: Double = 2 * Math.PI * radius
+}
+
+case class Rectangle(width: Double, height: Double) extends Shape {
+  override def area: Double = width * height
+  override def perimeter: Double = 2 * (width + height)
+}
+
+object ShapeFactory {
+  def createCircle(r: Double): Shape = Circle(r)
+  def createRectangle(w: Double, h: Double): Shape = Rectangle(w, h)
+}
diff --git a/tests/sample_project_scala/Utils.scala b/tests/sample_project_scala/Utils.scala
@@ -0,0 +1,29 @@
+package com.example.utils
+
+class Calculator {
+  def add(a: Int, b: Int): Int = a + b
+  
+  def subtract(a: Int, b: Int): Int = a - b
+  
+  def multiply(a: Int, b: Int): Int = a * b
+  
+  def divide(a: Double, b: Double): Double = {
+    if (b == 0) throw new IllegalArgumentException("Cannot divide by zero")
+    a / b
+  }
+}
+
+object StringUtils {
+  def reverse(s: String): String = s.reverse
+  
+  def isPalindrome(s: String): Boolean = {
+    val clean = s.toLowerCase.replaceAll("[^a-z0-9]", "")
+    clean == clean.reverse
+  }
+}
+
+// Generic class
+class Box[T](val content: T) {
+  def get: T = content
+  def map[U](f: T => U): Box[U] = new Box(f(content))
+}
diff --git a/tests/sample_project_scala/Variables.scala b/tests/sample_project_scala/Variables.scala
@@ -0,0 +1,14 @@
+package com.example.variables
+
+class Config {
+  val version: String = "1.0.0"
+  var buildNumber: Int = 1
+  
+  private val secretKey: String = "xyz"
+  protected val apiKey: String = "abc"
+  
+  lazy val expensiveCompute: Int = {
+    Thread.sleep(100)
+    42
+  }
+}

Original file line number	Diff line number	Diff line change
`@@ -46,6 +46,8 @@`
`46`	`46`	`"rust": "rust",`
`47`	`47`	`"kt": "kotlin",`
`48`	`48`	`"kotlin": "kotlin",`
	`49`	`+ "scala": "scala",`
	`50`	`+ ".scala": "scala",`
`49`	`51`	`}`
`50`	`52`
`51`	`53`