grep_ast_quickstart_pattern_search_with_syntax_context.py

1import sys
2from grep_ast import TreeContext, filename_to_lang
3
4def main():
5    if len(sys.argv) < 3:
6        print("Usage: python quickstart.py <filename> <pattern>")
7        return
8
9    file_path = sys.argv[1]
10    query_text = sys.argv[2]
11
12    # Read the file content
13    with open(file_path, "r") as f:
14        code = f.read()
15
16    # Determine language from filename
17    lang = filename_to_lang(file_path)
18
19    # Create a TreeContext object
20    tc = TreeContext(file_path, code, lang=lang)
21
22    # Find lines matching the pattern
23    matching_linenos = tc.grep(query_text)
24
25    if not matching_linenos:
26        print(f"No matches found for '{query_text}'")
27        return
28
29    # Add the matching lines to the context
30    tc.add_lines(matching_linenos)
31
32    # Optionally add some context around matches
33    tc.add_context()
34
35    # Get the formatted output
36    output = tc.format()
37    print(output)
38
39if __name__ == "__main__":
40    main()