fileuploader cmd

cmd/fileuploader/main.go

@@ -0,0 +1,88 @@
+// +build !nogrpcserver,!_test
+
+package main
+
+import (
+	"fmt"
+	"github.com/anytypeio/go-anytype-middleware/pkg/lib/pin"
+	"io/ioutil"
+	_ "net/http/pprof"
+	"os"
+	"path/filepath"
+	"strings"
+	"time"
+
+	"github.com/anytypeio/go-anytype-middleware/core"
+	"github.com/anytypeio/go-anytype-middleware/core/event"
+	"github.com/anytypeio/go-anytype-middleware/pb"
+	pb2 "github.com/anytypeio/go-anytype-middleware/pkg/lib/cafe/pb"
+)
+
+func main() {
+	if len(os.Args) > 2 {
+		invite := os.Args[1]
+		path := os.Args[2]
+		var files []string
+		mw := core.New()
+		rootPath, _ := ioutil.TempDir(os.TempDir(), "anytype_*")
+
+		defer os.RemoveAll(rootPath)
+
+		mw.EventSender = event.NewCallbackSender(func(event *pb.Event) {
+			// nothing to do
+		})
+
+		_ = mw.WalletCreate(&pb.RpcWalletCreateRequest{RootPath: rootPath})
+		_ = mw.AccountCreate(&pb.RpcAccountCreateRequest{Name: "profile", AlphaInviteCode: invite})
+
+		var cids []string
+		err := filepath.Walk(path, func(path string, info os.FileInfo, err error) error {
+			files = append(files, path)
+			if info.IsDir() {
+				return nil
+			}
+			if strings.HasSuffix(strings.ToLower(path), "ds_store") {
+				return nil
+			}
+
+			resp := mw.UploadFile(&pb.RpcUploadFileRequest{LocalPath: path, DisableEncryption: true})
+			if int(resp.Error.Code) != 0 {
+				return fmt.Errorf(resp.Error.Description)
+			}
+			cids = append(cids, resp.Hash)
+			fmt.Println(path)
+			fmt.Println(resp.Hash)
+			return nil
+		})
+		if err != nil {
+			panic(err.Error())
+		}
+
+		fs := mw.GetApp().MustComponent(pin.CName).(pin.FilePinService)
+		for {
+			r := fs.PinStatus(cids...)
+			var pinned int
+			var failed int
+			var inprog int
+			for k, f := range r {
+				if f.Status == pb2.PinStatus_Done {
+					pinned++
+				}
+				if f.Status == pb2.PinStatus_Failed {
+					failed++
+				}
+				if f.Status == pb2.PinStatus_Queued {
+					fmt.Printf("%s still in progress\n", k)
+					inprog++
+				}
+			}
+			fmt.Printf("%d pinned, %d in-progress, %d failed from %d\n", pinned, inprog, failed, len(r))
+
+			if len(r) == len(cids) {
+				fmt.Println("all pinned")
+				os.Exit(0)
+			}
+			time.Sleep(time.Second * 10)
+		}
+	}
+}