intro query

Author: Daniil159x

elixir/query.py

@@ -259,9 +259,12 @@ def query(self, cmd, *args):
 
             # DT bindings compatible strings are handled differently
             if family == 'B':
-                return self.get_idents_comps(version, ident)
+                defs, refs, docs = self.get_idents_comps(version, ident)
             else:
-                return self.get_idents_defs(version, ident, family)
+                defs, refs, docs = self.get_idents_defs(version, ident, family)
+
+            peeks = self.get_peeks_of_syms(version, defs, refs)
+            return defs, refs, docs, peeks
 
         else:
             return 'Unknown subcommand: ' + cmd + '\n'
@@ -402,10 +405,37 @@ def get_idents_defs(self, version, ident, family):
             symbol_doccomments.append(SymbolInstance(path, docline))
 
         return symbol_definitions, symbol_references, symbol_doccomments
+    
+    def get_peeks_of_syms(self, version, symbol_definitions, symbol_references):
+
+        peeks = {}
+
+        def request_peeks(syms):
+            if len(syms) > 100:
+                return
+            for sym in syms:
+                if sym.path not in peeks:
+                    peeks[sym.path] = {}
+
+                content = self.scriptLines('get-file', version, "/" + sym.path)
+
+                if type(sym.line) is int:
+                    lines = (sym.line,)
+                else:
+                    lines = map(int, sym.line.split(','))
+
+                for num in lines:
+                    index = num - 1
+                    if index >= 0 and index < len(content):
+                        peeks[sym.path][num] = decode(content[index]).strip()
+
+        request_peeks(symbol_definitions)
+        request_peeks(symbol_references)
+        return peeks
 
 
 def cmd_ident(q, version, ident, family, **kwargs):
-    symbol_definitions, symbol_references, symbol_doccomments = q.query("ident", version, ident, family)
+    symbol_definitions, symbol_references, symbol_doccomments, peeks = q.query("ident", version, ident, family)
     print("Symbol Definitions:")
     for symbol_definition in symbol_definitions:
         print(symbol_definition)
@@ -418,6 +448,11 @@ def cmd_ident(q, version, ident, family, **kwargs):
     for symbol_doccomment in symbol_doccomments:
         print(symbol_doccomment)
 
+    print("\nSymbol peeks:")
+    for file, content in peeks.items():
+        for num, line in content.items():
+            print(f"{file}:{num}: {line}")
+
 def cmd_file(q, version, path, **kwargs):
     code = q.query("file", version, path)
     print(code)