feat(kernel): simplify choice_fn, and make its interface closer to the unifier_plugin interface

Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>