Add call & callee

Author: wenkokke

data/playground/haskell/branch.hs

@@ -25,3 +25,10 @@ fromEither f g x = case x of
     Right r -> g r
 
 someFunction x (y1 : y2 : ys) (a, b, (c, [d])) = undefined
+
+compose :: (a -> b) -> (b -> c) -> (a -> c)
+compose f g x = g (f x)
+
+zipWith :: (a -> b -> c) -> [a] -> [b] -> [c]
+zipWith f [] [] = []
+zipWith f (x : xs) (y : ys) = f x y : zipWith f xs ys

queries/haskell.scm

@@ -2,6 +2,8 @@
 ;; import haskell/haskell.argumentOrParameter.iteration.scm
 ;; import haskell/haskell.branch.scm
 ;; import haskell/haskell.branch.iteration.scm
+;; import haskell/haskell.functionCall.scm
+;; import haskell/haskell.functionCallee.scm
 ;; import haskell/haskell.functionName.scm
 ;; import haskell/haskell.name.function.scm
 ;; import haskell/haskell.namedFunction.scm

queries/haskell/haskell.argumentOrParameter.scm

@@ -1,17 +1,26 @@
-;; argumentOrParameter: function
-(function
-  patterns: (patterns
-    (_) @argumentOrParameter
-  )
-)
+;; argumentOrParameter: actual argument
+(pat_apply) @argumentOrParameter
+(pat_as) @argumentOrParameter
+(pat_field) @argumentOrParameter
+(pat_fields) @argumentOrParameter
+(pat_infix) @argumentOrParameter
+(pat_irrefutable) @argumentOrParameter
+(pat_list) @argumentOrParameter
+(pat_literal) @argumentOrParameter
+(pat_name) @argumentOrParameter
+(pat_negation) @argumentOrParameter
+(pat_parens) @argumentOrParameter
+(pat_record) @argumentOrParameter
+(pat_strict) @argumentOrParameter
+(pat_tuple) @argumentOrParameter
+(pat_typed) @argumentOrParameter
+(pat_unboxed_tuple) @argumentOrParameter
+(pat_view) @argumentOrParameter
+(pat_wildcard) @argumentOrParameter
 
-;; argumentOrParameter: exp_case
-(exp_case
-  (alts
-    (alt
-      .
-      (_) @argumentOrParameter
-    )
-  )
+;; argumentOrParameter: formal argument
+(exp_apply
   .
+  (_)
+  (_) @argumentOrParameter
 )

queries/haskell/haskell.functionCall.scm

@@ -0,0 +1,2 @@
+;; functionCall: exp_apply
+(exp_apply) @functionCall

queries/haskell/haskell.functionCallee.scm

@@ -0,0 +1,5 @@
+;; functionCallee: exp_apply
+(exp_apply
+  .
+  (_) @functionCallee
+)