/**

* @name Print AST

* @description Outputs a representation of a file's Abstract Syntax Tree. This

* query is used by the VS Code extension.

* @id go/print-ast

* @kind graph

* @tags ide-contextual-queries/print-ast

*/

import go

import semmle.go.PrintAst

import ideContextual

/**

* Gets the source file to generate an AST from.

*/

external string selectedSourceFile();

/**

* A hook to customize the functions printed by this query.

*/

class Cfg extends PrintAstConfiguration {

override predicate shouldPrintFunction(FuncDecl func) { this.shouldPrintFile(func.getFile()) }

override predicate shouldPrintFile(File file) {

file = getFileBySourceArchiveName(selectedSourceFile())

}

override predicate shouldPrintComments(File file) { none() }

}