1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
|
<?php
/**
* $Header: /cvsroot/bitweaver/_bit_quota/LibertyQuota.php,v 1.12 2006/09/05 21:12:59 spiderr Exp $
*
* Copyright (c) 2004 bitweaver.org
* Copyright (c) 2003 tikwiki.org
* Copyright (c) 2002-2003, Luis Argerich, Garland Foster, Eduardo Polidor, et. al.
* All Rights Reserved. See copyright.txt for details and a complete list of authors.
* Licensed under the GNU LESSER GENERAL PUBLIC LICENSE. See license.txt for details
*
* $Id: LibertyQuota.php,v 1.12 2006/09/05 21:12:59 spiderr Exp $
* @package quota
*/
/**
* required setup
*/
require_once( LIBERTY_PKG_PATH.'LibertyAttachable.php' );
/**
* Quota class to illustrate best practices when creating a new bitweaver package that
* builds on core bitweaver functionality, such as the Liberty CMS engine
*
* @package quota
* @subpackage LibertyQuota
*
* created 2004/8/15
*
* @author spider <spider@steelsun.com>
*
* @version $Revision: 1.12 $ $Date: 2006/09/05 21:12:59 $ $Author: spiderr $
*/
class LibertyQuota extends LibertyBase {
/**
* Primary key for our mythical Quota class object & table
* @public
*/
var $mQuotaId;
/**
* During initialisation, be sure to call our base constructors
**/
function LibertyQuota( $pQuotaId=NULL, $pContentId=NULL ) {
$this->mQuotaId = $pQuotaId;
LibertyBase::LibertyBase();
}
/**
* Any method named Store inherently implies data will be written to the database
* @param pParamHash be sure to pass by reference in case we need to make modifcations to the hash
**/
function store( &$pParamHash ) {
if( $this->verify( $pParamHash ) ) {
$this->mDb->StartTrans();
$table = BIT_DB_PREFIX."quotas";
if( $this->mQuotaId ) {
$result = $this->mDb->associateUpdate( $table, $pParamHash['quota_store'], array( "quota_id" => $pParamHash['quota_id'] ) );
} else {
$this->mQuotaId = $this->mDb->GenID( 'quota_id_seq' );
$pParamHash['quota_store']['quota_id'] = $this->mQuotaId;
$result = $this->mDb->associateInsert( $table, $pParamHash['quota_store'] );
}
$this->load();
$this->mDb->CompleteTrans();
}
return( count( $this->mErrors ) == 0 );
}
/**
* Make sure the data is safe to store
* @param pParamHash be sure to pass by reference in case we need to make modifcations to the hash
**/
function verify( &$pParamHash ) {
if( isset( $pParamHash['description'] ) ) {
// insure we don't have column overflow, etc.
$pParamHash['quota_store']['description'] = trim( $pParamHash['description'], 0, 160 );
}
if( !empty( $pParamHash['title'] ) ) {
// insure we don't have column overflow, etc.
$pParamHash['quota_store']['title'] = substr( trim( $pParamHash['title'] ), 0, 160 );
} else {
$this->mErrors['title'] = "Your quota needs a title";
}
if( empty( $pParamHash['disk_usage'] ) || !is_numeric( $pParamHash['disk_usage'] ) ) {
$this->mErrors['disk_usage'] = "Invalid disk usage quantity";
} else {
$pParamHash['quota_store']['disk_usage'] = $pParamHash['disk_usage'] * 1000000;
}
if( empty( $pParamHash['monthly_transfer'] ) || !is_numeric( $pParamHash['monthly_transfer'] ) ) {
$this->mErrors['monthly_transfer'] = "Invalid disk usage quantity";
} else {
$pParamHash['quota_store']['monthly_transfer'] = $pParamHash['monthly_transfer'] * 1000000;
}
return( count( $this->mErrors ) == 0 );
}
/**
* Load the data from the database
* @param pParamHash be sure to pass by reference in case we need to make modifcations to the hash
**/
function load() {
if( $this->mQuotaId ) {
// LibertyContent::load() assumes you have joined already, and will not execute any sql!
// This is a significant performance optimization
$query = "SELECT qo.* FROM `".BIT_DB_PREFIX."quotas` qo WHERE qo.`quota_id`=?";
$result = $this->mDb->query( $query, array( $this->mQuotaId ) );
if ( $result && $result->numRows() ) {
$this->mInfo = $result->fields;
$query = "SELECT ug.`group_id`, ug.* FROM `".BIT_DB_PREFIX."users_groups` ug INNER JOIN `".BIT_DB_PREFIX."quotas_group_map` qgm ON( ug.`group_id`=qgm.`group_id` ) WHERE qgm.`quota_id`=?";
if( $rs = $this->mDb->query( $query, array( $this->mQuotaId ) ) ) {
$this->mInfo['quota_groups'] = $rs->fields;
}
}
}
return( count( $this->mInfo ) == 0 );
}
/**
*
**/
function getList() {
$query = "SELECT qo.`quota_id`, qo.* FROM `".BIT_DB_PREFIX."quotas` qo";
$ret = $this->mDb->getAssoc($query);
return ( $ret );
}
/**
*
**/
function getQuotaMenu( $pName='quota_menu', $pSelectId=NULL ) {
$query = "SELECT qo.`title`, qo.`quota_id` FROM `".BIT_DB_PREFIX."quotas` qo";
if( $rs = $this->mDb->query($query) ) {
$ret = $rs->GetMenu2( $pName, $pSelectId );
}
return ( $ret );
}
function getQuotaGroups() {
$sql = "SELECT ug.`group_id`, ug.*, qgm.`quota_id`
FROM `".BIT_DB_PREFIX."users_groups` ug LEFT OUTER JOIN `".BIT_DB_PREFIX."quotas_group_map` qgm ON( qgm.`group_id`=ug.`group_id` )
WHERE ug.`user_id`=".ROOT_USER_ID."
ORDER BY ug.`group_name` ASC";
return $this->mDb->getAssoc( $sql );
}
/**
*
**/
function assignQuotaToGroup( $pQuotaId, $pGroupId ) {
if( is_numeric( $pQuotaId ) && is_numeric( $pGroupId ) ) {
$hasRow = $this->mDb->getOne( 'SELECT `quota_id` FROM `'.BIT_DB_PREFIX.'quotas_group_map` WHERE `group_id`=?',array( $pGroupId ) );
if( $hasRow ) {
$query = 'UPDATE `'.BIT_DB_PREFIX.'quotas_group_map` SET `quota_id`=? WHERE `group_id`=?';
$rs = $this->mDb->query( $query, array( $pQuotaId, $pGroupId ) );
} else {
$query = 'INSERT INTO `'.BIT_DB_PREFIX.'quotas_group_map` (`quota_id`, `group_id`) VALUES (?,?)';
$rs = $this->mDb->query( $query, array( $pQuotaId, $pGroupId ) );
}
} elseif( is_numeric( $pGroupId ) && empty( $pQuotaId ) ) {
$query = 'DELETE FROM `'.BIT_DB_PREFIX.'quotas_group_map` WHERE `group_id`=?';
$rs = $this->mDb->query( $query, array( $pGroupId ) );
}
}
/**
* returns the quota and consumption if a user is under usage level
**/
function isUserUnderQuota( $pUserId ) {
$ret = FALSE;
if( is_numeric( $pUserId ) ) {
$query = 'SELECT MAX(qo.`disk_usage`) AS `disk_usage`
FROM `'.BIT_DB_PREFIX.'users_users` uu
INNER JOIN `'.BIT_DB_PREFIX.'users_groups_map` ugm ON ( ugm.`user_id`=uu.`user_id` )
INNER JOIN `'.BIT_DB_PREFIX.'quotas_group_map` qgm ON( qgm.`group_id`=ugm.`group_id` )
INNER JOIN `'.BIT_DB_PREFIX.'quotas` qo ON( qo.`quota_id`=qgm.`quota_id` )
WHERE uu.`user_id`=?';
if( $rs = $this->mDb->query( $query, array( $pUserId ) ) ) {
$diskQuota = $rs->fields['disk_usage'];
$diskConsumed = $this->getUserUsage( $pUserId );
if( $diskQuota == NULL || $diskQuota > $diskConsumed ) {
$ret = array($diskQuota, $diskConsumed);
}
}
}
return $ret;
}
/**
* Given a user_id, this will return the max quota for the given user. If the user belongs to more than one group, it will chose the max values
* @param pUserId user_id of the user for usage to be calculated for
* @returns an integer of the total bytes used
*/
function getUserQuota( $pUserId ) {
$ret = 0;
if( is_numeric( $pUserId ) ) {
$query = 'SELECT MAX(qo.`disk_usage`) AS `disk_usage`
FROM `'.BIT_DB_PREFIX.'users_users` uu
INNER JOIN `'.BIT_DB_PREFIX.'users_groups_map` ugm ON ( ugm.`user_id`=uu.`user_id` )
INNER JOIN `'.BIT_DB_PREFIX.'quotas_group_map` qgm ON( qgm.`group_id`=ugm.`group_id` )
INNER JOIN `'.BIT_DB_PREFIX.'quotas` qo ON( qo.`quota_id`=qgm.`quota_id` )
WHERE uu.`user_id`=?';
$ret = $this->mDb->getOne( $query, array( $pUserId ) );
}
return $ret;
}
/**
* Given a user_id, this will return this disk space used for the given user
* @param pUserId user_id of the user for usage to be calculated for
* @returns an integer of the total bytes used
*/
function getUserUsage( $pUserId ) {
$ret = 0;
if( is_numeric( $pUserId ) ) {
$ret = $this->mDb->getOne( "SELECT SUM(`file_size`) FROM `".BIT_DB_PREFIX."liberty_files` WHERE `user_id`=?", array( $pUserId ) );
}
return $ret;
}
/**
* Generates the URL to the quota page
* @param pExistsHash the hash that was returned by LibertyContent::pageExists
* @return the link to display the page.
*/
function getDisplayUrl() {
$ret = NULL;
if( @BitBase::verifyId( $this->mQuotaId ) ) {
$ret = QUOTA_PKG_URL."index.php?quota_id=".$this->mQuotaId;
}
return $ret;
}
function isValid() {
return( @BitBase::verifyId( $this->mQuotaId ) );
}
}
?>
|
