In the meanwhile, two of the Applescripts in Omni's Useful Scripts collection may be useful.

One collapses all subtrees to a numerically specified level, and the other collapses everything to the level of the currently selected node.