mirror of
https://github.com/Luzifer/tex-api.git
synced 2024-11-09 16:50:03 +00:00
Move builders to own file
Signed-off-by: Knut Ahlers <knut@ahlers.me>
This commit is contained in:
parent
3da686bec0
commit
81e4d4b2d7
GPG key ID:
DC2729FDD34BE99E
3 changed files with 105 additions and 95 deletions
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
|
|
@ -0,0 +1 @@
|
|||
tex-api
|
||||
104
assets.go
Normal file
104
assets.go
Normal file
|
|
@ -0,0 +1,104 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"archive/tar"
|
||||
"archive/zip"
|
||||
"bytes"
|
||||
"io"
|
||||
"os"
|
||||
"path"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
|
||||
"github.com/Luzifer/go_helpers/str"
|
||||
uuid "github.com/satori/go.uuid"
|
||||
)
|
||||
|
||||
func shouldPackFile(extension string) bool {
|
||||
return str.StringInSlice(extension, []string{
|
||||
".log",
|
||||
".pdf",
|
||||
})
|
||||
}
|
||||
|
||||
func buildAssetsZIP(uid uuid.UUID) (io.Reader, error) {
|
||||
buf := new(bytes.Buffer)
|
||||
w := zip.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
|
||||
}
|
||||
|
||||
zipInfo, err := zip.FileInfoHeader(info)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
zipInfo.Name = strings.TrimLeft(strings.Replace(p, basePath, "", 1), "/\\")
|
||||
zipFile, err := w.CreateHeader(zipInfo)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
osFile, err := os.Open(p)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
io.Copy(zipFile, osFile)
|
||||
osFile.Close()
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
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()
|
||||
}
|
||||
95
main.go
95
main.go
|
|
@ -1,9 +1,6 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"archive/tar"
|
||||
"archive/zip"
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
|
|
@ -13,12 +10,9 @@ import (
|
|||
"os"
|
||||
"os/exec"
|
||||
"path"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/Luzifer/go_helpers/str"
|
||||
"github.com/Luzifer/rconfig"
|
||||
log "github.com/Sirupsen/logrus"
|
||||
"github.com/gorilla/mux"
|
||||
|
|
@ -234,95 +228,6 @@ func waitForJob(res http.ResponseWriter, r *http.Request) {
|
|||
}
|
||||
}
|
||||
|
||||
func shouldPackFile(extension string) bool {
|
||||
return str.StringInSlice(extension, []string{
|
||||
".log",
|
||||
".pdf",
|
||||
})
|
||||
}
|
||||
|
||||
func buildAssetsZIP(uid uuid.UUID) (io.Reader, error) {
|
||||
buf := new(bytes.Buffer)
|
||||
w := zip.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
|
||||
}
|
||||
|
||||
zipInfo, err := zip.FileInfoHeader(info)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
zipInfo.Name = strings.TrimLeft(strings.Replace(p, basePath, "", 1), "/\\")
|
||||
zipFile, err := w.CreateHeader(zipInfo)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
osFile, err := os.Open(p)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
io.Copy(zipFile, osFile)
|
||||
osFile.Close()
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
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) {
|
||||
vars := mux.Vars(r)
|
||||
uid, err := uuid.FromString(vars["uid"])
|
||||
|
|
|
|||
Loading…
Reference in a new issue