mirror of
https://github.com/Luzifer/tex-api.git
synced 2024-11-09 16:50:03 +00:00
Support TAR output format
Signed-off-by: Knut Ahlers <knut@ahlers.me>
This commit is contained in:
parent
1d28bd2c1a
commit
f2d676c7db
1 changed files with 45 additions and 0 deletions
45
main.go
45
main.go
|
@ -1,6 +1,7 @@
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"archive/tar"
|
||||||
"archive/zip"
|
"archive/zip"
|
||||||
"bytes"
|
"bytes"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
@ -299,6 +300,47 @@ func buildAssetsZIP(uid uuid.UUID) (io.Reader, error) {
|
||||||
return buf, w.Close()
|
return buf, w.Close()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func buildAssetsTAR(uid uuid.UUID) (io.Reader, error) {
|
||||||
|
buf := new(bytes.Buffer)
|
||||||
|
w := tar.NewWriter(buf)
|
||||||
|
|
||||||
|
basePath := pathFromUUID(uid, filenameOutputDir)
|
||||||
|
err := filepath.Walk(basePath, func(p string, info os.FileInfo, err error) error {
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !shouldPackFile(path.Ext(info.Name())) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
tarInfo, err := tar.FileInfoHeader(info, "")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
tarInfo.Name = strings.TrimLeft(strings.Replace(p, basePath, "", 1), "/\\")
|
||||||
|
err = w.WriteHeader(tarInfo)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
osFile, err := os.Open(p)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
io.Copy(w, osFile)
|
||||||
|
osFile.Close()
|
||||||
|
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return buf, w.Close()
|
||||||
|
}
|
||||||
|
|
||||||
func downloadAssets(res http.ResponseWriter, r *http.Request) {
|
func downloadAssets(res http.ResponseWriter, r *http.Request) {
|
||||||
vars := mux.Vars(r)
|
vars := mux.Vars(r)
|
||||||
uid, err := uuid.FromString(vars["uid"])
|
uid, err := uuid.FromString(vars["uid"])
|
||||||
|
@ -313,6 +355,9 @@ func downloadAssets(res http.ResponseWriter, r *http.Request) {
|
||||||
)
|
)
|
||||||
|
|
||||||
switch r.Header.Get("Accept") {
|
switch r.Header.Get("Accept") {
|
||||||
|
case "application/tar", "application/x-tar", "applicaton/x-gtar", "multipart/x-tar", "application/x-compress", "application/x-compressed":
|
||||||
|
content, err = buildAssetsTAR(uid)
|
||||||
|
filename = uid.String() + ".tar"
|
||||||
default:
|
default:
|
||||||
content, err = buildAssetsZIP(uid)
|
content, err = buildAssetsZIP(uid)
|
||||||
filename = uid.String() + ".zip"
|
filename = uid.String() + ".zip"
|
||||||
|
|
Loading…
Reference in a new issue