Custom success / failure conditions for exit codes

We have a few tools that produce different exit codes depending on their outcome and we would like to choose if a particular set of exit codes means a success or failure for our tasks. I understand we can create a script to wrap our commands to the tool and return 0 or 1 based on our conditions, but I feel this would be better served inside the Bamboo task to make it more visible and useful.