#[cfg(feature = "scala-ast")]
impl Default for ScalaComplexityAnalyzer {
fn default() -> Self {
Self::new()
}
}
#[cfg(feature = "scala-ast")]
impl ScalaComplexityAnalyzer {
#[must_use]
pub fn new() -> Self {
Self {
cyclomatic_complexity: 0,
cognitive_complexity: 0,
}
}
pub fn analyze_complexity(&mut self, source: &str) -> Result<(u32, u32), String> {
self.cyclomatic_complexity = 1;
self.cognitive_complexity = 1;
let lines: Vec<&str> = source.lines().collect();
for line in lines {
let trimmed = line.trim();
self.analyze_complexity_for_line(trimmed);
}
Ok((self.cyclomatic_complexity, self.cognitive_complexity))
}
fn analyze_complexity_for_line(&mut self, line: &str) {
if line.contains("if ") || line.contains(" if ") {
self.cyclomatic_complexity += 1;
self.cognitive_complexity += 1;
}
if line.contains("match ") || line.contains(" match ") {
self.cyclomatic_complexity += 1;
self.cognitive_complexity += 1;
}
if line.contains("case ") && !line.contains("case class ") {
self.cyclomatic_complexity += 1;
self.cognitive_complexity += 1;
}
if line.contains(" && ") || line.contains(" || ") {
self.cyclomatic_complexity += 1;
}
if line.contains(" for ") || line.contains("while ") {
self.cyclomatic_complexity += 1;
self.cognitive_complexity += 1;
}
if line.contains("try ") || line.contains("catch ") {
self.cyclomatic_complexity += 1;
self.cognitive_complexity += 1;
}
}
}