106 lines
4.3 KiB
Go
106 lines
4.3 KiB
Go
package handler
|
|
|
|
import (
|
|
"errors"
|
|
"log"
|
|
"net/url"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
// Config provides a way to configure the Handler depending on your needs.
|
|
type Config struct {
|
|
// StoreComposer points to the store composer from which the core data store
|
|
// and optional dependencies should be taken. May only be nil if DataStore is
|
|
// set.
|
|
// TODO: Remove pointer?
|
|
StoreComposer *StoreComposer
|
|
// MaxSize defines how many bytes may be stored in one single upload. If its
|
|
// value is is 0 or smaller no limit will be enforced.
|
|
MaxSize int64
|
|
// BasePath defines the URL path used for handling uploads, e.g. "/files/".
|
|
// If no trailing slash is presented it will be added. You may specify an
|
|
// absolute URL containing a scheme, e.g. "http://tus.io"
|
|
BasePath string
|
|
isAbs bool
|
|
// DisableDownload indicates whether the server will refuse downloads of the
|
|
// uploaded file, by not mounting the GET handler.
|
|
DisableDownload bool
|
|
// DisableTermination indicates whether the server will refuse termination
|
|
// requests of the uploaded file, by not mounting the DELETE handler.
|
|
DisableTermination bool
|
|
// NotifyCompleteUploads indicates whether sending notifications about
|
|
// completed uploads using the CompleteUploads channel should be enabled.
|
|
NotifyCompleteUploads bool
|
|
// NotifyTerminatedUploads indicates whether sending notifications about
|
|
// terminated uploads using the TerminatedUploads channel should be enabled.
|
|
NotifyTerminatedUploads bool
|
|
// NotifyUploadProgress indicates whether sending notifications about
|
|
// the upload progress using the UploadProgress channel should be enabled.
|
|
NotifyUploadProgress bool
|
|
// NotifyCreatedUploads indicates whether sending notifications about
|
|
// the upload having been created using the CreatedUploads channel should be enabled.
|
|
NotifyCreatedUploads bool
|
|
// UploadProgressInterval specifies the interval at which the upload progress
|
|
// notifications are sent to the UploadProgress channel, if enabled.
|
|
// Defaults to 1s.
|
|
UploadProgressInterval time.Duration
|
|
// Logger is the logger to use internally, mostly for printing requests.
|
|
Logger *log.Logger
|
|
// Respect the X-Forwarded-Host, X-Forwarded-Proto and Forwarded headers
|
|
// potentially set by proxies when generating an absolute URL in the
|
|
// response to POST requests.
|
|
RespectForwardedHeaders bool
|
|
// PreUploadCreateCallback will be invoked before a new upload is created, if the
|
|
// property is supplied. If the callback returns no error, the upload will be created
|
|
// and optional values from HTTPResponse will be contained in the HTTP response.
|
|
// If the error is non-nil, the upload will not be created. This can be used to implement
|
|
// validation of upload metadata etc. Furthermore, HTTPResponse will be ignored and
|
|
// the error value can contain values for the HTTP response.
|
|
PreUploadCreateCallback func(hook HookEvent) (HTTPResponse, error)
|
|
// PreFinishResponseCallback will be invoked after an upload is completed but before
|
|
// a response is returned to the client. This can be used to implement post-processing validation.
|
|
// If the callback returns no error, optional values from HTTPResponse will be contained in the HTTP response.
|
|
// If the error is non-nil, the error will be forwarded to the client. Furthermore,
|
|
// HTTPResponse will be ignored and the error value can contain values for the HTTP response.
|
|
PreFinishResponseCallback func(hook HookEvent) (HTTPResponse, error)
|
|
}
|
|
|
|
func (config *Config) validate() error {
|
|
if config.Logger == nil {
|
|
config.Logger = log.New(os.Stdout, "[tusd] ", log.Ldate|log.Ltime)
|
|
}
|
|
|
|
base := config.BasePath
|
|
uri, err := url.Parse(base)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Ensure base path ends with slash to remove logic from absFileURL
|
|
if base != "" && string(base[len(base)-1]) != "/" {
|
|
base += "/"
|
|
}
|
|
|
|
// Ensure base path begins with slash if not absolute (starts with scheme)
|
|
if !uri.IsAbs() && len(base) > 0 && string(base[0]) != "/" {
|
|
base = "/" + base
|
|
}
|
|
config.BasePath = base
|
|
config.isAbs = uri.IsAbs()
|
|
|
|
if config.StoreComposer == nil {
|
|
return errors.New("tusd: StoreComposer must no be nil")
|
|
}
|
|
|
|
if config.StoreComposer.Core == nil {
|
|
return errors.New("tusd: StoreComposer in Config needs to contain a non-nil core")
|
|
}
|
|
|
|
if config.UploadProgressInterval <= 0 {
|
|
config.UploadProgressInterval = 1 * time.Second
|
|
}
|
|
|
|
return nil
|
|
}
|