mirror of
https://github.com/gotenberg/gotenberg.git
synced 2026-08-16 20:32:13 +01:00
minor refactoring of const + better timeout handling in pinter package + api package tests
This commit is contained in:
@@ -13,7 +13,6 @@ go test -race -cover -covermode=atomic github.com/thecodingmachine/gotenberg/int
|
|||||||
go test -race -cover -covermode=atomic github.com/thecodingmachine/gotenberg/internal/pkg/random
|
go test -race -cover -covermode=atomic github.com/thecodingmachine/gotenberg/internal/pkg/random
|
||||||
go test -race -cover -covermode=atomic github.com/thecodingmachine/gotenberg/internal/pkg/standarderror
|
go test -race -cover -covermode=atomic github.com/thecodingmachine/gotenberg/internal/pkg/standarderror
|
||||||
go test -race -cover -covermode=atomic github.com/thecodingmachine/gotenberg/internal/pkg/timeout
|
go test -race -cover -covermode=atomic github.com/thecodingmachine/gotenberg/internal/pkg/timeout
|
||||||
go test -race -cover -covermode=atomic github.com/thecodingmachine/gotenberg/internal/pkg/printer
|
|
||||||
go test -race -cover -covermode=atomic github.com/thecodingmachine/gotenberg/internal/app/api
|
go test -race -cover -covermode=atomic github.com/thecodingmachine/gotenberg/internal/app/api
|
||||||
|
|
||||||
# Finally testing processes shutdown.
|
# Finally testing processes shutdown.
|
||||||
|
|||||||
388
internal/app/api/api_test.go
Normal file
388
internal/app/api/api_test.go
Normal file
@@ -0,0 +1,388 @@
|
|||||||
|
package api
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/labstack/echo/v4"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"github.com/thecodingmachine/gotenberg/internal/app/api/pkg/handler"
|
||||||
|
"github.com/thecodingmachine/gotenberg/internal/app/api/pkg/middleware"
|
||||||
|
"github.com/thecodingmachine/gotenberg/internal/app/api/pkg/resource"
|
||||||
|
"github.com/thecodingmachine/gotenberg/internal/pkg/config"
|
||||||
|
"github.com/thecodingmachine/gotenberg/test"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestPing(t *testing.T) {
|
||||||
|
endpoint := handler.PingEndpoint
|
||||||
|
config, err := config.FromEnv()
|
||||||
|
assert.Nil(t, err)
|
||||||
|
srv := New(config)
|
||||||
|
// should be OK.
|
||||||
|
req := httptest.NewRequest(http.MethodGet, endpoint, nil)
|
||||||
|
test.AssertStatusCode(t, http.StatusOK, srv, req)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMerge(t *testing.T) {
|
||||||
|
os.Setenv(middleware.TestingTraceEnvVar, "1")
|
||||||
|
endpoint := handler.MergeEndpoint
|
||||||
|
config, err := config.FromEnv()
|
||||||
|
assert.Nil(t, err)
|
||||||
|
srv := New(config)
|
||||||
|
// should be OK.
|
||||||
|
body, contentType := test.PDFTestMultipartForm(t, nil)
|
||||||
|
req := httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusOK, srv, req)
|
||||||
|
// bad request.
|
||||||
|
body, contentType = test.PDFTestMultipartForm(t, map[string]string{resource.WaitTimeoutFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
// timeout.
|
||||||
|
body, contentType = test.PDFTestMultipartForm(t, map[string]string{resource.WaitTimeoutFormField: "0"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusRequestTimeout, srv, req)
|
||||||
|
// should have no more resources.
|
||||||
|
test.AssertDirectoryEmpty(t, middleware.TestsTracePrefix)
|
||||||
|
err = os.RemoveAll(middleware.TestsTracePrefix)
|
||||||
|
assert.Nil(t, err)
|
||||||
|
os.Unsetenv(middleware.TestingTraceEnvVar)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestHTML(t *testing.T) {
|
||||||
|
os.Setenv(middleware.TestingTraceEnvVar, "1")
|
||||||
|
endpoint := fmt.Sprintf("%s%s", handler.ConvertGroupEndpoint, handler.HTMLEndpoint)
|
||||||
|
config, err := config.FromEnv()
|
||||||
|
assert.Nil(t, err)
|
||||||
|
srv := New(config)
|
||||||
|
// should be OK.
|
||||||
|
body, contentType := test.HTMLTestMultipartForm(t, nil)
|
||||||
|
req := httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusOK, srv, req)
|
||||||
|
// bad request.
|
||||||
|
body, contentType = test.HTMLTestMultipartForm(t, map[string]string{resource.WaitTimeoutFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.HTMLTestMultipartForm(t, map[string]string{resource.WaitDelayFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.HTMLTestMultipartForm(t, map[string]string{resource.PaperWidthFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.HTMLTestMultipartForm(t, map[string]string{resource.PaperHeightFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.HTMLTestMultipartForm(t, map[string]string{resource.MarginTopFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.HTMLTestMultipartForm(t, map[string]string{resource.MarginBottomFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.HTMLTestMultipartForm(t, map[string]string{resource.MarginLeftFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.HTMLTestMultipartForm(t, map[string]string{resource.MarginRightFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.HTMLTestMultipartForm(t, map[string]string{resource.LandscapeFormField: "not a bool"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
// timeout.
|
||||||
|
body, contentType = test.HTMLTestMultipartForm(t, map[string]string{resource.WaitTimeoutFormField: "0"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusRequestTimeout, srv, req)
|
||||||
|
// should have no more resources.
|
||||||
|
test.AssertDirectoryEmpty(t, middleware.TestsTracePrefix)
|
||||||
|
err = os.RemoveAll(middleware.TestsTracePrefix)
|
||||||
|
assert.Nil(t, err)
|
||||||
|
os.Unsetenv(middleware.TestingTraceEnvVar)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMarkdown(t *testing.T) {
|
||||||
|
os.Setenv(middleware.TestingTraceEnvVar, "1")
|
||||||
|
endpoint := fmt.Sprintf("%s%s", handler.ConvertGroupEndpoint, handler.MarkdownEndpoint)
|
||||||
|
config, err := config.FromEnv()
|
||||||
|
assert.Nil(t, err)
|
||||||
|
srv := New(config)
|
||||||
|
// should be OK.
|
||||||
|
body, contentType := test.MarkdownTestMultipartForm(t, nil)
|
||||||
|
req := httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusOK, srv, req)
|
||||||
|
// bad request.
|
||||||
|
body, contentType = test.MarkdownTestMultipartForm(t, map[string]string{resource.WaitTimeoutFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.MarkdownTestMultipartForm(t, map[string]string{resource.WaitDelayFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.MarkdownTestMultipartForm(t, map[string]string{resource.PaperWidthFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.MarkdownTestMultipartForm(t, map[string]string{resource.PaperHeightFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.MarkdownTestMultipartForm(t, map[string]string{resource.MarginTopFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.MarkdownTestMultipartForm(t, map[string]string{resource.MarginBottomFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.MarkdownTestMultipartForm(t, map[string]string{resource.MarginLeftFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.MarkdownTestMultipartForm(t, map[string]string{resource.MarginRightFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.MarkdownTestMultipartForm(t, map[string]string{resource.LandscapeFormField: "not a bool"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
// timeout.
|
||||||
|
body, contentType = test.MarkdownTestMultipartForm(t, map[string]string{resource.WaitTimeoutFormField: "0"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusRequestTimeout, srv, req)
|
||||||
|
// should have no more resources.
|
||||||
|
test.AssertDirectoryEmpty(t, middleware.TestsTracePrefix)
|
||||||
|
err = os.RemoveAll(middleware.TestsTracePrefix)
|
||||||
|
assert.Nil(t, err)
|
||||||
|
os.Unsetenv(middleware.TestingTraceEnvVar)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestURL(t *testing.T) {
|
||||||
|
os.Setenv(middleware.TestingTraceEnvVar, "1")
|
||||||
|
endpoint := fmt.Sprintf("%s%s", handler.ConvertGroupEndpoint, handler.URLEndpoint)
|
||||||
|
config, err := config.FromEnv()
|
||||||
|
assert.Nil(t, err)
|
||||||
|
srv := New(config)
|
||||||
|
// should be OK.
|
||||||
|
body, contentType := test.URLTestMultipartForm(t, nil)
|
||||||
|
req := httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusOK, srv, req)
|
||||||
|
// bad request.
|
||||||
|
body, contentType = test.URLTestMultipartForm(t, map[string]string{resource.WaitTimeoutFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.URLTestMultipartForm(t, map[string]string{resource.WaitDelayFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.URLTestMultipartForm(t, map[string]string{resource.PaperWidthFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.URLTestMultipartForm(t, map[string]string{resource.PaperHeightFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.URLTestMultipartForm(t, map[string]string{resource.MarginTopFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.URLTestMultipartForm(t, map[string]string{resource.MarginBottomFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.URLTestMultipartForm(t, map[string]string{resource.MarginLeftFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.URLTestMultipartForm(t, map[string]string{resource.MarginRightFormField: "not a float"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
body, contentType = test.URLTestMultipartForm(t, map[string]string{resource.LandscapeFormField: "not a bool"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusBadRequest, srv, req)
|
||||||
|
// timeout.
|
||||||
|
body, contentType = test.URLTestMultipartForm(t, map[string]string{resource.WaitTimeoutFormField: "0"})
|
||||||
|
req = httptest.NewRequest(http.MethodPost, endpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusRequestTimeout, srv, req)
|
||||||
|
// should have no more resources.
|
||||||
|
test.AssertDirectoryEmpty(t, middleware.TestsTracePrefix)
|
||||||
|
err = os.RemoveAll(middleware.TestsTracePrefix)
|
||||||
|
assert.Nil(t, err)
|
||||||
|
os.Unsetenv(middleware.TestingTraceEnvVar)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestConcurrent(t *testing.T) {
|
||||||
|
const concurrentRequests int = 4
|
||||||
|
os.Setenv(middleware.TestingTraceEnvVar, "1")
|
||||||
|
config, err := config.FromEnv()
|
||||||
|
assert.Nil(t, err)
|
||||||
|
srv := New(config)
|
||||||
|
// Merge.
|
||||||
|
test.AssertConcurrent(
|
||||||
|
t,
|
||||||
|
func() error {
|
||||||
|
body, contentType := test.PDFTestMultipartForm(t, map[string]string{resource.WaitTimeoutFormField: "120"})
|
||||||
|
req := httptest.NewRequest(http.MethodPost, handler.MergeEndpoint, body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
srv.ServeHTTP(rec, req)
|
||||||
|
if rec.Code != http.StatusOK {
|
||||||
|
return fmt.Errorf("wrong status code: want '%d' got '%d'", http.StatusOK, rec.Code)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
concurrentRequests,
|
||||||
|
)
|
||||||
|
// HTML.
|
||||||
|
test.AssertConcurrent(
|
||||||
|
t,
|
||||||
|
func() error {
|
||||||
|
body, contentType := test.HTMLTestMultipartForm(t, map[string]string{resource.WaitTimeoutFormField: "120"})
|
||||||
|
req := httptest.NewRequest(http.MethodPost, fmt.Sprintf("%s%s", handler.ConvertGroupEndpoint, handler.HTMLEndpoint), body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
srv.ServeHTTP(rec, req)
|
||||||
|
if rec.Code != http.StatusOK {
|
||||||
|
return fmt.Errorf("wrong status code: want '%d' got '%d'", http.StatusOK, rec.Code)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
concurrentRequests,
|
||||||
|
)
|
||||||
|
// Markdown.
|
||||||
|
test.AssertConcurrent(
|
||||||
|
t,
|
||||||
|
func() error {
|
||||||
|
body, contentType := test.MarkdownTestMultipartForm(t, map[string]string{resource.WaitTimeoutFormField: "120"})
|
||||||
|
req := httptest.NewRequest(http.MethodPost, fmt.Sprintf("%s%s", handler.ConvertGroupEndpoint, handler.MarkdownEndpoint), body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
srv.ServeHTTP(rec, req)
|
||||||
|
if rec.Code != http.StatusOK {
|
||||||
|
return fmt.Errorf("wrong status code: want '%d' got '%d'", http.StatusOK, rec.Code)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
concurrentRequests,
|
||||||
|
)
|
||||||
|
// URL.
|
||||||
|
test.AssertConcurrent(
|
||||||
|
t,
|
||||||
|
func() error {
|
||||||
|
body, contentType := test.URLTestMultipartForm(t, map[string]string{resource.WaitTimeoutFormField: "120"})
|
||||||
|
req := httptest.NewRequest(http.MethodPost, fmt.Sprintf("%s%s", handler.ConvertGroupEndpoint, handler.URLEndpoint), body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
srv.ServeHTTP(rec, req)
|
||||||
|
if rec.Code != http.StatusOK {
|
||||||
|
return fmt.Errorf("wrong status code: want '%d' got '%d'", http.StatusOK, rec.Code)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
concurrentRequests,
|
||||||
|
)
|
||||||
|
// Office.
|
||||||
|
test.AssertConcurrent(
|
||||||
|
t,
|
||||||
|
func() error {
|
||||||
|
body, contentType := test.OfficeTestMultipartForm(t, map[string]string{resource.WaitTimeoutFormField: "120"})
|
||||||
|
req := httptest.NewRequest(http.MethodPost, fmt.Sprintf("%s%s", handler.ConvertGroupEndpoint, handler.OfficeEndpoint), body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
srv.ServeHTTP(rec, req)
|
||||||
|
if rec.Code != http.StatusOK {
|
||||||
|
return fmt.Errorf("wrong status code: want '%d' got '%d'", http.StatusOK, rec.Code)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
concurrentRequests,
|
||||||
|
)
|
||||||
|
// should have no more resources.
|
||||||
|
test.AssertDirectoryEmpty(t, middleware.TestsTracePrefix)
|
||||||
|
err = os.RemoveAll(middleware.TestsTracePrefix)
|
||||||
|
assert.Nil(t, err)
|
||||||
|
os.Unsetenv(middleware.TestingTraceEnvVar)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestWebhook(t *testing.T) {
|
||||||
|
status := make(chan error, 2)
|
||||||
|
rcv := echo.New()
|
||||||
|
rcv.POST("/foo", func(c echo.Context) error {
|
||||||
|
if c.Request().Header.Get("Content-type") != "application/pdf" {
|
||||||
|
status <- fmt.Errorf("wrong Content-type: got '%s' want '%s'", c.Request().Header.Get("Content-type"), "application/pdf")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
body, err := ioutil.ReadAll(c.Request().Body)
|
||||||
|
if err != nil {
|
||||||
|
status <- err
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if body == nil || len(body) == 0 {
|
||||||
|
status <- errors.New("empty body")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
status <- nil
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
go func() {
|
||||||
|
rcv.Start(":3001")
|
||||||
|
}()
|
||||||
|
os.Setenv(middleware.TestingTraceEnvVar, "1")
|
||||||
|
config, err := config.FromEnv()
|
||||||
|
assert.Nil(t, err)
|
||||||
|
srv := New(config)
|
||||||
|
body, contentType := test.PDFTestMultipartForm(t, map[string]string{resource.WebhookURLFormField: "http://localhost:3001/foo"})
|
||||||
|
req := httptest.NewRequest(http.MethodPost, "/merge", body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
test.AssertStatusCode(t, http.StatusOK, srv, req)
|
||||||
|
err = <-status
|
||||||
|
assert.NoError(t, err)
|
||||||
|
// should have no more resources.
|
||||||
|
test.AssertDirectoryEmpty(t, middleware.TestsTracePrefix)
|
||||||
|
err = os.RemoveAll(middleware.TestsTracePrefix)
|
||||||
|
assert.Nil(t, err)
|
||||||
|
os.Unsetenv(middleware.TestingTraceEnvVar)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestResultFilename(t *testing.T) {
|
||||||
|
os.Setenv(middleware.TestingTraceEnvVar, "1")
|
||||||
|
config, err := config.FromEnv()
|
||||||
|
assert.Nil(t, err)
|
||||||
|
srv := New(config)
|
||||||
|
body, contentType := test.PDFTestMultipartForm(t, map[string]string{resource.ResultFilenameFormField: "foo.pdf"})
|
||||||
|
req := httptest.NewRequest(http.MethodPost, "/merge", body)
|
||||||
|
req.Header.Set(echo.HeaderContentType, contentType)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
srv.ServeHTTP(rec, req)
|
||||||
|
assert.Equal(t, "attachment; filename=\"foo.pdf\"", rec.Header().Get("Content-Disposition"))
|
||||||
|
// should have no more resources.
|
||||||
|
test.AssertDirectoryEmpty(t, middleware.TestsTracePrefix)
|
||||||
|
err = os.RemoveAll(middleware.TestsTracePrefix)
|
||||||
|
assert.Nil(t, err)
|
||||||
|
os.Unsetenv(middleware.TestingTraceEnvVar)
|
||||||
|
}
|
||||||
@@ -36,7 +36,7 @@ func New(c echo.Context, logger *logger.Logger, config *config.Config) *Context
|
|||||||
// MustCastFromEchoContext cast an echo.Context to our custom
|
// MustCastFromEchoContext cast an echo.Context to our custom
|
||||||
// context. If something goes wrong, panic.
|
// context. If something goes wrong, panic.
|
||||||
func MustCastFromEchoContext(c echo.Context) *Context {
|
func MustCastFromEchoContext(c echo.Context) *Context {
|
||||||
const op = "context.MustCastFromEchoContext"
|
const op string = "context.MustCastFromEchoContext"
|
||||||
ctx, ok := c.(*Context)
|
ctx, ok := c.(*Context)
|
||||||
if !ok {
|
if !ok {
|
||||||
panic(fmt.Sprintf("%s: unable to cast an echo.Context to a custom context", op))
|
panic(fmt.Sprintf("%s: unable to cast an echo.Context to a custom context", op))
|
||||||
@@ -60,7 +60,7 @@ func (ctx *Context) Resource() *resource.Resource {
|
|||||||
|
|
||||||
// WithResource adds a resource to the context.
|
// WithResource adds a resource to the context.
|
||||||
func (ctx *Context) WithResource(resourceDirPath string) error {
|
func (ctx *Context) WithResource(resourceDirPath string) error {
|
||||||
const op = "context.WithResource"
|
const op string = "context.WithResource"
|
||||||
r, err := resource.New(ctx, ctx.logger, ctx.config, resourceDirPath)
|
r, err := resource.New(ctx, ctx.logger, ctx.config, resourceDirPath)
|
||||||
ctx.resource = r
|
ctx.resource = r
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -75,7 +75,7 @@ func (ctx *Context) WithResource(resourceDirPath string) error {
|
|||||||
// LogRequestResult logs the result of a request.
|
// LogRequestResult logs the result of a request.
|
||||||
// This method should only be used by a middleware!
|
// This method should only be used by a middleware!
|
||||||
func (ctx *Context) LogRequestResult(err error, isDebug bool) error {
|
func (ctx *Context) LogRequestResult(err error, isDebug bool) error {
|
||||||
const op = "context.LogRequestResult"
|
const op string = "context.LogRequestResult"
|
||||||
req := ctx.Request()
|
req := ctx.Request()
|
||||||
resp := ctx.Response()
|
resp := ctx.Response()
|
||||||
stopTime := time.Now()
|
stopTime := time.Now()
|
||||||
|
|||||||
@@ -35,7 +35,7 @@ const (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func convert(ctx *context.Context, p printer.Printer) error {
|
func convert(ctx *context.Context, p printer.Printer) error {
|
||||||
const op = "handler.convert"
|
const op string = "handler.convert"
|
||||||
r := ctx.Resource()
|
r := ctx.Resource()
|
||||||
logger := ctx.StandardLogger()
|
logger := ctx.StandardLogger()
|
||||||
baseFilename := random.Get()
|
baseFilename := random.Get()
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ import (
|
|||||||
// HTML is the endpoint for converting
|
// HTML is the endpoint for converting
|
||||||
// HTML to PDF.
|
// HTML to PDF.
|
||||||
func HTML(c echo.Context) error {
|
func HTML(c echo.Context) error {
|
||||||
const op = "handler.HTML"
|
const op string = "handler.HTML"
|
||||||
ctx := context.MustCastFromEchoContext(c)
|
ctx := context.MustCastFromEchoContext(c)
|
||||||
ctx.StandardLogger().DebugfOp(op, "html request")
|
ctx.StandardLogger().DebugfOp(op, "html request")
|
||||||
r := ctx.Resource()
|
r := ctx.Resource()
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ import (
|
|||||||
// Markdown is the endpoint for converting
|
// Markdown is the endpoint for converting
|
||||||
// Markdown to PDF.
|
// Markdown to PDF.
|
||||||
func Markdown(c echo.Context) error {
|
func Markdown(c echo.Context) error {
|
||||||
const op = "handler.Markdown"
|
const op string = "handler.Markdown"
|
||||||
ctx := context.MustCastFromEchoContext(c)
|
ctx := context.MustCastFromEchoContext(c)
|
||||||
ctx.StandardLogger().DebugfOp(op, "markdown request")
|
ctx.StandardLogger().DebugfOp(op, "markdown request")
|
||||||
r := ctx.Resource()
|
r := ctx.Resource()
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ import (
|
|||||||
// Merge is the endpoint for
|
// Merge is the endpoint for
|
||||||
// merging PDF files.
|
// merging PDF files.
|
||||||
func Merge(c echo.Context) error {
|
func Merge(c echo.Context) error {
|
||||||
const op = "handler.Merge"
|
const op string = "handler.Merge"
|
||||||
ctx := context.MustCastFromEchoContext(c)
|
ctx := context.MustCastFromEchoContext(c)
|
||||||
ctx.StandardLogger().DebugfOp(op, "merge request")
|
ctx.StandardLogger().DebugfOp(op, "merge request")
|
||||||
r := ctx.Resource()
|
r := ctx.Resource()
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ import (
|
|||||||
// Office is the endpoint for converting
|
// Office is the endpoint for converting
|
||||||
// Office files to PDF.
|
// Office files to PDF.
|
||||||
func Office(c echo.Context) error {
|
func Office(c echo.Context) error {
|
||||||
const op = "handler.Office"
|
const op string = "handler.Office"
|
||||||
ctx := context.MustCastFromEchoContext(c)
|
ctx := context.MustCastFromEchoContext(c)
|
||||||
ctx.StandardLogger().DebugfOp(op, "office request")
|
ctx.StandardLogger().DebugfOp(op, "office request")
|
||||||
r := ctx.Resource()
|
r := ctx.Resource()
|
||||||
|
|||||||
@@ -11,7 +11,7 @@ import (
|
|||||||
// URL is the endpoint for converting
|
// URL is the endpoint for converting
|
||||||
// a URL to PDF.
|
// a URL to PDF.
|
||||||
func URL(c echo.Context) error {
|
func URL(c echo.Context) error {
|
||||||
const op = "handler.URL"
|
const op string = "handler.URL"
|
||||||
ctx := context.MustCastFromEchoContext(c)
|
ctx := context.MustCastFromEchoContext(c)
|
||||||
ctx.StandardLogger().DebugfOp(op, "url request")
|
ctx.StandardLogger().DebugfOp(op, "url request")
|
||||||
r := ctx.Resource()
|
r := ctx.Resource()
|
||||||
|
|||||||
@@ -11,7 +11,7 @@ import (
|
|||||||
func Cleanup() echo.MiddlewareFunc {
|
func Cleanup() echo.MiddlewareFunc {
|
||||||
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
||||||
return func(c echo.Context) error {
|
return func(c echo.Context) error {
|
||||||
const op = "middleware.Cleanup"
|
const op string = "middleware.Cleanup"
|
||||||
err := next(c)
|
err := next(c)
|
||||||
ctx := context.MustCastFromEchoContext(c)
|
ctx := context.MustCastFromEchoContext(c)
|
||||||
r := ctx.Resource()
|
r := ctx.Resource()
|
||||||
|
|||||||
@@ -15,13 +15,13 @@ import (
|
|||||||
const (
|
const (
|
||||||
// TestingTraceEnvVar is an environment
|
// TestingTraceEnvVar is an environment
|
||||||
// variable used in some tests.
|
// variable used in some tests.
|
||||||
TestingTraceEnvVar = "TESTING_TRACE"
|
TestingTraceEnvVar string = "TESTING_TRACE"
|
||||||
// TestsTracePrefix helps
|
// TestsTracePrefix helps
|
||||||
// creating all resources inside a prefix.
|
// creating all resources inside a prefix.
|
||||||
// Only used in some tests
|
// Only used in some tests
|
||||||
// to check if the resources
|
// to check if the resources
|
||||||
// have been removed.
|
// have been removed.
|
||||||
TestsTracePrefix = "tmp"
|
TestsTracePrefix string = "tmp"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Context helps extending the default echo.Context with
|
// Context helps extending the default echo.Context with
|
||||||
|
|||||||
@@ -65,7 +65,7 @@ type Resource struct {
|
|||||||
|
|
||||||
// New creates a new resource.
|
// New creates a new resource.
|
||||||
func New(c echo.Context, logger *logger.Logger, config *config.Config, dirPath string) (*Resource, error) {
|
func New(c echo.Context, logger *logger.Logger, config *config.Config, dirPath string) (*Resource, error) {
|
||||||
const op = "resource.New"
|
const op string = "resource.New"
|
||||||
r := &Resource{
|
r := &Resource{
|
||||||
logger: logger,
|
logger: logger,
|
||||||
config: config,
|
config: config,
|
||||||
@@ -83,7 +83,7 @@ func New(c echo.Context, logger *logger.Logger, config *config.Config, dirPath s
|
|||||||
}
|
}
|
||||||
|
|
||||||
func formValues(c echo.Context, logger *logger.Logger) map[string]string {
|
func formValues(c echo.Context, logger *logger.Logger) map[string]string {
|
||||||
const op = "resource.formValues"
|
const op string = "resource.formValues"
|
||||||
v := make(map[string]string)
|
v := make(map[string]string)
|
||||||
fetch := func(formField string) string {
|
fetch := func(formField string) string {
|
||||||
value := c.FormValue(formField)
|
value := c.FormValue(formField)
|
||||||
@@ -110,7 +110,7 @@ func formValues(c echo.Context, logger *logger.Logger) map[string]string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func formFiles(c echo.Context, logger *logger.Logger, dirPath string) error {
|
func formFiles(c echo.Context, logger *logger.Logger, dirPath string) error {
|
||||||
const op = "resource.formFiles"
|
const op string = "resource.formFiles"
|
||||||
form, err := c.MultipartForm()
|
form, err := c.MultipartForm()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return &standarderror.Error{Op: op, Err: err}
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
@@ -153,7 +153,7 @@ func (r *Resource) DirPath() string {
|
|||||||
// Close deletes the working directory of the
|
// Close deletes the working directory of the
|
||||||
// resource if it exists.
|
// resource if it exists.
|
||||||
func (r *Resource) Close() error {
|
func (r *Resource) Close() error {
|
||||||
const op = "resource.Close"
|
const op string = "resource.Close"
|
||||||
if _, err := os.Stat(r.formFilesDirPath); os.IsNotExist(err) {
|
if _, err := os.Stat(r.formFilesDirPath); os.IsNotExist(err) {
|
||||||
r.logger.DebugfOp(op, "directory '%s' does not exist, nothing to remove", r.formFilesDirPath)
|
r.logger.DebugfOp(op, "directory '%s' does not exist, nothing to remove", r.formFilesDirPath)
|
||||||
return nil
|
return nil
|
||||||
@@ -171,7 +171,7 @@ const defaultHeaderFooterHTML string = "<html><head></head><body></body></html>"
|
|||||||
// thanks to the form values and form files from the request
|
// thanks to the form values and form files from the request
|
||||||
// plus the default values from the configuration.
|
// plus the default values from the configuration.
|
||||||
func (r *Resource) ChromePrinterOptions() (*printer.ChromeOptions, error) {
|
func (r *Resource) ChromePrinterOptions() (*printer.ChromeOptions, error) {
|
||||||
const op = "resource.ChromePrinterOptions"
|
const op string = "resource.ChromePrinterOptions"
|
||||||
waitTimeout, err := r.float64(WaitTimeoutFormField, r.config.DefaultWaitTimeout())
|
waitTimeout, err := r.float64(WaitTimeoutFormField, r.config.DefaultWaitTimeout())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, &standarderror.Error{Op: op, Err: err}
|
return nil, &standarderror.Error{Op: op, Err: err}
|
||||||
@@ -237,7 +237,7 @@ func (r *Resource) ChromePrinterOptions() (*printer.ChromeOptions, error) {
|
|||||||
// thanks to the form values from the request
|
// thanks to the form values from the request
|
||||||
// plus the default values from the configuration.
|
// plus the default values from the configuration.
|
||||||
func (r *Resource) OfficePrinterOptions() (*printer.OfficeOptions, error) {
|
func (r *Resource) OfficePrinterOptions() (*printer.OfficeOptions, error) {
|
||||||
const op = "resource.OfficePrinterOptions"
|
const op string = "resource.OfficePrinterOptions"
|
||||||
waitTimeout, err := r.float64(WaitTimeoutFormField, r.config.DefaultWaitTimeout())
|
waitTimeout, err := r.float64(WaitTimeoutFormField, r.config.DefaultWaitTimeout())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, &standarderror.Error{Op: op, Err: err}
|
return nil, &standarderror.Error{Op: op, Err: err}
|
||||||
@@ -258,7 +258,7 @@ func (r *Resource) OfficePrinterOptions() (*printer.OfficeOptions, error) {
|
|||||||
// thanks to the form values from the request
|
// thanks to the form values from the request
|
||||||
// plus the default values from the configuration.
|
// plus the default values from the configuration.
|
||||||
func (r *Resource) MergePrinterOptions() (*printer.MergeOptions, error) {
|
func (r *Resource) MergePrinterOptions() (*printer.MergeOptions, error) {
|
||||||
const op = "resource.MergePrinterOptions"
|
const op string = "resource.MergePrinterOptions"
|
||||||
waitTimeout, err := r.float64(WaitTimeoutFormField, r.config.DefaultWaitTimeout())
|
waitTimeout, err := r.float64(WaitTimeoutFormField, r.config.DefaultWaitTimeout())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, &standarderror.Error{Op: op, Err: err}
|
return nil, &standarderror.Error{Op: op, Err: err}
|
||||||
@@ -289,7 +289,7 @@ func (r *Resource) hasFile(filename string) bool {
|
|||||||
|
|
||||||
// Get returns the form field value.
|
// Get returns the form field value.
|
||||||
func (r *Resource) Get(formField string) (string, error) {
|
func (r *Resource) Get(formField string) (string, error) {
|
||||||
const op = "resource.Get"
|
const op string = "resource.Get"
|
||||||
v, err := r.value(formField)
|
v, err := r.value(formField)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", &standarderror.Error{Op: op, Err: err}
|
return "", &standarderror.Error{Op: op, Err: err}
|
||||||
@@ -298,7 +298,7 @@ func (r *Resource) Get(formField string) (string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (r *Resource) value(formField string) (string, error) {
|
func (r *Resource) value(formField string) (string, error) {
|
||||||
const op = "resource.value"
|
const op string = "resource.value"
|
||||||
v, ok := r.formValues[formField]
|
v, ok := r.formValues[formField]
|
||||||
if !ok {
|
if !ok {
|
||||||
return "", &standarderror.Error{
|
return "", &standarderror.Error{
|
||||||
@@ -311,7 +311,7 @@ func (r *Resource) value(formField string) (string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (r *Resource) float64(formField string, defaultValue float64) (float64, error) {
|
func (r *Resource) float64(formField string, defaultValue float64) (float64, error) {
|
||||||
const op = "resource.float64"
|
const op string = "resource.float64"
|
||||||
if !r.Has(formField) {
|
if !r.Has(formField) {
|
||||||
return defaultValue, nil
|
return defaultValue, nil
|
||||||
}
|
}
|
||||||
@@ -331,7 +331,7 @@ func (r *Resource) float64(formField string, defaultValue float64) (float64, err
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (r *Resource) bool(formField string, defaultValue bool) (bool, error) {
|
func (r *Resource) bool(formField string, defaultValue bool) (bool, error) {
|
||||||
const op = "resource.bool"
|
const op string = "resource.bool"
|
||||||
if !r.Has(formField) {
|
if !r.Has(formField) {
|
||||||
return defaultValue, nil
|
return defaultValue, nil
|
||||||
}
|
}
|
||||||
@@ -353,7 +353,7 @@ func (r *Resource) bool(formField string, defaultValue bool) (bool, error) {
|
|||||||
// Fpath returns the path of the given filename.
|
// Fpath returns the path of the given filename.
|
||||||
// This filename should be the name of a form file.
|
// This filename should be the name of a form file.
|
||||||
func (r *Resource) Fpath(filename string) (string, error) {
|
func (r *Resource) Fpath(filename string) (string, error) {
|
||||||
const op = "resource.Fpath"
|
const op string = "resource.Fpath"
|
||||||
fpath := fmt.Sprintf("%s/%s", r.formFilesDirPath, filename)
|
fpath := fmt.Sprintf("%s/%s", r.formFilesDirPath, filename)
|
||||||
_, err := os.Stat(fpath)
|
_, err := os.Stat(fpath)
|
||||||
if os.IsNotExist(err) {
|
if os.IsNotExist(err) {
|
||||||
@@ -371,7 +371,7 @@ func (r *Resource) Fpath(filename string) (string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (r *Resource) content(filename string, defaultValue string) (string, error) {
|
func (r *Resource) content(filename string, defaultValue string) (string, error) {
|
||||||
const op = "resource.content"
|
const op string = "resource.content"
|
||||||
if !r.hasFile(filename) {
|
if !r.hasFile(filename) {
|
||||||
return defaultValue, nil
|
return defaultValue, nil
|
||||||
}
|
}
|
||||||
@@ -389,7 +389,7 @@ func (r *Resource) content(filename string, defaultValue string) (string, error)
|
|||||||
// Fpaths returns the list of files of the resource
|
// Fpaths returns the list of files of the resource
|
||||||
// according to given file extensions.
|
// according to given file extensions.
|
||||||
func (r *Resource) Fpaths(exts ...string) ([]string, error) {
|
func (r *Resource) Fpaths(exts ...string) ([]string, error) {
|
||||||
const op = "resource.Fpaths"
|
const op string = "resource.Fpaths"
|
||||||
var fpaths []string
|
var fpaths []string
|
||||||
err := filepath.Walk(r.formFilesDirPath, func(path string, info os.FileInfo, _ error) error {
|
err := filepath.Walk(r.formFilesDirPath, func(path string, info os.FileInfo, _ error) error {
|
||||||
if info.IsDir() {
|
if info.IsDir() {
|
||||||
|
|||||||
@@ -10,11 +10,11 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
defaultWaitTimeoutEnvVar = "DEFAULT_WAIT_TIMEOUT"
|
defaultWaitTimeoutEnvVar string = "DEFAULT_WAIT_TIMEOUT"
|
||||||
defaultListenPortEnvVar = "DEFAULT_LISTEN_PORT"
|
defaultListenPortEnvVar string = "DEFAULT_LISTEN_PORT"
|
||||||
disableGoogleChromeEnvVar = "DISABLE_GOOGLE_CHROME"
|
disableGoogleChromeEnvVar string = "DISABLE_GOOGLE_CHROME"
|
||||||
disableUnoconvEnvVar = "DISABLE_UNOCONV"
|
disableUnoconvEnvVar string = "DISABLE_UNOCONV"
|
||||||
logLevelEnvVar = "LOG_LEVEL"
|
logLevelEnvVar string = "LOG_LEVEL"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Config contains the application
|
// Config contains the application
|
||||||
@@ -40,7 +40,7 @@ func defaultConfig() *Config {
|
|||||||
// FromEnv fetches configuration
|
// FromEnv fetches configuration
|
||||||
// from environment variables.
|
// from environment variables.
|
||||||
func FromEnv() (*Config, error) {
|
func FromEnv() (*Config, error) {
|
||||||
const op = "config.FromEnv"
|
const op string = "config.FromEnv"
|
||||||
c := defaultConfig()
|
c := defaultConfig()
|
||||||
defaultWaitTimeout, err := defaultWaitTimeoutFromEnv(defaultWaitTimeoutEnvVar, c.DefaultWaitTimeout())
|
defaultWaitTimeout, err := defaultWaitTimeoutFromEnv(defaultWaitTimeoutEnvVar, c.DefaultWaitTimeout())
|
||||||
c.defaultWaitTimeout = defaultWaitTimeout
|
c.defaultWaitTimeout = defaultWaitTimeout
|
||||||
@@ -103,7 +103,7 @@ func (c *Config) LogLevel() logrus.Level {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func defaultWaitTimeoutFromEnv(envVar string, defaultValue float64) (float64, error) {
|
func defaultWaitTimeoutFromEnv(envVar string, defaultValue float64) (float64, error) {
|
||||||
const op = "config.defaultWaitTimeoutFromEnv"
|
const op string = "config.defaultWaitTimeoutFromEnv"
|
||||||
if v, ok := os.LookupEnv(envVar); ok {
|
if v, ok := os.LookupEnv(envVar); ok {
|
||||||
waitTimeout, err := strconv.ParseFloat(v, 64)
|
waitTimeout, err := strconv.ParseFloat(v, 64)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -119,7 +119,7 @@ func defaultWaitTimeoutFromEnv(envVar string, defaultValue float64) (float64, er
|
|||||||
}
|
}
|
||||||
|
|
||||||
func defaultListenPortFromEnv(envVar string, defaultValue string) (string, error) {
|
func defaultListenPortFromEnv(envVar string, defaultValue string) (string, error) {
|
||||||
const op = "config.defaultListenPortFromEnv"
|
const op string = "config.defaultListenPortFromEnv"
|
||||||
if v, ok := os.LookupEnv(envVar); ok {
|
if v, ok := os.LookupEnv(envVar); ok {
|
||||||
portAsUint, err := strconv.ParseUint(v, 10, 64)
|
portAsUint, err := strconv.ParseUint(v, 10, 64)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -142,7 +142,7 @@ func defaultListenPortFromEnv(envVar string, defaultValue string) (string, error
|
|||||||
}
|
}
|
||||||
|
|
||||||
func boolFromEnv(envVar string, defaultValue bool) (bool, error) {
|
func boolFromEnv(envVar string, defaultValue bool) (bool, error) {
|
||||||
const op = "config.boolFromEnv"
|
const op string = "config.boolFromEnv"
|
||||||
if v, ok := os.LookupEnv(envVar); ok {
|
if v, ok := os.LookupEnv(envVar); ok {
|
||||||
if v != "1" && v != "0" {
|
if v != "1" && v != "0" {
|
||||||
return defaultValue, &standarderror.Error{
|
return defaultValue, &standarderror.Error{
|
||||||
@@ -157,7 +157,7 @@ func boolFromEnv(envVar string, defaultValue bool) (bool, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func logLevelFromEnv(envVar string, defaultValue logrus.Level) (logrus.Level, error) {
|
func logLevelFromEnv(envVar string, defaultValue logrus.Level) (logrus.Level, error) {
|
||||||
const op = "config.logLevelFromEnv"
|
const op string = "config.logLevelFromEnv"
|
||||||
if v, ok := os.LookupEnv(envVar); ok {
|
if v, ok := os.LookupEnv(envVar); ok {
|
||||||
switch v {
|
switch v {
|
||||||
case "DEBUG":
|
case "DEBUG":
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ import (
|
|||||||
"github.com/thecodingmachine/gotenberg/internal/pkg/standarderror"
|
"github.com/thecodingmachine/gotenberg/internal/pkg/standarderror"
|
||||||
)
|
)
|
||||||
|
|
||||||
const chromeWarmupTime = 10 * time.Second
|
const chromeWarmupTime time.Duration = 10 * time.Second
|
||||||
|
|
||||||
type chrome struct {
|
type chrome struct {
|
||||||
manager *processManager
|
manager *processManager
|
||||||
@@ -28,7 +28,7 @@ func (p *chrome) Fullname() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (p *chrome) Start() error {
|
func (p *chrome) Start() error {
|
||||||
const op = "pm2.chrome.Start"
|
const op string = "pm2.chrome.Start"
|
||||||
if err := p.manager.start(p); err != nil {
|
if err := p.manager.start(p); err != nil {
|
||||||
return &standarderror.Error{Op: op, Err: err}
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
}
|
}
|
||||||
@@ -36,7 +36,7 @@ func (p *chrome) Start() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (p *chrome) Shutdown() error {
|
func (p *chrome) Shutdown() error {
|
||||||
const op = "pm2.chrome.Shutdown"
|
const op string = "pm2.chrome.Shutdown"
|
||||||
if err := p.manager.shutdown(p); err != nil {
|
if err := p.manager.shutdown(p); err != nil {
|
||||||
return &standarderror.Error{Op: op, Err: err}
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
}
|
}
|
||||||
@@ -67,7 +67,7 @@ func (p *chrome) name() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (p *chrome) viable() bool {
|
func (p *chrome) viable() bool {
|
||||||
const op = "pm2.chrome.viable"
|
const op string = "pm2.chrome.viable"
|
||||||
// check if Google Chrome is correctly running.
|
// check if Google Chrome is correctly running.
|
||||||
ctx, cancel := context.WithCancel(context.Background())
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
defer cancel()
|
defer cancel()
|
||||||
@@ -93,7 +93,7 @@ func (p *chrome) viable() bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (p *chrome) warmup() {
|
func (p *chrome) warmup() {
|
||||||
const op = "pm2.chrome.warmup"
|
const op string = "pm2.chrome.warmup"
|
||||||
p.manager.logger.DebugfOp(
|
p.manager.logger.DebugfOp(
|
||||||
op,
|
op,
|
||||||
"allowing %v to startup",
|
"allowing %v to startup",
|
||||||
|
|||||||
@@ -12,7 +12,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
stoppedState = iota
|
stoppedState int32 = iota
|
||||||
runningState
|
runningState
|
||||||
errorState
|
errorState
|
||||||
)
|
)
|
||||||
@@ -35,7 +35,7 @@ type processManager struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (m *processManager) start(p Process) error {
|
func (m *processManager) start(p Process) error {
|
||||||
const op = "pm2.start"
|
const op string = "pm2.start"
|
||||||
if err := m.pm2(p, "start"); err != nil {
|
if err := m.pm2(p, "start"); err != nil {
|
||||||
return &standarderror.Error{Op: op, Err: err}
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
}
|
}
|
||||||
@@ -63,7 +63,7 @@ func (m *processManager) start(p Process) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (m *processManager) shutdown(p Process) error {
|
func (m *processManager) shutdown(p Process) error {
|
||||||
const op = "pm2.shutdown"
|
const op string = "pm2.shutdown"
|
||||||
if m.heuristicState != runningState {
|
if m.heuristicState != runningState {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -76,7 +76,7 @@ func (m *processManager) shutdown(p Process) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (m *processManager) pm2(p Process, cmdName string) error {
|
func (m *processManager) pm2(p Process, cmdName string) error {
|
||||||
const op = "pm2.pm2"
|
const op string = "pm2.pm2"
|
||||||
cmdArgs := []string{
|
cmdArgs := []string{
|
||||||
cmdName,
|
cmdName,
|
||||||
p.name(),
|
p.name(),
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ import (
|
|||||||
"github.com/thecodingmachine/gotenberg/internal/pkg/standarderror"
|
"github.com/thecodingmachine/gotenberg/internal/pkg/standarderror"
|
||||||
)
|
)
|
||||||
|
|
||||||
const unoconvWarmupTime = 5 * time.Second
|
const unoconvWarmupTime time.Duration = 5 * time.Second
|
||||||
|
|
||||||
type unoconv struct {
|
type unoconv struct {
|
||||||
manager *processManager
|
manager *processManager
|
||||||
@@ -26,7 +26,7 @@ func (p *unoconv) Fullname() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (p *unoconv) Start() error {
|
func (p *unoconv) Start() error {
|
||||||
const op = "pm2.unoconv.Start"
|
const op string = "pm2.unoconv.Start"
|
||||||
if err := p.manager.start(p); err != nil {
|
if err := p.manager.start(p); err != nil {
|
||||||
return &standarderror.Error{Op: op, Err: err}
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
}
|
}
|
||||||
@@ -34,7 +34,7 @@ func (p *unoconv) Start() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (p *unoconv) Shutdown() error {
|
func (p *unoconv) Shutdown() error {
|
||||||
const op = "pm2.unoconv.Shutdown"
|
const op string = "pm2.unoconv.Shutdown"
|
||||||
if err := p.manager.shutdown(p); err != nil {
|
if err := p.manager.shutdown(p); err != nil {
|
||||||
return &standarderror.Error{Op: op, Err: err}
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
}
|
}
|
||||||
@@ -60,7 +60,7 @@ func (p *unoconv) viable() bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (p *unoconv) warmup() {
|
func (p *unoconv) warmup() {
|
||||||
const op = "pm2.unoconv.warmup"
|
const op string = "pm2.unoconv.warmup"
|
||||||
p.manager.logger.DebugfOp(
|
p.manager.logger.DebugfOp(
|
||||||
op,
|
op,
|
||||||
"allowing %v to startup",
|
"allowing %v to startup",
|
||||||
|
|||||||
@@ -39,83 +39,89 @@ type ChromeOptions struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (p *chrome) Print(destination string) error {
|
func (p *chrome) Print(destination string) error {
|
||||||
const op = "printer.chrome.Print"
|
const op string = "printer.chrome.Print"
|
||||||
ctx, cancel := timeout.Context(p.opts.WaitTimeout + p.opts.WaitDelay)
|
ctx, cancel := timeout.Context(p.opts.WaitTimeout + p.opts.WaitDelay)
|
||||||
defer cancel()
|
defer cancel()
|
||||||
devt, err := devtool.New("http://localhost:9222").Version(ctx)
|
resolver := func() error {
|
||||||
if err != nil {
|
devt, err := devtool.New("http://localhost:9222").Version(ctx)
|
||||||
return handleErrContext(ctx, &standarderror.Error{Op: op, Err: err})
|
if err != nil {
|
||||||
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
|
}
|
||||||
|
// connect to WebSocket URL (page) that speaks the Chrome DevTools Protocol.
|
||||||
|
devtConn, err := rpcc.DialContext(ctx, devt.WebSocketDebuggerURL)
|
||||||
|
if err != nil {
|
||||||
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
|
}
|
||||||
|
defer devtConn.Close() // nolint: errcheck
|
||||||
|
// create a new CDP Client that uses conn.
|
||||||
|
devtClient := cdp.NewClient(devtConn)
|
||||||
|
newContextTarget, err := devtClient.Target.CreateBrowserContext(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
|
}
|
||||||
|
// create a new blank target with the new browser context.
|
||||||
|
createTargetArgs := target.
|
||||||
|
NewCreateTargetArgs("about:blank").
|
||||||
|
SetBrowserContextID(newContextTarget.BrowserContextID)
|
||||||
|
newTarget, err := devtClient.Target.CreateTarget(ctx, createTargetArgs)
|
||||||
|
if err != nil {
|
||||||
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
|
}
|
||||||
|
// connect the client to the new target.
|
||||||
|
newTargetWsURL := fmt.Sprintf("ws://127.0.0.1:9222/devtools/page/%s", newTarget.TargetID)
|
||||||
|
newContextConn, err := rpcc.DialContext(ctx, newTargetWsURL)
|
||||||
|
if err != nil {
|
||||||
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
|
}
|
||||||
|
defer newContextConn.Close() // nolint: errcheck
|
||||||
|
// create a new CDP Client that uses newContextConn.
|
||||||
|
targetClient := cdp.NewClient(newContextConn)
|
||||||
|
closeTargetArgs := target.NewCloseTargetArgs(newTarget.TargetID)
|
||||||
|
// close the target when done.
|
||||||
|
defer targetClient.Target.CloseTarget(ctx, closeTargetArgs) // nolint: errcheck
|
||||||
|
if err := runBatch(
|
||||||
|
// enable all the domain events that we're interested in.
|
||||||
|
func() error { return targetClient.DOM.Enable(ctx) },
|
||||||
|
func() error { return targetClient.Network.Enable(ctx, network.NewEnableArgs()) },
|
||||||
|
func() error { return targetClient.Page.Enable(ctx) },
|
||||||
|
func() error { return targetClient.Runtime.Enable(ctx) },
|
||||||
|
); err != nil {
|
||||||
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
|
}
|
||||||
|
if err := p.navigate(ctx, targetClient); err != nil {
|
||||||
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
|
}
|
||||||
|
print, err := targetClient.Page.PrintToPDF(
|
||||||
|
ctx,
|
||||||
|
page.NewPrintToPDFArgs().
|
||||||
|
SetPaperWidth(p.opts.PaperWidth).
|
||||||
|
SetPaperHeight(p.opts.PaperHeight).
|
||||||
|
SetMarginTop(p.opts.MarginTop).
|
||||||
|
SetMarginBottom(p.opts.MarginBottom).
|
||||||
|
SetMarginLeft(p.opts.MarginLeft).
|
||||||
|
SetMarginRight(p.opts.MarginRight).
|
||||||
|
SetLandscape(p.opts.Landscape).
|
||||||
|
SetDisplayHeaderFooter(true).
|
||||||
|
SetHeaderTemplate(p.opts.HeaderHTML).
|
||||||
|
SetFooterTemplate(p.opts.FooterHTML).
|
||||||
|
SetPrintBackground(true),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
|
}
|
||||||
|
if err := ioutil.WriteFile(destination, print.Data, 0644); err != nil {
|
||||||
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
// connect to WebSocket URL (page) that speaks the Chrome DevTools Protocol.
|
if err := resolver(); err != nil {
|
||||||
devtConn, err := rpcc.DialContext(ctx, devt.WebSocketDebuggerURL)
|
return timeout.Err(ctx, err)
|
||||||
if err != nil {
|
|
||||||
return handleErrContext(ctx, &standarderror.Error{Op: op, Err: err})
|
|
||||||
}
|
|
||||||
defer devtConn.Close() // nolint: errcheck
|
|
||||||
// create a new CDP Client that uses conn.
|
|
||||||
devtClient := cdp.NewClient(devtConn)
|
|
||||||
newContextTarget, err := devtClient.Target.CreateBrowserContext(ctx)
|
|
||||||
if err != nil {
|
|
||||||
return handleErrContext(ctx, &standarderror.Error{Op: op, Err: err})
|
|
||||||
}
|
|
||||||
// create a new blank target with the new browser context.
|
|
||||||
createTargetArgs := target.
|
|
||||||
NewCreateTargetArgs("about:blank").
|
|
||||||
SetBrowserContextID(newContextTarget.BrowserContextID)
|
|
||||||
newTarget, err := devtClient.Target.CreateTarget(ctx, createTargetArgs)
|
|
||||||
if err != nil {
|
|
||||||
return handleErrContext(ctx, &standarderror.Error{Op: op, Err: err})
|
|
||||||
}
|
|
||||||
// connect the client to the new target.
|
|
||||||
newTargetWsURL := fmt.Sprintf("ws://127.0.0.1:9222/devtools/page/%s", newTarget.TargetID)
|
|
||||||
newContextConn, err := rpcc.DialContext(ctx, newTargetWsURL)
|
|
||||||
if err != nil {
|
|
||||||
return handleErrContext(ctx, &standarderror.Error{Op: op, Err: err})
|
|
||||||
}
|
|
||||||
defer newContextConn.Close() // nolint: errcheck
|
|
||||||
// create a new CDP Client that uses newContextConn.
|
|
||||||
targetClient := cdp.NewClient(newContextConn)
|
|
||||||
closeTargetArgs := target.NewCloseTargetArgs(newTarget.TargetID)
|
|
||||||
// close the target when done.
|
|
||||||
defer targetClient.Target.CloseTarget(ctx, closeTargetArgs) // nolint: errcheck
|
|
||||||
if err := runBatch(
|
|
||||||
// enable all the domain events that we're interested in.
|
|
||||||
func() error { return targetClient.DOM.Enable(ctx) },
|
|
||||||
func() error { return targetClient.Network.Enable(ctx, network.NewEnableArgs()) },
|
|
||||||
func() error { return targetClient.Page.Enable(ctx) },
|
|
||||||
func() error { return targetClient.Runtime.Enable(ctx) },
|
|
||||||
); err != nil {
|
|
||||||
return handleErrContext(ctx, &standarderror.Error{Op: op, Err: err})
|
|
||||||
}
|
|
||||||
if err := p.navigate(ctx, targetClient); err != nil {
|
|
||||||
return handleErrContext(ctx, &standarderror.Error{Op: op, Err: err})
|
|
||||||
}
|
|
||||||
print, err := targetClient.Page.PrintToPDF(
|
|
||||||
ctx,
|
|
||||||
page.NewPrintToPDFArgs().
|
|
||||||
SetPaperWidth(p.opts.PaperWidth).
|
|
||||||
SetPaperHeight(p.opts.PaperHeight).
|
|
||||||
SetMarginTop(p.opts.MarginTop).
|
|
||||||
SetMarginBottom(p.opts.MarginBottom).
|
|
||||||
SetMarginLeft(p.opts.MarginLeft).
|
|
||||||
SetMarginRight(p.opts.MarginRight).
|
|
||||||
SetLandscape(p.opts.Landscape).
|
|
||||||
SetDisplayHeaderFooter(true).
|
|
||||||
SetHeaderTemplate(p.opts.HeaderHTML).
|
|
||||||
SetFooterTemplate(p.opts.FooterHTML).
|
|
||||||
SetPrintBackground(true),
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
return handleErrContext(ctx, &standarderror.Error{Op: op, Err: err})
|
|
||||||
}
|
|
||||||
if err := ioutil.WriteFile(destination, print.Data, 0644); err != nil {
|
|
||||||
return &standarderror.Error{Op: op, Err: err}
|
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (p *chrome) navigate(ctx context.Context, client *cdp.Client) error {
|
func (p *chrome) navigate(ctx context.Context, client *cdp.Client) error {
|
||||||
const op = "printer.chrome.navigate"
|
const op string = "printer.chrome.navigate"
|
||||||
// make sure Page events are enabled.
|
// make sure Page events are enabled.
|
||||||
if err := client.Page.Enable(ctx); err != nil {
|
if err := client.Page.Enable(ctx); err != nil {
|
||||||
return &standarderror.Error{Op: op, Err: err}
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ import (
|
|||||||
|
|
||||||
// NewMarkdown returns a Markdown printer.
|
// NewMarkdown returns a Markdown printer.
|
||||||
func NewMarkdown(fpath string, opts *ChromeOptions) (Printer, error) {
|
func NewMarkdown(fpath string, opts *ChromeOptions) (Printer, error) {
|
||||||
const op = "printer.NewMarkdown"
|
const op string = "printer.NewMarkdown"
|
||||||
tmpl, err := template.
|
tmpl, err := template.
|
||||||
New(filepath.Base(fpath)).
|
New(filepath.Base(fpath)).
|
||||||
Funcs(template.FuncMap{"toHTML": markdownToHTML}).
|
Funcs(template.FuncMap{"toHTML": markdownToHTML}).
|
||||||
@@ -46,7 +46,7 @@ type templateData struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func markdownToHTML(dirPath, filename string) (template.HTML, error) {
|
func markdownToHTML(dirPath, filename string) (template.HTML, error) {
|
||||||
const op = "printer.markdownToHTML"
|
const op string = "printer.markdownToHTML"
|
||||||
fpath := fmt.Sprintf("%s/%s", dirPath, filename)
|
fpath := fmt.Sprintf("%s/%s", dirPath, filename)
|
||||||
b, err := ioutil.ReadFile(fpath)
|
b, err := ioutil.ReadFile(fpath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
@@ -29,19 +29,25 @@ func NewMerge(fpaths []string, opts *MergeOptions) Printer {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (p *merge) Print(destination string) error {
|
func (p *merge) Print(destination string) error {
|
||||||
const op = "printer.merge.Print"
|
const op string = "printer.merge.Print"
|
||||||
if p.ctx == nil {
|
if p.ctx == nil {
|
||||||
ctx, cancel := timeout.Context(p.opts.WaitTimeout)
|
ctx, cancel := timeout.Context(p.opts.WaitTimeout)
|
||||||
defer cancel()
|
defer cancel()
|
||||||
p.ctx = ctx
|
p.ctx = ctx
|
||||||
}
|
}
|
||||||
var cmdArgs []string
|
resolver := func() error {
|
||||||
cmdArgs = append(cmdArgs, p.fpaths...)
|
var cmdArgs []string
|
||||||
cmdArgs = append(cmdArgs, "cat", "output", destination)
|
cmdArgs = append(cmdArgs, p.fpaths...)
|
||||||
cmd := exec.CommandContext(p.ctx, "pdftk", cmdArgs...)
|
cmdArgs = append(cmdArgs, "cat", "output", destination)
|
||||||
_, err := cmd.Output()
|
cmd := exec.CommandContext(p.ctx, "pdftk", cmdArgs...)
|
||||||
if err != nil {
|
_, err := cmd.Output()
|
||||||
return handleErrContext(p.ctx, &standarderror.Error{Op: op, Err: err})
|
if err != nil {
|
||||||
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if err := resolver(); err != nil {
|
||||||
|
return timeout.Err(p.ctx, err)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -34,18 +34,24 @@ func NewOffice(fpaths []string, opts *OfficeOptions) Printer {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (p *office) Print(destination string) error {
|
func (p *office) Print(destination string) error {
|
||||||
const op = "printer.office.Print"
|
const op string = "printer.office.Print"
|
||||||
ctx, cancel := timeout.Context(p.opts.WaitTimeout)
|
ctx, cancel := timeout.Context(p.opts.WaitTimeout)
|
||||||
defer cancel()
|
defer cancel()
|
||||||
fpaths := make([]string, len(p.fpaths))
|
fpaths := make([]string, len(p.fpaths))
|
||||||
dirPath := filepath.Dir(destination)
|
resolver := func() error {
|
||||||
for i, fpath := range p.fpaths {
|
dirPath := filepath.Dir(destination)
|
||||||
baseFilename := random.String(32)
|
for i, fpath := range p.fpaths {
|
||||||
tmpDest := fmt.Sprintf("%s/%d%s.pdf", dirPath, i, baseFilename)
|
baseFilename := random.String(32)
|
||||||
if err := unoconv(ctx, fpath, tmpDest, p.opts); err != nil {
|
tmpDest := fmt.Sprintf("%s/%d%s.pdf", dirPath, i, baseFilename)
|
||||||
return handleErrContext(ctx, &standarderror.Error{Op: op, Err: err})
|
if err := unoconv(ctx, fpath, tmpDest, p.opts); err != nil {
|
||||||
|
return &standarderror.Error{Op: op, Err: err}
|
||||||
|
}
|
||||||
|
fpaths[i] = tmpDest
|
||||||
}
|
}
|
||||||
fpaths[i] = tmpDest
|
return nil
|
||||||
|
}
|
||||||
|
if err := resolver(); err != nil {
|
||||||
|
return timeout.Err(ctx, err)
|
||||||
}
|
}
|
||||||
if len(fpaths) == 1 {
|
if len(fpaths) == 1 {
|
||||||
if err := os.Rename(fpaths[0], destination); err != nil {
|
if err := os.Rename(fpaths[0], destination); err != nil {
|
||||||
@@ -67,7 +73,7 @@ func (p *office) Print(destination string) error {
|
|||||||
var mu sync.Mutex
|
var mu sync.Mutex
|
||||||
|
|
||||||
func unoconv(ctx context.Context, fpath, destination string, opts *OfficeOptions) error {
|
func unoconv(ctx context.Context, fpath, destination string, opts *OfficeOptions) error {
|
||||||
const op = "printer.unoconv"
|
const op string = "printer.unoconv"
|
||||||
mu.Lock()
|
mu.Lock()
|
||||||
defer mu.Unlock()
|
defer mu.Unlock()
|
||||||
cmdArgs := []string{
|
cmdArgs := []string{
|
||||||
|
|||||||
@@ -1,39 +1,7 @@
|
|||||||
package printer
|
package printer
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"fmt"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/thecodingmachine/gotenberg/internal/pkg/standarderror"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Printer is a type that can create a PDF file from a source.
|
// Printer is a type that can create a PDF file from a source.
|
||||||
// The source is defined in the underlying implementation.
|
// The source is defined in the underlying implementation.
|
||||||
type Printer interface {
|
type Printer interface {
|
||||||
Print(destination string) error
|
Print(destination string) error
|
||||||
}
|
}
|
||||||
|
|
||||||
func handleErrContext(ctx context.Context, previousErr error) error {
|
|
||||||
const op = "printer.handleErrContext"
|
|
||||||
if previousErr == nil {
|
|
||||||
panic(fmt.Sprintf("%s: previous error should not be nil", op))
|
|
||||||
}
|
|
||||||
err := ctx.Err()
|
|
||||||
if err == nil {
|
|
||||||
return previousErr
|
|
||||||
}
|
|
||||||
if strings.Contains(err.Error(), context.DeadlineExceeded.Error()) {
|
|
||||||
return &standarderror.Error{
|
|
||||||
Code: standarderror.Timeout,
|
|
||||||
Message: "context has timed out",
|
|
||||||
Op: op,
|
|
||||||
Err: previousErr,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return &standarderror.Error{
|
|
||||||
Message: "context finished with an error",
|
|
||||||
Op: op,
|
|
||||||
Err: previousErr,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,35 +0,0 @@
|
|||||||
package printer
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"testing"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
|
||||||
"github.com/thecodingmachine/gotenberg/internal/pkg/standarderror"
|
|
||||||
"github.com/thecodingmachine/gotenberg/internal/pkg/timeout"
|
|
||||||
"github.com/thecodingmachine/gotenberg/test"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestHandlerErr(t *testing.T) {
|
|
||||||
previousErr := errors.New("previous error")
|
|
||||||
// should be OK.
|
|
||||||
ctx, cancel := timeout.Context(5)
|
|
||||||
defer cancel()
|
|
||||||
assert.NotNil(t, handleErrContext(ctx, previousErr))
|
|
||||||
// should timeout.
|
|
||||||
ctx, cancel = timeout.Context(0.5)
|
|
||||||
defer cancel()
|
|
||||||
time.Sleep(timeout.Duration(1))
|
|
||||||
err := handleErrContext(ctx, previousErr)
|
|
||||||
assert.NotNil(t, err)
|
|
||||||
standardized := test.RequireStandardError(t, err)
|
|
||||||
assert.Equal(t, standarderror.Timeout, standardized.Code)
|
|
||||||
// should failed.
|
|
||||||
ctx, cancel = timeout.Context(5)
|
|
||||||
cancel()
|
|
||||||
err = handleErrContext(ctx, previousErr)
|
|
||||||
assert.NotNil(t, err)
|
|
||||||
standardized = test.RequireStandardError(t, err)
|
|
||||||
assert.Equal(t, standarderror.Internal, standarderror.Code(err))
|
|
||||||
}
|
|
||||||
@@ -64,7 +64,7 @@ func Code(err error) string {
|
|||||||
return Internal
|
return Internal
|
||||||
}
|
}
|
||||||
|
|
||||||
const defaultMessage = "an internal error has occurred: please contact technical support"
|
const defaultMessage string = "an internal error has occurred: please contact technical support"
|
||||||
|
|
||||||
// Message returns the human-readable message of the error, if available.
|
// Message returns the human-readable message of the error, if available.
|
||||||
// Otherwise returns a generic error message.
|
// Otherwise returns a generic error message.
|
||||||
|
|||||||
@@ -1,3 +1,3 @@
|
|||||||
// Package timeout helps creating
|
// Package timeout helps managing
|
||||||
// context with timeout.
|
// context with timeout.
|
||||||
package timeout
|
package timeout
|
||||||
|
|||||||
@@ -2,7 +2,11 @@ package timeout
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/thecodingmachine/gotenberg/internal/pkg/standarderror"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Context creates a context with timeout for
|
// Context creates a context with timeout for
|
||||||
@@ -15,3 +19,29 @@ func Context(seconds float64) (context.Context, context.CancelFunc) {
|
|||||||
func Duration(seconds float64) time.Duration {
|
func Duration(seconds float64) time.Duration {
|
||||||
return time.Duration(1000*seconds) * time.Millisecond
|
return time.Duration(1000*seconds) * time.Millisecond
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Err checks if there is an error in the given context
|
||||||
|
// and wraps the previous error inside a standarderror.Error.
|
||||||
|
func Err(ctx context.Context, previousErr error) error {
|
||||||
|
const op string = "timeout.Err"
|
||||||
|
if previousErr == nil {
|
||||||
|
panic(fmt.Sprintf("%s: previous error should not be nil", op))
|
||||||
|
}
|
||||||
|
err := ctx.Err()
|
||||||
|
if err == nil {
|
||||||
|
return previousErr
|
||||||
|
}
|
||||||
|
if strings.Contains(err.Error(), context.DeadlineExceeded.Error()) {
|
||||||
|
return &standarderror.Error{
|
||||||
|
Code: standarderror.Timeout,
|
||||||
|
Message: "context has timed out",
|
||||||
|
Op: op,
|
||||||
|
Err: previousErr,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return &standarderror.Error{
|
||||||
|
Message: "context finished with an error",
|
||||||
|
Op: op,
|
||||||
|
Err: previousErr,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,20 +1,40 @@
|
|||||||
package timeout
|
package timeout
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
|
"github.com/thecodingmachine/gotenberg/internal/pkg/standarderror"
|
||||||
|
"github.com/thecodingmachine/gotenberg/test"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestContext(t *testing.T) {
|
|
||||||
ctx, cancel := Context(1.5)
|
|
||||||
assert.NotNil(t, ctx)
|
|
||||||
assert.NotNil(t, cancel)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDuration(t *testing.T) {
|
func TestDuration(t *testing.T) {
|
||||||
expected := time.Duration(1500) * time.Millisecond
|
expected := time.Duration(1500) * time.Millisecond
|
||||||
result := Duration(1.5)
|
result := Duration(1.5)
|
||||||
assert.Equal(t, expected.String(), result.String())
|
assert.Equal(t, expected.String(), result.String())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestErr(t *testing.T) {
|
||||||
|
previousErr := errors.New("previous error")
|
||||||
|
// should be OK.
|
||||||
|
ctx, cancel := Context(5)
|
||||||
|
defer cancel()
|
||||||
|
assert.NotNil(t, Err(ctx, previousErr))
|
||||||
|
// should timeout.
|
||||||
|
ctx, cancel = Context(0.5)
|
||||||
|
defer cancel()
|
||||||
|
time.Sleep(Duration(1))
|
||||||
|
err := Err(ctx, previousErr)
|
||||||
|
assert.NotNil(t, err)
|
||||||
|
standardized := test.RequireStandardError(t, err)
|
||||||
|
assert.Equal(t, standarderror.Timeout, standardized.Code)
|
||||||
|
// should failed.
|
||||||
|
ctx, cancel = Context(5)
|
||||||
|
cancel()
|
||||||
|
err = Err(ctx, previousErr)
|
||||||
|
assert.NotNil(t, err)
|
||||||
|
standardized = test.RequireStandardError(t, err)
|
||||||
|
assert.Equal(t, standarderror.Internal, standarderror.Code(err))
|
||||||
|
}
|
||||||
|
|||||||
@@ -37,6 +37,9 @@ func AssertDirectoryEmpty(t *testing.T, directory string) {
|
|||||||
assert.Nil(t, err)
|
assert.Nil(t, err)
|
||||||
defer f.Close() // nolint: errcheck
|
defer f.Close() // nolint: errcheck
|
||||||
_, err = f.Readdir(1)
|
_, err = f.Readdir(1)
|
||||||
|
if err == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
assert.Equal(t, io.EOF, err)
|
assert.Equal(t, io.EOF, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user