Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
041406a340 | ||
|
|
b3311cbf94 | ||
|
|
da281ca4ab | ||
|
|
187de30351 |
No files matched your search
@@ -17,6 +17,8 @@ Built-in commands:
|
|||||||
fg [%n] - Continue job n (default: current job) in the foreground
|
fg [%n] - Continue job n (default: current job) in the foreground
|
||||||
bg [%n] - Continue stopped job n in the background
|
bg [%n] - Continue stopped job n in the background
|
||||||
wait [%n] - Wait for job n or all background jobs
|
wait [%n] - Wait for job n or all background jobs
|
||||||
|
alias [name[=value] ...] - Define or show aliases (alias ll='ls -l')
|
||||||
|
unalias name - Remove an alias (unalias -a: all aliases)
|
||||||
help - Show this help
|
help - Show this help
|
||||||
|
|
||||||
Features:
|
Features:
|
||||||
@@ -28,13 +30,17 @@ Features:
|
|||||||
override with $GOSH_HISTFILE); lines starting with a space are not saved
|
override with $GOSH_HISTFILE); lines starting with a space are not saved
|
||||||
- Tab completion for commands and file paths
|
- Tab completion for commands and file paths
|
||||||
- Tilde expansion (~, ~/path) at the start of unquoted words
|
- Tilde expansion (~, ~/path) at the start of unquoted words
|
||||||
|
- Brace expansion: a{b,c}d -> abd acd, {1..5}, {01..10..2}, {a..e}
|
||||||
- Wildcards: * ? [abc] [a-z] [!abc] match file names (not hidden files
|
- Wildcards: * ? [abc] [a-z] [!abc] match file names (not hidden files
|
||||||
unless the pattern starts with a dot); quoted wildcards and values of
|
unless the pattern starts with a dot); quoted wildcards and values of
|
||||||
variables are not expanded, a pattern without match is kept as is
|
variables are not expanded, a pattern without match is kept as is
|
||||||
- Colored prompt (user@host green, red for root; directory blue),
|
- Prompt user@host:dir $, with [N] before the $ if the last command
|
||||||
configurable with $GOSH_COLORS, e.g. GOSH_COLORS='user=1;33:path=36'
|
failed with exit status N
|
||||||
(keys user, root, path; values are ANSI SGR codes, empty = no color);
|
- Colored prompt (user@host green, red for root; directory blue; exit
|
||||||
disabled with $NO_COLOR, TERM=dumb or when output is not a terminal
|
status red), configurable with $GOSH_COLORS, e.g.
|
||||||
|
GOSH_COLORS='user=1;33:path=36' (keys user, root, path, error; values
|
||||||
|
are ANSI SGR codes, empty = no color); disabled with $NO_COLOR,
|
||||||
|
TERM=dumb or when output is not a terminal
|
||||||
- Environment variable expansion ($VAR, ${VAR}, ${VAR:-default});
|
- Environment variable expansion ($VAR, ${VAR}, ${VAR:-default});
|
||||||
using an unset variable is an error, use ${VAR:-} for an empty default
|
using an unset variable is an error, use ${VAR:-} for an empty default
|
||||||
- Command lists: a ; b (run both), a && b (b only if a succeeded),
|
- Command lists: a ; b (run both), a && b (b only if a succeeded),
|
||||||
@@ -43,6 +49,8 @@ Features:
|
|||||||
2>> file, 2>&1, >&2, &> file (stdout and stderr), &>> file
|
2>> file, 2>&1, >&2, &> file (stdout and stderr), &>> file
|
||||||
- Job control: command & runs in the background, Ctrl+Z stops the
|
- Job control: command & runs in the background, Ctrl+Z stops the
|
||||||
foreground job; jobs, fg, bg and wait manage the jobs
|
foreground job; jobs, fg, bg and wait manage the jobs
|
||||||
|
- Aliases replace the first word of a command in interactive shells
|
||||||
|
(define them in ~/.goshrc); \name or 'name' skips the alias
|
||||||
- Non-interactive use: gosh -c 'command', gosh script, command | gosh;
|
- Non-interactive use: gosh -c 'command', gosh script, command | gosh;
|
||||||
the exit status is the one of the last command
|
the exit status is the one of the last command
|
||||||
- Startup files: login shells (-l or argv[0] starting with -) read
|
- Startup files: login shells (-l or argv[0] starting with -) read
|
||||||
@@ -96,6 +104,10 @@ Beispiel für `~/.goshrc`:
|
|||||||
export PATH=$HOME/bin:$HOME/go/bin:$PATH
|
export PATH=$HOME/bin:$HOME/go/bin:$PATH
|
||||||
export EDITOR=vim
|
export EDITOR=vim
|
||||||
|
|
||||||
|
# Aliase
|
||||||
|
alias ll='ls -l'
|
||||||
|
alias ..='cd ..'
|
||||||
|
|
||||||
# Prompt-Farben (Anführungszeichen wegen des ;)
|
# Prompt-Farben (Anführungszeichen wegen des ;)
|
||||||
export GOSH_COLORS='user=1;38;5;208:path=38;5;205'
|
export GOSH_COLORS='user=1;38;5;208:path=38;5;205'
|
||||||
|
|
||||||
@@ -134,6 +146,35 @@ Status 127
|
|||||||
|
|
||||||
Schlägt ein Programm fehl, gibt gosh selbst keine Meldung mehr aus (das Programm meldet den Fehler selbst), nur bei Fehlern von gosh, z. B. wenn ein Befehl nicht gefunden wird.
|
Schlägt ein Programm fehl, gibt gosh selbst keine Meldung mehr aus (das Programm meldet den Fehler selbst), nur bei Fehlern von gosh, z. B. wenn ein Befehl nicht gefunden wird.
|
||||||
|
|
||||||
|
### Klammer-Expansion
|
||||||
|
|
||||||
|
Wie in bash erzeugt `{…}` aus einem Wort mehrere Wörter:
|
||||||
|
|
||||||
|
| Ausdruck | ergibt |
|
||||||
|
|-------------------|---------------------------------|
|
||||||
|
| `a{b,c}d` | `abd acd` |
|
||||||
|
| `file{,.bak}` | `file file.bak` |
|
||||||
|
| `{a,b}{1,2}` | `a1 a2 b1 b2` |
|
||||||
|
| `x{a,{b,c}}` | `xa xb xc` (verschachtelt) |
|
||||||
|
| `{1..5}` | `1 2 3 4 5` |
|
||||||
|
| `{5..1}` | `5 4 3 2 1` |
|
||||||
|
| `{1..10..3}` | `1 4 7 10` (Schrittweite) |
|
||||||
|
| `{01..10..4}` | `01 05 09` (mit führenden Nullen) |
|
||||||
|
| `{a..e}` | `a b c d e` |
|
||||||
|
|
||||||
|
```
|
||||||
|
$ mkdir -p projekt/{src,doc,test}
|
||||||
|
$ cp config.yaml{,.bak} # Sicherungskopie
|
||||||
|
$ ls *.{go,md}
|
||||||
|
$ touch log-{01..12}.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
- Die Klammer-Expansion kommt zuerst, die Ergebnisse durchlaufen danach `~`, Variablen und Wildcards: `*.{go,md}` wird zu `*.go *.md`, `{${A},b}` zu `$A b`.
|
||||||
|
- Ohne Komma (und ohne gültige Folge) bleiben Klammern stehen: `{a}`, `{}`, `{a..1}`. Auch `${VAR}` wird nicht angefasst.
|
||||||
|
- In Anführungszeichen oder mit `\` davor sind die Zeichen normale Zeichen: `"{a,b}"`, `\{a,b\}`. Ein Komma lässt sich mit `\,` schützen: `{a\,b,c}` ergibt `a,b c`.
|
||||||
|
- Bei einer Umleitung muss genau ein Wort herauskommen, `> {a,b}` meldet `ambiguous redirect`.
|
||||||
|
- Ein Wort darf höchstens 100 000 Wörter erzeugen.
|
||||||
|
|
||||||
### Wildcards
|
### Wildcards
|
||||||
|
|
||||||
Wörter mit `*`, `?` oder `[...]` werden durch die passenden Dateinamen ersetzt, sortiert:
|
Wörter mit `*`, `?` oder `[...]` werden durch die passenden Dateinamen ersetzt, sortiert:
|
||||||
@@ -186,6 +227,34 @@ $ grep -c error < build.log
|
|||||||
- Der Exit-Status einer Pipeline ist der des letzten Befehls. Beendet sich ein Befehl am Ende (z. B. `head`), bekommt der schreibende Befehl davor `SIGPIPE`, `yes | head -1` endet also sofort.
|
- Der Exit-Status einer Pipeline ist der des letzten Befehls. Beendet sich ein Befehl am Ende (z. B. `head`), bekommt der schreibende Befehl davor `SIGPIPE`, `yes | head -1` endet also sofort.
|
||||||
- Die Built-ins `pwd` und `help` lassen sich umleiten und in Pipes verwenden (`help | less`). `cd`, `set`, `export`, `exit` und `quit` verändern die Shell selbst und sind in einer Pipeline nicht erlaubt.
|
- Die Built-ins `pwd` und `help` lassen sich umleiten und in Pipes verwenden (`help | less`). `cd`, `set`, `export`, `exit` und `quit` verändern die Shell selbst und sind in einer Pipeline nicht erlaubt.
|
||||||
|
|
||||||
|
### Aliase
|
||||||
|
|
||||||
|
Ein Alias ist ein Kurzname für einen Befehl, am besten in `~/.goshrc` definiert:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
alias ll='ls -l'
|
||||||
|
alias la='ls -la'
|
||||||
|
alias ..='cd ..'
|
||||||
|
alias gs='git status'
|
||||||
|
alias ls='ls --color=auto' # ein Alias darf den eigenen Namen enthalten
|
||||||
|
alias please='sudo ' # endet mit Leerzeichen: das nächste Wort wird auch ersetzt
|
||||||
|
```
|
||||||
|
|
||||||
|
| Befehl | Bedeutung |
|
||||||
|
|-------------------------|---------------------------------------------------|
|
||||||
|
| `alias name='wert'` | definiert einen Alias (mehrere in einem Aufruf möglich) |
|
||||||
|
| `alias` | listet alle Aliase (in einer Form, die sich wieder einlesen lässt) |
|
||||||
|
| `alias name` | zeigt einen Alias |
|
||||||
|
| `unalias name` | entfernt einen Alias, `unalias -a` alle |
|
||||||
|
|
||||||
|
- Ersetzt wird nur das erste Wort eines Befehls, also am Zeilenanfang und nach `;`, `&&`, `||`, `|` und `&`: `ll | less` funktioniert, `echo ll` gibt `ll` aus.
|
||||||
|
- Der Wert darf selbst mehrere Befehle enthalten (`alias up='cd .. && ls'`) und wird wiederum auf Aliase geprüft. Ein Alias, der gerade ersetzt wird, wird darin nicht noch einmal ersetzt, deshalb funktioniert `alias ls='ls --color=auto'`.
|
||||||
|
- Endet der Wert mit einem Leerzeichen, wird auch das nächste Wort als Alias geprüft: Mit den Aliasen oben wird `please ll` zu `sudo ls --color=auto -l`.
|
||||||
|
- Mit `\ll` oder `'ll'` umgeht man den Alias und ruft das Programm direkt auf.
|
||||||
|
- Aliase gelten nur in interaktiven Shells (wie in bash), also auch in `~/.goshrc`, aber nicht in Skripten und bei `gosh -c`. Ein Alias wirkt erst ab der nächsten Zeile, nicht in der Zeile, in der er definiert wird.
|
||||||
|
- Erlaubt sind Namen aus Buchstaben, Ziffern und `_ - . + : , @ % ^`.
|
||||||
|
- Aliase werden auch mit Tab vervollständigt.
|
||||||
|
|
||||||
### Job-Control
|
### Job-Control
|
||||||
|
|
||||||
In einer interaktiven Shell läuft jede Befehlszeile (bzw. jede Pipeline) als eigener *Job* in einer eigenen Prozessgruppe:
|
In einer interaktiven Shell läuft jede Befehlszeile (bzw. jede Pipeline) als eigener *Job* in einer eigenen Prozessgruppe:
|
||||||
@@ -220,8 +289,25 @@ $ bg %2 # oder: angehaltenen Job im Hintergrund weiterlauf
|
|||||||
- `exit` (oder Ctrl+D) mit angehaltenen Jobs warnt einmal (`there are stopped jobs`), ein zweites `exit` direkt danach beendet gosh.
|
- `exit` (oder Ctrl+D) mit angehaltenen Jobs warnt einmal (`there are stopped jobs`), ein zweites `exit` direkt danach beendet gosh.
|
||||||
- `cd`, `set`, `export`, `exit`, `source`, `fg`, `bg` und `wait` lassen sich nicht mit `&` im Hintergrund starten.
|
- `cd`, `set`, `export`, `exit`, `source`, `fg`, `bg` und `wait` lassen sich nicht mit `&` im Hintergrund starten.
|
||||||
- In Skripten und mit `gosh -c` gibt es keine Job-Control (kein Ctrl+Z, kein `fg`/`bg`), `&` und `wait` funktionieren aber. Hintergrund-Jobs lesen dort von `/dev/null`.
|
- In Skripten und mit `gosh -c` gibt es keine Job-Control (kein Ctrl+Z, kein `fg`/`bg`), `&` und `wait` funktionieren aber. Hintergrund-Jobs lesen dort von `/dev/null`.
|
||||||
|
- Wird ein angehaltener Job von außen fortgesetzt (`kill -CONT`), zeigt `jobs` ihn weiter als `Stopped`, bis er endet oder erneut angehalten wird; `fg` und `bg` funktionieren trotzdem.
|
||||||
- Einen Job beenden lässt sich mit `kill` über die Prozess-ID (z. B. `kill 12345`); `kill %1` versteht gosh noch nicht.
|
- Einen Job beenden lässt sich mit `kill` über die Prozess-ID (z. B. `kill 12345`); `kill %1` versteht gosh noch nicht.
|
||||||
|
|
||||||
|
### Prompt
|
||||||
|
|
||||||
|
Der Prompt zeigt Benutzer, Rechner und Verzeichnis. Ist der letzte Befehl fehlgeschlagen, steht sein Exit-Status in eckigen Klammern davor, bis wieder ein Befehl erfolgreich war:
|
||||||
|
|
||||||
|
```
|
||||||
|
flow@host:~/src $ false
|
||||||
|
flow@host:~/src [1] $ gibtsnicht
|
||||||
|
gosh: gibtsnicht: command not found
|
||||||
|
flow@host:~/src [127] $ sleep 100 # mit Ctrl+C abgebrochen
|
||||||
|
^C
|
||||||
|
flow@host:~/src [130] $ true
|
||||||
|
flow@host:~/src $
|
||||||
|
```
|
||||||
|
|
||||||
|
Typische Werte: `1` allgemeiner Fehler, `2` Syntaxfehler, `126` nicht ausführbar, `127` Befehl nicht gefunden, `128+n` durch Signal `n` beendet (`130` = Ctrl+C, `148` = mit Ctrl+Z angehalten unter Linux). Eine leere Eingabe ändert den Status nicht.
|
||||||
|
|
||||||
### Farben
|
### Farben
|
||||||
|
|
||||||
Der Prompt wird über die Umgebungsvariable `GOSH_COLORS` eingefärbt. Sie besteht aus durch `:` getrennten Einträgen `schlüssel=code`:
|
Der Prompt wird über die Umgebungsvariable `GOSH_COLORS` eingefärbt. Sie besteht aus durch `:` getrennten Einträgen `schlüssel=code`:
|
||||||
@@ -231,6 +317,7 @@ Der Prompt wird über die Umgebungsvariable `GOSH_COLORS` eingefärbt. Sie beste
|
|||||||
| `user` | `user@host` | `1;32` (fett grün) |
|
| `user` | `user@host` | `1;32` (fett grün) |
|
||||||
| `root` | `user@host`, wenn man root ist | `1;31` (fett rot) |
|
| `root` | `user@host`, wenn man root ist | `1;31` (fett rot) |
|
||||||
| `path` | aktuelles Verzeichnis | `1;34` (fett blau) |
|
| `path` | aktuelles Verzeichnis | `1;34` (fett blau) |
|
||||||
|
| `error` | Exit-Status `[N]` nach einem Fehler | `1;31` (fett rot) |
|
||||||
|
|
||||||
Die Werte sind [ANSI-SGR-Codes](https://en.wikipedia.org/wiki/ANSI_escape_code#SGR_(Select_Graphic_Rendition)_parameters), mehrere werden mit `;` kombiniert:
|
Die Werte sind [ANSI-SGR-Codes](https://en.wikipedia.org/wiki/ANSI_escape_code#SGR_(Select_Graphic_Rendition)_parameters), mehrere werden mit `;` kombiniert:
|
||||||
|
|
||||||
@@ -268,6 +355,9 @@ GOSH_COLORS='user=1;38;2;136;192;208:path=38;2;216;222;233'
|
|||||||
# Ozean: türkis und kursiv blau
|
# Ozean: türkis und kursiv blau
|
||||||
GOSH_COLORS='user=1;96:path=3;94'
|
GOSH_COLORS='user=1;96:path=3;94'
|
||||||
|
|
||||||
|
# Fehler-Status gelb auf rotem Hintergrund
|
||||||
|
GOSH_COLORS='error=1;33;41'
|
||||||
|
|
||||||
# Root-Warnung: weiß auf rotem Hintergrund
|
# Root-Warnung: weiß auf rotem Hintergrund
|
||||||
GOSH_COLORS='root=1;97;41'
|
GOSH_COLORS='root=1;97;41'
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,191 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
// Aliases: like in bash, an alias replaces the first word of a command and
|
||||||
|
// is only expanded in interactive shells (including the startup files), not
|
||||||
|
// in scripts or with gosh -c.
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
aliasMu sync.Mutex
|
||||||
|
aliases = map[string]string{}
|
||||||
|
)
|
||||||
|
|
||||||
|
// isAliasName reports whether s can be an alias name: letters, digits and a
|
||||||
|
// few punctuation characters, but nothing that is special to the shell.
|
||||||
|
func isAliasName(s string) bool {
|
||||||
|
if s == "" {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
for _, r := range s {
|
||||||
|
switch {
|
||||||
|
case r >= 'a' && r <= 'z', r >= 'A' && r <= 'Z', r >= '0' && r <= '9':
|
||||||
|
case strings.ContainsRune("_-.+:,@%^", r):
|
||||||
|
default:
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// lookupAlias returns the value of an alias.
|
||||||
|
func lookupAlias(name string) (string, bool) {
|
||||||
|
aliasMu.Lock()
|
||||||
|
defer aliasMu.Unlock()
|
||||||
|
value, ok := aliases[name]
|
||||||
|
return value, ok
|
||||||
|
}
|
||||||
|
|
||||||
|
// aliasNames returns the names of all aliases, sorted.
|
||||||
|
func aliasNames() []string {
|
||||||
|
aliasMu.Lock()
|
||||||
|
defer aliasMu.Unlock()
|
||||||
|
names := make([]string, 0, len(aliases))
|
||||||
|
for name := range aliases {
|
||||||
|
names = append(names, name)
|
||||||
|
}
|
||||||
|
sort.Strings(names)
|
||||||
|
return names
|
||||||
|
}
|
||||||
|
|
||||||
|
// expandAliases replaces aliases at the start of every command in line (at
|
||||||
|
// the start and after ;, &, &&, || and |). The value of an alias is expanded
|
||||||
|
// again, except for aliases that are already being expanded, so that e.g.
|
||||||
|
// alias ls='ls --color' works. If the value ends with a blank, the next word
|
||||||
|
// is checked for an alias too. Quoted or escaped words (\ls, 'ls') are never
|
||||||
|
// replaced.
|
||||||
|
func expandAliases(line string) string {
|
||||||
|
return expandAliasesIn(line, map[string]bool{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func expandAliasesIn(line string, active map[string]bool) string {
|
||||||
|
var (
|
||||||
|
out strings.Builder
|
||||||
|
quote byte
|
||||||
|
escaped bool
|
||||||
|
commandPos = true // the next word is the name of a command
|
||||||
|
)
|
||||||
|
|
||||||
|
for i := 0; i < len(line); i++ {
|
||||||
|
ch := line[i]
|
||||||
|
if quote == 0 && !escaped && commandPos {
|
||||||
|
if ch == ' ' || ch == '\t' {
|
||||||
|
out.WriteByte(ch)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
commandPos = false
|
||||||
|
end := i
|
||||||
|
for end < len(line) && strings.IndexByte(" \t\n;&|<>", line[end]) < 0 {
|
||||||
|
end++
|
||||||
|
}
|
||||||
|
name := line[i:end]
|
||||||
|
if value, ok := lookupAlias(name); ok && isAliasName(name) && !active[name] {
|
||||||
|
active[name] = true
|
||||||
|
out.WriteString(expandAliasesIn(value, active))
|
||||||
|
delete(active, name)
|
||||||
|
// A trailing blank makes the next word a command name too
|
||||||
|
commandPos = strings.HasSuffix(value, " ") || strings.HasSuffix(value, "\t")
|
||||||
|
i = end - 1
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
out.WriteByte(ch)
|
||||||
|
switch {
|
||||||
|
case escaped:
|
||||||
|
escaped = false
|
||||||
|
case quote == '\'':
|
||||||
|
if ch == '\'' {
|
||||||
|
quote = 0
|
||||||
|
}
|
||||||
|
case ch == '\\':
|
||||||
|
escaped = true
|
||||||
|
case quote == '"':
|
||||||
|
if ch == '"' {
|
||||||
|
quote = 0
|
||||||
|
}
|
||||||
|
case ch == '"' || ch == '\'':
|
||||||
|
quote = ch
|
||||||
|
case ch == '#' && (i == 0 || strings.IndexByte(" \t", line[i-1]) >= 0):
|
||||||
|
// Comment until the end of the line
|
||||||
|
out.WriteString(line[i+1:])
|
||||||
|
return out.String()
|
||||||
|
case ch == ';' || ch == '|':
|
||||||
|
commandPos = true
|
||||||
|
case ch == '&':
|
||||||
|
// & and && separate commands, &>, >& and <& do not
|
||||||
|
if !(i+1 < len(line) && line[i+1] == '>') && !(i > 0 && strings.IndexByte("<>", line[i-1]) >= 0) {
|
||||||
|
commandPos = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// quoteAlias quotes a value for the output of alias, so that it can be read
|
||||||
|
// back: 'value' with ' written as '\”.
|
||||||
|
func quoteAlias(value string) string {
|
||||||
|
return "'" + strings.ReplaceAll(value, "'", `'\''`) + "'"
|
||||||
|
}
|
||||||
|
|
||||||
|
// builtinAlias defines aliases (alias name=value ...), prints them (alias
|
||||||
|
// name ...) or lists all of them (alias).
|
||||||
|
func builtinAlias(args []string, stdout io.Writer) error {
|
||||||
|
if len(args) == 1 {
|
||||||
|
for _, name := range aliasNames() {
|
||||||
|
value, _ := lookupAlias(name)
|
||||||
|
fmt.Fprintf(stdout, "alias %s=%s\n", name, sanitize(quoteAlias(value)))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var err error
|
||||||
|
for _, arg := range args[1:] {
|
||||||
|
name, value, define := strings.Cut(arg, "=")
|
||||||
|
if !isAliasName(name) {
|
||||||
|
err = fmt.Errorf("alias: `%s': invalid alias name", sanitize(name))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if define {
|
||||||
|
aliasMu.Lock()
|
||||||
|
aliases[name] = value
|
||||||
|
aliasMu.Unlock()
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if value, ok := lookupAlias(name); ok {
|
||||||
|
fmt.Fprintf(stdout, "alias %s=%s\n", name, sanitize(quoteAlias(value)))
|
||||||
|
} else {
|
||||||
|
err = fmt.Errorf("alias: %s: not found", sanitize(name))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// builtinUnalias removes aliases (unalias name ...) or all of them
|
||||||
|
// (unalias -a).
|
||||||
|
func builtinUnalias(args []string) error {
|
||||||
|
if len(args) < 2 {
|
||||||
|
return fmt.Errorf("usage: unalias [-a] name ...")
|
||||||
|
}
|
||||||
|
aliasMu.Lock()
|
||||||
|
defer aliasMu.Unlock()
|
||||||
|
if args[1] == "-a" {
|
||||||
|
aliases = map[string]string{}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var err error
|
||||||
|
for _, name := range args[1:] {
|
||||||
|
if _, ok := aliases[name]; !ok {
|
||||||
|
err = fmt.Errorf("unalias: %s: not found", sanitize(name))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
delete(aliases, name)
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
+164
@@ -0,0 +1,164 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"reflect"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
// withAliases sets the aliases for a test and restores them afterwards.
|
||||||
|
func withAliases(t *testing.T, defs map[string]string) {
|
||||||
|
t.Helper()
|
||||||
|
aliasMu.Lock()
|
||||||
|
saved := aliases
|
||||||
|
aliases = map[string]string{}
|
||||||
|
for name, value := range defs {
|
||||||
|
aliases[name] = value
|
||||||
|
}
|
||||||
|
aliasMu.Unlock()
|
||||||
|
t.Cleanup(func() {
|
||||||
|
aliasMu.Lock()
|
||||||
|
aliases = saved
|
||||||
|
aliasMu.Unlock()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestExpandAliases(t *testing.T) {
|
||||||
|
withAliases(t, map[string]string{
|
||||||
|
"ll": "ls -l",
|
||||||
|
"ls": "ls --color",
|
||||||
|
"..": "cd ..",
|
||||||
|
"e": "echo ",
|
||||||
|
"w": "world",
|
||||||
|
"a": "b",
|
||||||
|
"b": "a",
|
||||||
|
"multi": "cd /tmp && ls",
|
||||||
|
})
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
input, want string
|
||||||
|
}{
|
||||||
|
{"ll", "ls --color -l"},
|
||||||
|
{" ll -a", " ls --color -l -a"},
|
||||||
|
{"ls", "ls --color"},
|
||||||
|
{"..", "cd .."},
|
||||||
|
// Only the command name is replaced
|
||||||
|
{"echo ll", "echo ll"},
|
||||||
|
{"ll ll", "ls --color -l ll"},
|
||||||
|
// ... also after ;, &&, ||, | and &
|
||||||
|
{"ll; ll && ll || ll | ll & ll", "ls --color -l; ls --color -l && ls --color -l || ls --color -l | ls --color -l & ls --color -l"},
|
||||||
|
// but not after redirections with &
|
||||||
|
{"ll 2>&1 ll", "ls --color -l 2>&1 ll"},
|
||||||
|
{"ll &> ll", "ls --color -l &> ll"},
|
||||||
|
// Quoted or escaped words are not replaced
|
||||||
|
{`\ll`, `\ll`},
|
||||||
|
{`'ll'`, `'ll'`},
|
||||||
|
{`"ll"`, `"ll"`},
|
||||||
|
{`echo "a; ll"`, `echo "a; ll"`},
|
||||||
|
{`echo 'a | ll'`, `echo 'a | ll'`},
|
||||||
|
// A trailing blank makes the next word a command name too
|
||||||
|
{"e w", "echo world"},
|
||||||
|
{"e e w", "echo echo world"},
|
||||||
|
// Recursion stops at aliases that are already being expanded
|
||||||
|
{"a", "a"},
|
||||||
|
{"b", "b"},
|
||||||
|
// The value can contain several commands
|
||||||
|
{"multi", "cd /tmp && ls --color"},
|
||||||
|
// Comments are left alone
|
||||||
|
{"echo x # ll", "echo x # ll"},
|
||||||
|
}
|
||||||
|
for _, tc := range tests {
|
||||||
|
if got := expandAliases(tc.input); got != tc.want {
|
||||||
|
t.Errorf("expandAliases(%q) = %q, want %q", tc.input, got, tc.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBuiltinAlias(t *testing.T) {
|
||||||
|
withAliases(t, nil)
|
||||||
|
|
||||||
|
run := func(args ...string) (string, error) {
|
||||||
|
var out bytes.Buffer
|
||||||
|
err := builtinAlias(append([]string{"alias"}, args...), &out)
|
||||||
|
return out.String(), err
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := run("ll=ls -l", "q=echo it's", "..=cd .."); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
out, _ := run()
|
||||||
|
want := "alias ..='cd ..'\nalias ll='ls -l'\nalias q='echo it'\\''s'\n"
|
||||||
|
if out != want {
|
||||||
|
t.Errorf("alias = %q, want %q", out, want)
|
||||||
|
}
|
||||||
|
if out, _ := run("ll"); out != "alias ll='ls -l'\n" {
|
||||||
|
t.Errorf("alias ll = %q", out)
|
||||||
|
}
|
||||||
|
if _, err := run("nope"); err == nil || err.Error() != "alias: nope: not found" {
|
||||||
|
t.Errorf("alias nope: %v", err)
|
||||||
|
}
|
||||||
|
for _, name := range []string{"a/b", "a=b", "$x", "a b", ""} {
|
||||||
|
if isAliasName(name) {
|
||||||
|
t.Errorf("isAliasName(%q) = true", name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if _, err := run("a/b=x"); err == nil || !strings.Contains(err.Error(), "invalid alias name") {
|
||||||
|
t.Errorf("alias a/b=x: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := builtinUnalias([]string{"unalias", "ll", "nope"}); err == nil || err.Error() != "unalias: nope: not found" {
|
||||||
|
t.Errorf("unalias nope: %v", err)
|
||||||
|
}
|
||||||
|
if _, ok := lookupAlias("ll"); ok {
|
||||||
|
t.Error("ll still defined after unalias")
|
||||||
|
}
|
||||||
|
builtinUnalias([]string{"unalias", "-a"})
|
||||||
|
if names := aliasNames(); len(names) != 0 {
|
||||||
|
t.Errorf("aliases after unalias -a: %q", names)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestExecInput_Aliases(t *testing.T) {
|
||||||
|
withAliases(t, nil)
|
||||||
|
origDir, _ := os.Getwd()
|
||||||
|
defer os.Chdir(origDir)
|
||||||
|
dir := t.TempDir()
|
||||||
|
os.Chdir(dir)
|
||||||
|
defer func(saved bool) { interactiveShell = saved }(interactiveShell)
|
||||||
|
|
||||||
|
// Aliases are expanded in interactive shells only
|
||||||
|
interactiveShell = true
|
||||||
|
execInput(`alias say='echo hello' to=out`)
|
||||||
|
execInput("say world > out")
|
||||||
|
if got, _ := os.ReadFile("out"); string(got) != "hello world\n" {
|
||||||
|
t.Errorf("alias in interactive shell: out = %q", got)
|
||||||
|
}
|
||||||
|
|
||||||
|
interactiveShell = false
|
||||||
|
os.Remove("out")
|
||||||
|
if err := execInput("say world > out"); err == nil {
|
||||||
|
t.Error("alias was expanded in a non-interactive shell")
|
||||||
|
}
|
||||||
|
|
||||||
|
// alias works in pipelines (listing), unalias does not
|
||||||
|
interactiveShell = true
|
||||||
|
execInput("alias | grep say > out")
|
||||||
|
if got, _ := os.ReadFile(filepath.Join(dir, "out")); string(got) != "alias say='echo hello'\n" {
|
||||||
|
t.Errorf("alias | grep = %q", got)
|
||||||
|
}
|
||||||
|
if err := execInput("echo | unalias say"); err == nil {
|
||||||
|
t.Error("unalias in a pipeline must fail")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestShellCompleter_Aliases(t *testing.T) {
|
||||||
|
withAliases(t, map[string]string{"gst": "git status"})
|
||||||
|
t.Setenv("PATH", t.TempDir())
|
||||||
|
got, length := completeLine("gs")
|
||||||
|
if !reflect.DeepEqual(got, []string{"t "}) || length != 2 {
|
||||||
|
t.Errorf("completion of gs = %q, %d", got, length)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,268 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
// Brace expansion like in bash: a{b,c}d becomes abd acd, {1..5} becomes
|
||||||
|
// 1 2 3 4 5. It happens first, on the text of the command line, so that the
|
||||||
|
// resulting words go through the other expansions (~, variables, wildcards).
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// maxBraceWords limits the number of words a single word may expand to.
|
||||||
|
const maxBraceWords = 100000
|
||||||
|
|
||||||
|
// expandBraceWords applies brace expansion to every word of a command (the
|
||||||
|
// text of a pipeline). Quoted or escaped braces and ${...} are left alone.
|
||||||
|
// A redirection target must expand to a single word.
|
||||||
|
func expandBraceWords(input string) (string, error) {
|
||||||
|
var (
|
||||||
|
out strings.Builder
|
||||||
|
afterRedirect bool // the next word is the target of a redirection
|
||||||
|
)
|
||||||
|
|
||||||
|
for i := 0; i < len(input); {
|
||||||
|
ch := input[i]
|
||||||
|
switch {
|
||||||
|
case ch == ' ' || ch == '\t' || ch == '\n':
|
||||||
|
out.WriteByte(ch)
|
||||||
|
i++
|
||||||
|
case strings.IndexByte(";&|<>", ch) >= 0:
|
||||||
|
if ch == '<' || ch == '>' {
|
||||||
|
afterRedirect = true
|
||||||
|
}
|
||||||
|
out.WriteByte(ch)
|
||||||
|
i++
|
||||||
|
case ch == '#':
|
||||||
|
// A comment, until the end of the line
|
||||||
|
out.WriteString(input[i:])
|
||||||
|
return out.String(), nil
|
||||||
|
default:
|
||||||
|
end := wordEnd(input, i)
|
||||||
|
words, err := expandBraces(input[i:end])
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
if afterRedirect && len(words) > 1 {
|
||||||
|
return "", fmt.Errorf("%s: ambiguous redirect", sanitize(input[i:end]))
|
||||||
|
}
|
||||||
|
afterRedirect = false
|
||||||
|
out.WriteString(strings.Join(words, " "))
|
||||||
|
i = end
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out.String(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// wordEnd returns the end of the word starting at i: the next unquoted blank
|
||||||
|
// or operator character.
|
||||||
|
func wordEnd(s string, i int) int {
|
||||||
|
var quote byte
|
||||||
|
for ; i < len(s); i++ {
|
||||||
|
ch := s[i]
|
||||||
|
switch {
|
||||||
|
case quote == '\'':
|
||||||
|
if ch == '\'' {
|
||||||
|
quote = 0
|
||||||
|
}
|
||||||
|
case ch == '\\':
|
||||||
|
i++
|
||||||
|
case ch == '$' && i+1 < len(s) && s[i+1] == '{':
|
||||||
|
if j := strings.IndexByte(s[i:], '}'); j >= 0 {
|
||||||
|
i += j
|
||||||
|
}
|
||||||
|
case quote == '"':
|
||||||
|
if ch == '"' {
|
||||||
|
quote = 0
|
||||||
|
}
|
||||||
|
case ch == '"' || ch == '\'':
|
||||||
|
quote = ch
|
||||||
|
case strings.IndexByte(" \t\n;&|<>", ch) >= 0:
|
||||||
|
return i
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return len(s)
|
||||||
|
}
|
||||||
|
|
||||||
|
// braceScan calls fn for every unquoted, unescaped character of s that is
|
||||||
|
// not part of a ${...} reference, with its index and the brace depth.
|
||||||
|
// Scanning stops when fn returns false.
|
||||||
|
func braceScan(s string, fn func(i, depth int) bool) {
|
||||||
|
var quote byte
|
||||||
|
depth := 0
|
||||||
|
for i := 0; i < len(s); i++ {
|
||||||
|
ch := s[i]
|
||||||
|
switch {
|
||||||
|
case quote == '\'':
|
||||||
|
if ch == '\'' {
|
||||||
|
quote = 0
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
case ch == '\\':
|
||||||
|
i++
|
||||||
|
continue
|
||||||
|
case ch == '$' && i+1 < len(s) && s[i+1] == '{':
|
||||||
|
if j := strings.IndexByte(s[i:], '}'); j >= 0 {
|
||||||
|
i += j
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
case quote == '"':
|
||||||
|
if ch == '"' {
|
||||||
|
quote = 0
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
case ch == '"' || ch == '\'':
|
||||||
|
quote = ch
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if ch == '}' {
|
||||||
|
depth--
|
||||||
|
}
|
||||||
|
if !fn(i, depth) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if ch == '{' {
|
||||||
|
depth++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// expandBraces expands the first valid brace expression of a word and then
|
||||||
|
// recursively the results: a{b,c}d -> abd acd, x{1..3} -> x1 x2 x3. A brace
|
||||||
|
// expression is valid if it contains an unquoted comma at its top level or
|
||||||
|
// is a sequence; other braces ({}, {a}, ${x}) are kept as they are.
|
||||||
|
func expandBraces(word string) ([]string, error) {
|
||||||
|
// Find the candidates: unquoted { with their matching }
|
||||||
|
var opens []int
|
||||||
|
pairs := map[int]int{}
|
||||||
|
braceScan(word, func(i, depth int) bool {
|
||||||
|
switch word[i] {
|
||||||
|
case '{':
|
||||||
|
opens = append(opens, i)
|
||||||
|
case '}':
|
||||||
|
if len(opens) > 0 {
|
||||||
|
pairs[opens[len(opens)-1]] = i
|
||||||
|
opens = opens[:len(opens)-1]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
|
for start := 0; start < len(word); start++ {
|
||||||
|
end, ok := pairs[start]
|
||||||
|
if !ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
inner := word[start+1 : end]
|
||||||
|
alternatives := splitTopLevelCommas(inner)
|
||||||
|
if len(alternatives) < 2 {
|
||||||
|
seq, ok, err := braceSequence(inner)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if !ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
alternatives = seq
|
||||||
|
}
|
||||||
|
|
||||||
|
prefix, suffix := word[:start], word[end+1:]
|
||||||
|
var words []string
|
||||||
|
for _, alt := range alternatives {
|
||||||
|
expanded, err := expandBraces(prefix + alt + suffix)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
words = append(words, expanded...)
|
||||||
|
if len(words) > maxBraceWords {
|
||||||
|
return nil, errors.New("brace expansion: too many words")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return words, nil
|
||||||
|
}
|
||||||
|
return []string{word}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// splitTopLevelCommas splits s at its unquoted commas that are not inside
|
||||||
|
// nested braces.
|
||||||
|
func splitTopLevelCommas(s string) []string {
|
||||||
|
var parts []string
|
||||||
|
last := 0
|
||||||
|
braceScan(s, func(i, depth int) bool {
|
||||||
|
if s[i] == ',' && depth == 0 {
|
||||||
|
parts = append(parts, s[last:i])
|
||||||
|
last = i + 1
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
return append(parts, s[last:])
|
||||||
|
}
|
||||||
|
|
||||||
|
// braceSequence expands a sequence expression: x..y or x..y..step with
|
||||||
|
// integers (zero-padded to the same width if one of them has a leading zero)
|
||||||
|
// or single letters. ok is false if s is no sequence.
|
||||||
|
func braceSequence(s string) (words []string, ok bool, err error) {
|
||||||
|
parts := strings.Split(s, "..")
|
||||||
|
if len(parts) != 2 && len(parts) != 3 {
|
||||||
|
return nil, false, nil
|
||||||
|
}
|
||||||
|
step := 1
|
||||||
|
if len(parts) == 3 {
|
||||||
|
step, err = strconv.Atoi(parts[2])
|
||||||
|
if err != nil {
|
||||||
|
return nil, false, nil
|
||||||
|
}
|
||||||
|
if step < 0 {
|
||||||
|
step = -step
|
||||||
|
}
|
||||||
|
if step == 0 {
|
||||||
|
step = 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
from, errFrom := strconv.Atoi(parts[0])
|
||||||
|
to, errTo := strconv.Atoi(parts[1])
|
||||||
|
isLetter := func(p string) bool {
|
||||||
|
return len(p) == 1 && (p[0] >= 'a' && p[0] <= 'z' || p[0] >= 'A' && p[0] <= 'Z')
|
||||||
|
}
|
||||||
|
format := func(n int) string { return strconv.Itoa(n) }
|
||||||
|
switch {
|
||||||
|
case errFrom == nil && errTo == nil:
|
||||||
|
// {01..10}: pad to the width of the longer number
|
||||||
|
if hasLeadingZero(parts[0]) || hasLeadingZero(parts[1]) {
|
||||||
|
width := max(len(parts[0]), len(parts[1]))
|
||||||
|
format = func(n int) string {
|
||||||
|
if n < 0 {
|
||||||
|
return fmt.Sprintf("-%0*d", width-1, -n)
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%0*d", width, n)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case isLetter(parts[0]) && isLetter(parts[1]):
|
||||||
|
from, to = int(parts[0][0]), int(parts[1][0])
|
||||||
|
format = func(n int) string { return string(rune(n)) }
|
||||||
|
default:
|
||||||
|
return nil, false, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
count := (max(from, to)-min(from, to))/step + 1
|
||||||
|
if count > maxBraceWords {
|
||||||
|
return nil, false, errors.New("brace expansion: too many words")
|
||||||
|
}
|
||||||
|
if to < from {
|
||||||
|
step = -step
|
||||||
|
}
|
||||||
|
for n := from; (step > 0 && n <= to) || (step < 0 && n >= to); n += step {
|
||||||
|
words = append(words, format(n))
|
||||||
|
}
|
||||||
|
return words, true, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// hasLeadingZero reports whether a number is written with a leading zero
|
||||||
|
// (01, -05), which makes a sequence zero-padded.
|
||||||
|
func hasLeadingZero(s string) bool {
|
||||||
|
s = strings.TrimPrefix(s, "-")
|
||||||
|
return len(s) > 1 && s[0] == '0'
|
||||||
|
}
|
||||||
+118
@@ -0,0 +1,118 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"reflect"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestExpandBraces(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
word string
|
||||||
|
want []string
|
||||||
|
}{
|
||||||
|
{"a{b,c}d", []string{"abd", "acd"}},
|
||||||
|
{"{a,b}{1,2}", []string{"a1", "a2", "b1", "b2"}},
|
||||||
|
{"x{a,{b,c}}y", []string{"xay", "xby", "xcy"}},
|
||||||
|
{"file{,.bak}", []string{"file", "file.bak"}},
|
||||||
|
{"{1..5}", []string{"1", "2", "3", "4", "5"}},
|
||||||
|
{"{3..1}", []string{"3", "2", "1"}},
|
||||||
|
{"{1..10..3}", []string{"1", "4", "7", "10"}},
|
||||||
|
{"{10..1..-4}", []string{"10", "6", "2"}},
|
||||||
|
{"{01..10..4}", []string{"01", "05", "09"}},
|
||||||
|
{"{8..010..1}", []string{"008", "009", "010"}},
|
||||||
|
{"{-2..2}", []string{"-2", "-1", "0", "1", "2"}},
|
||||||
|
{"{-01..1}", []string{"-01", "000", "001"}},
|
||||||
|
{"{a..e}", []string{"a", "b", "c", "d", "e"}},
|
||||||
|
{"{C..A}", []string{"C", "B", "A"}},
|
||||||
|
{"*.{go,md}", []string{"*.go", "*.md"}},
|
||||||
|
{`{"a b",c}`, []string{`"a b"`, "c"}},
|
||||||
|
{`{a\,b,c}`, []string{`a\,b`, "c"}},
|
||||||
|
{"{$HOME,~}/x", []string{"$HOME/x", "~/x"}},
|
||||||
|
// Not expanded: quoted, escaped, no comma, ${...}, invalid sequences
|
||||||
|
{`"{a,b}"`, []string{`"{a,b}"`}},
|
||||||
|
{`'{a,b}'`, []string{`'{a,b}'`}},
|
||||||
|
{`\{a,b\}`, []string{`\{a,b\}`}},
|
||||||
|
{"{a}", []string{"{a}"}},
|
||||||
|
{"{}", []string{"{}"}},
|
||||||
|
{"{a,b", []string{"{a,b"}},
|
||||||
|
{"a,b}", []string{"a,b}"}},
|
||||||
|
{"${HOME}", []string{"${HOME}"}},
|
||||||
|
{"${X:-a,b}", []string{"${X:-a,b}"}},
|
||||||
|
{"{a..1}", []string{"{a..1}"}},
|
||||||
|
{"{1..2..x}", []string{"{1..2..x}"}},
|
||||||
|
{"{ab..cd}", []string{"{ab..cd}"}},
|
||||||
|
// An invalid brace before a valid one
|
||||||
|
{"{a}{b,c}", []string{"{a}b", "{a}c"}},
|
||||||
|
}
|
||||||
|
for _, tc := range tests {
|
||||||
|
got, err := expandBraces(tc.word)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("expandBraces(%q) returned error: %v", tc.word, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if !reflect.DeepEqual(got, tc.want) {
|
||||||
|
t.Errorf("expandBraces(%q) = %q, want %q", tc.word, got, tc.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := expandBraces("{1..1000000}"); err == nil {
|
||||||
|
t.Error("expected an error for a too large sequence")
|
||||||
|
}
|
||||||
|
if _, err := expandBraces("{1..11}{1..100}{1..100}"); err == nil {
|
||||||
|
t.Error("expected an error for too many words")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestExpandBraceWords(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
input, want string
|
||||||
|
}{
|
||||||
|
{"echo a{b,c} d", "echo ab ac d"},
|
||||||
|
{"echo {1..3}|wc -w", "echo 1 2 3|wc -w"},
|
||||||
|
{"echo a > {f}", "echo a > {f}"},
|
||||||
|
{"echo {a,b} 2>&1", "echo a b 2>&1"},
|
||||||
|
{"echo x # {a,b}", "echo x # {a,b}"},
|
||||||
|
{`echo "{a,b} c" {d,e}`, `echo "{a,b} c" d e`},
|
||||||
|
}
|
||||||
|
for _, tc := range tests {
|
||||||
|
got, err := expandBraceWords(tc.input)
|
||||||
|
if err != nil || got != tc.want {
|
||||||
|
t.Errorf("expandBraceWords(%q) = %q, %v; want %q", tc.input, got, err, tc.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, input := range []string{"echo a > {f,g}", "cat < x{1..2}", "echo a 2>> {a,b}"} {
|
||||||
|
if _, err := expandBraceWords(input); err == nil || !strings.Contains(err.Error(), "ambiguous redirect") {
|
||||||
|
t.Errorf("expandBraceWords(%q): %v, want ambiguous redirect", input, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestExecInput_BraceExpansion(t *testing.T) {
|
||||||
|
origDir, _ := os.Getwd()
|
||||||
|
defer os.Chdir(origDir)
|
||||||
|
dir := t.TempDir()
|
||||||
|
os.Chdir(dir)
|
||||||
|
|
||||||
|
if err := execInput("mkdir -p d/{x,y}/{1..2} && touch d/x/1/f{a,b}.txt"); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
for _, path := range []string{"d/x/1", "d/x/2", "d/y/1", "d/y/2", "d/x/1/fa.txt", "d/x/1/fb.txt"} {
|
||||||
|
if _, err := os.Stat(path); err != nil {
|
||||||
|
t.Errorf("%s was not created", path)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Brace expansion comes before wildcards and variables
|
||||||
|
t.Setenv("GOSH_BRACE", "v")
|
||||||
|
execInput("echo d/x/1/*.{txt,go} {${GOSH_BRACE},w}1 > out")
|
||||||
|
if got, _ := os.ReadFile("out"); string(got) != "d/x/1/fa.txt d/x/1/fb.txt d/x/1/*.go v1 w1\n" {
|
||||||
|
t.Errorf("out = %q", got)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := execInput("echo a > {f,g}"); err == nil || lastStatus != 1 {
|
||||||
|
t.Errorf("ambiguous redirect: %v, $? = %d", err, lastStatus)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -162,9 +162,7 @@ func (j *job) update(pid int, ws syscall.WaitStatus) {
|
|||||||
switch {
|
switch {
|
||||||
case ws.Stopped():
|
case ws.Stopped():
|
||||||
p.stopped = true
|
p.stopped = true
|
||||||
case ws.Continued():
|
case ws.Exited() || ws.Signaled():
|
||||||
p.stopped = false
|
|
||||||
default:
|
|
||||||
p.finished, p.stopped = true, false
|
p.finished, p.stopped = true, false
|
||||||
p.err = waitStatusError(ws)
|
p.err = waitStatusError(ws)
|
||||||
p.cmd.Process.Release()
|
p.cmd.Process.Release()
|
||||||
@@ -185,17 +183,30 @@ func waitStatusError(ws syscall.WaitStatus) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// wait waits until no process of the job is running any more (all finished
|
// waitFlags returns the options for wait4: WUNTRACED to see stopped
|
||||||
// or, with job control, stopped). With nohang it only collects the status
|
// processes (with job control) and WNOHANG to not block.
|
||||||
// changes that are already available and does not block.
|
//
|
||||||
func (j *job) wait(nohang bool) {
|
// WCONTINUED is not used: Go's syscall.WaitStatus decodes the "continued"
|
||||||
|
// status wrongly on FreeBSD (as killed by SIGCONT) and macOS (as stopped).
|
||||||
|
// The shell marks jobs as running itself when it continues them (fg, bg); a
|
||||||
|
// job continued from outside (kill -CONT) stays "Stopped" in the job table
|
||||||
|
// until it stops again or exits.
|
||||||
|
func waitFlags(j *job, nohang bool) int {
|
||||||
flags := 0
|
flags := 0
|
||||||
if jobControl && j.pgid != 0 {
|
if jobControl && j.pgid != 0 {
|
||||||
flags |= syscall.WUNTRACED | syscall.WCONTINUED
|
flags |= syscall.WUNTRACED
|
||||||
}
|
}
|
||||||
if nohang {
|
if nohang {
|
||||||
flags |= syscall.WNOHANG
|
flags |= syscall.WNOHANG
|
||||||
}
|
}
|
||||||
|
return flags
|
||||||
|
}
|
||||||
|
|
||||||
|
// wait waits until no process of the job is running any more (all finished
|
||||||
|
// or, with job control, stopped). With nohang it only collects the status
|
||||||
|
// changes that are already available and does not block.
|
||||||
|
func (j *job) wait(nohang bool) {
|
||||||
|
flags := waitFlags(j, nohang)
|
||||||
|
|
||||||
for {
|
for {
|
||||||
// With job control all processes of the job are in its process
|
// With job control all processes of the job are in its process
|
||||||
|
|||||||
@@ -105,6 +105,21 @@ func (s *ptyShell) expect(want string) {
|
|||||||
s.t.Fatalf("timeout waiting for %q, output:\n%s", want, s.out.String())
|
s.t.Fatalf("timeout waiting for %q, output:\n%s", want, s.out.String())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestWaitFlags_NoWCONTINUED(t *testing.T) {
|
||||||
|
defer func(saved bool) { jobControl = saved }(jobControl)
|
||||||
|
jobControl = true
|
||||||
|
j := &job{pgid: 1}
|
||||||
|
for _, nohang := range []bool{false, true} {
|
||||||
|
flags := waitFlags(j, nohang)
|
||||||
|
if flags&syscall.WCONTINUED != 0 {
|
||||||
|
t.Errorf("waitFlags(nohang=%v) contains WCONTINUED, which Go decodes wrongly on BSD", nohang)
|
||||||
|
}
|
||||||
|
if flags&syscall.WUNTRACED == 0 {
|
||||||
|
t.Errorf("waitFlags(nohang=%v) misses WUNTRACED", nohang)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestJobControl(t *testing.T) {
|
func TestJobControl(t *testing.T) {
|
||||||
s := startPtyShell(t)
|
s := startPtyShell(t)
|
||||||
|
|
||||||
@@ -120,6 +135,13 @@ func TestJobControl(t *testing.T) {
|
|||||||
s.send("bg\n")
|
s.send("bg\n")
|
||||||
s.expect("[1]+ Running sleep 30 &")
|
s.expect("[1]+ Running sleep 30 &")
|
||||||
|
|
||||||
|
// After bg the job keeps running and is not reported as finished
|
||||||
|
// (regression: on FreeBSD the "continued" status was read as killed by
|
||||||
|
// SIGCONT, "Exit 147")
|
||||||
|
time.Sleep(300 * time.Millisecond)
|
||||||
|
s.send("jobs\n")
|
||||||
|
s.expect("[1]+ Running sleep 30 &")
|
||||||
|
|
||||||
// fg brings it back, Ctrl+C reaches only the job
|
// fg brings it back, Ctrl+C reaches only the job
|
||||||
s.send("fg %1\n")
|
s.send("fg %1\n")
|
||||||
s.expect("sleep 30")
|
s.expect("sleep 30")
|
||||||
|
|||||||
@@ -612,7 +612,7 @@ func getPrompt() string {
|
|||||||
c := parseColors(os.Getenv("GOSH_COLORS"))
|
c := parseColors(os.Getenv("GOSH_COLORS"))
|
||||||
colors = &c
|
colors = &c
|
||||||
}
|
}
|
||||||
return buildPrompt(username, hostname, wd, os.Geteuid() == 0, colors)
|
return buildPrompt(username, hostname, wd, os.Geteuid() == 0, lastStatus, colors)
|
||||||
}
|
}
|
||||||
|
|
||||||
// promptColors holds the SGR parameters (e.g. "1;32") for the parts of the
|
// promptColors holds the SGR parameters (e.g. "1;32") for the parts of the
|
||||||
@@ -621,12 +621,14 @@ type promptColors struct {
|
|||||||
user string // user@host
|
user string // user@host
|
||||||
root string // user@host when running as root
|
root string // user@host when running as root
|
||||||
path string // working directory
|
path string // working directory
|
||||||
|
error string // exit status of the last command, if it failed
|
||||||
}
|
}
|
||||||
|
|
||||||
// defaultColors: bold green user, bold red root, bold blue directory
|
// defaultColors: bold green user, bold red root, bold blue directory, bold
|
||||||
var defaultColors = promptColors{user: "1;32", root: "1;31", path: "1;34"}
|
// red exit status
|
||||||
|
var defaultColors = promptColors{user: "1;32", root: "1;31", path: "1;34", error: "1;31"}
|
||||||
|
|
||||||
// parseColors reads a GOSH_COLORS value like "user=1;33:root=1;31:path=36"
|
// parseColors reads a GOSH_COLORS value like "user=1;33:path=36:error=31"
|
||||||
// and overrides the corresponding defaults. Unknown keys and values that are
|
// and overrides the corresponding defaults. Unknown keys and values that are
|
||||||
// not plain SGR parameters (digits and ';') are ignored, so the variable
|
// not plain SGR parameters (digits and ';') are ignored, so the variable
|
||||||
// cannot be used to write arbitrary escape sequences to the terminal.
|
// cannot be used to write arbitrary escape sequences to the terminal.
|
||||||
@@ -644,6 +646,8 @@ func parseColors(spec string) promptColors {
|
|||||||
colors.root = value
|
colors.root = value
|
||||||
case "path":
|
case "path":
|
||||||
colors.path = value
|
colors.path = value
|
||||||
|
case "error":
|
||||||
|
colors.error = value
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return colors
|
return colors
|
||||||
@@ -669,20 +673,28 @@ func colorize(text, sgr string) string {
|
|||||||
return "\x1b[" + sgr + "m" + text + "\x1b[0m"
|
return "\x1b[" + sgr + "m" + text + "\x1b[0m"
|
||||||
}
|
}
|
||||||
|
|
||||||
// buildPrompt formats the prompt as user@host:path $. The parts are sanitized
|
// buildPrompt formats the prompt as user@host:path $, or user@host:path [N] $
|
||||||
|
// if the last command failed with exit status N. The parts are sanitized
|
||||||
// before the color codes are added, so only our own escape sequences end up
|
// before the color codes are added, so only our own escape sequences end up
|
||||||
// in the prompt. colors == nil disables colors.
|
// in the prompt. colors == nil disables colors.
|
||||||
func buildPrompt(username, hostname, path string, root bool, colors *promptColors) string {
|
func buildPrompt(username, hostname, path string, root bool, status int, colors *promptColors) string {
|
||||||
userHost := sanitize(username + "@" + hostname)
|
userHost := sanitize(username + "@" + hostname)
|
||||||
path = sanitize(path)
|
path = sanitize(path)
|
||||||
|
var failed string
|
||||||
|
if status != 0 {
|
||||||
|
failed = " [" + strconv.Itoa(status) + "]"
|
||||||
|
}
|
||||||
if colors == nil {
|
if colors == nil {
|
||||||
return userHost + ":" + path + " $ "
|
return userHost + ":" + path + failed + " $ "
|
||||||
}
|
}
|
||||||
userColor := colors.user
|
userColor := colors.user
|
||||||
if root {
|
if root {
|
||||||
userColor = colors.root
|
userColor = colors.root
|
||||||
}
|
}
|
||||||
return colorize(userHost, userColor) + ":" + colorize(path, colors.path) + " $ "
|
if failed != "" {
|
||||||
|
failed = " " + colorize(failed[1:], colors.error)
|
||||||
|
}
|
||||||
|
return colorize(userHost, userColor) + ":" + colorize(path, colors.path) + failed + " $ "
|
||||||
}
|
}
|
||||||
|
|
||||||
// useColor reports whether the prompt should be colored: stdout must be a
|
// useColor reports whether the prompt should be colored: stdout must be a
|
||||||
@@ -702,7 +714,7 @@ type shellCompleter struct {
|
|||||||
|
|
||||||
func newShellCompleter() *shellCompleter {
|
func newShellCompleter() *shellCompleter {
|
||||||
return &shellCompleter{
|
return &shellCompleter{
|
||||||
commands: []string{"cd", "pwd", "echo", "exit", "quit", "set", "export", "source", "jobs", "fg", "bg", "wait", "help"},
|
commands: []string{"cd", "pwd", "echo", "exit", "quit", "set", "export", "source", "jobs", "fg", "bg", "wait", "alias", "unalias", "help"},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -726,7 +738,8 @@ func (c *shellCompleter) Do(line []rune, pos int) ([][]rune, int) {
|
|||||||
// Complete commands
|
// Complete commands
|
||||||
var candidates [][]rune
|
var candidates [][]rune
|
||||||
seen := make(map[string]bool)
|
seen := make(map[string]bool)
|
||||||
for _, cmd := range append(c.commands, pathExecutables(word)...) {
|
names := append(append(append([]string(nil), c.commands...), aliasNames()...), pathExecutables(word)...)
|
||||||
|
for _, cmd := range names {
|
||||||
if strings.HasPrefix(cmd, word) && !seen[cmd] {
|
if strings.HasPrefix(cmd, word) && !seen[cmd] {
|
||||||
seen[cmd] = true
|
seen[cmd] = true
|
||||||
candidates = append(candidates, []rune(escapeWord(cmd[len(word):])+" "))
|
candidates = append(candidates, []rune(escapeWord(cmd[len(word):])+" "))
|
||||||
@@ -814,7 +827,7 @@ func currentWord(s string) (word string, start int, quote byte) {
|
|||||||
func escapeWord(s string) string {
|
func escapeWord(s string) string {
|
||||||
var b strings.Builder
|
var b strings.Builder
|
||||||
for _, r := range s {
|
for _, r := range s {
|
||||||
if strings.ContainsRune(" \t\\'\";|&<>*?[$", r) {
|
if strings.ContainsRune(" \t\\'\";|&<>*?[${", r) {
|
||||||
b.WriteByte('\\')
|
b.WriteByte('\\')
|
||||||
}
|
}
|
||||||
b.WriteRune(r)
|
b.WriteRune(r)
|
||||||
@@ -951,6 +964,8 @@ flags:
|
|||||||
script = func() int { return runScript(os.Stdin, "") }
|
script = func() int { return runScript(os.Stdin, "") }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Aliases are expanded in interactive shells, also in their startup files
|
||||||
|
interactiveShell = interactiveMode
|
||||||
for _, file := range startupFiles(login && !noProfile, interactiveMode && !noRC) {
|
for _, file := range startupFiles(login && !noProfile, interactiveMode && !noRC) {
|
||||||
sourceStartupFile(file)
|
sourceStartupFile(file)
|
||||||
}
|
}
|
||||||
@@ -1327,6 +1342,9 @@ scan:
|
|||||||
// and ||. It returns the error of the last command that was run; errors of
|
// and ||. It returns the error of the last command that was run; errors of
|
||||||
// earlier commands are printed. $? is updated after each command.
|
// earlier commands are printed. $? is updated after each command.
|
||||||
func execInput(input string) error {
|
func execInput(input string) error {
|
||||||
|
if interactiveShell {
|
||||||
|
input = expandAliases(input)
|
||||||
|
}
|
||||||
items, err := splitList(input)
|
items, err := splitList(input)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
lastStatus = exitStatus(err)
|
lastStatus = exitStatus(err)
|
||||||
@@ -1348,6 +1366,12 @@ func execInput(input string) error {
|
|||||||
// execCommand executes a pipeline of simple commands with redirections, in
|
// execCommand executes a pipeline of simple commands with redirections, in
|
||||||
// the background if background is set.
|
// the background if background is set.
|
||||||
func execCommand(input string, background bool) error {
|
func execCommand(input string, background bool) error {
|
||||||
|
// Brace expansion comes first, so that its results are expanded further
|
||||||
|
input, err := expandBraceWords(input)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
// Split the input into words and operators, expanding variables and ~ on
|
// Split the input into words and operators, expanding variables and ~ on
|
||||||
// the way.
|
// the way.
|
||||||
tokens, err := tokenize(input)
|
tokens, err := tokenize(input)
|
||||||
@@ -1649,6 +1673,8 @@ var builtins = map[string]struct{ changesState bool }{
|
|||||||
"fg": {changesState: true},
|
"fg": {changesState: true},
|
||||||
"bg": {changesState: true},
|
"bg": {changesState: true},
|
||||||
"wait": {changesState: true},
|
"wait": {changesState: true},
|
||||||
|
"alias": {},
|
||||||
|
"unalias": {changesState: true},
|
||||||
}
|
}
|
||||||
|
|
||||||
// runBuiltin executes a built-in command, writing its output to stdout.
|
// runBuiltin executes a built-in command, writing its output to stdout.
|
||||||
@@ -1728,6 +1754,10 @@ func runBuiltin(args []string, stdout io.Writer) error {
|
|||||||
}
|
}
|
||||||
os.Setenv(parts[0], parts[1])
|
os.Setenv(parts[0], parts[1])
|
||||||
return nil
|
return nil
|
||||||
|
case "alias":
|
||||||
|
return builtinAlias(args, stdout)
|
||||||
|
case "unalias":
|
||||||
|
return builtinUnalias(args)
|
||||||
case "jobs":
|
case "jobs":
|
||||||
return builtinJobs(stdout)
|
return builtinJobs(stdout)
|
||||||
case "fg":
|
case "fg":
|
||||||
@@ -1758,6 +1788,8 @@ Built-in commands:
|
|||||||
fg [%n] - Continue job n (default: current job) in the foreground
|
fg [%n] - Continue job n (default: current job) in the foreground
|
||||||
bg [%n] - Continue stopped job n in the background
|
bg [%n] - Continue stopped job n in the background
|
||||||
wait [%n] - Wait for job n or all background jobs
|
wait [%n] - Wait for job n or all background jobs
|
||||||
|
alias [name[=value] ...] - Define or show aliases (alias ll='ls -l')
|
||||||
|
unalias name - Remove an alias (unalias -a: all aliases)
|
||||||
help - Show this help
|
help - Show this help
|
||||||
|
|
||||||
Features:
|
Features:
|
||||||
@@ -1769,13 +1801,17 @@ Features:
|
|||||||
override with $GOSH_HISTFILE); lines starting with a space are not saved
|
override with $GOSH_HISTFILE); lines starting with a space are not saved
|
||||||
- Tab completion for commands and file paths
|
- Tab completion for commands and file paths
|
||||||
- Tilde expansion (~, ~/path) at the start of unquoted words
|
- Tilde expansion (~, ~/path) at the start of unquoted words
|
||||||
|
- Brace expansion: a{b,c}d -> abd acd, {1..5}, {01..10..2}, {a..e}
|
||||||
- Wildcards: * ? [abc] [a-z] [!abc] match file names (not hidden files
|
- Wildcards: * ? [abc] [a-z] [!abc] match file names (not hidden files
|
||||||
unless the pattern starts with a dot); quoted wildcards and values of
|
unless the pattern starts with a dot); quoted wildcards and values of
|
||||||
variables are not expanded, a pattern without match is kept as is
|
variables are not expanded, a pattern without match is kept as is
|
||||||
- Colored prompt (user@host green, red for root; directory blue),
|
- Prompt user@host:dir $, with [N] before the $ if the last command
|
||||||
configurable with $GOSH_COLORS, e.g. GOSH_COLORS='user=1;33:path=36'
|
failed with exit status N
|
||||||
(keys user, root, path; values are ANSI SGR codes, empty = no color);
|
- Colored prompt (user@host green, red for root; directory blue; exit
|
||||||
disabled with $NO_COLOR, TERM=dumb or when output is not a terminal
|
status red), configurable with $GOSH_COLORS, e.g.
|
||||||
|
GOSH_COLORS='user=1;33:path=36' (keys user, root, path, error; values
|
||||||
|
are ANSI SGR codes, empty = no color); disabled with $NO_COLOR,
|
||||||
|
TERM=dumb or when output is not a terminal
|
||||||
- Environment variable expansion ($VAR, ${VAR}, ${VAR:-default});
|
- Environment variable expansion ($VAR, ${VAR}, ${VAR:-default});
|
||||||
using an unset variable is an error, use ${VAR:-} for an empty default
|
using an unset variable is an error, use ${VAR:-} for an empty default
|
||||||
- Command lists: a ; b (run both), a && b (b only if a succeeded),
|
- Command lists: a ; b (run both), a && b (b only if a succeeded),
|
||||||
@@ -1784,6 +1820,8 @@ Features:
|
|||||||
2>> file, 2>&1, >&2, &> file (stdout and stderr), &>> file
|
2>> file, 2>&1, >&2, &> file (stdout and stderr), &>> file
|
||||||
- Job control: command & runs in the background, Ctrl+Z stops the
|
- Job control: command & runs in the background, Ctrl+Z stops the
|
||||||
foreground job; jobs, fg, bg and wait manage the jobs
|
foreground job; jobs, fg, bg and wait manage the jobs
|
||||||
|
- Aliases replace the first word of a command in interactive shells
|
||||||
|
(define them in ~/.goshrc); \name or 'name' skips the alias
|
||||||
- Non-interactive use: gosh -c 'command', gosh script, command | gosh;
|
- Non-interactive use: gosh -c 'command', gosh script, command | gosh;
|
||||||
the exit status is the one of the last command
|
the exit status is the one of the last command
|
||||||
- Startup files: login shells (-l or argv[0] starting with -) read
|
- Startup files: login shells (-l or argv[0] starting with -) read
|
||||||
|
|||||||
+29
-9
@@ -1054,24 +1054,24 @@ func TestGetPrompt_EscapesControlCharacters(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestBuildPrompt_Color(t *testing.T) {
|
func TestBuildPrompt_Color(t *testing.T) {
|
||||||
got := buildPrompt("flow", "host", "~/src", false, &defaultColors)
|
got := buildPrompt("flow", "host", "~/src", false, 0, &defaultColors)
|
||||||
want := "\x1b[1;32mflow@host\x1b[0m:\x1b[1;34m~/src\x1b[0m $ "
|
want := "\x1b[1;32mflow@host\x1b[0m:\x1b[1;34m~/src\x1b[0m $ "
|
||||||
if got != want {
|
if got != want {
|
||||||
t.Errorf("buildPrompt() = %q, want %q", got, want)
|
t.Errorf("buildPrompt() = %q, want %q", got, want)
|
||||||
}
|
}
|
||||||
|
|
||||||
got = buildPrompt("root", "host", "/", true, &defaultColors)
|
got = buildPrompt("root", "host", "/", true, 0, &defaultColors)
|
||||||
if !strings.HasPrefix(got, "\x1b[1;31mroot@host") {
|
if !strings.HasPrefix(got, "\x1b[1;31mroot@host") {
|
||||||
t.Errorf("expected red user for root, got %q", got)
|
t.Errorf("expected red user for root, got %q", got)
|
||||||
}
|
}
|
||||||
|
|
||||||
got = buildPrompt("flow", "host", "~/src", false, nil)
|
got = buildPrompt("flow", "host", "~/src", false, 0, nil)
|
||||||
if got != "flow@host:~/src $ " {
|
if got != "flow@host:~/src $ " {
|
||||||
t.Errorf("buildPrompt() without color = %q", got)
|
t.Errorf("buildPrompt() without color = %q", got)
|
||||||
}
|
}
|
||||||
|
|
||||||
colors := promptColors{user: "38;5;208", path: ""}
|
colors := promptColors{user: "38;5;208", path: ""}
|
||||||
got = buildPrompt("flow", "host", "~/src", false, &colors)
|
got = buildPrompt("flow", "host", "~/src", false, 0, &colors)
|
||||||
if got != "\x1b[38;5;208mflow@host\x1b[0m:~/src $ " {
|
if got != "\x1b[38;5;208mflow@host\x1b[0m:~/src $ " {
|
||||||
t.Errorf("buildPrompt() with custom colors = %q", got)
|
t.Errorf("buildPrompt() with custom colors = %q", got)
|
||||||
}
|
}
|
||||||
@@ -1083,13 +1083,14 @@ func TestParseColors(t *testing.T) {
|
|||||||
want promptColors
|
want promptColors
|
||||||
}{
|
}{
|
||||||
{"", defaultColors},
|
{"", defaultColors},
|
||||||
{"user=1;33", promptColors{user: "1;33", root: "1;31", path: "1;34"}},
|
{"user=1;33", promptColors{user: "1;33", root: "1;31", path: "1;34", error: "1;31"}},
|
||||||
{"user=33:root=41;97:path=38;5;208", promptColors{user: "33", root: "41;97", path: "38;5;208"}},
|
{"user=33:root=41;97:path=38;5;208", promptColors{user: "33", root: "41;97", path: "38;5;208", error: "1;31"}},
|
||||||
{"path=", promptColors{user: "1;32", root: "1;31", path: ""}},
|
{"error=33", promptColors{user: "1;32", root: "1;31", path: "1;34", error: "33"}},
|
||||||
|
{"path=", promptColors{user: "1;32", root: "1;31", path: "", error: "1;31"}},
|
||||||
{"path=0m\x1b[2J", defaultColors}, // escape sequences are rejected
|
{"path=0m\x1b[2J", defaultColors}, // escape sequences are rejected
|
||||||
{"path=red", defaultColors}, // only numeric codes
|
{"path=red", defaultColors}, // only numeric codes
|
||||||
{"foo=1:user", defaultColors}, // unknown key, missing value
|
{"foo=1:user", defaultColors}, // unknown key, missing value
|
||||||
{"user=35::path=36", promptColors{user: "35", root: "1;31", path: "36"}},
|
{"user=35::path=36", promptColors{user: "35", root: "1;31", path: "36", error: "1;31"}},
|
||||||
}
|
}
|
||||||
for _, tc := range tests {
|
for _, tc := range tests {
|
||||||
if got := parseColors(tc.spec); got != tc.want {
|
if got := parseColors(tc.spec); got != tc.want {
|
||||||
@@ -1098,8 +1099,27 @@ func TestParseColors(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestBuildPrompt_Status(t *testing.T) {
|
||||||
|
if got := buildPrompt("flow", "host", "~", false, 0, nil); got != "flow@host:~ $ " {
|
||||||
|
t.Errorf("status 0: %q", got)
|
||||||
|
}
|
||||||
|
if got := buildPrompt("flow", "host", "~", false, 127, nil); got != "flow@host:~ [127] $ " {
|
||||||
|
t.Errorf("status 127 without color: %q", got)
|
||||||
|
}
|
||||||
|
got := buildPrompt("flow", "host", "~", false, 1, &defaultColors)
|
||||||
|
want := "\x1b[1;32mflow@host\x1b[0m:\x1b[1;34m~\x1b[0m \x1b[1;31m[1]\x1b[0m $ "
|
||||||
|
if got != want {
|
||||||
|
t.Errorf("status 1 with color = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
colors := defaultColors
|
||||||
|
colors.error = ""
|
||||||
|
if got := buildPrompt("flow", "host", "~", false, 2, &colors); !strings.HasSuffix(got, " [2] $ ") {
|
||||||
|
t.Errorf("status without error color = %q", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestBuildPrompt_ColorEscapesControlCharacters(t *testing.T) {
|
func TestBuildPrompt_ColorEscapesControlCharacters(t *testing.T) {
|
||||||
got := buildPrompt("flow", "host", "evil\x1b[2J\x07", false, &defaultColors)
|
got := buildPrompt("flow", "host", "evil\x1b[2J\x07", false, 0, &defaultColors)
|
||||||
if strings.Contains(got, "\x1b[2J") || strings.Contains(got, "\x07") {
|
if strings.Contains(got, "\x1b[2J") || strings.Contains(got, "\x07") {
|
||||||
t.Errorf("prompt contains control characters from path: %q", got)
|
t.Errorf("prompt contains control characters from path: %q", got)
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in new issue
Block a user