Contextual links for Webform Blocks and Nodes

Here's a naive solution I've hacked together, based on notes at http://bleen.net/comment/3264

It directly adds a couple of links to the web form block. I'm sure it could be done better, and obviously it would be nicer to have the contextual links as standard in Webform.

function MODULENAME_contextual_links_view_alter(&$element, &$items) {
   if (isset($element['#element']['#block']) && $element['#element']['#block']->module == "webform"){
    $element['#links']['page-webform'] = array(
      'title' => 'Webform',
      'href' => 'node/' . $element['#element']['#node']->nid . '/webform',
    );
   $element['#links']['page-results'] = array(
      'title' => 'Results',
      'href' => 'node/' . $element['#element']['#node']->nid . '/webform-results',
    );
  }
}