luzifer/tex-api
1
0
Fork 0
mirror of https://github.com/Luzifer/tex-api.git synced 2024-09-19 01:22:57 +00:00
Code Issues Releases Wiki Activity

Move builders to own file

Browse Source 
Signed-off-by: Knut Ahlers <knut@ahlers.me>
This commit is contained in:
Knut Ahlers 2017-03-06 21:59:53 +01:00
parent 3da686bec0
commit 81e4d4b2d7
Signed by: luzifer
GPG Key ID: DC2729FDD34BE99E
3 changed files with 105 additions and 95 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored Normal file
View File

@ -0,0 +1 @@
tex-api

104
assets.go Normal file
View 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
View File

@ -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"])