classes/Status_network.php

   1 <?php
   2 /**
   3  * Table Definition for status_network
   4  *
   5  * StatusNet - the distributed open-source microblogging tool
   6  * Copyright (C) 2009, StatusNet, Inc.
   7  *
   8  * This program is free software: you can redistribute it and/or modify
   9  * it under the terms of the GNU Affero General Public License as published by
  10  * the Free Software Foundation, either version 3 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU Affero General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Affero General Public License
  19  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20  */
  21
  22 if (!defined('STATUSNET') && !defined('LACONICA')) { exit(1); }
  23
  24 class Status_network extends Safe_DataObject
  25 {
  26     ###START_AUTOCODE
  27     /* the code below is auto generated do not remove the above tag */
  28
  29     public $__table = 'status_network';                  // table name
  30     public $nickname;                        // varchar(64)  primary_key not_null
  31     public $hostname;                        // varchar(255)  unique_key
  32     public $pathname;                        // varchar(255)  unique_key
  33     public $dbhost;                          // varchar(255)
  34     public $dbuser;                          // varchar(255)
  35     public $dbpass;                          // varchar(255)
  36     public $dbname;                          // varchar(255)
  37     public $sitename;                        // varchar(255)
  38     public $theme;                           // varchar(255)
  39     public $logo;                            // varchar(255)
  40     public $created;                         // datetime()   not_null
  41     public $modified;                        // timestamp()   not_null default_CURRENT_TIMESTAMP
  42     public $tags;                            // text
  43
  44     /* Static get */
  45     function staticGet($k,$v=NULL) {
  46         $i = DB_DataObject::staticGet('Status_network',$k,$v);
  47
  48         // Don't use local process cache; if we're fetching multiple
  49         // times it's because we're reloading it in a long-running
  50         // process; we need a fresh copy!
  51         global $_DB_DATAOBJECT;
  52         unset($_DB_DATAOBJECT['CACHE']['status_network']);
  53         return $i;
  54     }
  55
  56     /* the code above is auto generated do not remove the tag below */
  57     ###END_AUTOCODE
  58
  59     static $cache = null;
  60     static $cacheInitialized = false;
  61     static $base = null;
  62     static $wildcard = null;
  63
  64     /**
  65      * @param string $dbhost
  66      * @param string $dbuser
  67      * @param string $dbpass
  68      * @param string $dbname
  69      * @param array $servers memcached servers to use for caching config info
  70      */
  71     static function setupDB($dbhost, $dbuser, $dbpass, $dbname, $servers)
  72     {
  73         global $config;
  74
  75         $config['db']['database_'.$dbname] = "mysqli://$dbuser:$dbpass@$dbhost/$dbname";
  76         $config['db']['ini_'.$dbname] = INSTALLDIR.'/classes/status_network.ini';
  77         $config['db']['table_status_network'] = $dbname;
  78
  79         if (class_exists('Memcache')) {
  80             self::$cache = new Memcache();
  81
  82             // If we're a parent command-line process we need
  83             // to be able to close out the connection after
  84             // forking, so disable persistence.
  85             //
  86             // We'll turn it back on again the second time
  87             // through which will either be in a child process,
  88             // or a single-process script which is switching
  89             // configurations.
  90             $persist = php_sapi_name() != 'cli' || self::$cacheInitialized;
  91             if (is_array($servers)) {
  92                 foreach($servers as $server) {
  93                     self::$cache->addServer($server, 11211, $persist);
  94                 }
  95             } else {
  96                 self::$cache->addServer($servers, 11211, $persist);
  97             }
  98             self::$cacheInitialized = true;
  99         }
 100
 101         self::$base = $dbname;
 102     }
 103
 104     static function cacheKey($k, $v) {
 105         return 'statusnet:' . self::$base . ':status_network:'.$k.':'.$v;
 106     }
 107
 108     static function memGet($k, $v)
 109     {
 110         if (!self::$cache) {
 111             return self::staticGet($k, $v);
 112         }
 113
 114         $ck = self::cacheKey($k, $v);
 115
 116         $sn = self::$cache->get($ck);
 117
 118         if (empty($sn)) {
 119             $sn = self::staticGet($k, $v);
 120             if (!empty($sn)) {
 121                 self::$cache->set($ck, clone($sn));
 122             }
 123         }
 124
 125         return $sn;
 126     }
 127
 128     function decache()
 129     {
 130         if (self::$cache) {
 131             $keys = array('nickname', 'hostname', 'pathname');
 132             foreach ($keys as $k) {
 133                 $ck = self::cacheKey($k, $this->$k);
 134                 self::$cache->delete($ck);
 135             }
 136         }
 137     }
 138
 139     function update($orig=null)
 140     {
 141         if (is_object($orig)) {
 142             $orig->decache(); # might be different keys
 143         }
 144         return parent::update($orig);
 145     }
 146
 147     function delete()
 148     {
 149         $this->decache(); # while we still have the values!
 150         return parent::delete();
 151     }
 152
 153     /**
 154      * @param string $servername hostname
 155      * @param string $wildcard hostname suffix to match wildcard config
 156      * @return mixed Status_network or null
 157      */
 158     static function getFromHostname($servername, $wildcard)
 159     {
 160         $sn = null;
 161         if (0 == strncasecmp(strrev($wildcard), strrev($servername), strlen($wildcard))) {
 162             // special case for exact match
 163             if (0 == strcasecmp($servername, $wildcard)) {
 164                 $sn = self::memGet('nickname', '');
 165             } else {
 166                 $parts = explode('.', $servername);
 167                 $sn = self::memGet('nickname', strtolower($parts[0]));
 168             }
 169         } else {
 170             $sn = self::memGet('hostname', strtolower($servername));
 171
 172             if (empty($sn)) {
 173                 // Try for a no-www address
 174                 if (0 == strncasecmp($servername, 'www.', 4)) {
 175                     $sn = self::memGet('hostname', strtolower(substr($servername, 4)));
 176                 }
 177             }
 178         }
 179         return $sn;
 180     }
 181
 182     /**
 183      * @param string $servername hostname
 184      * @param string $pathname URL base path
 185      * @param string $wildcard hostname suffix to match wildcard config
 186      */
 187     static function setupSite($servername, $pathname, $wildcard)
 188     {
 189         global $config;
 190
 191         $sn = null;
 192
 193         // XXX I18N, probably not crucial for hostnames
 194         // XXX This probably needs a tune up
 195         $sn = self::getFromHostname($servername, $wildcard);
 196
 197         if (!empty($sn)) {
 198
 199             // Redirect to the right URL
 200
 201             if (!empty($sn->hostname) &&
 202                 empty($_SERVER['HTTPS']) &&
 203                 0 != strcasecmp($sn->hostname, $servername)) {
 204                 $sn->redirectTo('http://'.$sn->hostname.$_SERVER['REQUEST_URI']);
 205             } else if (!empty($_SERVER['HTTPS']) &&
 206                        0 != strcasecmp($sn->hostname, $servername) &&
 207                        0 != strcasecmp($sn->nickname.'.'.$wildcard, $servername)) {
 208                 $sn->redirectTo('https://'.$sn->nickname.'.'.$wildcard.$_SERVER['REQUEST_URI']);
 209             }
 210
 211             $dbhost = (empty($sn->dbhost)) ? 'localhost' : $sn->dbhost;
 212             $dbuser = (empty($sn->dbuser)) ? $sn->nickname : $sn->dbuser;
 213             $dbpass = $sn->dbpass;
 214             $dbname = (empty($sn->dbname)) ? $sn->nickname : $sn->dbname;
 215
 216             $config['db']['database'] = "mysqli://$dbuser:$dbpass@$dbhost/$dbname";
 217
 218             $config['site']['name'] = $sn->sitename;
 219             $config['site']['nickname'] = $sn->nickname;
 220
 221             self::$wildcard = $wildcard;
 222
 223             $config['site']['wildcard'] =& self::$wildcard;
 224
 225             if (!empty($sn->hostname)) {
 226                 $config['site']['server'] = $sn->hostname;
 227             }
 228
 229             if (!empty($sn->theme)) {
 230                 $config['site']['theme'] = $sn->theme;
 231             }
 232             if (!empty($sn->logo)) {
 233                 $config['site']['logo'] = $sn->logo;
 234             }
 235
 236             return $sn;
 237         } else {
 238             return null;
 239         }
 240     }
 241
 242     // Code partially mooked from http://www.richler.de/en/php-redirect/
 243     // (C) 2006 by Heiko Richler  http://www.richler.de/
 244     // LGPL
 245
 246     function redirectTo($destination)
 247     {
 248         $old = 'http'.
 249           (($_SERVER['HTTPS'] == 'on') ? 'S' : '').
 250           '://'.
 251           $_SERVER['HTTP_HOST'].
 252           $_SERVER['REQUEST_URI'].
 253           $_SERVER['QUERY_STRING'];
 254         if ($old == $destination) { // this would be a loop!
 255             // error_log(...) ?
 256             return false;
 257         }
 258
 259         header('HTTP/1.1 301 Moved Permanently');
 260         header("Location: $destination");
 261
 262         print "<a href='$destination'>$destination</a>\n";
 263
 264         exit;
 265     }
 266
 267     function getServerName()
 268     {
 269         if (!empty($this->hostname)) {
 270             return $this->hostname;
 271         } else {
 272             return $this->nickname . '.' . self::$wildcard;
 273         }
 274     }
 275
 276     /**
 277      * Return site meta-info tags as an array
 278      * @return array of strings
 279      */
 280     function getTags()
 281     {
 282         return array_filter(explode("|", strval($this->tags)));
 283     }
 284
 285     /**
 286      * Check if this site record has a particular meta-info tag attached.
 287      * @param string $tag
 288      * @return bool
 289      */
 290     function hasTag($tag)
 291     {
 292         return in_array($tag, $this->getTags());
 293     }
 294 }