File: setup_windows.go

package info (click to toggle)
elvish 0.21.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 6,372 kB
  • sloc: javascript: 236; sh: 130; python: 104; makefile: 88; xml: 9
file content (61 lines) | stat: -rw-r--r-- 1,563 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
package term

import (
	"os"

	"golang.org/x/sys/windows"
	"src.elv.sh/pkg/errutil"
)

const (
	inMode = windows.ENABLE_WINDOW_INPUT |
		windows.ENABLE_MOUSE_INPUT | windows.ENABLE_PROCESSED_INPUT
	outMode = windows.ENABLE_PROCESSED_OUTPUT |
		windows.ENABLE_WRAP_AT_EOL_OUTPUT |
		windows.ENABLE_VIRTUAL_TERMINAL_PROCESSING
)

func setupForTUIOnce(_, _ *os.File) func() {
	return func() {}
}

func setupForTUI(in, out *os.File) (func() error, error) {
	hIn := windows.Handle(in.Fd())
	hOut := windows.Handle(out.Fd())

	var oldInMode, oldOutMode uint32
	err := windows.GetConsoleMode(hIn, &oldInMode)
	if err != nil {
		return nil, err
	}
	err = windows.GetConsoleMode(hOut, &oldOutMode)
	if err != nil {
		return nil, err
	}

	errSetIn := windows.SetConsoleMode(hIn, inMode)
	errSetOut := windows.SetConsoleMode(hOut, outMode)
	errVT := setupVT(out)

	return func() error {
		return errutil.Multi(
			restoreVT(out),
			windows.SetConsoleMode(hOut, oldOutMode),
			windows.SetConsoleMode(hIn, oldInMode))
	}, errutil.Multi(errSetIn, errSetOut, errVT)
}

// We need ENABLE_VIRTUAL_TERMINAL_PROCESSING for styled text to function. This
// includes texts created by the user (with the "styled" builtin) or by Elvish
// itself (like exception stack traces).
const outFlagForEval = windows.ENABLE_VIRTUAL_TERMINAL_PROCESSING

func setupForEval(_, out *os.File) func() {
	h := windows.Handle(out.Fd())
	var oldOutMode uint32
	err := windows.GetConsoleMode(h, &oldOutMode)
	if err == nil {
		windows.SetConsoleMode(h, oldOutMode|outFlagForEval)
	}
	return func() {}
}