feat: update create extractor

advanced-security · Aug 14, 2023 · 24b960d · 24b960d
1 parent 05385c6
commit 24b960d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/scripts/create-extractor-pack.sh b/scripts/create-extractor-pack.sh
@@ -13,6 +13,15 @@ else
   exit 1
 fi
 
+if which codeql >/dev/null; then
+  CODEQL_BINARY="codeql"
+elif gh codeql >/dev/null; then
+  CODEQL_BINARY="gh codeql"
+else
+  gh extension install github/gh-codeql
+  CODEQL_BINARY="gh codeql"
+fi
+
 cargo build --release
 cargo run --release --bin codeql-extractor-iac -- generate --dbscheme ql/lib/iac.dbscheme --library ql/lib/codeql/iac/ast/internal/TreeSitter.qll
 $CODEQL_BINARY query format -i ql/lib/codeql/iac/ast/internal/TreeSitter.qll