RooGetTrait.php
[Pman.Core] / RooGetTrait.php
1 <?php
2
3 trait Pman_Core_RooGetTrait {
4     /**
5      * GET method   Roo/TABLENAME.php
6      *
7      * Generally for SELECT or Single SELECT
8      *
9      * Single SELECT:
10      *    _id=value          single fetch based on primary id.
11      *                       can be '0' if you want to fetch a set of defaults
12      *                       Use in conjuntion with toRooSingleArray()
13      *                      
14      *    lookup[key]=value  single fetch based on a single key value lookup.
15      *                       multiple key/value can be used. eg. ontable+onid..
16      *    _columns           what to return.
17      *
18      *    
19      * JOINS:
20      *  - all tables are always autojoined.
21      * 
22      * Search SELECT
23      *    COLUMNS to fetch
24      *      _columns=a,b,c,d     comma seperated list of columns.
25      *      _columns_exclude=a,b,c,d   comma seperated list of columns.
26      *      _distinct=name        a distinct column lookup. you also have to use _columns with this.
27      *
28      *    WHERE (searches)
29      *       colname = ...              => colname = ....
30      *       !colname=....                 => colname != ....
31      *       !colname[0]=... !colname[1]=... => colname NOT IN (.....) ** only supports main table at present..
32      *       colname[0]=... colname[1]=... => colname IN (.....) ** only supports main table at present..
33      *
34      *    ORDER BY
35      *       sort=name          what to sort.
36      *       sort=a,b,d         can support multiple columns
37      *       dir=ASC            what direction
38      *       _multisort ={...}  JSON encoded { sort : { row : direction }, order : [ row, row, row ] }
39      *
40      *    LIMIT
41      *      start=0         limit start
42      *      limit=25        limit number 
43      * 
44      * 
45      *    Simple CSV support
46      *      csvCols[0] csvCols[1]....    = .... column titles for CSV output
47      *      csvTitles[0], csvTitles[1] ....  = columns to use for CSV output
48      *
49      *  Depricated  
50      *      _toggleActive !:!:!:! - this hsould not really be here..
51      *      query[add_blank] - add a line in with an empty option...  - not really needed???
52      *      _delete    = delete a list of ids element. (depricated.. this will be removed...)
53      * 
54      * DEBUGGING
55      *  _post   =1    = simulate a post with debuggin on.
56      *  _debug_post << This is prefered, as _post may overlap with accouting posts..
57      *  
58      *  _debug     = turn on DB_dataobject deubbing, must be admin at present..
59      *
60      *
61      * CALLS methods on dataobjects if they exist
62      *
63      * 
64      *   checkPerm('S' , $authuser)
65      *                      - can we list the stuff
66      *                      - return false to disallow...
67      *   applySort($au, $sortcol, $direction, $array_of_columns, $multisort)
68      *                     -- does not support multisort at present..
69      *   applyFilters($_REQUEST, $authUser, $roo)
70      *                     -- apply any query filters on data. and hide stuff not to be seen.
71      *                     -- can exit by calling $roo->jerr()
72      *   postListExtra($_REQUEST) : array(extra_name => data)
73      *                     - add extra column data on the results (like new messages etc.)
74      *   postListFilter($data, $authUser, $request) return $data
75      *                      - add extra data to an object
76      * 
77      *   
78      *   toRooSingleArray($authUser, $request) : array
79      *                       - called on single fetch only, add or maniuplate returned array data.
80      *                       - is also called when _id=0 is used (for fetching a default set.)
81      *   toRooArray($request) : array
82      *                      - called if singleArray is unavailable on single fetch.
83      *                      - always tried for mutiple results.
84      *   toArray()          - the default method if none of the others are found. 
85      *   
86      *   autoJoin($request) 
87      *                      - standard DataObject feature - causes all results to show all
88      *                        referenced data.
89      *
90      * PROPERTIES:
91      *    _extra_cols  -- if set, then filtering by column etc. will use them.
92      *
93      
94      */
95     function get($tab)
96     {
97         $this->init();
98         
99         HTML_FlexyFramework::get()->generateDataobjectsCache($this->isDev);
100         
101         if ( $this->checkDebugPost()) {
102             $_POST  = $_GET;
103             return $this->post($tab);
104         }
105         
106         $this->checkDebug();
107         
108         PEAR::setErrorHandling(PEAR_ERROR_CALLBACK, array($this, 'onPearError'));
109    
110         $tab = array_shift(explode('/', $tab));
111         
112         $x = $this->dataObject($tab);
113         
114         $_columns = !empty($_REQUEST['_columns']) ? explode(',', $_REQUEST['_columns']) : false;
115         
116         if (isset( $_REQUEST['lookup'] ) && is_array($_REQUEST['lookup'] )) { // single fetch based on key/value pairs
117              $this->selectSingle($x, $_REQUEST['lookup'],$_REQUEST);
118              // actually exits.
119         }
120         
121         
122         // single fetch (use '0' to fetch an empty object..)
123         if (isset($_REQUEST['_id']) && is_numeric($_REQUEST['_id'])) {
124              
125              $this->selectSingle($x, $_REQUEST['_id'],$_REQUEST);
126              // actually exits.
127         }
128         
129         // Depricated...
130
131        
132         if (isset($_REQUEST['_delete'])) {
133             $this->jerr("DELETE by GET has been removed - update the code to use POST");
134         } 
135         
136         
137         // Depricated...
138         
139         if (isset($_REQUEST['_toggleActive'])) {
140             // do we really delete stuff!?!?!?
141             if (!$this->hasPerm("Core.Staff", 'E'))  {
142                 $this->jerr("PERMISSION DENIED (ta)");
143             }
144             $clean = create_function('$v', 'return (int)$v;');
145             $bits = array_map($clean, explode(',', $_REQUEST['_toggleActive']));
146             if (in_array($this->authUser->id, $bits) && $this->authUser->active) {
147                 $this->jerr("you can not disable yourself");
148             }
149             $x->query('UPDATE Person SET active = !active WHERE id IN (' .implode(',', $bits).')');
150             $this->addEvent("USERTOGGLE", false, implode(',', $bits));
151             $this->jok("Updated");
152             
153         }
154        //DB_DataObject::debugLevel(1);
155        
156         
157         // sets map and countWhat
158         $this->loadMap($x, array(
159             'columns' => $_columns,
160             'distinct' => empty($_REQUEST['_distinct']) ? false:  $_REQUEST['_distinct'],
161             'exclude' => empty($_REQUEST['_exclude_columns']) ? false:  explode(',', $_REQUEST['_exclude_columns'])
162         ));
163         
164         
165         $this->setFilters($x,$_REQUEST);
166         
167         if (!$this->checkPerm($x,'S', $_REQUEST))  {
168             $this->jerr("PERMISSION DENIED (g)");
169         }
170         
171         $total = $x->count($this->countWhat);
172         // sorting..
173       //   
174         //var_dump($total);exit;
175         $this->applySort($x);
176         
177         $fake_limit = false;
178         
179         if (!empty($_REQUEST['_distinct']) && $total < 400) {
180             $fake_limit  = true;
181         }
182         
183         if (!$fake_limit) {
184  
185             $x->limit(
186                 empty($_REQUEST['start']) ? 0 : (int)$_REQUEST['start'],
187                 min(empty($_REQUEST['limit']) ? 25 : (int)$_REQUEST['limit'], 10000)
188             );
189         } 
190         $queryObj = clone($x);
191         //DB_DataObject::debuglevel(1);
192         
193         $this->sessionState(0);
194         $res = $x->find();
195         $this->sessionState(1);
196         
197         if (false === $res) {
198             $this->jerr($x->_lastError->toString());
199         }
200         
201         $ret = array();
202         
203         // ---------------- THESE ARE DEPRICATED.. they should be moved to the model...
204         
205         
206         if (!empty($_REQUEST['query']['add_blank'])) {
207             $ret[] = array( 'id' => 0, 'name' => '----');
208             $total+=1;
209         }
210          
211         $rooar = method_exists($x, 'toRooArray');
212         $_columnsf = $_columns  ? array_flip($_columns) : false;
213         while ($x->fetch()) {
214             //print_R($x);exit;
215             $add = $rooar  ? $x->toRooArray($_REQUEST) : $x->toArray();
216             if ($add === false) {
217                 continue;
218             }
219             $ret[] =  !$_columns ? $add : array_intersect_key($add, $_columnsf);
220         }
221         
222         if ($fake_limit) {
223             $ret = array_slice($ret,
224                    empty($_REQUEST['start']) ? 0 : (int)$_REQUEST['start'],
225                     min(empty($_REQUEST['limit']) ? 25 : (int)$_REQUEST['limit'], 10000)
226             );
227             
228         }
229         
230         
231         $extra = false;
232         if (method_exists($queryObj ,'postListExtra')) {
233             $extra = $queryObj->postListExtra($_REQUEST, $this);
234         }
235         
236         
237         // filter results, and add any data that is needed...
238         if (method_exists($x,'postListFilter')) {
239             $ret = $x->postListFilter($ret, $this->getAuthUser(), $_REQUEST);
240         }
241         
242         
243         
244         if (!empty($_REQUEST['csvCols']) && !empty($_REQUEST['csvTitles']) ) {
245             
246             
247             $this->toCsv($ret, $_REQUEST['csvCols'], $_REQUEST['csvTitles'],
248                         empty($_REQUEST['csvFilename']) ? '' : $_REQUEST['csvFilename']
249                          );
250             
251             
252         
253         }
254         
255         if (!empty($_REQUEST['_requestMeta']) &&  count($ret)) {
256             $meta = $this->meta($x, $ret);
257             if ($meta) {
258                 $extra['metaData'] = $meta;
259             }
260         }
261         // this make take some time...
262         $this->sessionState(0);
263        // echo "<PRE>"; print_r($ret);
264         $this->jdata($ret, max(count($ret), $total), $extra );
265
266     
267     }
268     
269     function checkDebugPost()
270     {
271         return (!empty($_GET['_post']) || !empty($_GET['_debug_post'])) && 
272                     $this->authUser && 
273                     method_exists($this->authUser,'groups') &&
274                     in_array('Administrators', $this->authUser->groups('name')); 
275         
276     }
277     
278     function selectSingle($x, $id, $req=false)
279     {
280         $_columns = !empty($req['_columns']) ? explode(',', $req['_columns']) : false;
281
282         if (!is_array($id) && empty($id)) {
283             
284             if (method_exists($x, 'toRooSingleArray')) {
285                 $this->jok($x->toRooSingleArray($this->getAuthUser(), $req));
286             }
287             
288             if (method_exists($x, 'toRooArray')) {
289                 $this->jok($x->toRooArray($req));
290             }
291             
292             $this->jok($x->toArray());
293         }
294        
295         
296         $this->loadMap($x, array(
297             'columns' => $_columns,
298         ));
299         
300         if ($req !== false) { 
301             $this->setFilters($x, $req);
302         }
303         
304         if (is_array($id)) {
305             // lookup...
306             $x->setFrom($req['lookup'] );
307             $x->limit(1);
308             if (!$x->find(true)) {
309                 if (!empty($id['_id'])) {
310                     // standardize this?
311                     $this->jok($x->toArray());
312                 }
313                 $this->jok(false);
314             }
315             
316         } else if (!$x->get($id)) {
317             $this->jerr("selectSingle: no such record ($id)");
318         }
319         
320         // ignore perms if comming from update/insert - as it's already done...
321         if ($req !== false && !$this->checkPerm($x,'S'))  {
322             $this->jerr("PERMISSION DENIED - si");
323         }
324         // different symantics on all these calls??
325         if (method_exists($x, 'toRooSingleArray')) {
326             $this->jok($x->toRooSingleArray($this->getAuthUser(), $req));
327         }
328         if (method_exists($x, 'toRooArray')) {
329             $this->jok($x->toRooArray($req));
330         }
331         
332         $this->jok($x->toArray());
333         
334         
335     }
336     
337     function setFilters($x, $q)
338     {
339         if (method_exists($x, 'applyFilters')) {
340            // DB_DataObject::debugLevel(1);
341             if (false === $x->applyFilters($q, $this->getAuthUser(), $this)) {
342                 return; 
343             } 
344         }
345         $q_filtered = array();
346         
347         $keys = $x->keys();
348         // var_dump($keys);exit;
349         foreach($q as $key=>$val) {
350             
351             if (in_array($key,$keys) && !is_array($val)) {
352                
353                 $x->$key  = $val;
354             }
355             
356              // handles name[]=fred&name[]=brian => name in ('fred', 'brian').
357             // value is an array..
358             if (is_array($val) ) {
359                 
360                 $pref = '';
361                 
362                 if ($key[0] == '!') {
363                     $pref = '!';
364                     $key = substr($key,1);
365                 }
366                 
367                 if (!in_array( $key,  array_keys($this->cols))) {
368                     continue;
369                 }
370                 
371                 // support a[0] a[1] ..... => whereAddIn(
372                 $ar = array();
373                 $quote = false;
374                 foreach($val as $k=>$v) {
375                     if (!is_numeric($k)) {
376                         $ar = array();
377                         break;
378                     }
379                     // FIXME: note this is not typesafe for anything other than mysql..
380                     
381                     if (!is_numeric($v) || !is_long($v)) {
382                         $quote = true;
383                     }
384                     $ar[] = $v;
385                     
386                 }
387                 if (count($ar)) {
388                     
389                     
390                     $x->whereAddIn($pref . (
391                         isset($this->colsJname[$key]) ? 
392                             $this->colsJname[$key] :
393                             ($x->tableName(). '.'.$key)),
394                         $ar, $quote ? 'string' : 'int');
395                 }
396                 
397                 continue;
398             }
399             
400             
401             // handles !name=fred => name not equal fred.
402             if ($key[0] == '!' && in_array(substr($key, 1), array_keys($this->cols))) {
403                 
404                 $key  = substr($key, 1) ;
405                 
406                 $x->whereAdd(   (
407                         isset($this->colsJname[$key]) ? 
408                             $this->colsJname[$key] :
409                             $x->tableName(). '.'.$key ) . ' != ' .
410                     (is_numeric($val) ? $val : "'".  $x->escape($val) . "'")
411                 );
412                 continue;
413                 
414             }
415
416             switch($key) {
417                     
418                 // Events and remarks -- fixme - move to events/remarsk...
419                 case 'on_id':  // where TF is this used...
420                     if (!empty($q['query']['original'])) {
421                       //  DB_DataObject::debugLevel(1);
422                         $o = (int) $q['query']['original'];
423                         $oid = (int) $val;
424                         $x->whereAdd("(on_id = $oid  OR 
425                                 on_id IN ( SELECT distinct(id) FROM Documents WHERE original = $o ) 
426                             )");
427                         continue;
428                                 
429                     }
430                     $x->on_id = $val;
431                 
432                 
433                 default:
434                     if (strlen($val) && $key[0] != '_') {
435                         $q_filtered[$key] = $val;
436                     }
437                     
438                     // subjoined columns = check the values.
439                     // note this is not typesafe for anything other than mysql..
440                     
441                     if (isset($this->colsJname[$key])) {
442                         $quote = false;
443                         if (!is_numeric($val) || !is_long($val)) {
444                             $quote = true;
445                         }
446                         $x->whereAdd( "{$this->colsJname[$key]} = " . ($quote ? "'". $x->escape($val) ."'" : $val));
447                         
448                     }
449                     
450                     
451                     continue;
452             }
453         }
454         if (!empty($q_filtered)) {
455             $x->setFrom($q_filtered);
456         }
457         
458         if (!empty($q['query']['name'])) {
459             if (in_array( 'name',  array_keys($x->table()))) {
460                 $x->whereAdd($x->tableName().".name LIKE '". $x->escape($q['query']['name']) . "%'");
461             }
462         }
463         
464     }
465     
466     function applySort($x, $sort = '', $dir ='')
467     {
468         $sort = empty($_REQUEST['sort']) ? $sort : $_REQUEST['sort'];
469         $dir = empty($_REQUEST['dir']) ? $dir : $_REQUEST['dir'];
470         $dir = $dir == 'ASC' ? 'ASC' : 'DESC';
471          
472         $ms = empty($_REQUEST['_multisort']) ? false : $_REQUEST['_multisort'];
473         //var_Dump($ms);exit;
474         $sorted = false;
475         if (method_exists($x, 'applySort')) {
476             $sorted = $x->applySort(
477                     $this->getAuthUser(),
478                     $sort,
479                     $dir,
480                     array_keys($this->cols),
481                     $ms ? json_decode($ms) : false
482             );
483         }
484         if ($ms !== false) {
485             return $this->multiSort($x);
486         }
487         
488         if ($sorted === false) {
489             
490             $cols = $x->table();
491             $excols = array_keys($this->cols);
492             
493             if (isset($x->_extra_cols)) {
494                 $excols = array_merge($excols, $x->_extra_cols);
495             }
496             $sort_ar = explode(',', $sort);
497             $sort_str = array();
498           
499             foreach($sort_ar as $sort) {
500                 
501                 if (strlen($sort) && isset($cols[$sort]) ) {
502                     $sort_str[] =  $x->tableName() .'.'.$sort . ' ' . $dir ;
503                     
504                 } else if (in_array($sort, $excols)) {
505                     $sort_str[] = $sort . ' ' . $dir ;
506                 }
507             }
508              
509             if ($sort_str) {
510                 $x->orderBy(implode(', ', $sort_str ));
511             }
512         }
513     }
514     
515     function toCsv($data, $cols, $titles, $filename, $addDate = true)
516     {
517         $this->sessionState(0); // turn off sessions  - no locking..
518
519         require_once 'Pman/Core/SimpleExcel.php';
520         
521         $fn = (empty($filename) ? 'list-export-' : urlencode($filename)) . (($addDate) ? date('Y-m-d') : '') ;
522         
523         
524         $se_config=  array(
525             'workbook' => substr($fn, 0, 31),
526             'cols' => array(),
527             'leave_open' => true
528         );
529         
530         
531         $se = false;
532         if (is_object($data)) {
533             $rooar = method_exists($data, 'toRooArray');
534             while($data->fetch()) {
535                 $x = $rooar  ? $data->toRooArray($q) : $data->toArray();
536                 
537                 
538                 if ($cols == '*') {  /// did we get cols sent to us?
539                     $cols = array_keys($x);
540                 }
541                 if ($titles== '*') {
542                     $titles= array_keys($x);
543                 }
544                 if ($titles !== false) {
545                     
546                     foreach($cols as $i=>$col) {
547                         $se_config['cols'][] = array(
548                             'header'=> isset($titles[$i]) ? $titles[$i] : $col,
549                             'dataIndex'=> $col,
550                             'width'=>  100
551                         );
552                          $se = new Pman_Core_SimpleExcel(array(), $se_config);
553        
554                         
555                     }
556                     
557                     
558                     //fputcsv($fh, $titles);
559                     $titles = false;
560                 }
561                 
562
563                 $se->addLine($se_config['workbook'], $x);
564                     
565                 
566             }
567             if(!$se){
568                 
569                 $this->jerr('no data found', false, 'text/plain');
570             }
571             $se->send($fn .'.xls');
572             exit;
573             
574         } 
575         
576         
577         foreach($data as $x) {
578             //echo "<PRE>"; print_r(array($_REQUEST['csvCols'], $x->toArray())); exit;
579             $line = array();
580             if ($titles== '*') {
581                 $titles= array_keys($x);
582             }
583             if ($cols== '*') {
584                 $cols= array_keys($x);
585             }
586             if ($titles !== false) {
587                 foreach($cols as $i=>$col) {
588                     $se_config['cols'][] = array(
589                         'header'=> isset($titles[$i]) ? $titles[$i] : $col,
590                         'dataIndex'=> $col,
591                         'width'=>  100,
592                        //     'renderer' => array($this, 'getThumb'),
593                          //   'color' => 'yellow', // set color for the cell which is a header element
594                           // 'fillBlank' => 'gray', // set 
595                     );
596                     $se = new Pman_Core_SimpleExcel(array(),$se_config);
597    
598                     
599                 }
600                 
601                 
602                 //fputcsv($fh, $titles);
603                 $titles = false;
604             }
605             
606             
607             
608             $se->addLine($se_config['workbook'], $x);
609         }
610         if(!$se){
611             $this->jerr('no data found');
612         }
613         $se->send($fn .'.xls');
614         exit;
615     
616         
617         
618     }
619 }