summaryrefslogtreecommitdiff
path: root/service/http/uploads_config.go
diff options
context:
space:
mode:
Diffstat (limited to 'service/http/uploads_config.go')
-rw-r--r--service/http/uploads_config.go39
1 files changed, 39 insertions, 0 deletions
diff --git a/service/http/uploads_config.go b/service/http/uploads_config.go
new file mode 100644
index 00000000..715de69a
--- /dev/null
+++ b/service/http/uploads_config.go
@@ -0,0 +1,39 @@
+package http
+
+import (
+ "strings"
+ "path"
+ "os"
+)
+
+// UploadsConfig describes file location and controls access to them.
+type UploadsConfig struct {
+ // Dir contains name of directory to control access to.
+ Dir string
+
+ // Forbid specifies list of file extensions which are forbidden for access.
+ // Example: .php, .exe, .bat, .htaccess and etc.
+ Forbid []string
+}
+
+// TmpDir returns temporary directory.
+func (cfg *UploadsConfig) TmpDir() string {
+ if cfg.Dir != "" {
+ return cfg.Dir
+ }
+
+ return os.TempDir()
+}
+
+// Forbid must return true if file extension is not allowed for the upload.
+func (cfg *UploadsConfig) Forbids(filename string) bool {
+ ext := strings.ToLower(path.Ext(filename))
+
+ for _, v := range cfg.Forbid {
+ if ext == v {
+ return true
+ }
+ }
+
+ return false
+}