Improve TODO/FIXME script for more filetypes

Use a regex to match filetypes instead of checking the filenames. This
makes expanding the script to support C++ files easier.