diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/core.go | 11 | ||||
| -rw-r--r-- | src/item.go | 6 | ||||
| -rw-r--r-- | src/terminal.go | 4 |
3 files changed, 17 insertions, 4 deletions
diff --git a/src/core.go b/src/core.go index d639769d..404af12a 100644 --- a/src/core.go +++ b/src/core.go @@ -476,8 +476,17 @@ func Run(opts *Options) (int, error) { if len(opts.Expect) > 0 { opts.Printer("") } + transformer := func(item *Item) string { + return item.AsString(opts.Ansi) + } + if opts.AcceptNth != nil { + fn := opts.AcceptNth(opts.Delimiter) + transformer = func(item *Item) string { + return item.acceptNth(opts.Ansi, opts.Delimiter, fn) + } + } for i := 0; i < count; i++ { - opts.Printer(val.Get(i).item.AsString(opts.Ansi)) + opts.Printer(transformer(val.Get(i).item)) } if count == 0 { exitCode = ExitNoMatch diff --git a/src/item.go b/src/item.go index ca32f1bd..19f0498e 100644 --- a/src/item.go +++ b/src/item.go @@ -51,3 +51,9 @@ func (item *Item) AsString(stripAnsi bool) string { } return item.text.ToString() } + +func (item *Item) acceptNth(stripAnsi bool, delimiter Delimiter, transformer func([]Token, int32) string) string { + tokens := Tokenize(item.AsString(stripAnsi), delimiter) + transformed := transformer(tokens, item.Index()) + return StripLastDelimiter(transformed, delimiter) +} diff --git a/src/terminal.go b/src/terminal.go index a762853e..2f00a398 100644 --- a/src/terminal.go +++ b/src/terminal.go @@ -1575,9 +1575,7 @@ func (t *Terminal) output() bool { } if t.acceptNth != nil { transform = func(item *Item) string { - tokens := Tokenize(item.AsString(t.ansi), t.delimiter) - transformed := t.acceptNth(tokens, item.Index()) - return StripLastDelimiter(transformed, t.delimiter) + return item.acceptNth(t.ansi, t.delimiter, t.acceptNth) } } found := len(t.selected) > 0 |