updates on #29
This commit is contained in:
parent
3379e6e67c
commit
9453dc246a
|
@ -5,15 +5,19 @@ import (
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
|
||||||
"github.com/mholt/archiver"
|
"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) {
|
func (i *Info) DownloadAs(w http.ResponseWriter, query string) (int, error) {
|
||||||
var (
|
var (
|
||||||
extension string
|
extension string
|
||||||
temp string
|
temp string
|
||||||
err error
|
err error
|
||||||
|
tempfile string
|
||||||
)
|
)
|
||||||
|
|
||||||
temp, err = ioutil.TempDir("", "")
|
temp, err = ioutil.TempDir("", "")
|
||||||
|
@ -21,15 +25,18 @@ func (i *Info) DownloadAs(w http.ResponseWriter, query string) (int, error) {
|
||||||
return http.StatusInternalServerError, err
|
return http.StatusInternalServerError, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
defer os.RemoveAll(temp)
|
||||||
|
tempfile = filepath.Join(temp, "temp")
|
||||||
|
|
||||||
switch query {
|
switch query {
|
||||||
case "zip":
|
case "zip":
|
||||||
extension, err = ".zip", archiver.Zip.Make(temp+"/temp", []string{i.Path})
|
extension, err = ".zip", archiver.Zip.Make(tempfile, []string{i.Path})
|
||||||
case "tar":
|
case "tar":
|
||||||
extension, err = ".tar", archiver.Tar.Make(temp+"/temp", []string{i.Path})
|
extension, err = ".tar", archiver.Tar.Make(tempfile, []string{i.Path})
|
||||||
case "targz":
|
case "targz":
|
||||||
extension, err = ".tar.gz", archiver.TarGz.Make(temp+"/temp", []string{i.Path})
|
extension, err = ".tar.gz", archiver.TarGz.Make(tempfile, []string{i.Path})
|
||||||
case "tarbz2":
|
case "tarbz2":
|
||||||
extension, err = ".tar.bz2", archiver.TarBz2.Make(temp+"/temp", []string{i.Path})
|
extension, err = ".tar.bz2", archiver.TarBz2.Make(tempfile, []string{i.Path})
|
||||||
default:
|
default:
|
||||||
return http.StatusNotImplemented, nil
|
return http.StatusNotImplemented, nil
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue