actions/twittersettings.php

   1 <?php
   2 /*
   3  * Laconica - a distributed open-source microblogging tool
   4  * Copyright (C) 2008, Controlez-Vous, Inc.
   5  *
   6  * This program is free software: you can redistribute it and/or modify
   7  * it under the terms of the GNU Affero General Public License as published by
   8  * the Free Software Foundation, either version 3 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU Affero General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Affero General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 if (!defined('LACONICA')) { exit(1); }
  21
  22 require_once(INSTALLDIR.'/lib/settingsaction.php');
  23
  24 define('SUBSCRIPTIONS', 80);
  25
  26 class TwittersettingsAction extends SettingsAction {
  27
  28         function get_instructions() {
  29                 return _('Add your Twitter account to automatically send your notices to Twitter, ' .
  30                         'and subscribe to Twitter friends already here.');
  31         }
  32
  33         function show_form($msg=NULL, $success=false) {
  34                 $user = common_current_user();
  35                 $profile = $user->getProfile();
  36                 $fuser = NULL;
  37                 $flink = Foreign_link::getByUserID($user->id, 1); // 1 == Twitter
  38
  39                 if ($flink) {
  40                         $fuser = $flink->getForeignUser();
  41                 }
  42
  43                 $this->form_header(_('Twitter settings'), $msg, $success);
  44                 common_element_start('form', array('method' => 'post',
  45                                                                                    'id' => 'twittersettings',
  46                                                                                    'action' =>
  47                                                                                    common_local_url('twittersettings')));
  48                 common_hidden('token', common_session_token());
  49
  50                 common_element('h2', NULL, _('Twitter Account'));
  51
  52                 if ($fuser) {
  53                         common_element_start('p');
  54
  55                         common_element('span', 'twitter_user', $fuser->nickname);
  56                         common_element('a', array('href' => $fuser->uri),  $fuser->uri);
  57                         common_element('span', 'input_instructions',
  58                                        _('Current verified Twitter account.'));
  59                         common_hidden('flink_foreign_id', $flink->foreign_id);
  60                         common_element_end('p');
  61                         common_submit('remove', _('Remove'));
  62                 } else {
  63                         common_input('twitter_username', _('Twitter user name'),
  64                                                  ($this->arg('twitter_username')) ? $this->arg('twitter_username') : $profile->nickname,
  65                                                  _('No spaces, please.')); // hey, it's what Twitter says
  66
  67                         common_password('twitter_password', _('Twitter password'));
  68                 }
  69
  70                 common_element('h2', NULL, _('Preferences'));
  71
  72                 common_checkbox('noticesync', _('Automatically send my notices to Twitter.'),
  73                                                 ($flink) ? ($flink->noticesync & FOREIGN_NOTICE_SEND) : true);
  74
  75                 common_checkbox('replysync', _('Send local "@" replies to Twitter.'),
  76                                                 ($flink) ? ($flink->noticesync & FOREIGN_NOTICE_SEND_REPLY) : true);
  77
  78                 common_checkbox('friendsync', _('Subscribe to my Twitter friends here.'),
  79                                                 ($flink) ? ($flink->friendsync & FOREIGN_FRIEND_RECV) : false);
  80
  81                 if ($flink) {
  82                         common_submit('save', _('Save'));
  83                 } else {
  84                         common_submit('add', _('Add'));
  85                 }
  86
  87                 $this->show_twitter_subscriptions();
  88
  89                 common_element_end('form');
  90
  91                 common_show_footer();
  92         }
  93
  94         function subscribed_twitter_users() {
  95
  96                 $current_user = common_current_user();
  97
  98                 $qry = 'SELECT user.* ' .
  99                         'FROM subscription ' .
 100                         'JOIN user ON subscription.subscribed = user.id ' .
 101                         'JOIN foreign_link ON foreign_link.user_id = user.id ' .
 102                         'WHERE subscriber = %d ' .
 103                         'ORDER BY user.nickname';
 104
 105                 $user = new User();
 106
 107                 $user->query(sprintf($qry, $current_user->id));
 108
 109                 $users = array();
 110
 111                 while ($user->fetch()) {
 112
 113                         // Don't include the user's own self-subscription
 114                         if ($user->id != $current_user->id) {
 115                                 $users[] = clone($user);
 116                         }
 117                 }
 118
 119                 return $users;
 120         }
 121
 122         function show_twitter_subscriptions() {
 123
 124                 $friends = $this->subscribed_twitter_users();
 125                 $friends_count = count($friends);
 126
 127                 if ($friends_count > 0) {
 128
 129                         common_element('h3', NULL, _('Twitter Friends'));
 130                         common_element_start('div', array('id' => 'subscriptions'));
 131                         common_element_start('ul', array('id' => 'subscriptions_avatars'));
 132
 133                         for ($i = 0; $i < min($friends_count, SUBSCRIPTIONS); $i++) {
 134
 135                                 $other = Profile::staticGet($friends[$i]->id);
 136
 137                                 if (!$other) {
 138                                         common_log_db_error($subs, 'SELECT', __FILE__);
 139                                         continue;
 140                                 }
 141
 142                                 common_element_start('li');
 143                                 common_element_start('a', array('title' => ($other->fullname) ?
 144                                                                                                 $other->fullname :
 145                                                                                                 $other->nickname,
 146                                                                                                 'href' => $other->profileurl,
 147                                                                                                 'rel' => 'contact',
 148                                                                                                 'class' => 'subscription'));
 149                                 $avatar = $other->getAvatar(AVATAR_MINI_SIZE);
 150                                 common_element('img', array('src' => (($avatar) ? common_avatar_display_url($avatar) :  common_default_avatar(AVATAR_MINI_SIZE)),
 151                                                                                         'width' => AVATAR_MINI_SIZE,
 152                                                                                         'height' => AVATAR_MINI_SIZE,
 153                                                                                         'class' => 'avatar mini',
 154                                                                                         'alt' =>  ($other->fullname) ?
 155                                                                                         $other->fullname :
 156                                                                                         $other->nickname));
 157                                 common_element_end('a');
 158                                 common_element_end('li');
 159
 160                         }
 161
 162                         common_element_end('ul');
 163                         common_element_end('div');
 164
 165                 }
 166
 167                 // XXX Figure out a way to show all Twitter friends... ?
 168
 169                 /*
 170                 if ($subs_count > SUBSCRIPTIONS) {
 171                         common_element_start('p', array('id' => 'subscriptions_viewall'));
 172
 173                         common_element('a', array('href' => common_local_url('subscriptions',
 174                                                                                                                                  array('nickname' => $profile->nickname)),
 175                                                                           'class' => 'moresubscriptions'),
 176                                                    _('All subscriptions'));
 177                         common_element_end('p');
 178                 }
 179                 */
 180
 181         }
 182
 183         function handle_post() {
 184
 185                 # CSRF protection
 186                 $token = $this->trimmed('token');
 187                 if (!$token || $token != common_session_token()) {
 188                         $this->show_form(_('There was a problem with your session token. Try again, please.'));
 189                         return;
 190                 }
 191
 192                 if ($this->arg('save')) {
 193                         $this->save_preferences();
 194                 } else if ($this->arg('add')) {
 195                         $this->add_twitter_acct();
 196                 } else if ($this->arg('remove')) {
 197                         $this->remove_twitter_acct();
 198                 } else {
 199                         $this->show_form(_('Unexpected form submission.'));
 200                 }
 201         }
 202
 203         function add_twitter_acct() {
 204
 205                 $screen_name = $this->trimmed('twitter_username');
 206                 $password = $this->trimmed('twitter_password');
 207                 $noticesync = $this->boolean('noticesync');
 208                 $replysync = $this->boolean('replysync');
 209                 $friendsync = $this->boolean('friendsync');
 210
 211                 if (!Validate::string($screen_name,
 212                                 array(  'min_length' => 1,
 213                                                 'max_length' => 15,
 214                                                 'format' => VALIDATE_NUM . VALIDATE_ALPHA . '_'))) {
 215                         $this->show_form(
 216                                 _('Username must have only numbers, upper- and lowercase letters, and underscore (_). 15 chars max.'));
 217                         return;
 218                 }
 219
 220                 if (!$this->verify_credentials($screen_name, $password)) {
 221                         $this->show_form(_('Could not verify your Twitter credentials!'));
 222                         return;
 223                 }
 224
 225                 $twit_user = twitter_user_info($screen_name, $password);
 226
 227                 if (!$twit_user) {
 228                         $this->show_form(sprintf(_('Unable to retrieve account information for "%s" from Twitter.'),
 229                                 $screen_name));
 230                         return;
 231                 }
 232
 233                 if (!save_twitter_user($twit_user->id, $screen_name)) {
 234                         $this->show_form(_('Unable to save your Twitter settings!'));
 235                         return;
 236                 }
 237
 238                 $user = common_current_user();
 239
 240                 $flink = DB_DataObject::factory('foreign_link');
 241                 $flink->user_id = $user->id;
 242                 $flink->foreign_id = $twit_user->id;
 243                 $flink->service = 1; // Twitter
 244                 $flink->credentials = $password;
 245                 $flink->created = common_sql_now();
 246
 247                 $this->set_flags($flink, $noticesync, $replysync, $friendsync);
 248
 249                 $flink_id = $flink->insert();
 250
 251                 if (!$flink_id) {
 252                         common_log_db_error($flink, 'INSERT', __FILE__);
 253                         $this->show_form(_('Unable to save your Twitter settings!'));
 254                         return;
 255                 }
 256
 257                 if ($friendsync) {
 258                         save_twitter_friends($user, $twit_user->id, $screen_name, $password);
 259                 }
 260
 261                 $this->show_form(_('Twitter settings saved.'), true);
 262         }
 263
 264         function remove_twitter_acct() {
 265
 266                 $user = common_current_user();
 267                 $flink = Foreign_link::getByUserID($user->id, 1);
 268                 $flink_foreign_id = $this->arg('flink_foreign_id');
 269
 270                 # Maybe an old tab open...?
 271                 if ($flink->foreign_id != $flink_foreign_id) {
 272                     $this->show_form(_('That is not your Twitter account.'));
 273                     return;
 274                 }
 275
 276                 $result = $flink->delete();
 277
 278                 if (!$result) {
 279                         common_log_db_error($flink, 'DELETE', __FILE__);
 280                         common_server_error(_('Couldn\'t remove Twitter user.'));
 281                         return;
 282                 }
 283
 284                 $this->show_form(_('Twitter account removed.'), TRUE);
 285         }
 286
 287         function save_preferences() {
 288
 289                 $noticesync = $this->boolean('noticesync');
 290                 $friendsync = $this->boolean('friendsync');
 291                 $replysync = $this->boolean('replysync');
 292
 293                 $user = common_current_user();
 294
 295                 $flink = Foreign_link::getByUserID($user->id, 1);
 296
 297                 if (!$flink) {
 298                         common_log_db_error($flink, 'SELECT', __FILE__);
 299                         $this->show_form(_('Couldn\'t save Twitter preferences.'));
 300                         return;
 301                 }
 302
 303                 $twitter_id = $flink->foreign_id;
 304                 $password = $flink->credentials;
 305
 306                 $fuser = $flink->getForeignUser();
 307
 308                 if (!$fuser) {
 309                         common_log_db_error($fuser, 'SELECT', __FILE__);
 310                         $this->show_form(_('Couldn\'t save Twitter preferences.'));
 311                         return;
 312                 }
 313
 314                 $screen_name = $fuser->nickname;
 315
 316                 $original = clone($flink);
 317                 $this->set_flags($flink, $noticesync, $replysync, $friendsync);
 318                 $result = $flink->update($original);
 319
 320                 if ($result === FALSE) {
 321                         common_log_db_error($flink, 'UPDATE', __FILE__);
 322                         $this->show_form(_('Couldn\'t save Twitter preferences.'));
 323                         return;
 324                 }
 325
 326                 if ($friendsync) {
 327                         save_twitter_friends($user, $flink->foreign_id, $screen_name, $password);
 328                 }
 329
 330                 $this->show_form(_('Twitter preferences saved.'));
 331         }
 332
 333         function verify_credentials($screen_name, $password) {
 334                 $uri = 'http://twitter.com/account/verify_credentials.json';
 335                 $data = get_twitter_data($uri, $screen_name, $password);
 336
 337                 if (!$data) {
 338                         return false;
 339                 }
 340
 341                 $user = json_decode($data);
 342
 343                 if (!$user) {
 344                         return false;
 345                 }
 346
 347                 $twitter_id = $user->status->id;
 348
 349                 if ($twitter_id) {
 350                         return $twitter_id;
 351                 }
 352
 353                 return false;
 354         }
 355
 356         function set_flags(&$flink, $noticesync, $replysync, $friendsync) {
 357                 if ($noticesync) {
 358                         $flink->noticesync |= FOREIGN_NOTICE_SEND;
 359                 } else {
 360                         $flink->noticesync &= ~FOREIGN_NOTICE_SEND;
 361                 }
 362
 363                 if ($replysync) {
 364                         $flink->noticesync |= FOREIGN_NOTICE_SEND_REPLY;
 365                 } else {
 366                         $flink->noticesync &= ~FOREIGN_NOTICE_SEND_REPLY;
 367                 }
 368
 369                 if ($friendsync) {
 370                         $flink->friendsync |= FOREIGN_FRIEND_RECV;
 371                 } else {
 372                         $flink->friendsync &= ~FOREIGN_FRIEND_RECV;
 373                 }
 374
 375                 $flink->profilesync = 0;
 376         }
 377
 378 }