golang web 文件服务器 WFS

驭鲸·环球

已于 2024-12-29 00:00:18 修改

阅读量1.5k

点赞数

分类专栏： Golang 文章标签：服务器 golang 前端

于 2022-01-28 01:22:33 首次发布

知识因分享而具有价值，回忆昨天，把握今天，期待明天。 —— 驭鲸·环球

本文链接：https://blog.csdn.net/HighproX/article/details/122725881

版权

Golang 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了如何利用Golang的标准库简洁地创建一个Web文件服务器WFS，该服务器允许用户通过浏览器查看本地托管目录的内容。通过实现自定义的ResponseBodyWriter并在Write方法中修改响应，只需一行命令即可启动并访问服务器。

摘要由CSDN通过智能技术生成

golang 使用标准库 30行代码实现 web 文件托管服务器

WFS（Web File Server）是一个可以通过浏览器浏览本地托管目录的在线文件服务器。

使用自定义的 ResponseBodyWriter 结构在 Write 函数中重写 http.ResponseWriter 响应内容。

运行程序，访问 http://127.0.0.1:8080/ 即可访问 exe 根目录内容。

在这里插入图片描述

package main

import (
	"bytes"
	"log"
	"net/http"
	"os"
	"path/filepath"
)

func codeStarts() string {
	return `<!DOCTYPE html>
<html>
<head>
    <title>Title</title>
    <meta charset="utf-8">
    <meta name="viewport" content="width=device-width, initial-scale=1">
    <!-- UIkit CSS -->
    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/uikit@3.10.1/dist/css/uikit.min.css"/>

    <!-- UIkit JS -->
    <script src="https://cdn.jsdelivr.net/npm/uikit@3.10.1/dist/js/uikit.min.js"></script>
    <script src="https://cdn.jsdelivr.net/npm/uikit@3.10.1/dist/js/uikit-icons.min.js"></script>
</head>
<body class="uk-container uk-width-2xlarge">
<div class="uk-card uk-card-default uk-card-body">
	<header class="uk-label">FileServer - 驭鲸环球</header>
    <ul class="uk-list uk-list-striped uk-list-collapse">`
}

func codeEnd() string {
	return `</ul>
</div>
</body>
</html>`
}

type ResponseBodyWriter struct {
	http.ResponseWriter
	body *bytes.Buffer
}

func (r ResponseBodyWriter) Write(b []byte) (int, error) {
	if bytes.Equal(b, []byte{60, 112, 114, 101, 62, 10} /*<pre>*/) {
		return r.ResponseWriter.Write([]byte(codeStarts()))
	} else if bytes.Equal(b, []byte{60, 47, 112, 114, 101, 62, 10} /*</pre>*/) {
		return r.ResponseWriter.Write([]byte(codeEnd()))
	} else if bytes.HasPrefix(b, []byte("<a")) || bytes.HasSuffix(b, []byte("</a>")) {
		s := append([]byte("<li>"), b...)
		s = append(s, []byte("</li>")...)
		//r.body.Write(s)
		return r.ResponseWriter.Write(s)
	}
	return r.ResponseWriter.Write(b)
}

func middleware(next http.Handler) http.Handler {
	return http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
		w := &ResponseBodyWriter{ResponseWriter: writer, body: &bytes.Buffer{}}
		writer = w
		next.ServeHTTP(writer, request)
	})
}

func main() {
	path, _ := filepath.Abs(filepath.Dir(os.Args[0]))
	//设置wfs根目录
	//path = "D:/Games/Naraka"
	fileHandler := http.FileServer(http.Dir(path))
	http.Handle("/", middleware(fileHandler))
	err := http.ListenAndServe(":8080", nil)
	if err != nil {
		log.Println(err)
	}
}