提交

cool/coolfile/coolfile.go

@@ -0,0 +1,48 @@
+package coolfile
+
+import (
+	"github.com/cool-team-official/cool-admin-go/cool/coolconfig"
+	"github.com/gogf/gf/v2/errors/gerror"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+type Driver interface {
+	New() Driver
+	GetMode() (data interface{}, err error)
+	Upload(ctx g.Ctx) (string, error)
+}
+
+var (
+	// FileMap is the map for registered file drivers.
+	FileMap = map[string]Driver{}
+)
+
+func NewFile() (d Driver) {
+	if driver, ok := FileMap[coolconfig.Config.File.Mode]; ok {
+		return driver.New()
+	}
+	errorMsg := "\n"
+	errorMsg += `无法找到指定文件上传类型 "%s"`
+	errorMsg += `，您是否拼写错误了类型名称 "%s" 或者忘记导入上传支持包？`
+	errorMsg += `参考:https://github.com/cool-team-official/cool-admin-go/tree/master/contrib/files`
+	err := gerror.Newf(errorMsg, coolconfig.Config.File.Mode, coolconfig.Config.File.Mode)
+
+	panic(err)
+
+}
+
+// Register registers custom file driver to cool.
+func Register(name string, driver Driver) error {
+	FileMap[name] = driver
+	return nil
+}
+
+// func init() {
+// 	// Register("local", &Local{})
+// 	// Register("oss", &Oss{})
+// 	file, err := NewFile()
+// 	if err != nil {
+// 		panic(err)
+// 	}
+// 	File = file
+// }