ChWsbExport.php

Go to the documentation of this file.

<?php
 
ch_import('ChWsbExportQuery');
 
class ChWsbExport
{
    protected $_aSystem = array (); 
    protected $_oQuery = null;
    protected $_aTables = array(); 
    protected $_sFilesBaseDir = ''; 
    protected $_aTablesWithFiles = array(); 
 
    protected function __construct($aSystem)
    {
        $this->_aSystem = $aSystem;
        $this->_oQuery = new ChWsbExportQuery($this->_aSystem);
    }
 
    static public function getObjectInstance($sObject)
    {
        if (isset($GLOBALS['chWsbClasses']['ChWsbExport!'.$sObject]))
            return $GLOBALS['chWsbClasses']['ChWsbExport!'.$sObject];
 
        $aSystems =& self::getSystems ();
        if (!$aSystems || !isset($aSystems[$sObject]))
            return false;
 
        $aObject = $aSystems[$sObject];
 
        if (!($sClass = $aObject['class_name']))
            return false;
 
        if (!empty($aObject['class_file']))
            require_once(CH_DIRECTORY_PATH_ROOT . $aObject['class_file']);
        else
            ch_import($sClass);
 
        $o = new $sClass($aObject);
 
        return ($GLOBALS['chWsbClasses']['ChWsbExport!'.$sObject] = $o);
    }
 
    static public function & getSystems () {
        if (!isset($GLOBALS['ch_dol_export_systems']))
            $GLOBALS['ch_dol_export_systems'] = ChWsbExportQuery::getAllActiveSystemsFromCache ();
        return $GLOBALS['ch_dol_export_systems'];
    }
 
    static public function generateAllExports ($iProfileId)
    {
        $iProfileId = (int)$iProfileId;
        if (!($aProfile = getProfileInfo($iProfileId)))
            return "Profile($iProfileId) doesn't exist";
 
        $aSystems =& self::getSystems ();
        $aExports = array ();
        foreach ($aSystems as $sSystem => $aSystem) {
            if (!($o = self::getObjectInstance($sSystem)))
                continue;
 
            $a = $o->export($iProfileId);
            if (!empty($a['sql']) || !empty($a['files']))
                $aExports[$sSystem] = $a;
        }
 
        $sFileName = self::createZip($iProfileId, $aExports);
        if (!$sFileName)
            return 'Export zip file creation failed';
 
 
        if (!self::sendEmailNotification($aProfile, $sFileName))
            return "Send notification email to user($iProfileId) failed";
 
        return true;
    }
 
    static public function sendEmailNotification($aProfile, $sFilename)
    {
        $oEmailTemplate = new ChWsbEmailTemplates();
        $aTemplate = $oEmailTemplate->getTemplate('t_ExportReady', $aProfile['ID']);
        $aTemplateVars = array (
            'FileUrl' => CH_WSB_URL_CACHE_PUBLIC . $sFilename,
        );
        return sendMail($aProfile['Email'], $aTemplate['Subject'], $aTemplate['Body'], $aProfile['ID'], $aTemplateVars);
    }
 
    static public function createZip($iProfileId, $aExports)
    {
        if (!class_exists('ZipArchive'))
            return false;
 
        $sFileName = 'export-' . $iProfileId . '-' . $GLOBALS['site']['ver'] . '.' . $GLOBALS['site']['build'] . '-' . genRndPwd(8, false) . '.zip';
        $sFilePath = CH_DIRECTORY_PATH_CACHE_PUBLIC . $sFileName;
 
        $oZip = new ZipArchive();
        if ($oZip->open($sFilePath, ZipArchive::CREATE)!==TRUE)
            return false;
 
        // collect data
 
        $sSqlDump = "-- Cheetah user data export\n";
        $sSqlDump .= "-- Profile ID: $iProfileId\n";
        $sSqlDump .= "-- Profile Username: " . getUsername($iProfileId) . "\n";
        $sSqlDump .= "-- Cheetah Version: " . $GLOBALS['site']['ver'] . '.' . $GLOBALS['site']['build'] . "\n";
        foreach ($aExports as $sSystem => $a) {
            // sql
            if (!empty($a['sql']))
                $sSqlDump .= "\n\n-- " . $sSystem . "\n\n" . $a['sql'];
 
            // files
            if (!empty($a['files']))
                chdir(CH_DIRECTORY_PATH_ROOT);
                foreach ($a['files'] as $sFile)
                    $oZip->addGlob($sFile);
        }
 
        // add DB dump
        $oZip->addFromString('/dump.sql', $sSqlDump);
 
        $oZip->close();
 
        return $sFileName;
    }
 
    public function export ($iProfileId)
    {
        return array(
            'sql' => $this->exportSQL($iProfileId),
            'files' => $this->exportFiles($iProfileId),
        );
    }
 
    public function exportFiles($iProfileId)
    {
        $a = array();
        foreach ($this->_aTablesWithFiles as $sTableName => $aFields)
            $a = array_merge($a, $this->_getFiles($sTableName, $aFields, $iProfileId));
        return $a;
    }
 
    public function exportSQL($iProfileId)
    {
        $s = '';
        foreach ($this->_aTables as $sTableName => $mixedCond)
            $s .= $this->_getRows($sTableName, $mixedCond, $iProfileId);
        return $s;
    }
 
    protected function _getFiles($sTableName, $aFields, $iProfileId)
    {
        $mixedCond = $this->_aTables[$sTableName];
        $sQuery = $this->_getQuery($sTableName, $mixedCond, $iProfileId);
        $oStmt = $this->_oQuery->res($sQuery);
        return $this->_getFilesFromStmt($sTableName, $oStmt, $aFields);
    }
 
    protected function _getRows($sTableName, $mixedCond, $iProfileId)
    {
        $sQuery = $this->_getQuery($sTableName, $mixedCond, $iProfileId);
        $oStmt = $this->_oQuery->res($sQuery);
        return $this->_getRowsFromStmt($sTableName, $oStmt);
    }
 
    protected function _getQuery($sTableName, $mixedCond, $iProfileId)
    {
        if (is_string($mixedCond)) {
            $sWhere = str_replace('{profile_id}', $iProfileId, $mixedCond);
            $sQuery = "SELECT * FROM `$sTableName` WHERE $sWhere";
        }
        elseif (is_array($mixedCond) && isset($mixedCond['query'])) {
            $sQuery = str_replace('{profile_id}', $iProfileId, $mixedCond['query']);
        }
        return $sQuery;
    }
 
    protected function _getFilePath($sTableName, $sField, $sFileName, $sPrefix, $sExt)
    {
        return $this->_sFilesBaseDir . (is_string($sPrefix) ? $sPrefix : '') . $sFileName . $sExt;
    }
 
    protected function _getFilesFromStmt($sTableName, $oStmt, $aFields)
    {
        if (!$oStmt->rowCount())
            return array();
 
        $aFiles = array();
        while ($r = $oStmt->fetch(PDO::FETCH_ASSOC)) {
            if(is_a($aFields, 'ChWsbExportFiles'))
                $aFields->perform($r, $aFiles);
            else
                foreach ($aFields as $sField => $aPrefix2Ext) {
                    foreach ($aPrefix2Ext as $sPrefix => $sExt) {
                        $sPath = $this->_getFilePath($sTableName, $sField, $r[$sField], $sPrefix, $sExt);
                        if (file_exists($sPath))
                            $aFiles[] = $sPath;
                    }
                }
        }
 
        return $aFiles;
    }
 
    protected function _getRowsFromStmt($sTableName, $oStmt)
    {
        if (!$oStmt->rowCount())
            return '';
 
        $s .= "INSERT INTO `{$sTableName}` VALUES\n";
        while ($r = $oStmt->fetch(PDO::FETCH_NUM)) {
            $s .= "(";
            for ($j = 0; $j < count($r); $j++ ) {
                if (is_null($r[$j]))
                    $s .= "NULL, ";
                else
                    $s .= $this->_oQuery->escape($r[$j], true) . ", ";
            }
            $s = trim($s, ', ');
            $s .= "),\n";
        }
        $s = trim($s, ",\n");
        $s .= ";\n\n";
        return $s;
    }
}
 
class ChWsbExportFiles
{
    protected $_sBaseDir;
 
    public function __construct($sBaseDir)
    {
        $this->_sBaseDir = $sBaseDir;
    }
 
    public function perform($aRow, &$aFiles) {}
}