don't get a count from query

[quix0rs-gnu-social.git] / classes / User.php

diff --git a/classes/User.php b/classes/User.php
index e7d297b3c6947eaab61481301b88d85f5deb897d..b6689761eec4159faf62fbd15da775fe02a2f197 100644 (file)
--- a/classes/User.php
+++ b/classes/User.php
@@ -1,27 +1,28 @@
 <?php
-/* 
+/*
  * Laconica - a distributed open-source microblogging tool
  * Copyright (C) 2008, Controlez-Vous, 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('LACONICA')) { exit(1) }
+if (!defined('LACONICA')) { exit(1); }
 /**
  * Table Definition for user
  */
 require_once 'DB/DataObject.php';
+require_once 'Validate.php';
 
 class User extends DB_DataObject 
 {
@@ -30,8 +31,23 @@ class User extends DB_DataObject
 
     public $__table = 'user';                            // table name
     public $id;                              // int(4)  primary_key not_null
+    public $nickname;                        // varchar(64)  unique_key
     public $password;                        // varchar(255)  
     public $email;                           // varchar(255)  unique_key
+    public $incomingemail;                   // varchar(255)  unique_key
+    public $emailnotifysub;                  // tinyint(1)   default_1
+    public $emailpost;                       // tinyint(1)   default_1
+    public $jabber;                          // varchar(255)  unique_key
+    public $jabbernotify;                    // tinyint(1)  
+    public $jabberreplies;                   // tinyint(1)  
+    public $updatefrompresence;              // tinyint(1)  
+    public $sms;                             // varchar(64)  unique_key
+    public $carrier;                         // int(4)  
+    public $smsnotify;                       // tinyint(1)  
+    public $smsreplies;                      // tinyint(1)  
+    public $smsemail;                        // varchar(255)  
+    public $uri;                             // varchar(255)  unique_key
+    public $autosubscribe;                   // tinyint(1)  
     public $created;                         // datetime()   not_null
     public $modified;                        // timestamp()   not_null default_CURRENT_TIMESTAMP
 
@@ -40,11 +56,17 @@ class User extends DB_DataObject
 
     /* the code above is auto generated do not remove the tag below */
     ###END_AUTOCODE
-       
+
        function getProfile() {
-               return Profile::staticGet($this->$id);
+               $profile = DB_DataObject::factory('profile');
+               $profile->id = $this->id;
+               if ($profile->find()) {
+                       $profile->fetch();
+                       return $profile;
+               }
+               return NULL;
        }
-       
+
        function isSubscribed($other) {
                assert(!is_null($other));
                $sub = DB_DataObject::factory('subscription');
@@ -52,4 +74,71 @@ class User extends DB_DataObject
                $sub->subscribed = $other->id;
                return $sub->find();
        }
+
+       # 'update' won't write key columns, so we have to do it ourselves.
+
+       function updateKeys(&$orig) {
+               $parts = array();
+               foreach (array('nickname', 'email', 'jabber', 'incomingemail', 'sms', 'carrier', 'smsemail') as $k) {
+                       if (strcmp($this->$k, $orig->$k) != 0) {
+                               $parts[] = $k . ' = ' . $this->_quote($this->$k);
+                       }
+               }
+               if (count($parts) == 0) {
+                       # No changes
+                       return true;
+               }
+               $toupdate = implode(', ', $parts);
+               $qry = 'UPDATE ' . $this->tableName() . ' SET ' . $toupdate .
+                 ' WHERE id = ' . $this->id;
+               return $this->query($qry);
+       }
+
+       function allowed_nickname($nickname) {
+               # XXX: should already be validated for size, content, etc.
+               static $blacklist = array('rss', 'xrds', 'doc', 'main',
+                                                                 'settings', 'notice', 'user',
+                                                                 'search', 'avatar');
+               $merged = array_merge($blacklist, common_config('nickname', 'blacklist'));
+               return !in_array($nickname, $merged);
+       }
+
+       function getCurrentNotice($dt=NULL) {
+               $profile = $this->getProfile();
+               if (!$profile) {
+                       return NULL;
+               }
+               return $profile->getCurrentNotice($dt);
+       }
+       
+       function getCarrier() {
+               return Sms_carrier::staticGet($this->carrier);
+       }
+       
+       function subscribeTo($other) {
+               $sub = new Subscription();
+               $sub->subscriber = $this->id;
+               $sub->subscribed = $other->id;
+
+               $sub->created = DB_DataObject_Cast::dateTime(); # current time
+
+               if (!$sub->insert()) {
+                       return false;
+               }
+               
+               return true;
+       }
+
+       function noticesWithFriends($offset=0, $limit=20) {
+               
+               $notice = new Notice();
+               
+               $notice->query('SELECT notice.* ' .
+                                          'FROM notice JOIN subscription on notice.profile_id = subscription.subscribed ' .
+                                          'WHERE subscription.subscriber = ' . $this->id . ' ' .
+                                          'ORDER BY created DESC, notice.id DESC ' .
+                                          'LIMIT ' . $offset . ', ' . $limit);
+               
+               return $notice;
+       }
 }