Top Level Namespace

Defined Under Namespace

Modules: CodeBlockHelper, DaliboHelper, EnumHelper, FlamegraphHelper, Markdown, MarkdownRun, PostgresHelper, ResultHelper, TestSilencer Classes: CodeBlockParser, CodeExecutor, ExecutionDecider, FrontmatterParser, LanguageResolver, MarkdownFileWriter, MarkdownProcessor, PostgreSQLFlameGraphSVG

Constant Summary

JS_CONFIG =

{
  command: proc { |temp_file_path: nil, **|
    # Check if bun is available
    bun_exists = system("command -v bun > /dev/null 2>&1")
    if bun_exists
      [ "bun #{temp_file_path}", {} ]
    else
      # Fallback to node if bun is not available
      [ "node #{temp_file_path}", {} ]
    end
  },
  temp_file_suffix: ".js",
  error_handling: :js_specific # For specific stderr appending on error
}.freeze

SQLITE_CONFIG =

{
  command: proc { |code_content: nil, temp_file_path: nil, **| [ "sqlite3 #{temp_file_path}", { stdin_data: code_content } ] },
  temp_file_suffix: ".db" # Temp file is the database
}.freeze

SUPPORTED_LANGUAGES =

{
  "psql" => {
    command: proc { |code_content: nil, explain: false, flamegraph: false, **|
      unless PostgresHelper.available?
        if PostgresHelper.psql_command && PostgresHelper.using_docker?
          TestSilencer.abort_unless_testing "Error: PostgreSQL is running in Docker but required environment variables (PGUSER, PGDATABASE) are not set. Please set these variables before running psql commands."
        else
          TestSilencer.abort_unless_testing "Error: psql command not found. Please install PostgreSQL locally, ensure psql is in your PATH, or run PostgreSQL in a Docker container."
        end
      end

      PostgresHelper.validate_env_vars!
      psql_cmd = PostgresHelper.psql_command

      # Modify the SQL query if explain or flamegraph option is enabled
      if explain || flamegraph
        # Wrap the query with EXPLAIN (ANALYZE, BUFFERS, FORMAT JSON)
        # Remove any trailing semicolons and whitespace, then add our EXPLAIN wrapper
        clean_query = code_content.strip.gsub(/;\s*$/, "")
        explained_query = "EXPLAIN (ANALYZE, BUFFERS, FORMAT JSON) #{clean_query};"
        [ "#{psql_cmd} -A -t -X", { stdin_data: explained_query } ]
      else
        [ "#{psql_cmd} -A -t -X", { stdin_data: code_content } ]
      end
    }
  },
  "ruby" => {
    command: proc { |temp_file_path: nil, **|
      xmpfilter_exists = system("command -v xmpfilter > /dev/null 2>&1")
      unless xmpfilter_exists
        TestSilencer.abort_unless_testing "Error: xmpfilter command not found. Please install xmpfilter or ensure it is in your PATH."
      end
      [ "xmpfilter #{temp_file_path}", {} ]
    },
    temp_file_suffix: ".rb",
    result_block_type: "ruby" # For special '```ruby RESULT' blocks
  },
  "js" => JS_CONFIG,
  "javascript" => JS_CONFIG, # Alias for js
  "sql" => SQLITE_CONFIG,
  "sqlite" => SQLITE_CONFIG,
  "sqlite3" => SQLITE_CONFIG, # Alias for sqlite
  "bash" => {
    command: proc { |temp_file_path: nil, **|
      bash_exists = system("command -v bash > /dev/null 2>&1")
      unless bash_exists
        TestSilencer.abort_unless_testing "Error: bash command not found. Please ensure bash is in your PATH."
      end
      [ "bash #{temp_file_path}", {} ]
    },
    temp_file_suffix: ".sh"
  },
  "zsh" => {
    command: proc { |temp_file_path: nil, **|
      zsh_exists = system("command -v zsh > /dev/null 2>&1")
      unless zsh_exists
        TestSilencer.abort_unless_testing "Error: zsh command not found. Please ensure zsh is in your PATH."
      end
      [ "zsh #{temp_file_path}", {} ]
    },
    temp_file_suffix: ".zsh"
  },
  "sh" => {
    command: proc { |temp_file_path: nil, **|
      sh_exists = system("command -v sh > /dev/null 2>&1")
      unless sh_exists
        TestSilencer.abort_unless_testing "Error: sh command not found. Please ensure sh is in your PATH."
      end
      [ "sh #{temp_file_path}", {} ]
    },
    temp_file_suffix: ".sh"
  },
  "mermaid" => {
    command: proc { |temp_file_path: nil, input_file_path: nil, **|
      mmdc_exists = system("command -v mmdc > /dev/null 2>&1")
      unless mmdc_exists
        TestSilencer.abort_unless_testing "Error: mmdc command not found. Please install @mermaid-js/mermaid-cli: npm install -g @mermaid-js/mermaid-cli"
      end

      # Generate SVG output file path with directory structure based on markdown file
      if input_file_path
        # Extract markdown file basename without extension
        md_basename = File.basename(input_file_path, ".*")

        # Create directory named after the markdown file
        output_dir = File.join(File.dirname(input_file_path), md_basename)
        Dir.mkdir(output_dir) unless Dir.exist?(output_dir)

        # Generate unique filename with markdown basename prefix
        timestamp = Time.now.strftime("%Y%m%d-%H%M%S")
        random_suffix = SecureRandom.hex(6)
        svg_filename = "#{md_basename}-#{timestamp}-#{random_suffix}.svg"
        output_path = File.join(output_dir, svg_filename)
      else
        # Fallback to old behavior if no input file path provided
        input_dir = File.dirname(temp_file_path)
        base_name = File.basename(temp_file_path, ".*")
        output_path = File.join(input_dir, "#{base_name}.svg")
      end

      [ "mmdc -i #{temp_file_path} -o #{output_path}", { output_path: output_path } ]
    },
    temp_file_suffix: ".mmd",
    result_handling: :mermaid_svg # Special handling for SVG generation
  }
}.freeze

LANGUAGE_REGEX_PART =

SUPPORTED_LANGUAGES.keys.map { |lang| Regexp.escape(lang) }.join("|").freeze

CODE_BLOCK_START_REGEX =

/^```(#{LANGUAGE_REGEX_PART})$/i