From ca12b60ec54e8a3ba99e205f6a52446c6f47cea0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=A0=81=E9=BE=8D?= <695798354@qq.com>
Date: Thu, 1 Dec 2022 06:13:48 +0000
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E4=B8=8B=E8=BD=BD=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6=E5=93=8D=E5=BA=94?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: 码龍 <695798354@qq.com>
---
 library/think/response/Download.php | 148 ++++++++++++++++++++++++++++
 1 file changed, 148 insertions(+)
 create mode 100644 library/think/response/Download.php

diff --git a/library/think/response/Download.php b/library/think/response/Download.php
new file mode 100644
index 00000000..babee6ce
--- /dev/null
+++ b/library/think/response/Download.php
@@ -0,0 +1,148 @@
+<?php
+// +----------------------------------------------------------------------
+// | ThinkPHP [ WE CAN DO IT JUST THINK ]
+// +----------------------------------------------------------------------
+// | Copyright (c) 2006~2018 http://thinkphp.cn All rights reserved.
+// +----------------------------------------------------------------------
+// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
+// +----------------------------------------------------------------------
+// | Author: liu21st <liu21st@gmail.com>
+// +----------------------------------------------------------------------
+
+namespace think\response;
+
+use think\Exception;
+use think\Response;
+
+class Download extends Response
+{
+    protected $expire = 360;
+    protected $name;
+    protected $mimeType;
+    protected $isContent = false;
+    protected $openinBrowser = false;
+    /**
+     * 处理数据
+     * @access protected
+     * @param  mixed $data 要处理的数据
+     * @return mixed
+     * @throws \Exception
+     */
+    protected function output($data)
+    {
+        if (!$this->isContent && !is_file($data)) {
+            throw new Exception('file not exists:' . $data);
+        }
+
+        ob_end_clean();
+
+        if (!empty($this->name)) {
+            $name = $this->name;
+        } else {
+            $name = !$this->isContent ? pathinfo($data, PATHINFO_BASENAME) : '';
+        }
+
+        if ($this->isContent) {
+            $mimeType = $this->mimeType;
+            $size     = strlen($data);
+        } else {
+            $mimeType = $this->getMimeType($data);
+            $size     = filesize($data);
+        }
+
+        $this->header['Pragma']                    = 'public';
+        $this->header['Content-Type']              = $mimeType ?: 'application/octet-stream';
+        $this->header['Cache-control']             = 'max-age=' . $this->expire;
+        $this->header['Content-Disposition']       = $this->openinBrowser ? 'inline' : 'attachment; filename="' . $name . '"';
+        $this->header['Content-Length']            = $size;
+        $this->header['Content-Transfer-Encoding'] = 'binary';
+        $this->header['Expires']                   = gmdate("D, d M Y H:i:s", time() + $this->expire) . ' GMT';
+
+        $this->lastModified(gmdate('D, d M Y H:i:s', time()) . ' GMT');
+
+        $data = $this->isContent ? $data : file_get_contents($data);
+        return $data;
+    }
+
+    /**
+     * 设置是否为内容 必须配合mimeType方法使用
+     * @access public
+     * @param  bool $content
+     * @return $this
+     */
+    public function isContent($content = true)
+    {
+        $this->isContent = $content;
+        return $this;
+    }
+
+    /**
+     * 设置有效期
+     * @access public
+     * @param  integer $expire 有效期
+     * @return $this
+     */
+    public function expire($expire)
+    {
+        $this->expire = $expire;
+        return $this;
+    }
+
+    /**
+     * 设置文件类型
+     * @access public
+     * @param  string $filename 文件名
+     * @return $this
+     */
+    public function mimeType($mimeType)
+    {
+        $this->mimeType = $mimeType;
+        return $this;
+    }
+
+    /**
+     * 获取文件类型信息
+     * @access public
+     * @param  string $filename 文件名
+     * @return string
+     */
+    protected function getMimeType($filename)
+    {
+        if (!empty($this->mimeType)) {
+            return $this->mimeType;
+        }
+
+        $finfo = finfo_open(FILEINFO_MIME_TYPE);
+
+        return finfo_file($finfo, $filename);
+    }
+
+    /**
+     * 设置下载文件的显示名称
+     * @access public
+     * @param  string $filename 文件名
+     * @param  bool   $extension 后缀自动识别
+     * @return $this
+     */
+    public function name($filename, $extension = true)
+    {
+        $this->name = $filename;
+
+        if ($extension && false === strpos($filename, '.')) {
+            $this->name .= '.' . pathinfo($this->data, PATHINFO_EXTENSION);
+        }
+
+        return $this;
+    }
+
+    /**
+     * 设置是否在浏览器中显示文件
+     * @access public
+     * @param  bool  $openinBrowser 是否在浏览器中显示文件
+     * @return $this
+     */
+    public function openinBrowser($openinBrowser) {
+        $this->openinBrowser = $openinBrowser;
+        return $this;
+    }
+}
\ No newline at end of file
-- 
Gitee