github.com/grantbow/fit@v0.7.1-0.20220916164603-1f7c88ac81e6/fitapp/Tag.go

package fitapp

import (
	"fmt"
	bugs "github.com/grantbow/fit/issues"
	"os"
	"sort"
	"strconv"
	"strings"
)

//var dops = bugs.Directory(os.PathSeparator)
//var sops = string(os.PathSeparator)

// getAllTags returns all the tags
func getAllTags(config bugs.Config) map[string]int {
	bugs := bugs.GetAllIssues(config)
	//fmt.Printf("%+v\n", bugs)
	tagMap := make(map[string]int, 0)
	// Put all the tags in a map, values are count of occurrences
	for _, bug := range bugs {
		for _, tag := range bug.Tags() {
			tagMap[strings.ToLower(string(tag))] += 1
		}
	}
	return tagMap
}

func uniqueTagList(config bugs.Config) []string {
	get := getAllTags(config)
	var tags []string
	// iterate over map keys. results are unique. discard values.
	for k := range get {
		tags = append(tags, k)
	}
	sort.Strings(tags)
	return tags
}

func uniqueTagListWithValues(config bugs.Config) []string {
	get := getAllTags(config)
	var tags []string
	// iterate over map keys. results are unique.
	for k, v := range get {
		tags = append(tags, k+" "+strconv.Itoa(v))
	}
	sort.Strings(tags)
	return tags
}

// TagsNone is a subcommand to print issues with no assigned tags.
func TagsNone(config bugs.Config) {
	fitdir := bugs.FitDirer(config)
	issues := readIssues(string(fitdir))
	sort.Sort(byDir(issues))
	var wantTags bool = false

	allbugs := bugs.GetAllIssues(config)
	tagMap := make(map[string]int, 0)
	for _, bug := range allbugs {
		if len(bug.Tags()) == 0 {
			title := bug.Dir.ShortNamer()
			tagMap[string(title)] += 1
		}
	}

	//keys := make([]string, 0, len(tagMap))
	/*for k, _ := range tagMap {
		//fmt.Printf("%v\n", k)
		name := issueNamer(b, idx) // Issue x:
		fmt.Printf("%v\n", k)
		//keys = append(keys, k) // TODO: should just append not tagmap intermediary
	} */

	fmt.Printf("No tags assigned:\n")
	//fmt.Printf("%v\n", len(issues))
	for idx, issue := range issues {
		//fmt.Printf("%v\n", issue)
		for k := range tagMap {
			if issue.Name() == k {
				//fmt.Printf("1in: %v\n2tm: %v\n", issue.Name(), k)
				var dir bugs.Directory = fitdir + dops + bugs.Directory(issue.Name())
				//fmt.Printf("dir %v\n", dir)
				b := bugs.Issue{Dir: dir, DescriptionFileName: config.DescriptionFileName}
				name := issueNamer(b, idx) // Issue x:
				//fmt.Printf("name %v\n", name)
				if wantTags == false { // always
					fmt.Printf("%s: %s\n", name, b.Title(""))
					//keys = append(keys, fmt.Sprintf("%s: %s\n", name, b.Title("")))
				} else {
					fmt.Printf("%s: %s\n", name, b.Title("tags"))
					//keys = append(keys, fmt.Sprintf("%s: %s\n", name, b.Title("tags")))
				}
			}
		}
	}
	//return keys
	return
}

// TagsAssigned is a subcommand to print the assigned tags.
func TagsAssigned(Args argumentList, config bugs.Config) {
	outputCount := false
	if len(Args) == 1 &&
		(Args[0] == "-c" || Args[0] == "--count") {
		outputCount = true
	}
	//fmt.Printf("here\n")
	get := uniqueTagList(config)
	if len(get) > 0 {
		if outputCount {
			fmt.Printf("Tags used in current tree: <key:value> <count>\n")
			fmt.Printf("%s\n", strings.Join(uniqueTagListWithValues(config), "\n"))
		} else {
			fmt.Printf("Tags used in current tree: <key:value>\n")
			fmt.Printf("%s\n", strings.Join(get, "\n"))
		}
	} else {
		fmt.Print("<none assigned yet>\n")
	}
}

// TODO: need a new Tag subcommand with tag_key_value behavior

// Tag was a subcommand to assign a bool true/false tag to an issue.
func Tag(Args argumentList, config bugs.Config) {
	if len(Args) < 2 {
		fmt.Printf("Usage: %s tag [--rm] <IssueID> <tagname> [more tagnames]\n", os.Args[0])
		fmt.Printf("\nBoth issue number and tagname to set are required.\n")
		var tags = uniqueTagList(config)
		fmt.Printf("\nCurrently used tags in entire tree: %s\n", strings.Join(tags, "\n"))
		return
	}
	var removeTags bool = false
	if Args[0] == "--rm" {
		removeTags = true
		Args = Args[1:]
	}

	b, err := bugs.LoadIssueByHeuristic(Args[0], config)

	if err != nil {
		fmt.Printf("Could not load issue: %s\n", err.Error())
		return
	}
	for _, tag := range Args[1:] {
		if removeTags {
			b.RemoveTag(bugs.TagBoolTrue(tag), config)
		} else {
			b.TagIssue(bugs.TagBoolTrue(tag), config)
		}
	}

}