71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
|
package handlers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
"github.com/hacdias/caddy-filemanager/config"
|
||
|
|
"github.com/hacdias/caddy-filemanager/file"
|
||
|
|
"github.com/mholt/archiver"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Download creates an archieve in one of the supported formats (zip, tar,
|
||
|
|
// tar.gz or tar.bz2) and sends it to be downloaded.
|
||
|
|
func Download(w http.ResponseWriter, r *http.Request, c *config.Config, i *file.Info) (int, error) {
|
||
|
|
query := r.URL.Query().Get("download")
|
||
|
|
|
||
|
|
if !i.IsDir() {
|
||
|
|
w.Header().Set("Content-Disposition", "attachment; filename="+i.Name())
|
||
|
|
http.ServeFile(w, r, i.Path)
|
||
|
|
return 0, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
if query == "true" {
|
||
|
|
query = "zip"
|
||
|
|
}
|
||
|
|
|
||
|
|
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
|
||
|
|
}
|