filebrowser/http/download.go

103 lines
2.5 KiB
Go
Raw Normal View History

2017-08-18 08:00:32 +00:00
package http
2017-06-27 18:00:58 +00:00
import (
"net/http"
"net/url"
"os"
"path/filepath"
"strings"
fb "github.com/filebrowser/filebrowser"
2017-07-27 20:39:23 +00:00
"github.com/hacdias/fileutils"
2017-06-27 18:00:58 +00:00
"github.com/mholt/archiver"
)
2017-07-02 16:40:52 +00:00
// downloadHandler creates an archive in one of the supported formats (zip, tar,
2017-06-27 18:00:58 +00:00
// tar.gz or tar.bz2) and sends it to be downloaded.
func downloadHandler(c *fb.Context, w http.ResponseWriter, r *http.Request) (int, error) {
2017-07-26 10:25:42 +00:00
// If the file isn't a directory, serve it using http.ServeFile. We display it
// inline if it is requested.
if !c.File.IsDir {
return downloadFileHandler(c, w, r)
2017-06-27 18:00:58 +00:00
}
query := r.URL.Query().Get("format")
2017-06-27 18:00:58 +00:00
files := []string{}
names := strings.Split(r.URL.Query().Get("files"), ",")
2017-07-26 10:25:42 +00:00
// If there are files in the query, sanitize their names.
// Otherwise, just append the current path.
2017-06-27 18:00:58 +00:00
if len(names) != 0 {
for _, name := range names {
2017-07-26 10:25:42 +00:00
// Unescape the name.
2017-06-27 18:00:58 +00:00
name, err := url.QueryUnescape(name)
if err != nil {
return http.StatusInternalServerError, err
}
2017-07-26 10:25:42 +00:00
// Clean the slashes.
2017-07-27 20:39:23 +00:00
name = fileutils.SlashClean(name)
files = append(files, filepath.Join(c.File.Path, name))
2017-06-27 18:00:58 +00:00
}
} else {
files = append(files, c.File.Path)
2017-06-27 18:00:58 +00:00
}
var (
extension string
ar archiver.Archiver
2017-06-27 18:00:58 +00:00
)
switch query {
// If the format is true, just set it to "zip".
case "zip", "true", "":
extension, ar = ".zip", archiver.Zip
2017-06-27 18:00:58 +00:00
case "tar":
extension, ar = ".tar", archiver.Tar
2017-06-27 18:00:58 +00:00
case "targz":
extension, ar = ".tar.gz", archiver.TarGz
2017-06-27 18:00:58 +00:00
case "tarbz2":
extension, ar = ".tar.bz2", archiver.TarBz2
2017-06-27 18:00:58 +00:00
case "tarxz":
extension, ar = ".tar.xz", archiver.TarXZ
2017-06-27 18:00:58 +00:00
default:
return http.StatusNotImplemented, nil
}
2017-07-26 10:25:42 +00:00
// Defines the file name.
name := c.File.Name
2017-06-27 18:00:58 +00:00
if name == "." || name == "" {
name = "archive"
2017-06-27 18:00:58 +00:00
}
2017-07-26 10:25:42 +00:00
name += extension
w.Header().Set("Content-Disposition", "attachment; filename*=utf-8''"+url.PathEscape(name))
err := ar.Write(w, files)
2017-06-27 18:00:58 +00:00
2017-07-26 10:25:42 +00:00
return 0, err
2017-06-27 18:00:58 +00:00
}
func downloadFileHandler(c *fb.Context, w http.ResponseWriter, r *http.Request) (int, error) {
file, err := os.Open(c.File.Path)
if err != nil {
return http.StatusInternalServerError, err
}
defer file.Close()
stat, err := file.Stat()
if err != nil {
return http.StatusInternalServerError, err
}
if r.URL.Query().Get("inline") == "true" {
w.Header().Set("Content-Disposition", "inline")
} else {
// As per RFC6266 section 4.3
w.Header().Set("Content-Disposition", "attachment; filename*=utf-8''"+url.PathEscape(c.File.Name))
}
http.ServeContent(w, r, stat.Name(), stat.ModTime(), file)
return 0, nil
}