feat: add smb driver (close #1746) (#2114)

* feat: add smb driver (close #1746)

* Update driver.go

drivers/smb/util.go

@@ -0,0 +1,122 @@
+package smb
+
+import (
+	"io"
+	"io/fs"
+	"net"
+	"os"
+	"path/filepath"
+
+	"github.com/alist-org/alist/v3/internal/model"
+	"github.com/hirochachacha/go-smb2"
+)
+
+func (d *SMB) initFS() error {
+	conn, err := net.Dial("tcp", d.Address)
+	if err != nil {
+		return err
+	}
+	dialer := &smb2.Dialer{
+		Initiator: &smb2.NTLMInitiator{
+			User:     d.Username,
+			Password: d.Password,
+		},
+	}
+	s, err := dialer.Dial(conn)
+	if err != nil {
+		return err
+	}
+	d.fs, err = s.Mount(d.ShareName)
+	if err != nil {
+		return err
+	}
+	return err
+}
+
+func (d *SMB) getSMBPath(dir model.Obj) string {
+	fullPath := dir.GetPath()
+	if fullPath[0:1] != "." {
+		fullPath = "." + fullPath
+	}
+	return fullPath
+}
+
+// CopyFile File copies a single file from src to dst
+func (d *SMB) CopyFile(src, dst string) error {
+	var err error
+	var srcfd *smb2.File
+	var dstfd *smb2.File
+	var srcinfo fs.FileInfo
+
+	if srcfd, err = d.fs.Open(src); err != nil {
+		return err
+	}
+	defer srcfd.Close()
+
+	if dstfd, err = d.CreateNestedFile(dst); err != nil {
+		return err
+	}
+	defer dstfd.Close()
+
+	if _, err = io.Copy(dstfd, srcfd); err != nil {
+		return err
+	}
+	if srcinfo, err = d.fs.Stat(src); err != nil {
+		return err
+	}
+	return d.fs.Chmod(dst, srcinfo.Mode())
+}
+
+// CopyDir Dir copies a whole directory recursively
+func (d *SMB) CopyDir(src string, dst string) error {
+	var err error
+	var fds []fs.FileInfo
+	var srcinfo fs.FileInfo
+
+	if srcinfo, err = d.fs.Stat(src); err != nil {
+		return err
+	}
+	if err = d.fs.MkdirAll(dst, srcinfo.Mode()); err != nil {
+		return err
+	}
+	if fds, err = d.fs.ReadDir(src); err != nil {
+		return err
+	}
+	for _, fd := range fds {
+		srcfp := filepath.Join(src, fd.Name())
+		dstfp := filepath.Join(dst, fd.Name())
+
+		if fd.IsDir() {
+			if err = d.CopyDir(srcfp, dstfp); err != nil {
+				return err
+			}
+		} else {
+			if err = d.CopyFile(srcfp, dstfp); err != nil {
+				return err
+			}
+		}
+	}
+	return nil
+}
+
+// Exists determine whether the file exists
+func (d *SMB) Exists(name string) bool {
+	if _, err := d.fs.Stat(name); err != nil {
+		if os.IsNotExist(err) {
+			return false
+		}
+	}
+	return true
+}
+
+// CreateNestedFile create nested file
+func (d *SMB) CreateNestedFile(path string) (*smb2.File, error) {
+	basePath := filepath.Dir(path)
+	if !d.Exists(basePath) {
+		err := d.fs.MkdirAll(basePath, 0700)
+		if err != nil {
+			return nil, err
+		}
+	}
+	return d.fs.Create(path)
+}