Beginning stub of Yammer message->notice import

This commit is contained in:
Brion Vibber 2010-09-21 13:56:30 -07:00
parent 6d0ac28460
commit 599f4fe121
3 changed files with 90 additions and 146 deletions

View File

@ -65,10 +65,12 @@ class YammerImportPlugin extends Plugin
{
$base = dirname(__FILE__);
$lower = strtolower($cls);
switch ($cls) {
switch ($lower) {
case 'sn_yammerclient':
case 'yammerimporter':
case 'yammerimqueuehandler':
case 'importyammeraction':
require_once $base . $lower . '.php';
require_once "$base/$lower.php";
return false;
default:
return true;

View File

@ -8,150 +8,24 @@ define('INSTALLDIR', dirname(dirname(dirname(__FILE__))));
require INSTALLDIR . "/scripts/commandline.inc";
class SN_YammerClient
{
protected $apiBase = "https://www.yammer.com";
protected $consumerKey, $consumerSecret;
protected $token, $tokenSecret;
public function __construct($consumerKey, $consumerSecret, $token=null, $tokenSecret=null)
{
$this->consumerKey = $consumerKey;
$this->consumerSecret = $consumerSecret;
$this->token = $token;
$this->tokenSecret = $tokenSecret;
}
/**
* Make an HTTP hit with OAuth headers and return the response body on success.
*
* @param string $path URL chunk for the API method
* @param array $params
* @return array
*
* @throws Exception for HTTP error
*/
protected function fetch($path, $params=array())
{
$url = $this->apiBase . '/' . $path;
if ($params) {
$url .= '?' . http_build_query($params, null, '&');
}
$headers = array('Authorization: ' . $this->authHeader());
var_dump($headers);
$client = HTTPClient::start();
$response = $client->get($url, $headers);
if ($response->isOk()) {
return $response->getBody();
} else {
throw new Exception("Yammer API returned HTTP code " . $response->getStatus() . ': ' . $response->getBody());
}
}
/**
* Hit the main Yammer API point and decode returned JSON data.
*
* @param string $method
* @param array $params
* @return array from JSON data
*
* @throws Exception for HTTP error or bad JSON return
*/
protected function api($method, $params=array())
{
$body = $this->fetch("api/v1/$method.json", $params);
$data = json_decode($body, true);
if (!$data) {
throw new Exception("Invalid JSON response from Yammer API");
}
return $data;
}
/**
* Build an Authorization header value from the keys we have available.
*/
protected function authHeader()
{
// token
// token_secret
$params = array('realm' => '',
'oauth_consumer_key' => $this->consumerKey,
'oauth_signature_method' => 'PLAINTEXT',
'oauth_timestamp' => time(),
'oauth_nonce' => time(),
'oauth_version' => '1.0');
if ($this->token) {
$params['oauth_token'] = $this->token;
}
if ($this->tokenSecret) {
$params['oauth_signature'] = $this->consumerSecret . '&' . $this->tokenSecret;
} else {
$params['oauth_signature'] = $this->consumerSecret . '&';
}
if ($this->verifier) {
$params['oauth_verifier'] = $this->verifier;
}
$parts = array_map(array($this, 'authHeaderChunk'), array_keys($params), array_values($params));
return 'OAuth ' . implode(', ', $parts);
}
/**
* @param string $key
* @param string $val
*/
protected function authHeaderChunk($key, $val)
{
return urlencode($key) . '="' . urlencode($val) . '"';
}
/**
* @return array of oauth return data; should contain nice things
*/
public function requestToken()
{
if ($this->token || $this->tokenSecret) {
throw new Exception("Requesting a token, but already set up with a token");
}
$data = $this->fetch('oauth/request_token');
$arr = array();
parse_str($data, $arr);
return $arr;
}
/**
* @return array of oauth return data; should contain nice things
*/
public function accessToken($verifier)
{
$this->verifier = $verifier;
$data = $this->fetch('oauth/access_token');
$this->verifier = null;
$arr = array();
parse_str($data, $arr);
return $arr;
}
/**
* Give the URL to send users to to authorize a new app setup
*
* @param string $token as returned from accessToken()
* @return string URL
*/
public function authorizeUrl($token)
{
return $this->apiBase . '/oauth/authorize?oauth_token=' . urlencode($token);
}
public function messages($params)
{
return $this->api('messages', $params);
}
}
// temp stuff
require 'yam-config.php';
$yam = new SN_YammerClient($consumerKey, $consumerSecret, $token, $tokenSecret);
var_dump($yam->messages());
$imp = new YammerImporter();
$data = $yam->messages();
/*
["messages"]=>
["meta"]=> // followed_user_ids, current_user_id, etc
["threaded_extended"]=> // empty!
["references"]=> // lists the users, threads, replied messages, tags
*/
// 1) we're getting messages in descending order, but we'll want to process ascending
// 2) we'll need to pull out all those referenced items too?
// 3) do we need to page over or anything?
foreach ($data['messages'] as $message) {
$notice = $imp->messageToNotice($message);
var_dump($notice);
}

View File

@ -0,0 +1,68 @@
<?php
/*
* StatusNet - the distributed open-source microblogging tool
* Copyright (C) 2010, StatusNet, Inc.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
/**
* Basic client class for Yammer's OAuth/JSON API.
*
* @package YammerImportPlugin
* @author Brion Vibber <brion@status.net>
*/
class YammerImporter
{
function messageToNotice($message)
{
$messageId = $message['id'];
$messageUrl = $message['url'];
$profile = $this->findImportedProfile($message['sender_id']);
$content = $message['body']['plain'];
$source = 'yammer';
$options = array();
if ($message['replied_to_id']) {
$replyto = $this->findImportedNotice($message['replied_to_id']);
if ($replyto) {
$options['replyto'] = $replyto;
}
}
$options['created'] = common_sql_date(strtotime($message['created_at']));
// Parse/save rendered text?
// Save liked info?
// @todo attachments?
return array('orig_id' => $messageId,
'profile' => $profile,
'content' => $content,
'source' => $source,
'options' => $options);
}
function findImportedProfile($userId)
{
// @fixme
return $userId;
}
function findImportedNotice($messageId)
{
// @fixme
return $messageId;
}
}