A little more cleanup...

This commit is contained in:
Brion Vibber 2010-09-23 12:52:58 -07:00
parent e8526b2e1f
commit a78f184652
4 changed files with 141 additions and 34 deletions

View File

@ -64,7 +64,7 @@ class YammerImportPlugin extends Plugin
'Yammer_notice', 'Yammer_notice',
'Yammer_notice_stub'); 'Yammer_notice_stub');
foreach ($tables as $table) { foreach ($tables as $table) {
$schema->ensureTable($table, $table::schemaDef()); $schema->ensureTable(strtolower($table), $table::schemaDef());
} }
return true; return true;
@ -121,6 +121,14 @@ class YammerImportPlugin extends Plugin
case 'yammeradminpanelaction': case 'yammeradminpanelaction':
require_once "$base/actions/yammeradminpanel.php"; require_once "$base/actions/yammeradminpanel.php";
return false; return false;
case 'yammer_state':
case 'yammer_notice_stub':
case 'yammer_common':
case 'yammer_user':
case 'yammer_group':
case 'yammer_notice':
require_once "$base/classes/$cls.php";
return false;
default: default:
return true; return true;
} }

View File

@ -51,7 +51,7 @@ class Yammer_notice_stub extends Memcached_DataObject
/** /**
* Return schema definition to set this table up in onCheckSchema * Return schema definition to set this table up in onCheckSchema
*/ */
static function schemaDef($field) static function schemaDef()
{ {
return array(new ColumnDef('id', 'bigint', null, return array(new ColumnDef('id', 'bigint', null,
false, 'PRI'), false, 'PRI'),
@ -73,7 +73,7 @@ class Yammer_notice_stub extends Memcached_DataObject
function table() function table()
{ {
return array('id' => DB_DATAOBJECT_INT + DB_DATAOBJECT_NOTNULL, return array('id' => DB_DATAOBJECT_INT + DB_DATAOBJECT_NOTNULL,
'json_data' => DB_DATAOBJECT_INT + DB_DATAOBJECT_NOTNULL, 'json_data' => DB_DATAOBJECT_STR + DB_DATAOBJECT_NOTNULL,
'created' => DB_DATAOBJECT_STR + DB_DATAOBJECT_DATE + DB_DATAOBJECT_TIME + DB_DATAOBJECT_NOTNULL); 'created' => DB_DATAOBJECT_STR + DB_DATAOBJECT_DATE + DB_DATAOBJECT_TIME + DB_DATAOBJECT_NOTNULL);
} }

View File

@ -1,37 +1,136 @@
<?php <?php
/**
* Data class for remembering Yammer import state
*
* PHP version 5
*
* @category Data
* @package StatusNet
* @author Brion Vibber <brion@status.net>
* @license http://www.fsf.org/licensing/licenses/agpl.html AGPLv3
* @link http://status.net/
*
* 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/>.
*/
/* if (!defined('STATUSNET')) {
exit(1);
}
yammer_state class Yammer_state extends Memcached_DataObject
id {
state public $__table = 'yammer_state'; // table name
request_token public $id; // int primary_key not_null
oauth_token public $state; // import state key
users_page public $request_token; // oauth request token; clear when auth is complete.
groups_page public $oauth_token; // actual oauth token! clear when import is done?
messages_oldest public $users_page; // last page of users we've fetched
created public $groups_page; // last page of groups we've fetched
modified public $messages_oldest; // oldest message ID we've fetched
public $messages_newest; // newest message ID we've imported
public $created; // datetime
public $modified; // datetime
yammer_user /**
id * Return schema definition to set this table up in onCheckSchema
user_id */
created static function schemaDef()
{
return array(new ColumnDef('id', 'int', null,
false, 'PRI'),
new ColumnDef('state', 'text'),
new ColumnDef('request_token', 'text'),
new ColumnDef('oauth_token', 'text'),
new ColumnDef('users_page', 'int'),
new ColumnDef('groups_page', 'int'),
new ColumnDef('messages_oldest', 'bigint'),
new ColumnDef('messages_newest', 'bigint'),
new ColumnDef('created', 'datetime'),
new ColumnDef('modified', 'datetime'));
}
yammer_group /**
id * return table definition for DB_DataObject
group_id *
created * DB_DataObject needs to know something about the table to manipulate
* instances. This method provides all the DB_DataObject needs to know.
*
* @return array array of column definitions
*/
yammer_notice function table()
id {
notice_id return array('id' => DB_DATAOBJECT_INT + DB_DATAOBJECT_NOTNULL,
created 'state' => DB_DATAOBJECT_STR,
'request_token' => DB_DATAOBJECT_STR,
'oauth_token' => DB_DATAOBJECT_STR,
'users_page' => DB_DATAOBJECT_INT,
'groups_page' => DB_DATAOBJECT_INT,
'messages_oldest' => DB_DATAOBJECT_INT,
'messages_newest' => DB_DATAOBJECT_INT,
'created' => DB_DATAOBJECT_STR + DB_DATAOBJECT_DATE + DB_DATAOBJECT_TIME + DB_DATAOBJECT_NOTNULL);
}
yammer_notice_stub /**
id * return key definitions for DB_DataObject
json_data *
created * DB_DataObject needs to know about keys that the table has, since it
* won't appear in StatusNet's own keys list. In most cases, this will
* simply reference your keyTypes() function.
*
* @return array list of key field names
*/
*/ function keys()
{
return array_keys($this->keyTypes());
}
/**
* return key definitions for Memcached_DataObject
*
* Our caching system uses the same key definitions, but uses a different
* method to get them. This key information is used to store and clear
* cached data, so be sure to list any key that will be used for static
* lookups.
*
* @return array associative array of key definitions, field name to type:
* 'K' for primary key: for compound keys, add an entry for each component;
* 'U' for unique keys: compound keys are not well supported here.
*/
function keyTypes()
{
return array('id' => 'K');
}
/**
* Magic formula for non-autoincrementing integer primary keys
*
* If a table has a single integer column as its primary key, DB_DataObject
* assumes that the column is auto-incrementing and makes a sequence table
* to do this incrementation. Since we don't need this for our class, we
* overload this method and return the magic formula that DB_DataObject needs.
*
* @return array magic three-false array that stops auto-incrementing.
*/
function sequenceKey()
{
return array(false, false, false);
}
}

View File

@ -13,6 +13,7 @@ require 'yam-config.php';
$yam = new SN_YammerClient($consumerKey, $consumerSecret, $token, $tokenSecret); $yam = new SN_YammerClient($consumerKey, $consumerSecret, $token, $tokenSecret);
$imp = new YammerImporter($yam); $imp = new YammerImporter($yam);
/*
$data = $yam->users(); $data = $yam->users();
var_dump($data); var_dump($data);
// @fixme follow paging // @fixme follow paging
@ -20,9 +21,9 @@ foreach ($data as $item) {
$user = $imp->prepUser($item); $user = $imp->prepUser($item);
var_dump($user); var_dump($user);
} }
*/
/* $data = $yam->messages(array('newer_than' => 1));
$data = $yam->messages();
var_dump($data); var_dump($data);
// @fixme follow paging // @fixme follow paging
$messages = $data['messages']; $messages = $data['messages'];
@ -31,4 +32,3 @@ foreach ($messages as $message) {
$notice = $imp->prepNotice($message); $notice = $imp->prepNotice($message);
var_dump($notice); var_dump($notice);
} }
*/