如何避免通过 Go 提供模板文件

Question

我正在用 Go 编写小型网站，但我发现了一些我不知道如何解决的问题。所以... 基本想法是为主题创建一个文件夹，名为/themes/，我们将在其中放置所有主题，f.e. classic、modern等 目录树如下所示：

/themes/
    classic/
        index.html
        header.html
        footer.html
        css/
            style.css
        js/
            lib.js
    modern/
        index.html
        header.html
        footer.html
        css/
            style.css
        js/
            lib.js

所以，我的 http 处理程序：

func main() {
    reloadConfig()

    http.HandleFunc("/", homeHandler)

    http.HandleFunc("/reloadConfigHandler/", reloadConfigHandler)

    // TODO: Theme loads html files also
    http.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir("themes/"+config.Theme+"/"))))
    http.ListenAndServe(":80", nil)
}

问题

问题是我的模板文件可以从外部打开，如果我打开路径http://localhost/static/index.html，所以我需要解决方案：

1. 拒绝/static/，显示404。
2. 拒绝/static/*.html，显示404。
3. 允许/static/{folder_name}/{file_name}，这样以后我可以添加img 文件夹或fonts 文件夹，其中的内容将由服务器提供给客户端。

感谢您的建议。

Answer 1

简单的方法是实现你自己的http.FileSystem:

type fileSystem struct {
    http.FileSystem
}

func (fs fileSystem) Open(name string) (http.File, error) {
    f, err := fs.FileSystem.Open(name)
    if err != nil {
        return nil, err
    }

    stat, err := f.Stat()
    if err != nil {
        return nil, err
    }

    // This denies access to the directory listing
    if stat.IsDir() {
        return nil, os.ErrNotExist
    }

    // This denies access to anything but <prefix>/css/...
    if !strings.HasPrefix(name, "/css/") {
        return nil, os.ErrNotExist
    }

    return f, nil
}

现在您可以像这样在 main 中使用它：

fs := http.FileServer(fileSystem{http.Dir("themes/"+config.Theme+"/")})    
http.Handle("/static/", http.StripPrefix("/static/", fs))