1
0
mirror of https://github.com/jesseduffield/lazygit.git synced 2025-08-06 11:02:41 +03:00

build: add github.com/fsmiamoto/git-todo-parser

This commit is contained in:
Francisco Miamoto
2022-05-22 13:42:45 -03:00
parent 2bccbee32b
commit 0b08a0b298
5 changed files with 222 additions and 0 deletions

View File

@@ -0,0 +1,141 @@
package todo
import (
"bufio"
"errors"
"fmt"
"io"
"strings"
)
var (
ErrUnexpectedCommand = errors.New("unexpected command")
ErrMissingLabel = errors.New("missing label")
ErrMissingCommit = errors.New("missing commit")
ErrMissingExecCmd = errors.New("missing command for exec")
)
func Parse(f io.Reader) ([]Todo, error) {
var result []Todo
scanner := bufio.NewScanner(f)
scanner.Split(bufio.ScanLines)
for scanner.Scan() {
line := scanner.Text()
trimmed := strings.TrimSpace(line)
if trimmed == "" {
continue
}
cmd, err := parseLine(line)
if err != nil {
return nil, fmt.Errorf("failed to parse line %q: %w", line, err)
}
result = append(result, cmd)
}
if err := scanner.Err(); err != nil {
return nil, fmt.Errorf("failed to parse input: %w", err)
}
return result, nil
}
func parseLine(line string) (Todo, error) {
var todo Todo
if strings.HasPrefix(line, CommentChar) {
todo.Command = Comment
todo.Comment = strings.TrimLeft(line, CommentChar)
return todo, nil
}
fields := strings.Fields(line)
for i := TodoCommand(Pick); i < Comment; i++ {
if isCommand(i, fields[0]) {
todo.Command = TodoCommand(i)
fields = fields[1:]
break
}
}
if todo.Command == 0 {
// unexpected command
return todo, ErrUnexpectedCommand
}
if todo.Command == Break {
return todo, nil
}
if todo.Command == Label || todo.Command == Reset {
if len(fields) == 0 {
return todo, ErrMissingLabel
}
todo.Label = fields[0]
return todo, nil
}
if todo.Command == Exec {
if len(fields) == 0 {
return todo, ErrMissingExecCmd
}
todo.ExecCommand = strings.Join(fields, " ")
return todo, nil
}
if todo.Command == Merge {
if fields[0] == "-C" || fields[0] == "-c" {
fields = fields[1:]
if len(fields) == 0 {
return todo, ErrMissingCommit
}
todo.Commit = fields[0]
fields = fields[1:]
}
if len(fields) == 0 {
return todo, ErrMissingLabel
}
todo.Label = fields[0]
fields = fields[1:]
if fields[0] == "#" {
fields = fields[1:]
todo.Msg = strings.Join(fields, " ")
}
return todo, nil
}
if todo.Command == Fixup {
if len(fields) == 0 {
return todo, ErrMissingCommit
}
// Skip flags
if fields[0] == "-C" || fields[0] == "-c" {
fields = fields[1:]
}
}
if len(fields) == 0 {
return todo, ErrMissingCommit
}
todo.Commit = fields[0]
fields = fields[1:]
// Trim # and whitespace
todo.Msg = strings.TrimPrefix(strings.Join(fields, " "), CommentChar+" ")
return todo, nil
}
func isCommand(i TodoCommand, s string) bool {
if i < 0 || i > Comment {
return false
}
return len(s) > 0 &&
(todoCommandInfo[i].cmd == s || todoCommandInfo[i].nickname == s)
}

View File

@@ -0,0 +1,75 @@
package todo
type TodoCommand int
const (
Pick TodoCommand = iota + 1
Revert
Edit
Reword
Fixup
Squash
Exec
Break
Label
Reset
Merge
NoOp
Drop
Comment
)
const CommentChar = "#"
type Todo struct {
Command TodoCommand
Commit string
Comment string
ExecCommand string
Label string
Msg string
}
func (t TodoCommand) String() string {
return commandToString[t]
}
var commandToString = map[TodoCommand]string{
Pick: "pick",
Revert: "revert",
Edit: "edit",
Reword: "reword",
Fixup: "fixup",
Squash: "squash",
Exec: "exec",
Break: "break",
Label: "label",
Reset: "reset",
Merge: "merge",
NoOp: "noop",
Drop: "drop",
Comment: "comment",
}
var todoCommandInfo = [14]struct {
nickname string
cmd string
}{
{"", ""}, // dummy value since we're using 1-based indexing
{"p", "pick"},
{"", "revert"},
{"e", "edit"},
{"r", "reword"},
{"f", "fixup"},
{"s", "squash"},
{"x", "exec"},
{"b", "break"},
{"l", "label"},
{"t", "reset"},
{"m", "merge"},
{"", "noop"},
{"d", "drop"},
}