File size: 9,216 Bytes
ca533c7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
<?php
class FileShare {
    private $db;
    private $settings;
    private $upload_dir;
    private $cleanup_interval = 3600; // 清理间隔:1小时
    
    public function __construct() {
        $this->settings = new Settings();
        $this->db = new SQLite3($this->settings->getSetting('db_path'));
        $this->upload_dir = __DIR__ . '/uploads/files/';
        
        // 确保上传目录存在并创建必要的保护文件
        $this->initializeUploadDirectory();
        
        // 检查是否需要清理过期文件
        $this->checkAndCleanup();
    }
    
    // 初始化上传目录和保护文件
    private function initializeUploadDirectory() {
        // 创建上传目录
        if (!is_dir($this->upload_dir)) {
            if (!mkdir($this->upload_dir, 0755, true)) {
                throw new Exception('无法创建上传目录,请检查权限');
            }
            
            // 创建 .htaccess 文件来保护上传目录
            $htaccess = $this->upload_dir . '.htaccess';
            if (!file_exists($htaccess)) {
                $htaccess_content = "Options -Indexes\n";
                $htaccess_content .= "DirectoryIndex 403.html\n";
                $htaccess_content .= "AddType text/plain .php\n";
                $htaccess_content .= "AddType text/plain .html\n";
                $htaccess_content .= "AddType text/plain .htm\n";
                $htaccess_content .= "AddType text/plain .htaccess\n";
                file_put_contents($htaccess, $htaccess_content);
            }
            
            // 创建 index.html 文件来防止目录列表
            $index_html = $this->upload_dir . 'index.html';
            if (!file_exists($index_html)) {
                file_put_contents($index_html, '<!DOCTYPE html><html><head><title>403 Forbidden</title></head><body><h1>403 Forbidden</h1></body></html>');
            }
            
            // 创建 403.html 文件
            $forbidden_page = $this->upload_dir . '403.html';
            if (!file_exists($forbidden_page)) {
                file_put_contents($forbidden_page, '<!DOCTYPE html><html><head><title>403 Forbidden</title></head><body><h1>403 Forbidden</h1><p>Access to this directory is forbidden.</p></body></html>');
            }
        }
        
        // 设置目录权限
        chmod($this->upload_dir, 0755);
    }
    
    // 检查并清理过期文件
    private function checkAndCleanup() {
        $lastCleanup = $this->settings->getSetting('last_cleanup', 0);
        
        // 使用中国时间
        date_default_timezone_set('Asia/Shanghai');
        $currentTime = time();
        
        if ($currentTime - $lastCleanup > $this->cleanup_interval) {
            $this->settings->setSetting('last_cleanup', $currentTime);
            
            $stmt = $this->db->prepare('
                SELECT code, type, filepath 
                FROM files 
                WHERE expires_at < :current_time 
                OR (max_downloads > 0 AND current_downloads >= max_downloads)
            ');
            $stmt->bindValue(':current_time', $currentTime, SQLITE3_INTEGER);
            $result = $stmt->execute();
            
            // 删除过期的文件和记录
            while ($row = $result->fetchArray(SQLITE3_ASSOC)) {
                if ($row['type'] === 'file' && $row['filepath'] && file_exists($row['filepath'])) {
                    unlink($row['filepath']);
                }
                
                $deleteStmt = $this->db->prepare('DELETE FROM files WHERE code = :code');
                $deleteStmt->bindValue(':code', $row['code'], SQLITE3_TEXT);
                $deleteStmt->execute();
            }
            
            // 清理空目录
            $this->cleanEmptyDirectories($this->upload_dir);
        }
    }
    
    // 清理空目录
    private function cleanEmptyDirectories($dir) {
        if (!is_dir($dir)) {
            return;
        }
        
        $files = scandir($dir);
        foreach ($files as $file) {
            if ($file === '.' || $file === '..' || $file === '.htaccess' || 
                $file === 'index.html' || $file === '403.html') {
                continue;
            }
            
            $path = $dir . '/' . $file;
            if (is_dir($path)) {
                $this->cleanEmptyDirectories($path);
                // 如果目录为空(只包含 . 和 ..),则删除
                $subFiles = scandir($path);
                if (count($subFiles) <= 2) { // 只有 . 和 ..
                    rmdir($path);
                }
            }
        }
    }
    
    // 生成随机提取码
    private function generateCode($length = 6) {
        $chars = '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ';
        $code = '';
        for ($i = 0; $i < $length; $i++) {
            $code .= $chars[random_int(0, strlen($chars) - 1)];
        }
        return $code;
    }
    
    // 保存文件或文本
    public function save($data) {
        $this->checkAndCleanup();
        
        $code = $this->generateCode();
        
        // 使用中国时间
        date_default_timezone_set('Asia/Shanghai');
        $created_at = time(); // 当前中国时间
        $expires_at = $created_at + intval($data['expire_time']); // 过期时间也是中国时间
        
        if (isset($_FILES['file'])) {
            // 处理文件上传
            $file = $_FILES['file'];
            $filename = $file['name'];
            $filepath = $this->upload_dir . uniqid() . '_' . bin2hex(random_bytes(4)) . '_' . $filename;
            
            if (!move_uploaded_file($file['tmp_name'], $filepath)) {
                throw new Exception('文件上传失败');
            }
            
            $type = 'file';
            $content = null;
        } else {
            // 处理文本内容
            $filename = null;
            $filepath = null;
            $type = 'text';
            $content = $data['content'];
        }
        
        $stmt = $this->db->prepare('
            INSERT INTO files (code, filename, filepath, content, type, created_at, expires_at, max_downloads)
            VALUES (:code, :filename, :filepath, :content, :type, :created_at, :expires_at, :max_downloads)
        ');
        
        $stmt->bindValue(':code', $code, SQLITE3_TEXT);
        $stmt->bindValue(':filename', $filename, SQLITE3_TEXT);
        $stmt->bindValue(':filepath', $filepath, SQLITE3_TEXT);
        $stmt->bindValue(':content', $content, SQLITE3_TEXT);
        $stmt->bindValue(':type', $type, SQLITE3_TEXT);
        $stmt->bindValue(':created_at', $created_at, SQLITE3_INTEGER);
        $stmt->bindValue(':expires_at', $expires_at, SQLITE3_INTEGER);
        $stmt->bindValue(':max_downloads', intval($data['max_downloads']), SQLITE3_INTEGER);
        
        if ($stmt->execute()) {
            return $code;
        }
        throw new Exception('保存失败');
    }
    
    // 获取文件或文本
    public function get($code) {
        $this->checkAndCleanup();
        
        $stmt = $this->db->prepare('SELECT * FROM files WHERE code = :code');
        $stmt->bindValue(':code', $code, SQLITE3_TEXT);
        $result = $stmt->execute();
        
        if ($row = $result->fetchArray(SQLITE3_ASSOC)) {
            // 使用中国时间进行比较
            date_default_timezone_set('Asia/Shanghai');
            if (time() > $row['expires_at']) {
                $this->delete($code);
                return null;
            }
            
            // 检查下载次数
            if ($row['max_downloads'] > 0 && $row['current_downloads'] >= $row['max_downloads']) {
                $this->delete($code);
                return null;
            }
            
            // 更新下载次数
            $stmt = $this->db->prepare('
                UPDATE files 
                SET current_downloads = current_downloads + 1 
                WHERE code = :code
            ');
            $stmt->bindValue(':code', $code, SQLITE3_TEXT);
            $stmt->execute();
            
            // 返回更多文件信息
            return [
                'type' => $row['type'],
                'content' => $row['content'],
                'filename' => $row['filename'],
                'filepath' => $row['filepath'],
                'current_downloads' => $row['current_downloads'],
                'max_downloads' => $row['max_downloads'],
                'expires_at' => $row['expires_at']
            ];
        }
        return null;
    }
    
    // 删除文件或文本
    private function delete($code) {
        $stmt = $this->db->prepare('SELECT * FROM files WHERE code = :code');
        $stmt->bindValue(':code', $code, SQLITE3_TEXT);
        $result = $stmt->execute();
        
        if ($row = $result->fetchArray(SQLITE3_ASSOC)) {
            if ($row['type'] === 'file' && $row['filepath'] && file_exists($row['filepath'])) {
                unlink($row['filepath']);
            }
        }
        
        $stmt = $this->db->prepare('DELETE FROM files WHERE code = :code');
        $stmt->bindValue(':code', $code, SQLITE3_TEXT);
        $stmt->execute();
    }
}