updates on #29

This commit is contained in:
Henrique Dias 2016-10-20 21:58:53 +01:00
parent 3379e6e67c
commit 9453dc246a
1 changed files with 11 additions and 4 deletions

View File

@ -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
} }