alternc_quota_check($uid) that * may by exported by each service class.
* each class may also export a function alternc_quota_names() * that returns an array with the quotas names managed by this class. * * @copyright AlternC-Team 2001-2005 http://alternc.org/ * */ class m_quota { var $disk=Array( /* Liste des ressources disque soumises a quota */ "web"=>"web"); var $quotas; var $clquota; // Which class manage which quota. /* ----------------------------------------------------------------- */ /** * Constructor */ function m_quota() { } /* ----------------------------------------------------------------- */ /** Check if a user can use a ressource. * @param string $ressource the ressource name (a named quota) * @Return TRUE if the user can create a ressource (= is there any quota left ?) */ function cancreate($ressource="") { $t=$this->getquota($ressource); return $t["u"]<$t["t"]; } /* ----------------------------------------------------------------- */ /** * @Return an array with the list of quota-managed services in the server */ function qlist() { global $classes; $qlist=array(); reset($this->disk); while (list($key,$val)=each($this->disk)) { $qlist[$key]=_("quota_".$key); // those are specific disks quotas. } foreach($classes as $c) { if (method_exists($GLOBALS[$c],"alternc_quota_names")) { $res=$GLOBALS[$c]->alternc_quota_names(); // returns a string or an array. if($res != "") { if (is_array($res)) { foreach($res as $k) { $qlist[$k]=_("quota_".$k); $this->clquota[$k]=$c; } } else { $qlist[$res]=_("quota_".$res); $this->clquota[$res]=$c; } } } } return $qlist; } /* ----------------------------------------------------------------- */ /** Return a ressource usage (u) and total quota (t) * @param string $ressource ressource to get quota of * @Return array the quota used and total for this ressource (or for all ressource if unspecified) */ function getquota($ressource="") { global $db,$err,$cuid; $err->log("quota","getquota",$ressource); $this->qlist(); // Generate the quota list. $db->query("select * from quotas where uid='$cuid';"); if ($db->num_rows()==0) { return array("t"=>0, "u"=>0); } else { while ($db->next_record()) { $ttmp[]=$db->Record; } foreach ($ttmp as $tt) { $g=array("t"=>$tt["total"],"u"=>0); if (method_exists($GLOBALS[$this->clquota[$tt["name"]]],"alternc_get_quota")) { $g["u"]=$GLOBALS[$this->clquota[$tt["name"]]]->alternc_get_quota($tt["name"]); } $this->quotas[$tt["name"]]=$g; } } reset($this->disk); while (list($key,$val)=each($this->disk)) { $a=array(); exec("/usr/lib/alternc/quota_get ".$cuid." ".$val,$a); $this->quotas[$val]=array("t"=>$a[1],"u"=>$a[0]); } if ($ressource) { return $this->quotas[$ressource]; } else { return $this->quotas; } } /* ----------------------------------------------------------------- */ /** Set the quota for a user (and for a ressource) * @param string $ressource ressource to set quota of * @param integer size of the quota (available or used) */ function setquota($ressource,$size) { global $err,$db,$cuid; $err->log("quota","setquota",$ressource."/".$size); if (floatval($size)==0) $size="0"; if ($this->disk[$ressource]) { // It's a disk resource, update it with shell command exec("/usr/lib/alternc/quota_edit $cuid $size"); // Now we check that the value has been written properly : exec("/usr/lib/alternc/quota_get ".$cuid,$a); if ($size!=$a[1]) { $err->raise("quota",1); return false; } } // We check that this ressource exists for this client : $db->query("SELECT * FROM quotas WHERE uid='$cuid' AND name='$ressource'"); if ($db->num_rows()) { $db->query("UPDATE quotas SET total='$size' WHERE uid='$cuid' AND name='$ressource';"); } else { $db->query("INSERT INTO quotas (uid,name,total) VALUES ('$cuid','$ressource','$size');"); } return true; } /* ----------------------------------------------------------------- */ /** * Erase all quota information about the user. */ function delquotas() { global $db,$err,$cuid; $err->log("quota","delquota"); $db->query("DELETE FROM quotas WHERE uid='$cuid';"); return true; } /* ----------------------------------------------------------------- */ /** * Get the default quotas as an associative array * @return array the array of the default quotas */ function getdefaults() { global $db; $c=array(); $db->query("SELECT type,quota FROM defquotas WHERE type='default'"); if(!$db->next_record()) $this->addtype('default'); $db->query("SELECT value,quota,type FROM defquotas ORDER BY type,quota"); while($db->next_record()) { $type = $db->f("type"); $c[$type][$db->f("quota")] = $db->f("value"); } return $c; } /* ----------------------------------------------------------------- */ /** * Set the default quotas * @param array associative array of quota (key=>val) */ function setdefaults($newq) { global $db; $qlist=$this->qlist(); foreach($newq as $type => $quotas) { foreach($quotas as $qname => $value) { if(array_key_exists($qname, $qlist)) { if(!$db->query("REPLACE INTO defquotas (value,quota,type) VALUES ($value,'$qname','$type');")) return false; } } } return true; } /* ----------------------------------------------------------------- */ /** * Add an account type for quotas * @param string $type account type to be added * @return boolean true if all went ok */ function addtype($type) { global $db; $qlist=$this->qlist(); reset($qlist); if(empty($type)) return false; while (list($key,$val)=each($qlist)) { if(!$db->query("INSERT IGNORE INTO defquotas (quota,type) VALUES('$key', '$type');") || $db->affected_rows() == 0) return false; } return true; } /* ----------------------------------------------------------------- */ /** * Delete an account type for quotas * @param string $type account type to be deleted * @return boolean true if all went ok */ function deltype($type) { global $db; $qlist=$this->qlist(); reset($qlist); if($db->query("UPDATE membres SET type='default' WHERE type='$type'") && $db->query("DELETE FROM defquotas WHERE type='$type'")) { return true; } else { return false; } } /* ----------------------------------------------------------------- */ /** * Create default quotas entries for a new user. * The user we are talking about is in the global $cuid. */ function addquotas() { global $db,$err,$cuid; $err->log("quota","addquota"); $ql=$this->qlist(); reset($ql); $db->query("SELECT type,quota FROM defquotas WHERE type='default'"); if(!$db->next_record()) $this->addtype('default'); $db->query("SELECT type FROM membres WHERE uid='$cuid'"); $db->next_record(); $t = $db->f("type"); foreach($ql as $res => $val) { $db->query("SELECT value FROM defquotas WHERE quota='$res' AND type='$t'"); $q = $db->next_record() ? $db->f("value") : 0; $this->setquota($res, $q); } return true; } /* ----------------------------------------------------------------- */ /** Return a quota value with its unit (when it is a space quota) * in MB, GB, TB ... * @param string $type The quota type * @param integer $value The quota value * @return string a quota value with its unit. */ function display_val($type, $value) { switch ($type) { case 'bw_web': return format_size($value); case 'web': return format_size($value*1024); default: return $value; } } /* ----------------------------------------------------------------- */ /** Hook function call when a user is deleted * AlternC's standard function called when a user is deleted */ function alternc_del_member() { $this->delquotas(); } /* ----------------------------------------------------------------- */ /** Hook function called when a user is created * This function initialize the user's quotas. */ function alternc_add_member() { $this->addquotas(); } /* ----------------------------------------------------------------- */ /** * Exports all the quota related information for an account. * @access private * EXPERIMENTAL 'sid' function ;) */ function alternc_export($tmpdir) { global $db,$err; $err->log("quota","export"); $str="\n"; $q=$this->getquota(); foreach ($q as $k=>$v) { $str.=" \n ".xml_entities($k)."\n"; $str.=" ".xml_entities($v)."\n \n"; } $str.="\n"; return $str; } } /* Class m_quota */ ?>