setup output

This commit is contained in:
2026-01-13 13:31:01 +00:00
parent 7344cfdfdf
commit d4dfa0ad4b

View File

@@ -21,6 +21,10 @@ const (
defaultServiceDirTemplate = "/root/pb/{service}"
defaultEnvFileTemplate = "/root/pb/{service}/.env"
totalSetupSteps = 5
remoteWindowSize = 10
remoteIndent = " "
remoteLineColor = "\033[96m"
remoteColorReset = "\033[0m"
)
type pbToml struct {
@@ -490,6 +494,7 @@ func filterEnvStream(r io.Reader, w io.Writer) error {
scanner := bufio.NewScanner(r)
scanner.Buffer(make([]byte, 0, 64*1024), 2*1024*1024)
skipping := true
window := newRemoteWindow(w)
for scanner.Scan() {
line := scanner.Text()
if skipping && isEnvLine(line) {
@@ -498,7 +503,7 @@ func filterEnvStream(r io.Reader, w io.Writer) error {
if skipping {
skipping = false
}
if _, err := fmt.Fprintln(w, line); err != nil {
if err := window.add(line); err != nil {
return err
}
}
@@ -508,6 +513,39 @@ func filterEnvStream(r io.Reader, w io.Writer) error {
return nil
}
type remoteWindow struct {
w io.Writer
lines []string
prevCount int
}
func newRemoteWindow(w io.Writer) *remoteWindow {
return &remoteWindow{w: w}
}
func (rw *remoteWindow) add(line string) error {
rw.lines = append(rw.lines, line)
if len(rw.lines) > remoteWindowSize {
rw.lines = rw.lines[1:]
}
return rw.flush()
}
func (rw *remoteWindow) flush() error {
if rw.prevCount > 0 {
if _, err := fmt.Fprintf(rw.w, "\033[%dA", rw.prevCount); err != nil {
return err
}
}
for _, line := range rw.lines {
if _, err := fmt.Fprintf(rw.w, "\033[2K\r%s%s%s%s\n", remoteLineColor, remoteIndent, line, remoteColorReset); err != nil {
return err
}
}
rw.prevCount = len(rw.lines)
return nil
}
func stripLeadingEnvLines(input string) string {
scanner := bufio.NewScanner(strings.NewReader(input))
scanner.Buffer(make([]byte, 0, 64*1024), 2*1024*1024)