From 6900e5f704110e3dbaab844c81530e82f271f7da Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alvaro=20Mun=CC=83oz?= <alvaro@pwntester.com>
Date: Fri, 6 Nov 2020 23:10:59 +0100
Subject: [PATCH] minor improvments for QL

---
 queries/ql/highlights.scm | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/queries/ql/highlights.scm b/queries/ql/highlights.scm
index bc58a7e59..b996d627c 100644
--- a/queries/ql/highlights.scm
+++ b/queries/ql/highlights.scm
@@ -102,7 +102,8 @@
   "|"
 ] @punctuation.delimiter
 
-(className) @type
+(dataclass name: (className) @type)
+(datatype name: (className) @type)
 
 (varName) @variable
 
@@ -111,6 +112,7 @@
 
 (string) @string
 
+(charpred (className) @function)
 (aritylessPredicateExpr (literalId) @function)
 (predicateName) @function