In the comments, the original author of the Haskell code (Richard Smith) pops up and offers a terrifying 'golfed' one-liner version ...