refactor

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
+}