Function extract_annotations_from_sql

src/parser/sql_parser.rs:172–229

Extract annotations from SQL content and return cleaned query Supports @name and @db annotations

(content: &str)

/// Extract annotations from SQL content and return cleaned query
/// Supports @name and @db annotations
fn extract_annotations_from_sql(content: &str) -> (Option<String>, Option<String>, String) {
  let mut query_name = None;
  let mut db_connection = None;
  let mut cleaned_lines = Vec::new();

  // Regex patterns for annotations
  let name_re = Regex::new(r"@name:\s*(.+)").unwrap();
  let db_re = Regex::new(r"@db:\s*(.+)").unwrap();

  for line in content.lines() {
    let trimmed_line = line.trim();

    // Skip empty lines
    if trimmed_line.is_empty() {
      cleaned_lines.push(line);
      continue;
    }

    // Check for name annotation in comments
    if let Some(stripped) = trimmed_line.strip_prefix("--") {
      let comment_content = &stripped.trim();

      if let Some(captures) = name_re.captures(comment_content) {
        query_name = Some(captures.get(1).unwrap().as_str().trim().to_string());
        continue; // Don't include annotation lines in final query
      }

      if let Some(captures) = db_re.captures(comment_content) {
        db_connection = Some(captures.get(1).unwrap().as_str().trim().to_string());
        continue; // Don't include annotation lines in final query
      }
    }

    // Check for annotations in multi-line comments
    if trimmed_line.starts_with("/*") || trimmed_line.contains("@name:") || trimmed_line.contains("@db:") {
      if let Some(captures) = name_re.captures(trimmed_line) {
        query_name = Some(captures.get(1).unwrap().as_str().trim().to_string());
      }

      if let Some(captures) = db_re.captures(trimmed_line) {
        db_connection = Some(captures.get(1).unwrap().as_str().trim().to_string());
      }

      // Only skip pure annotation lines
      if trimmed_line.starts_with("/*")
        && (trimmed_line.contains("@name:") || trimmed_line.contains("@db:"))
        && trimmed_line.ends_with("*/")
      {
        continue;
      }
    }

    cleaned_lines.push(line);
  }

  let cleaned_query = cleaned_lines.join("\n");
  (query_name, db_connection, cleaned_query)
}