client_signals.go RawHistoryFolder

1package app
2
3import (
4	"context"
5	"os"
6	"os/signal"
7	"syscall"
8
9	"github.com/bnema/vev/internal/usecase/client"
10)
11
12// clientSignalContext cancels an attached client on process signals. SIGHUP
13// (terminal closed) and SIGTERM use client.ErrClientClosed as the cause so the
14// attachment reports Detach{Closed} and the daemon applies
15// ephemeral.close-on-exit. SIGINT keeps a plain cancellation.
16func clientSignalContext(parent context.Context) (context.Context, context.CancelFunc) {
17	ctx, cancel := context.WithCancelCause(parent)
18	signals := make(chan os.Signal, 1)
19	signal.Notify(signals, syscall.SIGHUP, syscall.SIGTERM, syscall.SIGINT)
20	done := make(chan struct{})
21	go func() {
22		select {
23		case sig := <-signals:
24			if sig == syscall.SIGINT {
25				cancel(context.Canceled)
26				return
27			}
28			cancel(client.ErrClientClosed)
29		case <-done:
30		}
31	}()
32	return ctx, func() {
33		signal.Stop(signals)
34		close(done)
35		cancel(context.Canceled)
36	}
37}