77 lines
1.4 KiB
Go
77 lines
1.4 KiB
Go
package vvin
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/shu-go/gli"
|
|
)
|
|
|
|
type waitCmd struct {
|
|
_ struct{} `help:"[--close] {Title}"`
|
|
|
|
Closed bool `help:"wait until the window is closed"`
|
|
Interval gli.Duration `cli:"interval,i=DURATION" default:"1s"`
|
|
Timeout gli.Duration `cli:"timeout=DURATION" default:"0s" help:"zelo value means ininite"`
|
|
}
|
|
|
|
func (c waitCmd) Run(args []string) error {
|
|
if len(args) != 1 {
|
|
return errors.New("not one target")
|
|
}
|
|
|
|
an := ancestors()
|
|
t := strings.ToLower(args[0])
|
|
|
|
var ctx context.Context
|
|
if c.Timeout == 0 {
|
|
ctx = context.Background()
|
|
} else {
|
|
ctx, _ = context.WithTimeout(context.Background(), c.Timeout.Duration())
|
|
}
|
|
|
|
signalChan := make(chan os.Signal, 1)
|
|
signal.Notify(signalChan, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT, os.Interrupt)
|
|
|
|
fmt.Println("Press Ctrl+C to cancel.")
|
|
|
|
waitLoop:
|
|
for {
|
|
wins, err := listAllWindows()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
win := findFirstTarget(t, wins, an)
|
|
if c.Closed {
|
|
if win == nil {
|
|
break
|
|
}
|
|
} else {
|
|
if win != nil {
|
|
break
|
|
}
|
|
}
|
|
|
|
select {
|
|
case <-signalChan:
|
|
fmt.Fprintln(os.Stderr, "cancelled")
|
|
break waitLoop
|
|
case <-ctx.Done():
|
|
fmt.Fprintln(os.Stderr, "cancelled")
|
|
break waitLoop
|
|
default:
|
|
//nop
|
|
}
|
|
|
|
time.Sleep(c.Interval.Duration())
|
|
}
|
|
|
|
return nil
|
|
}
|