summaryrefslogtreecommitdiff
path: root/plugins/http/config/uploads_config.go
diff options
context:
space:
mode:
authorValery Piashchynski <[email protected]>2021-01-21 13:25:36 +0300
committerValery Piashchynski <[email protected]>2021-01-21 13:25:36 +0300
commit7da6c78449776e1f3c6716250bca0b712a0423a4 (patch)
treef3512de66aca2bba408485a0ea2fc936c0e4fb9b /plugins/http/config/uploads_config.go
parent0ff05b2732b4fd0783f959c94c54d7e39169f979 (diff)
Uniform all configs
Add debug server Check nil's for all plugin intialization
Diffstat (limited to 'plugins/http/config/uploads_config.go')
-rw-r--r--plugins/http/config/uploads_config.go46
1 files changed, 46 insertions, 0 deletions
diff --git a/plugins/http/config/uploads_config.go b/plugins/http/config/uploads_config.go
new file mode 100644
index 00000000..5edb0ab7
--- /dev/null
+++ b/plugins/http/config/uploads_config.go
@@ -0,0 +1,46 @@
+package config
+
+import (
+ "os"
+ "path"
+ "strings"
+)
+
+// Uploads describes file location and controls access to them.
+type Uploads 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
+}
+
+// InitDefaults sets missing values to their default values.
+func (cfg *Uploads) InitDefaults() error {
+ cfg.Forbid = []string{".php", ".exe", ".bat"}
+ cfg.Dir = os.TempDir()
+ return nil
+}
+
+// TmpDir returns temporary directory.
+func (cfg *Uploads) TmpDir() string {
+ if cfg.Dir != "" {
+ return cfg.Dir
+ }
+
+ return os.TempDir()
+}
+
+// Forbids must return true if file extension is not allowed for the upload.
+func (cfg *Uploads) Forbids(filename string) bool {
+ ext := strings.ToLower(path.Ext(filename))
+
+ for _, v := range cfg.Forbid {
+ if ext == v {
+ return true
+ }
+ }
+
+ return false
+}