diff options
author | Michael Muré <batolettre@gmail.com> | 2022-12-27 19:40:40 +0100 |
---|---|---|
committer | Michael Muré <batolettre@gmail.com> | 2022-12-27 19:40:40 +0100 |
commit | e920987860dd9392fefc4b222aa4d2446b74f3d8 (patch) | |
tree | 7f67b211ed2a6342e4a3ca4e1011f3b91f0080bb /commands/bug/select/select.go | |
parent | 8cea6c7515e42cb8821a03c9bfebb0a8f63a01be (diff) | |
download | git-bug-e920987860dd9392fefc4b222aa4d2446b74f3d8.tar.gz |
commands: generic "select" code, move bug completion in bugcmd
Diffstat (limited to 'commands/bug/select/select.go')
-rw-r--r-- | commands/bug/select/select.go | 128 |
1 files changed, 0 insertions, 128 deletions
diff --git a/commands/bug/select/select.go b/commands/bug/select/select.go deleted file mode 100644 index 7096dde4..00000000 --- a/commands/bug/select/select.go +++ /dev/null @@ -1,128 +0,0 @@ -package _select - -import ( - "fmt" - "io" - "io/ioutil" - "os" - - "github.com/pkg/errors" - - "github.com/MichaelMure/git-bug/cache" - "github.com/MichaelMure/git-bug/entity" -) - -const selectFile = "select" - -var ErrNoValidId = errors.New("you must provide a bug id or use the \"select\" command first") - -// ResolveBug first try to resolve a bug using the first argument of the command -// line. If it fails, it falls back to the select mechanism. -// -// Returns: -// - the bug if any -// - the new list of command line arguments with the bug prefix removed if it -// has been used -// - an error if the process failed -func ResolveBug(repo *cache.RepoCache, args []string) (*cache.BugCache, []string, error) { - // At first, try to use the first argument as a bug prefix - if len(args) > 0 { - b, err := repo.Bugs().ResolvePrefix(args[0]) - - if err == nil { - return b, args[1:], nil - } - - if !entity.IsErrNotFound(err) { - return nil, nil, err - } - } - - // first arg is not a valid bug prefix, we can safely use the preselected bug if any - - b, err := selected(repo) - - // selected bug is invalid - if entity.IsErrNotFound(err) { - // we clear the selected bug - err = Clear(repo) - if err != nil { - return nil, nil, err - } - return nil, nil, ErrNoValidId - } - - // another error when reading the bug - if err != nil { - return nil, nil, err - } - - // bug is successfully retrieved - if b != nil { - return b, args, nil - } - - // no selected bug and no valid first argument - return nil, nil, ErrNoValidId -} - -// Select will select a bug for future use -func Select(repo *cache.RepoCache, id entity.Id) error { - f, err := repo.LocalStorage().OpenFile(selectFile, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666) - if err != nil { - return err - } - - _, err = f.Write([]byte(id.String())) - if err != nil { - return err - } - - return f.Close() -} - -// Clear will clear the selected bug, if any -func Clear(repo *cache.RepoCache) error { - return repo.LocalStorage().Remove(selectFile) -} - -func selected(repo *cache.RepoCache) (*cache.BugCache, error) { - f, err := repo.LocalStorage().Open(selectFile) - if err != nil { - if os.IsNotExist(err) { - return nil, nil - } else { - return nil, err - } - } - - buf, err := ioutil.ReadAll(io.LimitReader(f, 100)) - if err != nil { - return nil, err - } - if len(buf) == 100 { - return nil, fmt.Errorf("the select file should be < 100 bytes") - } - - id := entity.Id(buf) - if err := id.Validate(); err != nil { - err = repo.LocalStorage().Remove(selectFile) - if err != nil { - return nil, errors.Wrap(err, "error while removing invalid select file") - } - - return nil, fmt.Errorf("select file in invalid, removing it") - } - - b, err := repo.Bugs().Resolve(id) - if err != nil { - return nil, err - } - - err = f.Close() - if err != nil { - return nil, err - } - - return b, nil -} |