-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fixed build so alfred stuff is only included on darwin builds
- Loading branch information
Brent Beardsley
committed
Apr 19, 2020
1 parent
7599286
commit b8a2d13
Showing
3 changed files
with
234 additions
and
36 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,3 +1,5 @@ | ||
| // +build darwin | ||
|
|
||
| package main | ||
|
|
||
| import ( | ||
|
|
||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,228 @@ | ||
| // +build darwin | ||
|
|
||
| package main | ||
|
|
||
| import ( | ||
| "flag" | ||
| "fmt" | ||
| "log" | ||
| "os" | ||
| "regexp" | ||
| "strings" | ||
|
|
||
| "github.com/bbeardsley/histkeep" | ||
| ) | ||
|
|
||
| const version = "0.0.7" | ||
|
|
||
| func printUsage() { | ||
| fmt.Fprintln(os.Stderr, "Usage") | ||
| fmt.Fprintln(os.Stderr, " histkeep [options] <command> <file> <command arguments...>") | ||
| fmt.Fprintln(os.Stderr, "Version") | ||
| fmt.Fprintln(os.Stderr, " "+version) | ||
| fmt.Fprintln(os.Stderr, "Options") | ||
| flag.PrintDefaults() | ||
| fmt.Fprintln(os.Stderr, "Commands") | ||
| fmt.Fprintln(os.Stderr, " help -> show this help") | ||
| fmt.Fprintln(os.Stderr, " version -> print version number and exit") | ||
| fmt.Fprintln(os.Stderr, " add -> add value") | ||
| fmt.Fprintln(os.Stderr, " clear -> clear all values") | ||
| fmt.Fprintln(os.Stderr, " list -> list values") | ||
| fmt.Fprintln(os.Stderr, " remove -> remove value") | ||
| os.Exit(1) | ||
| } | ||
|
|
||
| var alfredGlobalVarFlags arrayFlags | ||
| var alfredItemVarFlags arrayFlags | ||
| var alfredCannedItemFlags arrayFlags | ||
| var alfredModItemFlags arrayFlags | ||
|
|
||
| func main() { | ||
| lastNPtr := flag.Int("last", 15, "keep the last specified number of values") | ||
| formatPtr := flag.String("format", "", "regex format for the values. Accepts NUMBER and UUID as shortcuts.") | ||
| versionPtr := flag.Bool("version", false, "print version number and exit") | ||
| filterPtr := flag.String("filter", "", "regex filter") | ||
| valuePtr := flag.String("value", "{{VALUE}}", "transforms the value before passing it on.") | ||
| ansiPtr := flag.Bool("ansi", false, "support ansi colors in value transformation") | ||
| reversePtr := flag.Bool("reverse", false, "list values in reverse order") | ||
|
|
||
| alfredPtr := flag.Bool("alfred", false, "output Alfred JSON list") | ||
| acopyPtr := flag.String("acopy", "", "text to copy in alfred when item in filter is copied. {{VALUE}} is replaced with the item value.") | ||
| aiconPtr := flag.String("aicon", "", "filename of icon to show in alfred for each item in filter. {{VALUE}} is replaced with item value.") | ||
| asubtitlePtr := flag.String("asubtitle", "", "subtitle to display for the item in alfred. {{VALUE}} is replaced with the item value.") | ||
| atitlePtr := flag.String("atitle", "{{VALUE}}", "item title in alfred. {{VALUE}} is replaced with the item value.") | ||
| aargPtr := flag.String("aarg", "{{VALUE}}", "item arg in alfred. {{VALUE}} is replaced with the item value.") | ||
| flag.Var(&alfredItemVarFlags, "avar", "name=value to be passed to alfred. {{VALUE}} is replaced with item value in value. Parameter can be specified multiple times for multiple variables.") | ||
| flag.Var(&alfredCannedItemFlags, "aitem", "item to include in alfred list. Parameter can be specified multiple times for multiple items") | ||
| flag.Var(&alfredGlobalVarFlags, "agvar", "name=value to be passed to alfred as a global variable. Parameter can be specified multiple times for multiple variables.") | ||
| flag.Var(&alfredModItemFlags, "amod", "(alt|cmd|ctrl|fn):(var|valid|arg|subtitle|icon):(value|name=value)") | ||
|
|
||
| flag.Parse() | ||
|
|
||
| if *versionPtr { | ||
| fmt.Println(version) | ||
| os.Exit(0) | ||
| } | ||
|
|
||
| command := strings.TrimSpace(flag.Arg(0)) | ||
| file := strings.TrimSpace(flag.Arg(1)) | ||
| value := strings.TrimSpace(flag.Arg(2)) | ||
| format := buildFormat(*formatPtr) | ||
| hist := histkeep.NewHistKeep(file, *lastNPtr, format) | ||
|
|
||
| switch command { | ||
| case "", "h", "-h", "--h", "/h", "/?", "help", "-help", "--help", "/help": | ||
| printUsage() | ||
| case "version", "-version", "--version", "/version": | ||
| fmt.Println(version) | ||
| case "add": | ||
| if file == "" || value == "" { | ||
| printUsage() | ||
| } | ||
|
|
||
| err := hist.AddValue(value) | ||
| if err != nil { | ||
| log.Fatal(err) | ||
| } | ||
| case "clear": | ||
| if file == "" { | ||
| printUsage() | ||
| } | ||
|
|
||
| err := hist.ClearValues() | ||
| if err != nil { | ||
| log.Fatal(err) | ||
| } | ||
| case "remove": | ||
| if file == "" || value == "" { | ||
| printUsage() | ||
| } | ||
|
|
||
| err := hist.RemoveValue(value) | ||
| if err != nil { | ||
| log.Fatal(err) | ||
| } | ||
| case "list": | ||
| if file == "" { | ||
| printUsage() | ||
| } | ||
|
|
||
| filterFunc := buildFilterFunc(*filterPtr) | ||
| lines, err := hist.GetFilteredValues(filterFunc) | ||
| if err != nil { | ||
| log.Fatal(err) | ||
| } | ||
|
|
||
| if *reversePtr { | ||
| lines = hist.ReverseValues(lines) | ||
| } | ||
|
|
||
| lines = replaceAllPlaceholders(lines, *valuePtr, "VALUE") | ||
|
|
||
| if *ansiPtr { | ||
| lines = handleAnsiCodes(lines) | ||
| } | ||
|
|
||
| if *alfredPtr { | ||
| a := alfred{ | ||
| itemTitle: *atitlePtr, | ||
| itemArg: *aargPtr, | ||
| itemSubtitle: *asubtitlePtr, | ||
| iconFilename: *aiconPtr, | ||
| copyText: *acopyPtr, | ||
| itemVars: alfredItemVarFlags, | ||
| cannedItems: alfredCannedItemFlags, | ||
| filter: *filterPtr, | ||
| filterFunc: filterFunc, | ||
| format: format, | ||
| globalVars: alfredGlobalVarFlags, | ||
| replacePlaceholder: replacePlaceholder, | ||
| itemModifiers: alfredModItemFlags, | ||
| } | ||
| a.list(lines) | ||
| } else { | ||
| listValues(lines) | ||
| } | ||
| default: | ||
| printUsage() | ||
| } | ||
|
|
||
| return | ||
| } | ||
|
|
||
| func replaceAllPlaceholders(values []string, template string, placeholder string) []string { | ||
| replaced := make([]string, len(values)) | ||
| for i, line := range values { | ||
| replaced[i] = replacePlaceholder(template, placeholder, line) | ||
| } | ||
| return replaced | ||
| } | ||
|
|
||
| func handleAnsiCodes(values []string) []string { | ||
| re := regexp.MustCompile("(?i)\\\\(e|033|x1b)") | ||
| replaced := make([]string, len(values)) | ||
| for i, line := range values { | ||
| replaced[i] = re.ReplaceAllString(line, "\x1B") | ||
| } | ||
| return replaced | ||
| } | ||
|
|
||
| func replacePlaceholder(input string, placeholder string, replacementValue string) string { | ||
| return strings.Replace(input, "{{"+placeholder+"}}", replacementValue, -1) | ||
| } | ||
|
|
||
| func buildFilterFunc(filter string) func(string) bool { | ||
| var filterFunc func(string) bool | ||
| if filter != "" { | ||
| filterRegex, err := regexp.Compile("^(?i)" + filter) | ||
| if err != nil { | ||
| log.Fatal(err) | ||
| } | ||
| filterFunc = func(line string) bool { | ||
| return filterRegex.MatchString(line) | ||
| } | ||
| } else { | ||
| filterFunc = func(line string) bool { | ||
| return true | ||
| } | ||
| } | ||
| return filterFunc | ||
| } | ||
|
|
||
| func listValues(values []string) { | ||
| for i, line := range values { | ||
| if i != 0 { | ||
| fmt.Println() | ||
| } | ||
| fmt.Print(line) | ||
| } | ||
| } | ||
|
|
||
| func buildFormat(formatStr string) *regexp.Regexp { | ||
| format := processedNamedFormats(formatStr) | ||
| return regexp.MustCompile("^" + format + "$") | ||
| } | ||
|
|
||
| func processedNamedFormats(formatStr string) string { | ||
| switch formatStr { | ||
| case "NUMBER": | ||
| return "\\d+" | ||
| case "UUID": | ||
| return "([a-fA-F0-9]{8}-[a-fA-F0-9]{4}-[a-fA-F0-9]{4}-[a-fA-F0-9]{4}-[a-fA-F0-9]{12}){1}" | ||
| case "": | ||
| return ".*" | ||
| default: | ||
| return formatStr | ||
| } | ||
| } | ||
|
|
||
| type arrayFlags []string | ||
|
|
||
| func (i *arrayFlags) String() string { | ||
| return "array flags" | ||
| } | ||
|
|
||
| func (i *arrayFlags) Set(value string) error { | ||
| *i = append(*i, value) | ||
| return nil | ||
| } |