tree-sitter-embedded-template

import os from tree_sitter import Language, Parser import tree_sitter_embedded_template as ts_embedded_template # Example ERB content CODE = b""" <h1>Hello, <%= name %>!</h1> <% if items.any? %> <ul> <% items.each do |item| %> <li><%= item %></li> <% end %> </ul> <% else %> <p>No items to display.</p> <% end %> """ # Ensure the tree-sitter library is built and loaded # For pre-compiled grammars like this, installation handles it. # For custom grammars, Language.build_library might be needed (but is deprecated in newer versions). # Load the embedded template language EMBEDDED_TEMPLATE_LANGUAGE = Language(ts_embedded_template.language()) # Create a parser and set its language parser = Parser() parser.set_language(EMBEDDED_TEMPLATE_LANGUAGE) # Parse the code tree = parser.parse(CODE) # Get the root node of the syntax tree root_node = tree.root_node # Print basic information about the root node print(f"Root Node Type: {root_node.type}") print(f"Root Node Text: {root_node.text.decode('utf8')}") print(f"Number of children: {len(root_node.children)}") # Example: Find all 'erb_interpolation' nodes # Note: This is a basic traversal. For complex queries, use Tree-sitter's query API. for child in root_node.children: if child.type == 'template_content': for grandchild in child.children: if grandchild.type == 'erb_interpolation': print(f"Found ERB Interpolation: {grandchild.text.decode('utf8')} (Type: {grandchild.type})") elif grandchild.type == 'erb_statement': print(f"Found ERB Statement: {grandchild.text.decode('utf8')} (Type: {grandchild.type})")