export.class.php

Go to the documentation of this file.

<?php
class export {
        
        private $_registry;
        
        private $_s;
        
        private $_fe;

        private $_table;
        
        private $_pkey;
        
        private $_sfields;
        
        private $_fkeys;

        private $_head;

        private $_fields;

        private $_rids;
        
        private $_order;
        
        private $_data;

        function __construct($opts=array()) {

                $this->_registry = registry::instance();

                $this->_s = gOpt($opts, 'separator', ',');
                $this->_fe = gOpt($opts, 'field_enclosure', '"');
                $this->_table = gOpt($opts, 'table', '');
                $this->_pkey = gOpt($opts, 'pkey', 'id');
                $this->_sfields = gOpt($opts, 'sfields', array());
                $this->_fkeys = gOpt($opts, 'fkeys', array());
                $this->_head = gOpt($opts, 'head', true);
                $this->_fields = gOpt($opts, 'fields', '*');
                $this->_rids = gOpt($opts, 'rids', '*');
                $this->_order = gOpt($opts, 'order', '');
                $this->_data = gOpt($opts, 'data', array());

        }

        public function setTable($table) {
                $this->_table = $table; 
        }
        
        public function setSfields($sfields) {
                $this->_sfields = $sfields;     
        }
        
        public function setFkeys($fkeys) {
                $this->_fkeys = $fkeys; 
        }

        public function setSeparator($s) {
                $this->_s = $s;
        }
        
        public function setFieldEnclosure($fe) {
                $this->_fe = $fe;
        }

        public function setFields($fields) {
                $this->_fields = $fields;
        }

        public function setHead($head) {
                $this->_head = $head;
        }
        
        public function setRids($rids) {
                $this->_rids = $rids;
        }

        public function setOrder($order) {
                $this->_order = $order;
        }

        public function setData($data) {
                $this->_data = $data;
        }

        public function exportData($filename, $extension, $output='stream') {

                if($extension=='csv') return $this->exportCsv($filename, $output);
                // maybe other extensions in the future 
        } 

        private function exportCsv($filename, $output) {
                
                $data = $this->getData();

                $csv = '';
                foreach($data as $row) {
                        $cell = array();
                        foreach($row as $v) $cell[] = $this->encloseField($v);
                        $csv .= implode($this->_s, $cell)."\r\n";
                }

                ob_clean();

                if($output=='stream') { 
                        header("Content-Type: plain/text");
                        header("Content-Disposition: Attachment; filename=$filename");

                        header("Pragma: no-cache");
                        echo $csv;
                        exit;
                }
                elseif($output=='file') {
                        $fo = fopen($filename, "w");
                        fwrite($fo, $csv);
                        fclose($fo);
                }

        }

        private function getData() {

                if($this->_data) return $this->_data;
                if(!$this->_table) return array();

                $tot_fk = count($this->_fkeys);
                $tot_sf = count($this->_sfields);
                $data = array();
                $table_structure = $this->_registry->db->getTableStructure($this->_table);

                $head_fields = $this->getHeadFields($table_structure);
                if(count($head_fields) && $this->_head) $data[] = $head_fields;

                if($this->_rids=='*') $where = '';
                elseif(is_array($this->_rids) && count($this->_rids)) 
                        $where = $this->_pkey."='".implode("' OR id='", $this->_rids)."'";
                elseif(is_string($this->_rids) && strlen($this->_rids)>0)       
                        $where = $this->_pkey."='".implode("' OR id='", explode(",",$this->_rids))."'";

                $order = $this->_order ? $this->_order : null;

                $at = new adminTable($this->_registry, $this->_table);
                $at->setForeignKeys($this->_fkeys);
                $at->setSpecialFields($this->_sfields);
                $results = $this->_registry->db->autoSelect($head_fields, $this->_table, $where, $order);
                foreach($results as $r) { 
                        if($tot_fk) $r = $at->parseForeignKeys($r);
                        if($tot_sf) $r = $at->parseSpecialFields($r, array("show_pwd"=>true, "mailto"=>false));
                        $data[] = $r;
                }

                return $data;

        }

        private function getHeadFields($table_structure) {
                
                if($this->_head && is_string($this->_fields) && preg_match("#\*#", $this->_fields)) {
                        preg_match("#\* -\((.*)\)#", $this->_fields, $matches);
                        $excluded_fields = isset($matches[1]) ? explode(",",$matches[1]):array();
                        $head_fields = array();
                        foreach($table_structure['fields'] as $field=>$info) {
                                if(!in_array($field, $excluded_fields)) $head_fields[] = $field;
                        }
                }
                elseif(is_string($this->_fields)) $head_fields = explode(",",$this->_fields);
                elseif(is_array($this->_fields)) $head_fields = $this->_fields;

                return $head_fields;

        }

        private function encloseField($field) {

                return $this->_fe.$field.$this->_fe;

        }

}

?>