feat: support alist as a storage

pkg/storage/local.go

@@ -6,22 +6,14 @@ import (
 	"io/ioutil"
 	"os"
 	"path/filepath"
-	"polaris/log"
 
+	"github.com/gabriel-vasile/mimetype"
 	"github.com/pkg/errors"
 )
 
-type Storage interface {
-	Move(src, dest string) error
-	Copy(src, dest string) error
-	ReadDir(dir string) ([]fs.FileInfo, error)
-	ReadFile(string) ([]byte, error)
-	WriteFile(string, []byte) error
-}
 
 func NewLocalStorage(dir string) (*LocalStorage, error) {
 	os.MkdirAll(dir, 0655)
-
 	return &LocalStorage{dir: dir}, nil
 }
 
@@ -30,57 +22,28 @@ type LocalStorage struct {
 }
 
 func (l *LocalStorage) Copy(src, destDir string) error {
-	os.MkdirAll(filepath.Join(l.dir, destDir), os.ModePerm)
-
-	targetBase := filepath.Join(l.dir, destDir, filepath.Base(src)) //文件的场景，要加上文件名, move filename ./dir/
-	info, err := os.Stat(src)
+	b, err := NewBase(src)
 	if err != nil {
-		return errors.Wrap(err, "read source dir")
+		return err
 	}
-	if info.IsDir() { //如果是路径，则只移动路径里面的文件，不管当前路径, 行为类似 move dirname/* target_dir/
-		targetBase = filepath.Join(l.dir, destDir)
-	}
-	log.Debugf("local storage target base dir is: %v", targetBase)
-
-	err = filepath.Walk(src, func(path string, info fs.FileInfo, err error) error {
-		if err != nil {
-			return err
-		}
-		rel, err := filepath.Rel(src, path)
-		if err != nil {
-			return errors.Wrapf(err, "relation between %s and %s", src, path)
-		}
-		destName := filepath.Join(targetBase, rel)
-
-		if info.IsDir() {
-			os.Mkdir(destName, os.ModePerm)
-		} else { //is file
-			if err := os.Link(path, destName); err != nil {
-				log.Warnf("hard link file error: %v, will try copy file, source: %s, dest: %s", err, path, destName)
-				if writer, err := os.OpenFile(destName, os.O_RDWR|os.O_CREATE|os.O_TRUNC, os.ModePerm); err != nil {
-					return errors.Wrapf(err, "create file %s", destName)
-				} else {
-					defer writer.Close()
-					if f, err := os.OpenFile(path, os.O_RDONLY, os.ModePerm); err != nil {
-						return errors.Wrapf(err, "read file %v", path)
-					} else { //open success
-						defer f.Close()
-						_, err := io.Copy(writer, f)
-						if err != nil {
-							return errors.Wrap(err, "transmitting data error")
-						}
-					}
-				}
 
+	baseDest := filepath.Join(l.dir, destDir)
+	uploadFunc := func(destPath string, destInfo fs.FileInfo, srcReader io.Reader, mimeType *mimetype.MIME) error {
+		if writer, err := os.OpenFile(destPath, os.O_RDWR|os.O_CREATE|os.O_TRUNC, os.ModePerm); err != nil {
+			return errors.Wrapf(err, "create file %s", destPath)
+		} else {
+			defer writer.Close()
+			_, err := io.Copy(writer, srcReader)
+			if err != nil {
+				return errors.Wrap(err, "transmitting data error")
 			}
+
 		}
-		log.Infof("file copy complete: %v", destName)
 		return nil
-	})
-	if err != nil {
-		return errors.Wrap(err, "move file error")
 	}
-	return nil
+	return b.Upload(baseDest, true, false, false, uploadFunc, func(s string) error {
+		return os.Mkdir(s, os.ModePerm)
+	})
 }
 
 func (l *LocalStorage) Move(src, destDir string) error {
@@ -103,3 +66,7 @@ func (l *LocalStorage) WriteFile(name string, data []byte) error {
 	os.MkdirAll(filepath.Dir(path), os.ModePerm)
 	return os.WriteFile(path, data, os.ModePerm)
 }
+
+func (l *LocalStorage) UploadProgress() float64 {
+	return 0
+}