updates

Former-commit-id: 54b88552d11f2151a165dba9debb4657dfa56cf8 [formerly 0ce53651a8e9660f9d5f977295f553b5b1d1e93a] [formerly 7ebca3a8896222091c95af86a9cf1d12550b8b76 [formerly 174330929ad7231b95b30acb98ad2033d697590f]]
Former-commit-id: 993d0cdb239f9969587d13a11ee8469fa8b91287 [formerly c22c911f944dd8d6597ab95589842d3c68d34869]
Former-commit-id: 44ed259fe50a085e8bcace3f1f14caafec97ce66

http/download.go

@@ -0,0 +1,113 @@
+package http
+
+import (
+	"io"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"os"
+	"path/filepath"
+	"strings"
+
+	fm "github.com/hacdias/filemanager"
+	"github.com/hacdias/fileutils"
+	"github.com/mholt/archiver"
+)
+
+// downloadHandler creates an archive in one of the supported formats (zip, tar,
+// tar.gz or tar.bz2) and sends it to be downloaded.
+func downloadHandler(c *fm.Context, w http.ResponseWriter, r *http.Request) (int, error) {
+	query := r.URL.Query().Get("format")
+
+	// If the file isn't a directory, serve it using http.ServeFile. We display it
+	// inline if it is requested.
+	if !c.File.IsDir {
+		if r.URL.Query().Get("inline") == "true" {
+			w.Header().Set("Content-Disposition", "inline")
+		} else {
+			w.Header().Set("Content-Disposition", "attachment; filename="+c.File.Name)
+		}
+
+		http.ServeFile(w, r, c.File.Path)
+		return 0, nil
+	}
+
+	files := []string{}
+	names := strings.Split(r.URL.Query().Get("files"), ",")
+
+	// If there are files in the query, sanitize their names.
+	// Otherwise, just append the current path.
+	if len(names) != 0 {
+		for _, name := range names {
+			// Unescape the name.
+			name, err := url.QueryUnescape(name)
+			if err != nil {
+				return http.StatusInternalServerError, err
+			}
+
+			// Clean the slashes.
+			name = fileutils.SlashClean(name)
+			files = append(files, filepath.Join(c.File.Path, name))
+		}
+	} else {
+		files = append(files, c.File.Path)
+	}
+
+	// If the format is true, just set it to "zip".
+	if query == "true" || query == "" {
+		query = "zip"
+	}
+
+	var (
+		extension string
+		temp      string
+		err       error
+		tempfile  string
+	)
+
+	// Create a temporary directory.
+	temp, err = ioutil.TempDir("", "")
+	if err != nil {
+		return http.StatusInternalServerError, err
+	}
+	defer os.RemoveAll(temp)
+
+	tempfile = filepath.Join(temp, "temp")
+
+	switch query {
+	case "zip":
+		extension, err = ".zip", archiver.Zip.Make(tempfile, files)
+	case "tar":
+		extension, err = ".tar", archiver.Tar.Make(tempfile, files)
+	case "targz":
+		extension, err = ".tar.gz", archiver.TarGz.Make(tempfile, files)
+	case "tarbz2":
+		extension, err = ".tar.bz2", archiver.TarBz2.Make(tempfile, files)
+	case "tarxz":
+		extension, err = ".tar.xz", archiver.TarXZ.Make(tempfile, files)
+	default:
+		return http.StatusNotImplemented, nil
+	}
+
+	if err != nil {
+		return http.StatusInternalServerError, err
+	}
+
+	// Defines the file name.
+	name := c.File.Name
+	if name == "." || name == "" {
+		name = "download"
+	}
+	name += extension
+
+	// Opens the file so it can be downloaded.
+	file, err := os.Open(temp + "/temp")
+	if err != nil {
+		return http.StatusInternalServerError, err
+	}
+	defer file.Close()
+
+	w.Header().Set("Content-Disposition", "attachment; filename="+name)
+	_, err = io.Copy(w, file)
+	return 0, err
+}