export_csv.modules.php

Go to the documentation of this file.

<?php
/* Copyright (C) 2006-2013 Laurent Destailleur  <eldy@users.sourceforge.net>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <https://www.gnu.org/licenses/>.
 */

require_once DOL_DOCUMENT_ROOT.'/core/modules/export/modules_export.php';

// avoid timeout for big export
set_time_limit(0);

class ExportCsv extends ModeleExports
{
  public $id;

  public $label;

  public $extension;

  public $version = 'dolibarr';

  public $label_lib;

  public $version_lib;

  public $separator;

  public $handle; // Handle fichier


  public function __construct($db)
  {
    global $conf, $langs;
    $this->db = $db;

    $this->separator = ',';
    if (!empty($conf->global->EXPORT_CSV_SEPARATOR_TO_USE)) $this->separator = $conf->global->EXPORT_CSV_SEPARATOR_TO_USE;
    $this->escape = '"';
    $this->enclosure = '"';

    $this->id = 'csv'; // Same value then xxx in file name export_xxx.modules.php
    $this->label = 'CSV'; // Label of driver
    $this->desc = $langs->trans("CSVFormatDesc", $this->separator, $this->enclosure, $this->escape);
    $this->extension = 'csv'; // Extension for generated file by this driver
    $this->picto = 'mime/other'; // Picto
    $this->version = '1.32'; // Driver version

    // If driver use an external library, put its name here
    $this->label_lib = 'Dolibarr';
    $this->version_lib = DOL_VERSION;
  }

  public function getDriverId()
  {
    return $this->id;
  }

  public function getDriverLabel()
  {
    return $this->label;
  }

  public function getDriverDesc()
  {
    return $this->desc;
  }

  public function getDriverExtension()
  {
    return $this->extension;
  }

  public function getDriverVersion()
  {
    return $this->version;
  }

  public function getLibLabel()
  {
    return $this->label_lib;
  }

  public function getLibVersion()
  {
    return $this->version_lib;
  }


  // phpcs:disable PEAR.NamingConventions.ValidFunctionName.ScopeNotCamelCaps
  public function open_file($file, $outputlangs)
  {
    // phpcs:enable
    global $langs;

    dol_syslog("ExportCsv::open_file file=".$file);

    $ret = 1;

    $outputlangs->load("exports");
    $this->handle = fopen($file, "wt");
    if (!$this->handle)
    {
      $langs->load("errors");
      $this->error = $langs->trans("ErrorFailToCreateFile", $file);
      $ret = -1;
    }

    return $ret;
  }

  // phpcs:disable PEAR.NamingConventions.ValidFunctionName.ScopeNotCamelCaps
  public function write_header($outputlangs)
  {
    // phpcs:enable
    return 0;
  }


  // phpcs:disable PEAR.NamingConventions.ValidFunctionName.ScopeNotCamelCaps
  public function write_title($array_export_fields_label, $array_selected_sorted, $outputlangs, $array_types)
  {
    // phpcs:enable
    global $conf;

    if (!empty($conf->global->EXPORT_CSV_FORCE_CHARSET))
    {
      $outputlangs->charset_output = $conf->global->EXPORT_CSV_FORCE_CHARSET;
    } else {
      $outputlangs->charset_output = 'ISO-8859-1';
    }

    foreach ($array_selected_sorted as $code => $value)
    {
      $newvalue = $outputlangs->transnoentities($array_export_fields_label[$code]); // newvalue is now $outputlangs->charset_output encoded
      $newvalue = $this->csvClean($newvalue, $outputlangs->charset_output);

      fwrite($this->handle, $newvalue.$this->separator);
    }
    fwrite($this->handle, "\n");
    return 0;
  }


  // phpcs:disable PEAR.NamingConventions.ValidFunctionName.ScopeNotCamelCaps
  public function write_record($array_selected_sorted, $objp, $outputlangs, $array_types)
  {
    // phpcs:enable
    global $conf;

    if (!empty($conf->global->EXPORT_CSV_FORCE_CHARSET))
    {
      $outputlangs->charset_output = $conf->global->EXPORT_CSV_FORCE_CHARSET;
    } else {
      $outputlangs->charset_output = 'ISO-8859-1';
    }

    $this->col = 0;

    $reg = array();

    foreach ($array_selected_sorted as $code => $value)
    {
      if (strpos($code, ' as ') == 0) $alias = str_replace(array('.', '-', '(', ')'), '_', $code);
      else $alias = substr($code, strpos($code, ' as ') + 4);
      if (empty($alias)) dol_print_error('', 'Bad value for field with key='.$code.'. Try to redefine export.');

      $newvalue = $outputlangs->convToOutputCharset($objp->$alias); // objp->$alias must be utf8 encoded as any var in memory // newvalue is now $outputlangs->charset_output encoded
      $typefield = isset($array_types[$code]) ? $array_types[$code] : '';

      // Translation newvalue
      if (preg_match('/^\((.*)\)$/i', $newvalue, $reg)) $newvalue = $outputlangs->transnoentities($reg[1]);

      $newvalue = $this->csvClean($newvalue, $outputlangs->charset_output);

      if (preg_match('/^Select:/i', $typefield, $reg) && $typefield = substr($typefield, 7))
      {
        $array = unserialize($typefield);
        $array = $array['options'];
        $newvalue = $array[$newvalue];
      }

      fwrite($this->handle, $newvalue.$this->separator);
      $this->col++;
    }

    fwrite($this->handle, "\n");
    return 0;
  }

  // phpcs:disable PEAR.NamingConventions.ValidFunctionName.ScopeNotCamelCaps
  public function write_footer($outputlangs)
  {
    // phpcs:enable
    return 0;
  }

  // phpcs:disable PEAR.NamingConventions.ValidFunctionName.ScopeNotCamelCaps
  public function close_file()
  {
    // phpcs:enable
    fclose($this->handle);
    return 0;
  }


  public function csvClean($newvalue, $charset)
  {
    global $conf;
    $addquote = 0;

    // Rule Dolibarr: No HTML
      //print $charset.' '.$newvalue."\n";
      //$newvalue=dol_string_nohtmltag($newvalue,0,$charset);
      $newvalue = dol_htmlcleanlastbr($newvalue);
      //print $charset.' '.$newvalue."\n";

    // Rule 1 CSV: No CR, LF in cells (except if USE_STRICT_CSV_RULES is on, we can keep record as it is but we must add quotes)
    $oldvalue = $newvalue;
    $newvalue = str_replace("\r", '', $newvalue);
    $newvalue = str_replace("\n", '\n', $newvalue);
    if (!empty($conf->global->USE_STRICT_CSV_RULES) && $oldvalue != $newvalue)
    {
      // If strict use of CSV rules, we just add quote
      $newvalue = $oldvalue;
      $addquote = 1;
    }

    // Rule 2 CSV: If value contains ", we must escape with ", and add "
    if (preg_match('/"/', $newvalue))
    {
      $addquote = 1;
      $newvalue = str_replace('"', '""', $newvalue);
    }

    // Rule 3 CSV: If value contains separator, we must add "
    if (preg_match('/'.$this->separator.'/', $newvalue))
    {
      $addquote = 1;
    }

    return ($addquote ? '"' : '').$newvalue.($addquote ? '"' : '');
  }
}