pkg/http/fs.go

package http

import (
	"fmt"
	"html/template"
	"net/http"
	"os"
	"path/filepath"
	"sort"
	"strings"

	"github.com/BytemanD/easygo/pkg/global/logging"
)

var HTML = `
<!DOCTYPE html>
<html lang="zh-CN">
	<head>
		<meta charset="UTF-8">
		<meta http-equiv="X-UA-Compatible" content="ie=edge">
		<link rel="stylesheet" href="https://cdn.staticfile.org/font-awesome/4.7.0/css/font-awesome.css">
		<title>SimpleHttpFS</title>
	</head>
	<body>
		<div style="margin-left: 50px">
			<table>
				<tr></th><th>类型</th><th>名称</th><th>大小</th></tr>
				{{ range $index, $webFile := . }}
					<tr>
						{{ if $webFile.IsDir }}
						<td><i class="fa fa-folder" style="color: orange;"></i></td>
						<td><a href="{{$webFile.WebPath}}" > {{ $webFile.Name }} </a></li></td>
						{{ else }}
						<td><i class="fa fa-file" style="color: grey;"></i></td>
						<td><a target="view_window" href="{{$webFile.WebPath}}" > {{ $webFile.Name }} </a></li></td>
						{{ end }}
						<td style="text-align:right">{{ $webFile.HumanSize }}</td>
					</tr>
				{{ end}}
			<table>
		</div>
</body>
<html>
`
var FSConfig HTTPFSConfig

type HTTPFSConfig struct {
	Port int16
	Root string
}

type WebFile struct {
	Dir  string
	Name string
	Size int64
}

const (
	KB = 1024
	MB = KB * 1024
	GB = MB * 1024
)

func (webFile *WebFile) LogicPath() string {
	return filepath.Join(webFile.Dir, webFile.Name)
}
func (webFile *WebFile) WebPath() string {
	return strings.ReplaceAll(webFile.LogicPath(), "\\", "/")
}

func (webFile *WebFile) IsDir() bool {
	filePath := filepath.Join(FSConfig.Root, webFile.LogicPath())
	fi, _ := os.Stat(filePath)
	return fi.IsDir()
}
func (webFile *WebFile) HumanSize() string {
	switch {
	case webFile.Size > GB:
		return fmt.Sprintf("%.1f G", float32(webFile.Size)/GB)
	case webFile.Size > MB:
		return fmt.Sprintf("%.1f M", float32(webFile.Size)/MB)
	case webFile.Size > KB:
		return fmt.Sprintf("%.1f K", float32(webFile.Size)/KB)
	default:
		return fmt.Sprintf("%d B", webFile.Size)
	}
}

func handleError(err error, respWriter http.ResponseWriter, request *http.Request) {
	logging.Error("目录处理异常, %s", err)
	switch {
	case os.IsNotExist(err):
		fmt.Fprintf(respWriter, "路径不存在")
	case os.IsPermission(err):
		respWriter.WriteHeader(http.StatusBadRequest)
		fmt.Fprintf(respWriter, "目录无访问权限")
	default:
		respWriter.WriteHeader(http.StatusBadRequest)
		fmt.Fprintf(respWriter, "访问目录失败 %s", request.URL.Path)
	}
}
func handleFileDownload(respWriter http.ResponseWriter, request *http.Request) {
	filePath := filepath.Join(FSConfig.Root, request.URL.Path)
	filePath = strings.ReplaceAll(filePath, "\\", "/")
	file, _ := os.Open(filePath)
	defer file.Close()

	logging.Info("下载文件: %s", file.Name())
	respWriter.Header().Set("Content-Disposition", "attachment; filename="+
		filepath.Base(file.Name()))
	http.ServeFile(respWriter, request, filePath)
}
func FilePathHandler(respWriter http.ResponseWriter, request *http.Request) {
	logging.Info("请求地址 %s", request.URL.Path)
	dirPath := filepath.Join(FSConfig.Root, request.URL.Path)
	dir, err := os.Stat(dirPath)
	if err != nil {
		handleError(err, respWriter, request)
		return
	}
	webFiles := []WebFile{}
	if dir.IsDir() {
		rd, err := os.ReadDir(dirPath)
		if err != nil {
			handleError(err, respWriter, request)
			return
		}
		for _, fi := range rd {
			webFile := WebFile{Dir: filepath.Join(request.URL.Path), Name: fi.Name()}
			fiInfo, err := fi.Info()
			if err == nil {
				webFile.Size = fiInfo.Size()
			} else {
				logging.Warning("get dir info failed, %v", err)
				webFile.Size = 0
			}
			webFiles = append(webFiles, webFile)
		}
		tmpl, _ := template.New("dirPage").Parse(HTML)
		sort.Slice(webFiles, func(i, j int) bool {
			return webFiles[i].IsDir()
		})
		tmpl.Execute(respWriter, webFiles)
	} else {
		handleFileDownload(respWriter, request)
	}
}

func SimpleHttpFS() error {
	http.HandleFunc("/", FilePathHandler) //设置访问的路由
	logging.Info("启动文件服务器 端口:%d, 目录: %s",
		FSConfig.Port, FSConfig.Root)
	if _, err := os.Stat(FSConfig.Root); err != nil {
		return err
	}
	return http.ListenAndServe(
		fmt.Sprintf(":%d", FSConfig.Port),
		nil)
}