(fn tbl? [x]
  (= :table (type x)))

(fn highlight [group-arg colset]
  "Add a highlighting group."
  (each [_ group (ipairs (if (tbl? group-arg) group-arg [group-arg]))]
      (vim.api.nvim_set_hl 0 group colset)))

{: highlight}