depot/nix/pkgs/seaweedfs/0002-shell-add-fs.meta.edit-for-editing-a-filer-metadata-.patch

135 lines
3.4 KiB
Diff
Raw Permalink Normal View History

From 74c486e9f8b2557e4bf764adbbd46642c175ea40 Mon Sep 17 00:00:00 2001
From: Luke Granger-Brown <git@lukegb.com>
Date: Sun, 20 Oct 2024 16:59:21 +0100
Subject: [PATCH 2/2] [shell] add fs.meta.edit for editing a filer metadata
entry
---
weed/shell/command_fs_meta_edit.go | 114 +++++++++++++++++++++++++++++
1 file changed, 114 insertions(+)
create mode 100644 weed/shell/command_fs_meta_edit.go
diff --git a/weed/shell/command_fs_meta_edit.go b/weed/shell/command_fs_meta_edit.go
new file mode 100644
index 000000000..ece203b4d
--- /dev/null
+++ b/weed/shell/command_fs_meta_edit.go
@@ -0,0 +1,114 @@
+package shell
+
+import (
+ "flag"
+ "fmt"
+ "io"
+ "os"
+ "os/exec"
+
+ "github.com/seaweedfs/seaweedfs/weed/pb/filer_pb"
+ "github.com/seaweedfs/seaweedfs/weed/util"
+ "google.golang.org/protobuf/encoding/prototext"
+)
+
+func init() {
+ Commands = append(Commands, &commandFsMetaEdit{})
+}
+
+type commandFsMetaEdit struct {
+ dirPrefix *string
+}
+
+func (c *commandFsMetaEdit) Name() string {
+ return "fs.meta.edit"
+}
+
+func (c *commandFsMetaEdit) Help() string {
+ return `edit single file's metadata
+
+ fs.meta.edit /path/to/single/file
+`
+}
+
+func (c *commandFsMetaEdit) HasTag(CommandTag) bool {
+ return false
+}
+
+func (c *commandFsMetaEdit) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) {
+
+ if len(args) == 0 {
+ fmt.Fprintf(writer, "missing a path to a file\n")
+ return nil
+ }
+
+ metaEditCommand := flag.NewFlagSet(c.Name(), flag.ContinueOnError)
+ if err = metaEditCommand.Parse(args); err != nil {
+ return nil
+ }
+ path, parseErr := commandEnv.parseUrl(findInputDirectory(metaEditCommand.Args()))
+ if parseErr != nil {
+ return parseErr
+ }
+ filerPath := util.FullPath(path)
+
+ // Fetch entry.
+ fmt.Fprintf(writer, "%#v\n", filerPath)
+ entry, err := filer_pb.GetEntry(commandEnv, filerPath)
+ if err != nil {
+ return fmt.Errorf("getting entry data: %w", err)
+ }
+
+ // Write entry to temp file.
+ f, err := os.CreateTemp("", "seaweed-entry-*.txtpb")
+ if err != nil {
+ return fmt.Errorf("creating temp file: %w", err)
+ }
+ defer os.Remove(f.Name())
+
+ if _, err := f.Write([]byte(prototext.Format(entry))); err != nil {
+ return fmt.Errorf("writing formatted entry data: %w", err)
+ }
+
+ // Launch an editor.
+ editor := os.Getenv("VISUAL")
+ if editor == "" {
+ editor = os.Getenv("EDITOR")
+ }
+ if editor == "" {
+ editor = "vim"
+ }
+ cmd := exec.Command("sh", "-c", fmt.Sprintf("%s %s", editor, f.Name()))
+ cmd.Stdin = os.Stdin
+ cmd.Stdout = os.Stdout
+ cmd.Stderr = os.Stderr
+ if err := cmd.Run(); err != nil {
+ return fmt.Errorf("editor failed: %w", err)
+ }
+
+ // Read edited entry.
+ if _, err := f.Seek(0, 0); err != nil {
+ return fmt.Errorf("seeking to beginning of file failed: %w", err)
+ }
+ editedBytes, err := io.ReadAll(f)
+ if err != nil {
+ return fmt.Errorf("reading edited entry: %w", err)
+ }
+ if err := prototext.Unmarshal(editedBytes, entry); err != nil {
+ return fmt.Errorf("parsing edited entry: %w", err)
+ }
+
+ // Write back entry to filer.
+ dir, _ := filerPath.DirAndName()
+ req := &filer_pb.UpdateEntryRequest{
+ Directory: dir,
+ Entry: entry,
+ }
+ if err := commandEnv.WithFilerClient(false, func(client filer_pb.SeaweedFilerClient) error {
+ return filer_pb.UpdateEntry(client, req)
+ }); err != nil {
+ return fmt.Errorf("updating entry in filer: %w", err)
+ }
+
+ return nil
+}
--
2.46.0