<?php
namespace Think\Upload\Driver\Qiniu;

class QiniuStorage
{

    public $QINIU_RSF_HOST = 'http://rsf.qbox.me';
    public $QINIU_RS_HOST  = 'http://rs.qbox.me';
    public $QINIU_UP_HOST  = 'http://up.qiniu.com';
    public $timeout        = '';

    public function __construct($config)
    {
        $this->sk      = $config['secretKey'];
        $this->ak      = $config['accessKey'];
        $this->domain  = $config['domain'];
        $this->bucket  = $config['bucket'];
        $this->timeout = isset($config['timeout']) ? $config['timeout'] : 3600;
    }

    public static function sign($sk, $ak, $data)
    {
        $sign = hash_hmac('sha1', $data, $sk, true);
        return $ak . ':' . self::qiniuEncode($sign);
    }

    public static function signWithData($sk, $ak, $data)
    {
        $data = self::qiniuEncode($data);
        return self::sign($sk, $ak, $data) . ':' . $data;
    }

    public function accessToken($url, $body = '')
    {
        $parsed_url = parse_url($url);
        $path       = $parsed_url['path'];
        $access     = $path;
        if (isset($parsed_url['query'])) {
            $access .= "?" . $parsed_url['query'];
        }
        $access .= "\n";

        if ($body) {
            $access .= $body;
        }
        return self::sign($this->sk, $this->ak, $access);
    }

    public function UploadToken($sk, $ak, $param)
    {
        $param['deadline'] = 0 == $param['Expires'] ? 3600 : $param['Expires'];
        $param['deadline'] += time();
        $data = array('scope' => $this->bucket, 'deadline' => $param['deadline']);
        if (!empty($param['CallbackUrl'])) {
            $data['callbackUrl'] = $param['CallbackUrl'];
        }
        if (!empty($param['CallbackBody'])) {
            $data['callbackBody'] = $param['CallbackBody'];
        }
        if (!empty($param['ReturnUrl'])) {
            $data['returnUrl'] = $param['ReturnUrl'];
        }
        if (!empty($param['ReturnBody'])) {
            $data['returnBody'] = $param['ReturnBody'];
        }
        if (!empty($param['AsyncOps'])) {
            $data['asyncOps'] = $param['AsyncOps'];
        }
        if (!empty($param['EndUser'])) {
            $data['endUser'] = $param['EndUser'];
        }
        $data = json_encode($data);
        return self::SignWithData($sk, $ak, $data);
    }

    public function upload($config, $file)
    {
        $uploadToken = $this->UploadToken($this->sk, $this->ak, $config);

        $url          = "{$this->QINIU_UP_HOST}";
        $mimeBoundary = md5(microtime());
        $header       = array('Content-Type' => 'multipart/form-data;boundary=' . $mimeBoundary);
        $data         = array();

        $fields = array(
            'token' => $uploadToken,
            'key'   => $config['saveName'] ?: $file['fileName'],
        );

        if (is_array($config['custom_fields']) && array() !== $config['custom_fields']) {
            $fields = array_merge($fields, $config['custom_fields']);
        }

        foreach ($fields as $name => $val) {
            array_push($data, '--' . $mimeBoundary);
            array_push($data, "Content-Disposition: form-data; name=\"$name\"");
            array_push($data, '');
            array_push($data, $val);
        }

        //文件
        array_push($data, '--' . $mimeBoundary);
        $name     = $file['name'];
        $fileName = $file['fileName'];
        $fileBody = $file['fileBody'];
        $fileName = self::qiniuEscapequotes($fileName);
        array_push($data, "Content-Disposition: form-data; name=\"$name\"; filename=\"$fileName\"");
        array_push($data, 'Content-Type: application/octet-stream');
        array_push($data, '');
        array_push($data, $fileBody);

        array_push($data, '--' . $mimeBoundary . '--');
        array_push($data, '');

        $body     = implode("\r\n", $data);
        $response = $this->request($url, 'POST', $header, $body);
        return $response;
    }

    public function dealWithType($key, $type)
    {
        $param = $this->buildUrlParam();
        $url   = '';

        switch ($type) {
            case 'img':
                $url = $this->downLink($key);
                if ($param['imageInfo']) {
                    $url .= '?imageInfo';
                } else if ($param['exif']) {
                    $url .= '?exif';
                } else if ($param['imageView']) {
                    $url .= '?imageView/' . $param['mode'];
                    if ($param['w']) {
                        $url .= "/w/{$param['w']}";
                    }

                    if ($param['h']) {
                        $url .= "/h/{$param['h']}";
                    }

                    if ($param['q']) {
                        $url .= "/q/{$param['q']}";
                    }

                    if ($param['format']) {
                        $url .= "/format/{$param['format']}";
                    }

                }
                break;
            case 'video': //TODO 视频处理
            case 'doc':
                $url = $this->downLink($key);
                $url .= '?md2html';
                if (isset($param['mode'])) {
                    $url .= '/' . (int) $param['mode'];
                }

                if ($param['cssurl']) {
                    $url .= '/' . self::qiniuEncode($param['cssurl']);
                }

                break;

        }
        return $url;
    }

    public function buildUrlParam()
    {
        return $_REQUEST;
    }

    //获取某个路径下的文件列表
    public function getList($query = array(), $path = '')
    {
        $query       = array_merge(array('bucket' => $this->bucket), $query);
        $url         = "{$this->QINIU_RSF_HOST}/list?" . http_build_query($query);
        $accessToken = $this->accessToken($url);
        $response    = $this->request($url, 'POST', array('Authorization' => "QBox $accessToken"));
        return $response;
    }

    //获取某个文件的信息
    public function info($key)
    {
        $key         = trim($key);
        $url         = "{$this->QINIU_RS_HOST}/stat/" . self::qiniuEncode("{$this->bucket}:{$key}");
        $accessToken = $this->accessToken($url);
        $response    = $this->request($url, 'POST', array(
            'Authorization' => "QBox $accessToken",
        ));
        return $response;
    }

    //获取文件下载资源链接
    public function downLink($key)
    {
        $key = urlencode($key);
        $key = self::qiniuEscapequotes($key);
        $url = "http://{$this->domain}/{$key}";
        return $url;
    }

    //重命名单个文件
    public function rename($file, $new_file)
    {
        $key = trim($file);
        $url = "{$this->QINIU_RS_HOST}/move/" . self::qiniuEncode("{$this->bucket}:{$key}") . '/' . self::qiniuEncode("{$this->bucket}:{$new_file}");
        trace($url);
        $accessToken = $this->accessToken($url);
        $response    = $this->request($url, 'POST', array('Authorization' => "QBox $accessToken"));
        return $response;
    }

    //删除单个文件
    public function del($file)
    {
        $key         = trim($file);
        $url         = "{$this->QINIU_RS_HOST}/delete/" . self::qiniuEncode("{$this->bucket}:{$key}");
        $accessToken = $this->accessToken($url);
        $response    = $this->request($url, 'POST', array('Authorization' => "QBox $accessToken"));
        return $response;
    }

    //批量删除文件
    public function delBatch($files)
    {
        $url = $this->QINIU_RS_HOST . '/batch';
        $ops = array();
        foreach ($files as $file) {
            $ops[] = "/delete/" . self::qiniuEncode("{$this->bucket}:{$file}");
        }
        $params = 'op=' . implode('&op=', $ops);
        $url .= '?' . $params;
        trace($url);
        $accessToken = $this->accessToken($url);
        $response    = $this->request($url, 'POST', array('Authorization' => "QBox $accessToken"));
        return $response;
    }

    public static function qiniuEncode($str)
    {
// URLSafeBase64Encode
        $find    = array('+', '/');
        $replace = array('-', '_');
        return str_replace($find, $replace, base64_encode($str));
    }

    public static function qiniuEscapequotes($str)
    {
        $find    = array("\\", "\"");
        $replace = array("\\\\", "\\\"");
        return str_replace($find, $replace, $str);
    }

    /**
     * 请求云服务器
     * @param  string   $path    请求的PATH
     * @param  string   $method  请求方法
     * @param  array    $headers 请求header
     * @param  resource $body    上传文件资源
     * @return boolean
     */
    private function request($path, $method, $headers = null, $body = null)
    {
        $ch = curl_init($path);

        $_headers = array('Expect:');
        if (!is_null($headers) && is_array($headers)) {
            foreach ($headers as $k => $v) {
                array_push($_headers, "{$k}: {$v}");
            }
        }

        $length = 0;
        $date   = gmdate('D, d M Y H:i:s \G\M\T');

        if (!is_null($body)) {
            if (is_resource($body)) {
                fseek($body, 0, SEEK_END);
                $length = ftell($body);
                fseek($body, 0);

                array_push($_headers, "Content-Length: {$length}");
                curl_setopt($ch, CURLOPT_INFILE, $body);
                curl_setopt($ch, CURLOPT_INFILESIZE, $length);
            } else {
                $length = @strlen($body);
                array_push($_headers, "Content-Length: {$length}");
                curl_setopt($ch, CURLOPT_POSTFIELDS, $body);
            }
        } else {
            array_push($_headers, "Content-Length: {$length}");
        }

        // array_push($_headers, 'Authorization: ' . $this->sign($method, $uri, $date, $length));
        array_push($_headers, "Date: {$date}");

        curl_setopt($ch, CURLOPT_HTTPHEADER, $_headers);
        curl_setopt($ch, CURLOPT_TIMEOUT, $this->timeout);
        curl_setopt($ch, CURLOPT_HEADER, 1);
        curl_setopt($ch, CURLOPT_RETURNTRANSFER, 1);
        curl_setopt($ch, CURLOPT_FOLLOWLOCATION, 0);
        curl_setopt($ch, CURLOPT_CUSTOMREQUEST, $method);

        if ('PUT' == $method || 'POST' == $method) {
            curl_setopt($ch, CURLOPT_POST, 1);
        } else {
            curl_setopt($ch, CURLOPT_POST, 0);
        }

        if ('HEAD' == $method) {
            curl_setopt($ch, CURLOPT_NOBODY, true);
        }

        $response = curl_exec($ch);
        $status   = curl_getinfo($ch, CURLINFO_HTTP_CODE);
        curl_close($ch);
        list($header, $body) = explode("\r\n\r\n", $response, 2);
        if (200 == $status) {
            if ('GET' == $method) {
                return $body;
            } else {
                return $this->response($response);
            }
        } else {
            $this->error($header, $body);
            return false;
        }
    }

    /**
     * 获取响应数据
     * @param  string $text 响应头字符串
     * @return array        响应数据列表
     */
    private function response($text)
    {
        $headers = explode(PHP_EOL, $text);
        $items   = array();
        foreach ($headers as $header) {
            $header = trim($header);
            if (strpos($header, '{') !== false) {
                $items = json_decode($header, 1);
                break;
            }
        }
        return $items;
    }

    /**
     * 获取请求错误信息
     * @param  string $header 请求返回头信息
     */
    private function error($header, $body)
    {
        list($status, $stash)     = explode("\r\n", $header, 2);
        list($v, $code, $message) = explode(" ", $status, 3);
        $message                  = is_null($message) ? 'File Not Found' : "[{$status}]:{$message}]";
        $this->error              = $message;
        $this->errorStr           = json_decode($body, 1);
        $this->errorStr           = $this->errorStr['error'];
    }
}
