mirror of
https://github.com/gotenberg/gotenberg.git
synced 2026-08-08 08:32:16 +01:00
106 lines
2.7 KiB
Go
106 lines
2.7 KiB
Go
// Package context provides functions for setting and getting values from
|
|
// a request's context.
|
|
package context
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"github.com/thecodingmachine/gotenberg/app/converter"
|
|
)
|
|
|
|
type key uint32
|
|
|
|
const (
|
|
requestIDKey key = iota
|
|
converterKey
|
|
resultFilePathKey
|
|
)
|
|
|
|
// WithRequestID populates a request's context with the given request ID
|
|
// and returns the updated request.
|
|
func WithRequestID(r *http.Request, requestID string) *http.Request {
|
|
ctx := r.Context()
|
|
ctx = context.WithValue(ctx, requestIDKey, requestID)
|
|
r = r.WithContext(ctx)
|
|
|
|
return r
|
|
}
|
|
|
|
type requestIDNotFoundError struct{}
|
|
|
|
const requestIDNotFoundErrorMessage = "the request ID was not found in request context"
|
|
|
|
func (e *requestIDNotFoundError) Error() string {
|
|
return requestIDNotFoundErrorMessage
|
|
}
|
|
|
|
// GetRequestID returns the request ID if found in
|
|
// the request's context. Otherwise throws an error.
|
|
func GetRequestID(r *http.Request) (string, error) {
|
|
ID, ok := r.Context().Value(requestIDKey).(string)
|
|
if !ok {
|
|
return "", &requestIDNotFoundError{}
|
|
}
|
|
|
|
return ID, nil
|
|
}
|
|
|
|
// WithConverter populates a request's context with the given converter
|
|
// and returns the updated request.
|
|
func WithConverter(r *http.Request, converter *converter.Converter) *http.Request {
|
|
ctx := r.Context()
|
|
ctx = context.WithValue(ctx, converterKey, converter)
|
|
r = r.WithContext(ctx)
|
|
|
|
return r
|
|
}
|
|
|
|
type converterNotFoundError struct{}
|
|
|
|
const converterNotFoundErrorMessage = "the converter was not found in request context"
|
|
|
|
func (e *converterNotFoundError) Error() string {
|
|
return converterNotFoundErrorMessage
|
|
}
|
|
|
|
// GetConverter returns the converter if found in
|
|
// the request's context. Otherwise throws an error.
|
|
func GetConverter(r *http.Request) (*converter.Converter, error) {
|
|
c, ok := r.Context().Value(converterKey).(*converter.Converter)
|
|
if !ok {
|
|
return nil, &converterNotFoundError{}
|
|
}
|
|
|
|
return c, nil
|
|
}
|
|
|
|
// WithResultFilePath populates a request's context with the given result file path
|
|
// and returns the updated request.
|
|
func WithResultFilePath(r *http.Request, resultFilePath string) *http.Request {
|
|
ctx := r.Context()
|
|
ctx = context.WithValue(ctx, resultFilePathKey, resultFilePath)
|
|
r = r.WithContext(ctx)
|
|
|
|
return r
|
|
}
|
|
|
|
type resultFilePathNotFoundError struct{}
|
|
|
|
const resultFilePathNotFoundErrorMessage = "the result file path was not found in request context"
|
|
|
|
func (e *resultFilePathNotFoundError) Error() string {
|
|
return resultFilePathNotFoundErrorMessage
|
|
}
|
|
|
|
// GetResultFilePath returns the result file path if found in
|
|
// the request's context. Otherwise throws an error.
|
|
func GetResultFilePath(r *http.Request) (string, error) {
|
|
path, ok := r.Context().Value(resultFilePathKey).(string)
|
|
if !ok {
|
|
return "", &resultFilePathNotFoundError{}
|
|
}
|
|
|
|
return path, nil
|
|
}
|