Files

348 líneas
13 KiB
Go

package httpfile
import (
"encoding/json"
"fmt"
"log"
"net/http"
"os"
"path/filepath"
"strings"
"ftp-server/config"
)
// Server represents the HTTP file server
type Server struct {
config *config.Config
}
// NewServer creates a new HTTP file server
func NewServer(cfg *config.Config) *Server {
return &Server{config: cfg}
}
// Start starts the HTTP file server
func (s *Server) Start() error {
if !s.config.HTTPFile.Enable {
return nil
}
rootDir := s.config.HTTPFile.RootDir
if err := os.MkdirAll(rootDir, 0755); err != nil {
return fmt.Errorf("failed to create root directory: %v", err)
}
mux := http.NewServeMux()
// Serve HTML page
mux.HandleFunc("/", s.handleFileBrowser)
// API endpoints
mux.HandleFunc("/api/list", s.apiListDir)
mux.HandleFunc("/api/download", s.apiDownload)
addr := fmt.Sprintf("%s:%d", s.config.HTTPFile.Host, s.config.HTTPFile.Port)
log.Printf("HTTP file server listening on http://%s", addr)
return http.ListenAndServe(addr, mux)
}
func (s *Server) handleFileBrowser(w http.ResponseWriter, r *http.Request) {
if r.URL.Path != "/" {
http.NotFound(w, r)
return
}
w.Header().Set("Content-Type", "text/html; charset=utf-8")
w.Header().Set("Cache-Control", "no-cache, no-store, must-revalidate")
w.Header().Set("Pragma", "no-cache")
w.Header().Set("Expires", "0")
w.Write([]byte(fileBrowserHTML))
}
func (s *Server) apiListDir(w http.ResponseWriter, r *http.Request) {
dir := r.URL.Query().Get("path")
if dir == "" {
dir = "/"
}
realPath := s.toRealPath(dir)
entries, err := os.ReadDir(realPath)
if err != nil {
http.Error(w, `{"error":"Cannot read directory"}`, http.StatusInternalServerError)
return
}
var files []map[string]interface{}
for _, entry := range entries {
info, _ := entry.Info()
modTime := info.ModTime().Format("2006-01-02 15:04:05")
size := info.Size()
file := map[string]interface{}{
"name": entry.Name(),
"modTime": modTime,
"size": formatSize(size),
"isDir": entry.IsDir(),
}
if !entry.IsDir() {
file["ext"] = strings.ToLower(filepath.Ext(entry.Name()))
}
files = append(files, file)
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(map[string]interface{}{
"files": files,
})
}
func (s *Server) apiDownload(w http.ResponseWriter, r *http.Request) {
file := r.URL.Query().Get("file")
if file == "" {
http.Error(w, "Missing file parameter", http.StatusBadRequest)
return
}
realPath := s.toRealPath(file)
if info, err := os.Stat(realPath); err != nil || info.IsDir() {
http.Error(w, "File not found", http.StatusNotFound)
return
}
http.ServeFile(w, r, realPath)
log.Printf("HTTP download: %s from %s", file, r.RemoteAddr)
}
func (s *Server) toRealPath(httpPath string) string {
rootDir := s.config.HTTPFile.RootDir
cleanPath := strings.TrimPrefix(httpPath, "/")
return filepath.Join(rootDir, cleanPath)
}
func formatSize(size int64) string {
switch {
case size < 1024:
return fmt.Sprintf("%d B", size)
case size < 1024*1024:
return fmt.Sprintf("%.1f KB", float64(size)/1024)
case size < 1024*1024*1024:
return fmt.Sprintf("%.1f MB", float64(size)/(1024*1024))
default:
return fmt.Sprintf("%.1f GB", float64(size)/(1024*1024*1024))
}
}
// fileBrowserHTML is the embedded HTML page
const fileBrowserHTML = `<!DOCTYPE html>
<html lang="zh-CN">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>HTTP 文件服务器</title>
<style>
* { margin: 0; padding: 0; box-sizing: border-box; }
body { font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); min-height: 100vh; }
.container { max-width: 1200px; margin: 0 auto; padding: 20px; }
.header { background: white; border-radius: 12px; padding: 24px; margin-bottom: 20px; box-shadow: 0 4px 20px rgba(0,0,0,0.1); }
.header h1 { font-size: 28px; color: #333; margin-bottom: 8px; }
.header p { color: #888; font-size: 14px; }
.breadcrumb { background: white; border-radius: 12px; padding: 16px 24px; margin-bottom: 20px; box-shadow: 0 4px 20px rgba(0,0,0,0.1); display: flex; align-items: center; gap: 8px; flex-wrap: wrap; }
.breadcrumb a { color: #667eea; text-decoration: none; font-weight: 500; padding: 4px 8px; border-radius: 6px; transition: background 0.2s; }
.breadcrumb a:hover { background: #f0f4ff; }
.breadcrumb span { color: #999; font-size: 20px; }
.actions { background: white; border-radius: 12px; padding: 16px 24px; margin-bottom: 20px; box-shadow: 0 4px 20px rgba(0,0,0,0.1); display: flex; gap: 12px; align-items: center; flex-wrap: wrap; }
.btn { padding: 10px 20px; border: none; border-radius: 8px; cursor: pointer; font-size: 14px; font-weight: 600; transition: all 0.3s; display: inline-flex; align-items: center; gap: 6px; }
.btn-primary { background: linear-gradient(135deg, #667eea, #764ba2); color: white; }
.btn-primary:hover { transform: translateY(-2px); box-shadow: 0 4px 15px rgba(102,126,234,0.4); }
.btn-success { background: #28a745; color: white; }
.btn-success:hover { background: #218838; }
.btn-sm { padding: 6px 12px; font-size: 12px; }
.file-list { background: white; border-radius: 12px; padding: 24px; box-shadow: 0 4px 20px rgba(0,0,0,0.1); }
table { width: 100%; border-collapse: collapse; }
th, td { padding: 12px 16px; text-align: left; border-bottom: 1px solid #f0f0f0; font-size: 14px; }
th { background: #fafafa; font-weight: 600; color: #555; }
tr:hover { background: #f8f9ff; }
.file-name { display: flex; align-items: center; gap: 8px; }
.file-icon { width: 32px; height: 32px; border-radius: 8px; display: flex; align-items: center; justify-content: center; font-size: 18px; }
.folder-icon { background: #ffeaa7; }
.file-icon-doc { background: #74b9ff; }
.file-icon-img { background: #ff7675; }
.file-icon-zip { background: #a29bfe; }
.file-icon-code { background: #55efc4; }
.file-icon-other { background: #dfe6e9; }
.file-link { color: #667eea; text-decoration: none; font-weight: 500; }
.file-link:hover { text-decoration: underline; }
.actions-cell { display: flex; gap: 8px; }
.empty-state { text-align: center; padding: 60px 20px; color: #999; }
.empty-state .icon { font-size: 64px; margin-bottom: 16px; }
</style>
</head>
<body>
<div class="container">
<div class="header">
<h1>HTTP 文件服务器</h1>
<p>在线浏览、下载文件</p>
</div>
<div class="breadcrumb" id="breadcrumb"></div>
<div class="actions">
<button class="btn btn-primary" onclick="refresh()">刷新</button>
</div>
<div class="file-list">
<table id="fileTable">
<thead>
<tr><th>名称</th><th>大小</th><th>修改时间</th><th>操作</th></tr>
</thead>
<tbody id="fileTableBody"></tbody>
</table>
<div class="empty-state" id="emptyState" style="display:none">
<div class="icon">&#128193;</div>
<div>目录为空</div>
</div>
</div>
</div>
<script>
var currentPath = '/';
function init() {
loadFiles();
}
function loadFiles() {
fetch('/api/list?path=' + encodeURIComponent(currentPath))
.then(function(r) { return r.json(); })
.then(function(data) {
var tbody = document.getElementById('fileTableBody');
tbody.innerHTML = '';
if (!data.files || data.files.length === 0) {
document.getElementById('emptyState').style.display = 'block';
document.getElementById('fileTable').style.display = 'none';
return;
}
document.getElementById('emptyState').style.display = 'none';
document.getElementById('fileTable').style.display = 'table';
data.files.forEach(function(f) {
var tr = document.createElement('tr');
var nameTd = document.createElement('td');
var nameDiv = document.createElement('div');
nameDiv.className = 'file-name';
var iconDiv = document.createElement('div');
var icon = f.isDir ? '&#128193;' : getFileIcon(f.name);
var iconClass = f.isDir ? 'folder-icon' : 'file-icon-' + getFileType(f.name);
iconDiv.className = 'file-icon ' + iconClass;
iconDiv.innerHTML = icon;
nameDiv.appendChild(iconDiv);
if (f.isDir) {
var link = document.createElement('a');
link.className = 'file-link';
link.href = '#';
link.textContent = f.name;
link.onclick = function() { enterFolder(f.name); return false; };
nameDiv.appendChild(link);
} else {
var span = document.createElement('span');
span.textContent = f.name;
nameDiv.appendChild(span);
}
nameTd.appendChild(nameDiv);
var sizeTd = document.createElement('td');
sizeTd.textContent = f.isDir ? '-' : f.size;
var timeTd = document.createElement('td');
timeTd.textContent = f.modTime;
var actionTd = document.createElement('td');
actionTd.className = 'actions-cell';
if (f.isDir) {
var btn = document.createElement('button');
btn.className = 'btn btn-sm btn-primary';
btn.textContent = '打开';
btn.onclick = function() { enterFolder(f.name); };
actionTd.appendChild(btn);
} else {
var link = document.createElement('a');
link.className = 'btn btn-sm btn-success';
link.href = '/api/download?file=' + encodeURIComponent(currentPath + f.name);
link.textContent = '下载';
actionTd.appendChild(link);
}
tr.appendChild(nameTd);
tr.appendChild(sizeTd);
tr.appendChild(timeTd);
tr.appendChild(actionTd);
tbody.appendChild(tr);
});
updateBreadcrumb();
});
}
function enterFolder(name) {
currentPath = currentPath === '/' ? '/' + name + '/' : currentPath + name + '/';
loadFiles();
}
function updateBreadcrumb() {
var bc = document.getElementById('breadcrumb');
bc.innerHTML = '<a href="#" onclick="goTo(\'/\');return false;">根目录</a>';
var parts = currentPath.split('/').filter(function(p) { return p; });
var path = '';
parts.forEach(function(p) {
path += p + '/';
bc.innerHTML += '<span>/</span><a href="#" onclick="goTo(\'' + path + '\');return false;">' + p + '</a>';
});
}
function goTo(path) {
currentPath = path;
loadFiles();
}
function refresh() {
loadFiles();
}
function getFileIcon(name) {
var ext = name.split('.').pop().toLowerCase();
var icons = {
'doc': '&#128196;', 'docx': '&#128196;', 'pdf': '&#128196;', 'txt': '&#128196;',
'xls': '&#128202;', 'xlsx': '&#128202;', 'csv': '&#128202;',
'jpg': '&#127748;', 'jpeg': '&#127748;', 'png': '&#127748;', 'gif': '&#127748;', 'bmp': '&#127748;',
'mp3': '&#127925;', 'wav': '&#127925;', 'flac': '&#127925;',
'mp4': '&#127916;', 'avi': '&#127916;', 'mkv': '&#127916;', 'mov': '&#127916;',
'zip': '&#128230;', 'rar': '&#128230;', '7z': '&#128230;', 'tar': '&#128230;', 'gz': '&#128230;',
'js': '&#128187;', 'ts': '&#128187;', 'py': '&#128187;', 'go': '&#128187;', 'java': '&#128187;', 'c': '&#128187;', 'cpp': '&#128187;', 'html': '&#128187;', 'css': '&#128187;', 'json': '&#128187;', 'xml': '&#128187;', 'yml': '&#128187;', 'yaml': '&#128187;'
};
return icons[ext] || '&#128196;';
}
function getFileType(name) {
var ext = name.split('.').pop().toLowerCase();
if (['jpg','jpeg','png','gif','bmp','svg'].indexOf(ext) !== -1) return 'img';
if (['zip','rar','7z','tar','gz'].indexOf(ext) !== -1) return 'zip';
if (['js','ts','py','go','java','c','cpp','html','css','json','xml','yml','yaml'].indexOf(ext) !== -1) return 'code';
return 'doc';
}
init();
</script>
</body>
</html>
`