Reimplement trace --peg-tokens using env.log.

Change-Id: I6e8bca0df7623017194bed16b32da2194d57772b