csv.inc

<?php
class CSV
{

    var $filepath='';

    var $_dir='';

    var $_filename='';

    var $deliminator = ',';

    var $field_headers=false;

    var $values='';

    var $line_count=0;


    function CSV($filepath='')
    {
        if ($filepath) $this->setFilepath($filepath);

    }//end constructor


    function setFieldHeaders(&$headers)
    {
        $this->field_headers = &$headers;

    }//end setFieldHeaders()


    function setValues(&$values)
    {
        $this->values = &$values;

    }//end setValues()


    function setDeliminator($deliminator)
    {
        $this->deliminator = $deliminator;

    }//end setDeliminator()


    function setFilename($filename='')
    {
        $this->_filename = $filename;

    }//end setFilename()


    function setFilepath($filepath='')
    {
        if ($filepath) {
            // full path to this file
            $this->filepath = $filepath;

            // directory of this file
            $this->_dir = dirname($filepath);
            if (!file_exists($this->_dir)) mkdir($this->_dir, 0775);

            // name of file without full path
            $this->setFilename(basename($filepath));

            if (!file_exists($this->_dir)) {
                trigger_error('CSV directory "'.$this->_dir.'" does not exist!', E_USER_WARNING);
                return;
            }
        }

    }//end setFilepath()


    function setHeader($header='')
    {
        if ($header) {
            // set custom headers
            header($header);
        } else {
            header('Content-Type: application/csv');
            header('Content-Disposition: attachment; filename='.$this->_filename.';');
            
            //This is known that Internet Explorer has problems with SSL connection (https).
            //http://support.microsoft.com/default.aspx?scid=kb;en-us;812935
            //http://support.microsoft.com/default.aspx?scid=kb;en-us;316431
            // do not send no cache header in https for IE

            // get protocol information
            $url_info = parse_url(current_url());
            $protocol = (isset($url_info['scheme'])) ? $url_info['scheme'] : NULL;
            $using_ie6 = strpos($_SERVER['HTTP_USER_AGENT'], 'MSIE 6.') !== FALSE;

            if (!is_null($protocol) && $protocol == 'https' && $using_ie6) {
                header('Cache-Control: private, max-age=0, must-revalidate');
                header('Pragma: private');
                header('Expires: '.gmdate('D, d M Y H:i:s', time()-3600).' GMT');
    
            }

        }

    }//end setHeader()


    function export($ordered=false)
    {
        if (!$this->filepath) {
            // no file to write to.....lets send the header and then the file
            $this->setHeader();
            $fp = null;
        } else {
            if (!$fp = fopen($this->filepath,'w')) {
                trigger_error('Unable to open file "'.$this->filepath.'" for writing', E_USER_WARNING);
                return;
            }
        }

        if ($this->field_headers) {
            $line = $this->formatLine($this->field_headers, $this->deliminator);
            if (isset($fp)) {
                fputs($fp, $line."\n");
            } else {
                echo $line."\n";
            }
        }

        $this->printValues($ordered, $fp);
        
        if (isset($fp)) fclose($fp);

    }//end export()

    private function printValues($ordered=false, $fp=null)
    {
        foreach ($this->values as $value) {
            // if field headers are there, use the ordered version of the
            // format line method to export
            if ($this->field_headers && $ordered) {
                $line = $this->formatOrderedLine($value, $this->deliminator);
            } else {
                $line = $this->formatLine($value, $this->deliminator);
            }
            if (isset($fp) && $fp !== null) {
                fputs($fp, $line."\n");
            } else {
                echo $line."\n";
            }
        }
    }

    function exportValues($ordered=false)
    {
        if ($this->filepath) {
            if (!$fp = fopen($this->filepath,'a')) {
                trigger_error('Unable to open file "'.$this->filepath.'" for writing', E_USER_WARNING);
                return;
            }
        } else {
            $fp = null;
        }
        $this->printValues($ordered, $fp);
        
        if (isset($fp)) fclose($fp);
    }

    function import($start_at=1, $lines=0)
    {
        // can only import from a file
        if (!$this->filepath) return;

        // check to see if the file exsts
        if (!is_file($this->filepath)) return;

        //set auto_detect_line_endings to detect weird old mac line endings and stuff
        $line_ending_cfg = ini_set('auto_detect_line_endings', 1);
        
        if (!$fp = fopen($this->filepath,'r')) {
            trigger_error('Unable to open file "'.$this->filepath.'" for reading', E_USER_WARNING);
            return;
        }

        $start_at = (int)$start_at;
        $lines    = (int)$lines;

        $line_count = 0;
        $lines_done = 0;

        while (($tmp = fgetcsv($fp, 65535, $this->deliminator)) != null) {
            $line_count++;
            if ($line_count >= $start_at) {
                $this->values[] = $tmp;
                $lines_done++;
                if ($lines && ($lines_done >= $lines)) break;
            }
        }

        fclose($fp);
        
        if (!$line_ending_cfg) {
            ini_set('auto_detect_line_endings', 0);
        }

    }//end import()


    function formatLine(&$array, $deliminator=',')
    {
        $line = '';
        foreach ($array as $val) {
            if (is_array($val)) {
                $line .= '"'.$this->formatLine($val,$deliminator).'"'.$deliminator;
            } else {
                if (preg_match("/[".preg_quote($deliminator, '/') . "\"\n\r]/", $val)) {
                    $val = '"'.str_replace('"', '""', $val).'"';
                }
                $line .= $val.$deliminator;
            }
        }

        // strip the last deliminator
        return $line = substr($line, 0, (strlen($deliminator) * -1));

    }//end formatLine()


    function formatOrderedLine(&$array, $deliminator=',')
    {
        if (!$this->field_headers) return '';
        $line_elements = Array();

        foreach($this->field_headers as $key => $val) {
            $this_val = '';
            if (isset($array[$key])) {
                if (is_array($array[$key])) {
                    $this_val = '"'.$this->formatLine($array[$key],$deliminator).'"';
                } else {
                    $this_val = $array[$key];
                }

                if (preg_match("/[".preg_quote($deliminator, '/') . "\"\n\r]/", $this_val)) {
                    $this_val = '"'.str_replace('"', '""', $this_val).'"';
                }
            }

            $line_elements[] = $this_val;
        }

        return implode($deliminator, $line_elements);

    }//end formatLine()


    function countLines($start_at=0)
    {
        // can only count from a file
        if (!$this->filepath) return;

        // check if the file even exists
        if (!is_file($this->filepath)) {
            $this->line_count = 0;
            return;
        }

        if (!$fp = fopen($this->filepath,'r')) {
            trigger_error('Unable to open file "'.$this->filepath.'" for counting', E_USER_WARNING);
            return;
        }

        $line_count = 0;

        while (($tmp = fgetcsv($fp, 65535, $this->deliminator)) != null) {
            $line_count++;
        }

        $this->line_count = $line_count - $start_at;

        fclose($fp);

    }//end countLines()


    function isUnique($string)
    {
        // can only import from a file
        if (!$this->filepath) return true;

        // check if the file even exists
        if (!is_file($this->filepath)) return true;

        if (!$fp = fopen($this->filepath,'r')) {
            trigger_error('Unable to open file "'.$this->filepath.'" for reading', E_USER_WARNING);
            return true;
        }

        while (($tmp = fgetcsv($fp, 65535, $this->deliminator)) != null) {
            if (in_array($string, $tmp)) return false;
        }

        fclose($fp);

        return true;

    }//end isUnique()


}//end class

?>