Go

Question

我在Go上写了一个小网站，我发现了一些问题，我不知道如何解决。所以... 基本的想法是为主题创建一个单独的文件夹，称为/themes/，我们将放置所有主题，例如： 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
}

现在你可以在你的主体中使用它：

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