151 lines
2.7 KiB
Go
151 lines
2.7 KiB
Go
package commands
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"sort"
|
|
"strings"
|
|
"time"
|
|
|
|
"git.sr.ht/~sircmpwn/aerc/widgets"
|
|
"github.com/gdamore/tcell"
|
|
"github.com/mitchellh/go-homedir"
|
|
)
|
|
|
|
// QuickTerm is an ephemeral terminal for running a single command and quiting.
|
|
func QuickTerm(aerc *widgets.Aerc, args []string, stdin io.Reader) (*widgets.Terminal, error) {
|
|
cmd := exec.Command(args[0], args[1:]...)
|
|
pipe, err := cmd.StdinPipe()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
term, err := widgets.NewTerminal(cmd)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
term.OnClose = func(err error) {
|
|
if err != nil {
|
|
aerc.PushError(" " + err.Error())
|
|
// remove the tab on error, otherwise it gets stuck
|
|
aerc.RemoveTab(term)
|
|
} else {
|
|
aerc.PushStatus("Process complete, press any key to close.",
|
|
10*time.Second)
|
|
term.OnEvent = func(event tcell.Event) bool {
|
|
aerc.RemoveTab(term)
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
|
|
term.OnStart = func() {
|
|
status := make(chan error, 1)
|
|
|
|
go func() {
|
|
_, err := io.Copy(pipe, stdin)
|
|
defer pipe.Close()
|
|
status <- err
|
|
}()
|
|
|
|
err := <-status
|
|
if err != nil {
|
|
aerc.PushError(" " + err.Error())
|
|
}
|
|
}
|
|
|
|
return term, nil
|
|
}
|
|
|
|
// CompletePath provides filesystem completions given a starting path.
|
|
func CompletePath(path string) []string {
|
|
if path == "" {
|
|
// default to cwd
|
|
cwd, err := os.Getwd()
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
path = cwd
|
|
}
|
|
|
|
path, err := homedir.Expand(path)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
// strip trailing slashes, etc.
|
|
path = filepath.Clean(path)
|
|
|
|
if _, err := os.Stat(path); os.IsNotExist(err) {
|
|
// if the path doesn't exist, it is likely due to it being a partial path
|
|
// in this case, we want to return possible matches (ie /hom* should match
|
|
// /home)
|
|
matches, err := filepath.Glob(fmt.Sprintf("%s*", path))
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
for i, m := range matches {
|
|
if isDir(m) {
|
|
matches[i] = m + "/"
|
|
}
|
|
}
|
|
|
|
sort.Strings(matches)
|
|
return matches
|
|
}
|
|
|
|
files := listDir(path, false)
|
|
|
|
for i, f := range files {
|
|
f = filepath.Join(path, f)
|
|
if isDir(f) {
|
|
f += "/"
|
|
}
|
|
|
|
files[i] = f
|
|
}
|
|
|
|
sort.Strings(files)
|
|
return files
|
|
}
|
|
|
|
func isDir(path string) bool {
|
|
info, err := os.Stat(path)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
return info.IsDir()
|
|
}
|
|
|
|
// return all filenames in a directory, optionally including hidden files
|
|
func listDir(path string, hidden bool) []string {
|
|
f, err := os.Open(path)
|
|
if err != nil {
|
|
return []string{}
|
|
}
|
|
|
|
files, err := f.Readdirnames(-1) // read all dir names
|
|
if err != nil {
|
|
return []string{}
|
|
}
|
|
|
|
if hidden {
|
|
return files
|
|
}
|
|
|
|
var filtered []string
|
|
for _, g := range files {
|
|
if !strings.HasPrefix(g, ".") {
|
|
filtered = append(filtered, g)
|
|
}
|
|
}
|
|
|
|
return filtered
|
|
}
|