working commit

2026-03-04 20:27:35 +02:00
parent 36a143c7c3
commit 9ecd25ed0b
12 changed files with 1065 additions and 0 deletions
@@ -0,0 +1,116 @@
+/*
+ * Copyright 2026 Oleg Borodin <onborodin@gmail.com>
+ *
+ * This work is published and licensed under a Creative Commons
+ * Attribution-NonCommercial-NoDerivatives 4.0 International License.
+ *
+ * Distribution of this work is permitted, but commercial use and
+ * modifications are strictly prohibited.
+ */
+package filecmd
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"io/fs"
+	"os"
+	"path/filepath"
+	"strings"
+	"time"
+
+	"github.com/spf13/cobra"
+
+	"mstore/pkg/filecli"
+)
+
+// ImportFiles
+type ImportFilesParams struct {
+	Source   string
+	Dest     string
+	Progress bool
+}
+type ImportFilesResult struct {
+	Files []string `json:"files,omitempty"`
+}
+
+func (util *FileUtil) ImportFiles(cmd *cobra.Command, args []string) {
+	util.importFilesParams.Source = args[0]
+	util.importFilesParams.Dest = args[1]
+
+	res, err := util.importFiles(&util.commonFileParams, &util.importFilesParams)
+	printResponse(res, err)
+}
+
+func (util *FileUtil) importFiles(common *CommonFileParams, params *ImportFilesParams) (*ImportFilesResult, error) {
+	var err error
+	res := &ImportFilesResult{
+		Files: make([]string, 0),
+	}
+
+	putErrors := make([]error, 0)
+	walcFunc := func(walkpath string, infoItem fs.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		mode := infoItem.Mode()
+		skip := mode == fs.ModeDevice
+		skip = skip || mode == fs.ModeNamedPipe
+		skip = skip || mode == fs.ModeCharDevice
+		if skip {
+			return nil
+		}
+
+		if !infoItem.IsDir() {
+			relpath, _ := strings.CutPrefix(walkpath, params.Source)
+			ref, err := filecli.ParsePath(params.Dest)
+			if err != nil {
+				putErrors = append(putErrors, err)
+				return nil
+			}
+                        ref.JoinResource(relpath)
+        //fmt.Printf("====%s %s\n", relpath, ref.Raw())
+        //return nil
+			file, err := os.OpenFile(walkpath, os.O_RDONLY, 0)
+			if err != nil {
+				putErrors = append(putErrors, err)
+				return nil
+			}
+			defer file.Close()
+			stat, err := file.Stat()
+			if err != nil {
+				putErrors = append(putErrors, err)
+				return nil
+			}
+			timeout := time.Duration(common.Timeout) * time.Second
+			ctx, _ := context.WithTimeout(context.Background(), timeout)
+
+			ref.SetUserinfo(common.Username, common.Password)
+			mw := filecli.NewBasicAuthMiddleware(ref.Userinfo())
+
+			cli := filecli.NewClient(nil, mw)
+			err = cli.PutFile(ctx, ref.Raw(), file, stat.Size())
+			if err != nil {
+				putErrors = append(putErrors, err)
+				fmt.Printf("- %s: error: %v \n", ref.Raw(), err)
+			} else {
+				res.Files = append(res.Files, relpath)
+				fmt.Printf("- %s: ok\n", ref.Raw())
+			}
+		}
+		return nil
+	}
+        err = filepath.Walk(params.Source, walcFunc)
+        if err != nil {
+                return res, err
+        }
+
+	for _, item := range putErrors {
+		err = errors.Join(err, item)
+	}
+	err = filepath.Walk(params.Source, walcFunc)
+	if err != nil {
+		return res, err
+	}
+	return res, err
+}