<?php



/**

 * W3 CDN FTP Class

 */

if (!defined('W3TC')) {

    die();

}



define('W3TC_CDN_FTP_CONNECT_TIMEOUT', 30);



w3_require_once(W3TC_LIB_W3_DIR . '/Cdn/Base.php');



/**

 * Class W3_Cdn_Ftp

 */

class W3_Cdn_Ftp extends W3_Cdn_Base {

    /**

     * FTP resource

     *

     * @var resource

     */

    var $_ftp = null;



    /**

     * PHP5 Constructor

     *

     * @param array $config

     */

    function __construct($config = array()) {

        $config = array_merge(array(

            'host' => '',

            'user' => '',

            'pass' => '',

            'path' => '',

            'pasv' => false,

            'domain' => array(),

        ), $config);



        $host_port = explode(':', $config['host']);

        if (sizeof($host_port) == 2) {

            $config['host'] = $host_port[0];

            $config['port'] = $host_port[1];

        }



        parent::__construct($config);

    }



    /**

     * Connects to FTP server

     *

     * @param string $error

     * @return boolean

     */

    function _connect(&$error) {

        if (empty($this->_config['host'])) {

            $error = 'Empty host.';



            return false;

        }



        if (empty($this->_config['port'])) {

            $this->_config['port'] = 21;

        }



        $this->_set_error_handler();



        $this->_ftp = @ftp_connect($this->_config['host'], (int) $this->_config['port'], W3TC_CDN_FTP_CONNECT_TIMEOUT);



        if (!$this->_ftp) {

            $error = sprintf('Unable to connect to %s:%d (%s).', $this->_config['host'], $this->_config['port'], $this->_get_last_error());



            $this->_restore_error_handler();



            return false;

        }



        if (!@ftp_login($this->_ftp, $this->_config['user'], $this->_config['pass'])) {

            $error = sprintf('Incorrect login or password (%s).', $this->_get_last_error());



            $this->_restore_error_handler();

            $this->_disconnect();



            return false;

        }



        if (!@ftp_pasv($this->_ftp, $this->_config['pasv'])) {

            $error = sprintf('Unable to change mode to passive (%s).', $this->_get_last_error());



            $this->_restore_error_handler();

            $this->_disconnect();



            return false;

        }



        if (!empty($this->_config['path']) && !@ftp_chdir($this->_ftp, $this->_config['path'])) {

            $error = sprintf('Unable to change directory to: %s (%s).', $this->_config['path'], $this->_get_last_error());



            $this->_restore_error_handler();

            $this->_disconnect();



            return false;

        }



        $this->_restore_error_handler();



        return true;

    }



    /**

     * Disconnects from FTP server

     */

    function _disconnect() {

        @ftp_close($this->_ftp);

    }



    /**

     * Sends MDTM command

     *

     * @param string $remote_file

     * @param integer $mtime

     * @return boolean

     */

    function _mdtm($remote_file, $mtime) {

        $command = sprintf('MDTM %s %s', date('YmdHis', $mtime), $remote_file);



        return @ftp_raw($this->_ftp, $command);

    }



    /**

     * Uploads files to FTP

     *

     * @param array $files

     * @param array $results

     * @param boolean $force_rewrite

     * @return boolean

     */

    function upload($files, &$results, $force_rewrite = false) {

        $error = null;



        if (!$this->_connect($error)) {

            $results = $this->_get_results($files, W3TC_CDN_RESULT_HALT, $error);



            return false;

        }



        $this->_set_error_handler();



        $home = @ftp_pwd($this->_ftp);



        if ($home === false) {

            $results = $this->_get_results($files, W3TC_CDN_RESULT_HALT, sprintf('Unable to get current directory (%s).', $this->_get_last_error()));



            $this->_restore_error_handler();

            $this->_disconnect();



            return false;

        }



        foreach ($files as $file) {

            $local_path = $file['local_path'];

            $remote_path = $file['remote_path'];



            if (!file_exists($local_path)) {

                $results[] = $this->_get_result($local_path, $remote_path, W3TC_CDN_RESULT_ERROR, 'Source file not found.');



                continue;

            }



            @ftp_chdir($this->_ftp, $home);



            $remote_dir = dirname($remote_path);

            $remote_dirs = preg_split('~\\/+~', $remote_dir);



            foreach ($remote_dirs as $dir) {

                if (!@ftp_chdir($this->_ftp, $dir)) {

                    if (!@ftp_mkdir($this->_ftp, $dir)) {

                        $results[] = $this->_get_result($local_path, $remote_path, W3TC_CDN_RESULT_ERROR, sprintf('Unable to create directory (%s).', $this->_get_last_error()));



                        continue 2;

                    }



                    if (!@ftp_chdir($this->_ftp, $dir)) {

                        $results[] = $this->_get_result($local_path, $remote_path, W3TC_CDN_RESULT_ERROR, sprintf('Unable to change directory (%s).', $this->_get_last_error()));



                        continue 2;

                    }

                }

            }



            $remote_file = basename($remote_path);

            $mtime = @filemtime($local_path);



            if (!$force_rewrite) {

                $size = @filesize($local_path);

                $ftp_size = @ftp_size($this->_ftp, $remote_file);

                $ftp_mtime = @ftp_mdtm($this->_ftp, $remote_file);



                if ($size === $ftp_size && $mtime === $ftp_mtime) {

                    $results[] = $this->_get_result($local_path, $remote_path, W3TC_CDN_RESULT_OK, 'File up-to-date.');



                    continue;

                }

            }



            $result = @ftp_put($this->_ftp, $remote_file, $local_path, FTP_BINARY);



            if ($result) {

                $this->_mdtm($remote_file, $mtime);



                $results[] = $this->_get_result($local_path, $remote_path, W3TC_CDN_RESULT_OK, 'OK');

            } else {

                $results[] = $this->_get_result($local_path, $remote_path, W3TC_CDN_RESULT_ERROR, sprintf('Unable to upload file (%s).', $this->_get_last_error()));

            }

        }



        $this->_restore_error_handler();

        $this->_disconnect();



        return !$this->_is_error($results);

    }



    /**

     * Deletes files from FTP

     *

     * @param array $files

     * @param array $results

     * @return boolean

     */

    function delete($files, &$results) {

        $error = null;



        if (!$this->_connect($error)) {

            $results = $this->_get_results($files, W3TC_CDN_RESULT_HALT, $error);



            return false;

        }



        $this->_set_error_handler();



        foreach ($files as $file) {

            $local_path = $file['local_path'];

            $remote_path = $file['remote_path'];



            $result = @ftp_delete($this->_ftp, $remote_path);



            if ($result) {

                $results[] = $this->_get_result($local_path, $remote_path, W3TC_CDN_RESULT_OK, 'OK');

            } else {

                $results[] = $this->_get_result($local_path, $remote_path, W3TC_CDN_RESULT_ERROR, sprintf('Unable to delete file (%s).', $this->_get_last_error()));

            }



            while (true) {

                $remote_path = dirname($remote_path);



                if ($remote_path == '.' || !@ftp_rmdir($this->_ftp, $remote_path)) {

                    break;

                }

            }

        }



        $this->_restore_error_handler();

        $this->_disconnect();



        return !$this->_is_error($results);

    }



    /**

     * Tests FTP server

     *

     * @param string $error

     * @return boolean

     */

    function test(&$error) {

        if (!parent::test($error)) {

            return false;

        }



        $rand = md5(time());

        $tmp_dir = 'test_dir_' . $rand;

        $tmp_file = 'test_file_' . $rand;

        $tmp_path = W3TC_CACHE_TMP_DIR . '/' . $tmp_file;



        if (!@file_put_contents($tmp_path, $rand)) {

            $error = sprintf('Unable to create file: %s.', $tmp_path);



            return false;

        }



        if (!$this->_connect($error)) {

            return false;

        }



        $this->_set_error_handler();



        if (!@ftp_mkdir($this->_ftp, $tmp_dir)) {

            $error = sprintf('Unable to make directory: %s (%s).', $tmp_dir, $this->_get_last_error());



            @unlink($tmp_path);



            $this->_restore_error_handler();

            $this->_disconnect();



            return false;

        }



        if (!@ftp_chdir($this->_ftp, $tmp_dir)) {

            $error = sprintf('Unable to change directory to: %s (%s).', $tmp_dir, $this->_get_last_error());



            @unlink($tmp_path);

            @ftp_rmdir($this->_ftp, $tmp_dir);



            $this->_restore_error_handler();

            $this->_disconnect();



            return false;

        }



        if (!@ftp_put($this->_ftp, $tmp_file, $tmp_path, FTP_BINARY)) {

            $error = sprintf('Unable to upload file: %s (%s).', $tmp_path, $this->_get_last_error());



            @unlink($tmp_path);

            @ftp_cdup($this->_ftp);

            @ftp_rmdir($this->_ftp, $tmp_dir);



            $this->_restore_error_handler();

            $this->_disconnect();



            return false;

        }



        @unlink($tmp_path);



        if (!@ftp_delete($this->_ftp, $tmp_file)) {

            $error = sprintf('Unable to delete file: %s (%s).', $tmp_path, $this->_get_last_error());



            @ftp_cdup($this->_ftp);

            @ftp_rmdir($this->_ftp, $tmp_dir);



            $this->_restore_error_handler();

            $this->_disconnect();



            return false;

        }



        @ftp_cdup($this->_ftp);



        if (!@ftp_rmdir($this->_ftp, $tmp_dir)) {

            $error = sprintf('Unable to remove directory: %s (%s).', $tmp_dir, $this->_get_last_error());



            $this->_restore_error_handler();

            $this->_disconnect();



            return false;

        }



        $this->_restore_error_handler();

        $this->_disconnect();



        return true;

    }



    /**

     * Returns array of CDN domains

     *

     * @return array

     */

    function get_domains() {

        if (!empty($this->_config['domain'])) {

            return (array) $this->_config['domain'];

        }



        return array();

    }



    /**

     * How and if headers should be set

     * @return string W3TC_CDN_HEADER_NONE, W3TC_CDN_HEADER_UPLOADABLE, W3TC_CDN_HEADER_MIRRORING

     */

    function headers_support() {

        return W3TC_CDN_HEADER_MIRRORING;

    }

}

