Proxy exec.Cmd Stdout/Stderr sans perdre ATS

Question

J'ai le code suivant qui exécute une commande shell arbitraire et canalise la stdout et stderr au terminal .:

c := exec.Command("/bin/sh", "-c", cmd) 
c.Stdin = os.Stdin 
c.Stdout = os.Stdout 
c.Stderr = os.Stderr 

Cependant, je dois traiter la sortie avant de l'imprimer, donc je l'ai enveloppé avec un proxy Interface io.Writer:

type ProxyWriter struct { 
    file *os.File 
} 

func NewProxyWriter(file *os.File) *ProxyWriter { 
    return &ProxyWriter{ 
     file: file, 
    } 
} 

func (w *ProxyWriter) Write(p []byte) (int, error) { 
    // ... do something with bytes first 
    fmt.Fprintf(w.file, "%s", string(p)) 
    return len(p), nil 
} 

Ainsi, le code d'origine est maintenant:

c := exec.Command("/bin/sh", "-c", cmd) 
c.Stdin = os.Stdin 
c.Stdout = NewProxyWriter(os.Stdout) 
c.Stderr = NewProxyWriter(os.Stderr) 

Cela fonctionne pour la plupart, cependant, stdout et stderr ne semblent plus être admissibles comme ATS. Toute sortie de style ou de couleur n'est plus stylisée ou colorée.

J'ai confirmé que ce n'est pas un problème avec mon ProxyWriter simplement jouer avec le formatage en définissant la commande à la suivante, qui affiche correctement le texte coloré.

c := exec.Command("echo", "\033[0;31mTEST\033[0m") 

Un test plus explicite:

c := exec.Command("/bin/sh", "-c", "if [ -t 1 ] ; then echo \"terminal\"; else echo \"not a terminal\"; fi") 

qui délivre en sortie:

not a terminal 

Y at-il de toute façon je peux envelopper les commandes stdout/stderr sans perdre le statut ATS?

Answer 1

Remplacer

func (w *ProxyWriter) Write(p []byte) (int, error) { 
    // ... do something with bytes first 
    fmt.Fprintf(w.file, "%s", string(p)) 
    return len(p), nil 
} 

Pour

func (w *ProxyWriter) Write(p []byte) (int, error) { 
    return w.Write(p) 
} 

fmt.Fprintf ont une certaine logique pour éviter de casser la borne.