153 lines
3.7 KiB
Go
153 lines
3.7 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"git.ctrlz.es/mgdelacroix/campaigner/campaign"
|
|
"git.ctrlz.es/mgdelacroix/campaigner/model"
|
|
)
|
|
|
|
func GrepAddCmd() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "grep",
|
|
Short: "Generates the tickets reading grep's output from stdin",
|
|
Long: "Generates tickets for the campaign reading from the standard input the output grep. The grep command must be run with the -n flag",
|
|
Example: ` grep -nriIF --include \*.go cobra.Command | campaigner add grep`,
|
|
Args: cobra.NoArgs,
|
|
Run: grepAddCmdF,
|
|
}
|
|
|
|
cmd.Flags().BoolP("file-only", "f", false, "generates one ticket per file instead of per match")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func AgAddCmd() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "ag",
|
|
Short: "Generates the tickets reading ag's output from stdin",
|
|
Long: "Generates tickets for the campaign reading from the standard input the output ag",
|
|
Example: ` ag cobra.Command | campaigner add ag`,
|
|
Args: cobra.NoArgs,
|
|
RunE: agAddCmdF,
|
|
}
|
|
|
|
cmd.Flags().BoolP("file-only", "f", false, "generates one ticket per file instead of per match")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func GovetAddCmd() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "govet",
|
|
Short: "Generates the tickets reading govet's output from stdin",
|
|
Long: "Generates tickets for the campaign reading from the standard input the output grep. The grep command must be run with the -json flag",
|
|
Example: ` govet -json ./... | campaigner add govet`,
|
|
Args: cobra.NoArgs,
|
|
RunE: govetAddCmdF,
|
|
}
|
|
|
|
cmd.Flags().BoolP("file-only", "f", false, "generates one ticket per file instead of per match")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func CsvAddCmd() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "csv",
|
|
Short: "Generates the tickets reading a csv file",
|
|
Example: ` campaigner add csv --file tickets.csv`,
|
|
Args: cobra.NoArgs,
|
|
RunE: csvAddCmdF,
|
|
}
|
|
|
|
cmd.Flags().BoolP("file-only", "f", false, "generates one ticket per file instead of per match")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func AddCmd() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "add",
|
|
Short: "Adds tickets to the campaign from the output of grep/ag/govet",
|
|
}
|
|
|
|
cmd.AddCommand(
|
|
GrepAddCmd(),
|
|
AgAddCmd(),
|
|
GovetAddCmd(),
|
|
CsvAddCmd(),
|
|
)
|
|
|
|
return cmd
|
|
}
|
|
|
|
func parseGrepLine(line string) (*model.Ticket, error) {
|
|
// ToDo: it would be great to be able to relate a line with its
|
|
// parent method, at least for JS and Golang
|
|
parts := strings.Split(line, ":")
|
|
if len(parts) < 3 {
|
|
return nil, fmt.Errorf("cannot parse line: %s", line)
|
|
}
|
|
|
|
filename := parts[0]
|
|
lineNo, err := strconv.Atoi(parts[1])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
text := strings.Join(parts[2:], "")
|
|
|
|
return &model.Ticket{
|
|
"filename": filename,
|
|
"lineNo": lineNo,
|
|
"text": text,
|
|
}, nil
|
|
}
|
|
|
|
func parseGrep() []*model.Ticket {
|
|
tickets := []*model.Ticket{}
|
|
scanner := bufio.NewScanner(os.Stdin)
|
|
for scanner.Scan() {
|
|
ticket, _ := parseGrepLine(scanner.Text())
|
|
if ticket != nil {
|
|
tickets = append(tickets, ticket)
|
|
}
|
|
}
|
|
return tickets
|
|
}
|
|
|
|
func grepAddCmdF(cmd *cobra.Command, _ []string) {
|
|
fileOnly, _ := cmd.Flags().GetBool("file-only")
|
|
|
|
tickets := parseGrep()
|
|
|
|
cmp, err := campaign.Read()
|
|
if err != nil {
|
|
ErrorAndExit(cmd, err)
|
|
}
|
|
|
|
cmp.Tickets = append(cmp.Tickets, tickets...)
|
|
cmp.Tickets = model.RemoveDuplicateTickets(cmp.Tickets, fileOnly)
|
|
|
|
if err := campaign.Save(cmp); err != nil {
|
|
ErrorAndExit(cmd, err)
|
|
}
|
|
}
|
|
|
|
func agAddCmdF(_ *cobra.Command, _ []string) error {
|
|
return fmt.Errorf("not implemented yet")
|
|
}
|
|
|
|
func govetAddCmdF(_ *cobra.Command, _ []string) error {
|
|
return fmt.Errorf("not implemented yet")
|
|
}
|
|
|
|
func csvAddCmdF(cmd *cobra.Command, _ []string) error {
|
|
return fmt.Errorf("not implemented yet")
|
|
}
|