FTP上传类

FTP上传类

<?php

namespace ftp;

/*$config = array(
        'host'     => '127.0.0.1', //服务器
        'port'     => 21, //端口
        'timeout'  => 90, //超时时间
        'username' => 'test', //用户名
        'password' => 'test', //密码
        'is_pasv'  => true,
        'domain'  => 'img.test.com' ,
    );

try {
    $ftp = new Ftp($config);
    $ftp->checkRootPath("/demo");
    $ftp->checkSavePath("abc");
    $ftp->upload('formfilename');
    var_dump($ftp->get_upload_info());
} catch (FtpException $e) {
    echo $e->getMessage();
    echo $e->getCode();
}*/

/**
 * 错误异常异常代码编号
 * 21000,链接登录FTP失败
 * 21001，上传根目录不存在
 * 21002，目录创建失败
 * 21003，文件后缀不可上传
 * 21004,上传失败
 */
class FtpException
{
}

class Ftp
{
    /**
     * 上传文件根目录
     * @var string
     */
    private $rootPath;

    private $savepath = '';
    private $filename = '';

    /**
     * FTP连接
     * @var resource
     */
    private $link;
    // 文件命名规则 (md5,timestamp,sha1,time);
    private $config = array(
        'host' => '', //服务器
        'port' => 21, //端口
        'timeout' => 60, //超时时间
        'username' => '', //用户名
        'password' => '', //密码
        'is_pasv' => false,//开启被动模式
        'filename' => 'md5',
        'allow_file_extension' => array('gif', 'png', 'jpg', 'jpeg', 'doc', 'excel'),
        'domain' => 'img.test.com',
    );

    /**
     * 构造函数，用于设置上传根路径
     * @param array $config FTP配置
     */
    public function __construct($config)
    {
        /* 默认FTP配置 */
        $this->config = array_merge($this->config, $config);

        /* 登录FTP服务器 */
        if (!$this->login()) {
            throw new FtpException($this->error, 21000);
        }
    }

    /**
     * 检测上传根目录
     * @param string $rootpath 根目录
     * @return boolean true-检测通过，false-检测失败,抛出异常
     */
    public function checkRootPath($rootpath)
    {
        /* 设置根目录 */
        $this->rootPath = @ftp_pwd($this->link) . ltrim($rootpath, '/');

        if (!@ftp_chdir($this->link, $this->rootPath)) {
            throw new FtpException('上传根目录不存在！(' . $this->rootPath . ')', 21001);
        }
        return true;
    }

    /**
     * 设置上传目录
     * @param  string $savepath 上传目录
     * @return boolean          检测结果，true-通过，false-失败
     */
    public function checkSavePath($savepath)
    {
        $this->savepath = $savepath = "/" . ltrim($savepath, '/');
        /* 检测并创建目录 */
        if (!$this->mkdir($savepath)) {
            return false;
        } else {
            //TODO:检测目录是否可写
            return true;
        }
    }

    /**
     * 保存指定文件
     * @param  array $file 保存的文件信息
     * @param  String $user_defined_filename 自定义文件名
     * @return boolean          保存状态，true-成功，false-失败
     */
    public function upload($file, $user_defined_filename = '')
    {
        if (empty($user_defined_filename)) {
            $filename = $this->filenamerule();
        } else {
            $filename = $user_defined_filename;
        }
        $uploadfile = $_FILES[$file];
        $file_ext = $this->_get_file_extension($uploadfile['name']);
        $this->filename = $filename . "." . $file_ext;
        $fileurl = $this->rootPath . $this->savepath . "/" . $this->filename;
        /* 移动文件 */
        if (!ftp_put($this->link, $fileurl, $uploadfile['tmp_name'], FTP_BINARY)) {
            throw new FtpException("{$filename}.{$file_ext} 上传失败", 21004);
        }
        return true;
    }

    public function get_upload_info()
    {
        @extract($this->config);
        $result['domain_all_path'] = "//" . $domain . $this->rootPath . $this->savepath . "/" . $this->filename;
        $result['all_path'] = $this->rootPath . $this->savepath . "/" . $this->filename;
        $result['save_path_file'] = $this->savepath . "/" . $this->filename;
        return $result;
    }

    private function filenamerule()
    {
        @extract($this->config);
        switch ($filename) {
            case 'sha1':
                $filename = sha1(microtime());
                break;
            case 'md5':
                $filename = md5(microtime());
                break;
            case 'time':
                $filename = date('YmdHis');
                break;
            case 'timestamp':
                $filename = time();
                break;
        }
        return $filename;
    }

    private function _get_file_extension($file)
    {
        @extract($this->config);
        $file_ext = pathinfo($file, PATHINFO_EXTENSION);
        if (in_array($file_ext, $allow_file_extension)) {
            return $file_ext;
        } else {
            throw new FtpException(" {$file_ext} 不可上传", 21003);
        }

    }

    /**
     * 创建目录
     * @param  string $savepath 要创建的目录
     * @return boolean          创建状态，true-成功，false-失败
     */
    public function mkdir($savepath)
    {
        $dir = $this->rootPath . $savepath;
        if (@ftp_chdir($this->link, $dir)) {
            return true;
        }

        if (@ftp_mkdir($this->link, $dir)) {
            return true;
        } elseif ($this->mkdir(dirname($savepath)) && @ftp_mkdir($this->link, $dir)) {
            return true;
        } else {
            throw new FtpException("目录 {$savepath} 创建失败,可能设置写入权限和权限组不对！", 21002);
        }
    }

    /**
     * 登录到FTP服务器
     * @return boolean true-登录成功，false-登录失败
     */
    private function login()
    {
        @extract($this->config);
        $this->link = @ftp_connect($host, $port, $timeout);
        if ($this->link) {
            if (@ftp_login($this->link, $username, $password)) {
                if ($is_pasv) {
                    @ftp_pasv($this->link, true);
                }
                return true;
            } else {
                $this->error = "无法登录到FTP服务器：username - {$username}";
            }
        } else {
            $this->error = "无法连接到FTP服务器：{$host}";
        }
        return false;
    }

    /**
     * 析构方法，用于断开当前FTP连接
     */
    public function __destruct()
    {
        @ftp_close($this->link);
    }

}