filebrowser/file/download.go

57 lines
1.3 KiB
Go

package file
import (
"io"
"io/ioutil"
"net/http"
"os"
"path/filepath"
"github.com/mholt/archiver"
)
// DownloadAs creates an archieve in one of the supported formats (zip, tar,
// tar.gz or tar.bz2) and sends it to be downloaded.
func (i *Info) DownloadAs(w http.ResponseWriter, query string) (int, error) {
var (
extension string
temp string
err error
tempfile string
)
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, []string{i.Path})
case "tar":
extension, err = ".tar", archiver.Tar.Make(tempfile, []string{i.Path})
case "targz":
extension, err = ".tar.gz", archiver.TarGz.Make(tempfile, []string{i.Path})
case "tarbz2":
extension, err = ".tar.bz2", archiver.TarBz2.Make(tempfile, []string{i.Path})
default:
return http.StatusNotImplemented, nil
}
if err != nil {
return http.StatusInternalServerError, err
}
file, err := os.Open(temp + "/temp")
if err != nil {
return http.StatusInternalServerError, err
}
w.Header().Set("Content-Disposition", "attachment; filename="+i.Name()+extension)
io.Copy(w, file)
return http.StatusOK, nil
}