[Minor] Add `in` and `not_in` transformations

1 files changed, 21 insertions, 0 deletions

diff --git a/lualib/lua_selectors.lua b/lualib/lua_selectors.lua
index 69a20b9c4..5f2f3564a 100644
--- a/lualib/lua_selectors.lua
+++ b/lualib/lua_selectors.lua
@@ -382,6 +382,27 @@ local transform_function = {
       return inp[args[1]](inp)
     end
   },
+  -- Boolean function in, returns either nil or its input if input is in args list
+  ['in'] = {
+    ['types'] = {
+      ['string'] = true,
+    },
+    ['map_type'] = 'string',
+    ['process'] = function(inp, _, args)
+      for _,a in ipairs(args) do if a == inp then return inp end end
+      return nil
+    end
+  },
+  ['not_in'] = {
+    ['types'] = {
+      ['string'] = true,
+    },
+    ['map_type'] = 'string',
+    ['process'] = function(inp, _, args)
+      for _,a in ipairs(args) do if a == inp then return nil end end
+      return inp
+    end
+  },
 }
 
 local function process_selector(task, sel)