Consider using setWindowSize API provided by Webdriver to implement setViewportSize command:

browser.setWindowSize(width, height);

This is needed in one of our visual regression tests so I'm going to handle this at some point 🙂