Refactor the shard map parsing so we can pass it down into the API object. (#564)

Right now the type itself is defined in the cli package, which means we can't
use it without an import cycle.

Signed-off-by: Dan Lorenc <lorenc.d@gmail.com>