Merge pull request #870 from MichaelMure/cli-reorg

commands: reorg into different packages

1 files changed, 77 insertions, 0 deletions

diff --git a/commands/bug/bug_comment_edit.go b/commands/bug/bug_comment_edit.go
new file mode 100644
index 00000000..8be7cb80
--- /dev/null
+++ b/commands/bug/bug_comment_edit.go
@@ -0,0 +1,77 @@
+package bugcmd
+
+import (
+	"github.com/spf13/cobra"
+
+	"github.com/MichaelMure/git-bug/commands/execenv"
+	"github.com/MichaelMure/git-bug/commands/input"
+)
+
+type bugCommentEditOptions struct {
+	messageFile    string
+	message        string
+	nonInteractive bool
+}
+
+func newBugCommentEditCommand() *cobra.Command {
+	env := execenv.NewEnv()
+	options := bugCommentEditOptions{}
+
+	cmd := &cobra.Command{
+		Use:     "edit [COMMENT_ID]",
+		Short:   "Edit an existing comment on a bug",
+		Args:    cobra.ExactArgs(1),
+		PreRunE: execenv.LoadBackendEnsureUser(env),
+		RunE: execenv.CloseBackend(env, func(cmd *cobra.Command, args []string) error {
+			return runBugCommentEdit(env, options, args)
+		}),
+	}
+
+	flags := cmd.Flags()
+	flags.SortFlags = false
+
+	flags.StringVarP(&options.messageFile, "file", "F", "",
+		"Take the message from the given file. Use - to read the message from the standard input")
+
+	flags.StringVarP(&options.message, "message", "m", "",
+		"Provide the new message from the command line")
+	flags.BoolVar(&options.nonInteractive, "non-interactive", false, "Do not ask for user input")
+
+	return cmd
+}
+
+func runBugCommentEdit(env *execenv.Env, opts bugCommentEditOptions, args []string) error {
+	b, commentId, err := env.Backend.ResolveComment(args[0])
+	if err != nil {
+		return err
+	}
+
+	if opts.messageFile != "" && opts.message == "" {
+		opts.message, err = input.BugCommentFileInput(opts.messageFile)
+		if err != nil {
+			return err
+		}
+	}
+
+	if opts.messageFile == "" && opts.message == "" {
+		if opts.nonInteractive {
+			env.Err.Println("No message given. Use -m or -F option to specify a message. Aborting.")
+			return nil
+		}
+		opts.message, err = input.BugCommentEditorInput(env.Backend, "")
+		if err == input.ErrEmptyMessage {
+			env.Err.Println("Empty message, aborting.")
+			return nil
+		}
+		if err != nil {
+			return err
+		}
+	}
+
+	_, err = b.EditComment(commentId, opts.message)
+	if err != nil {
+		return err
+	}
+
+	return b.Commit()
+}