Add interactive mode for tea pr create (#279)

refactor pull create into task & interact module

avoid creation of invalid PRs

refactor task.CreatePull

to make functionality reusable in interact module

implement interactive.CreatePull

Co-authored-by: Norwin Roosen <git@nroo.de>
Reviewed-on: https://gitea.com/gitea/tea/pulls/279
Reviewed-by: 6543 <6543@obermui.de>
Reviewed-by: techknowlogick <techknowlogick@gitea.io>
Co-Authored-By: Norwin <noerw@noreply.gitea.io>
Co-Committed-By: Norwin <noerw@noreply.gitea.io>

modules/interact/pull_create.go

@@ -0,0 +1,133 @@
+// Copyright 2020 The Gitea Authors. All rights reserved.
+// Use of this source code is governed by a MIT-style
+// license that can be found in the LICENSE file.
+
+package interact
+
+import (
+	"fmt"
+	"strings"
+
+	"code.gitea.io/tea/modules/config"
+	"code.gitea.io/tea/modules/git"
+	"code.gitea.io/tea/modules/task"
+
+	"github.com/AlecAivazis/survey/v2"
+)
+
+// CreatePull interactively creates a PR
+func CreatePull(login *config.Login, owner, repo string) error {
+	var base, head, title, description string
+
+	// owner, repo
+	owner, repo, err := promptRepoSlug(owner, repo)
+	if err != nil {
+		return err
+	}
+
+	// base
+	baseBranch, err := task.GetDefaultPRBase(login, owner, repo)
+	if err != nil {
+		return err
+	}
+	promptI := &survey.Input{Message: "Target branch [" + baseBranch + "]:"}
+	if err := survey.AskOne(promptI, &base); err != nil {
+		return err
+	}
+	if len(base) == 0 {
+		base = baseBranch
+	}
+
+	// head
+	localRepo, err := git.RepoForWorkdir()
+	if err != nil {
+		return err
+	}
+	promptOpts := survey.WithValidator(survey.Required)
+	headOwner, headBranch, err := task.GetDefaultPRHead(localRepo)
+	if err == nil {
+		promptOpts = nil
+	}
+	var headOwnerInput, headBranchInput string
+	promptI = &survey.Input{Message: "Source repo owner [" + headOwner + "]:"}
+	if err := survey.AskOne(promptI, &headOwnerInput); err != nil {
+		return err
+	}
+	if len(headOwnerInput) != 0 {
+		headOwner = headOwnerInput
+	}
+	promptI = &survey.Input{Message: "Source branch [" + headBranch + "]:"}
+	if err := survey.AskOne(promptI, &headBranchInput, promptOpts); err != nil {
+		return err
+	}
+	if len(headBranchInput) != 0 {
+		headBranch = headBranchInput
+	}
+
+	head = task.GetHeadSpec(headOwner, headBranch, owner)
+
+	// title
+	title = task.GetDefaultPRTitle(head)
+	promptOpts = survey.WithValidator(survey.Required)
+	if len(title) != 0 {
+		promptOpts = nil
+	}
+	promptI = &survey.Input{Message: "PR title [" + title + "]:"}
+	if err := survey.AskOne(promptI, &title, promptOpts); err != nil {
+		return err
+	}
+
+	// description
+	promptM := &survey.Multiline{Message: "PR description:"}
+	if err := survey.AskOne(promptM, &description); err != nil {
+		return err
+	}
+
+	return task.CreatePull(
+		login,
+		owner,
+		repo,
+		base,
+		head,
+		title,
+		description)
+}
+
+func promptRepoSlug(defaultOwner, defaultRepo string) (owner, repo string, err error) {
+	prompt := "Target repo:"
+	required := true
+	if len(defaultOwner) != 0 && len(defaultRepo) != 0 {
+		prompt = fmt.Sprintf("Target repo [%s/%s]:", defaultOwner, defaultRepo)
+		required = false
+	}
+	var repoSlug string
+
+	owner = defaultOwner
+	repo = defaultRepo
+
+	err = survey.AskOne(
+		&survey.Input{Message: prompt},
+		&repoSlug,
+		survey.WithValidator(func(input interface{}) error {
+			if str, ok := input.(string); ok {
+				if !required && len(str) == 0 {
+					return nil
+				}
+				split := strings.Split(str, "/")
+				if len(split) != 2 || len(split[0]) == 0 || len(split[1]) == 0 {
+					return fmt.Errorf("must follow the <owner>/<repo> syntax")
+				}
+			} else {
+				return fmt.Errorf("invalid result type")
+			}
+			return nil
+		}),
+	)
+
+	if err == nil && len(repoSlug) != 0 {
+		repoSlugSplit := strings.Split(repoSlug, "/")
+		owner = repoSlugSplit[0]
+		repo = repoSlugSplit[1]
+	}
+	return
+}