mirror of
https://github.com/aykhans/sarin.git
synced 2026-08-29 03:24:33 +00:00
aykhans
d08b6dc222
Import paths moved to charm.land/*/v2 and the v1 paths no longer resolve. Ports View() -> tea.View, AltScreen/MouseMode to View fields, KeyPressMsg, and the new progress/viewport constructors. Styled output outside Bubble Tea switched to lipgloss.Print* because v2 downsamples colour at the output layer. vendorHash refreshed.
56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package sarin
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"sync"
|
|
"sync/atomic"
|
|
|
|
tea "charm.land/bubbletea/v2"
|
|
)
|
|
|
|
const forceExitCode = 130
|
|
|
|
// StopController coordinates a two-stage shutdown.
|
|
//
|
|
// The first Stop call cancels the supplied context so workers and the job
|
|
// loop can drain. The second Stop call restores the terminal (if a bubbletea
|
|
// program has been attached) and calls os.Exit(forceExitCode), bypassing any
|
|
// in-flight captcha polls, Lua/JS scripts, or HTTP requests that would
|
|
// otherwise keep the process alive.
|
|
type StopController struct {
|
|
count atomic.Int32
|
|
cancel func()
|
|
mu sync.Mutex
|
|
program *tea.Program
|
|
}
|
|
|
|
func NewStopController(cancel func()) *StopController {
|
|
return &StopController{cancel: cancel}
|
|
}
|
|
|
|
// AttachProgram registers the active bubbletea program so the terminal state
|
|
// can be restored before os.Exit on the forced shutdown path. Pass nil to
|
|
// detach once the program has finished.
|
|
func (s *StopController) AttachProgram(program *tea.Program) {
|
|
s.mu.Lock()
|
|
s.program = program
|
|
s.mu.Unlock()
|
|
}
|
|
|
|
func (s *StopController) Stop() {
|
|
switch s.count.Add(1) {
|
|
case 1:
|
|
s.cancel()
|
|
case 2:
|
|
s.mu.Lock()
|
|
p := s.program
|
|
s.mu.Unlock()
|
|
if p != nil {
|
|
_ = p.ReleaseTerminal()
|
|
}
|
|
fmt.Fprintln(os.Stderr, "killing...")
|
|
os.Exit(forceExitCode)
|
|
}
|
|
}
|