github.com/kaydxh/golang@v0.0.131/pkg/logs/logrus/terminal_check_windows.go (about)

     1  //go:build !appengine && !js && windows
     2  // +build !appengine,!js,windows
     3  
     4  /*
     5   *Copyright (c) 2022, kaydxh
     6   *
     7   *Permission is hereby granted, free of charge, to any person obtaining a copy
     8   *of this software and associated documentation files (the "Software"), to deal
     9   *in the Software without restriction, including without limitation the rights
    10   *to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
    11   *copies of the Software, and to permit persons to whom the Software is
    12   *furnished to do so, subject to the following conditions:
    13   *
    14   *The above copyright notice and this permission notice shall be included in all
    15   *copies or substantial portions of the Software.
    16   *
    17   *THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
    18   *IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
    19   *FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
    20   *AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
    21   *LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
    22   *OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
    23   *SOFTWARE.
    24   */
    25  
    26  package logrus
    27  
    28  import (
    29  	"io"
    30  	"os"
    31  
    32  	"golang.org/x/sys/windows"
    33  )
    34  
    35  func checkIfTerminal(w io.Writer) bool {
    36  	switch v := w.(type) {
    37  	case *os.File:
    38  		handle := windows.Handle(v.Fd())
    39  		var mode uint32
    40  		if err := windows.GetConsoleMode(handle, &mode); err != nil {
    41  			return false
    42  		}
    43  		mode |= windows.ENABLE_VIRTUAL_TERMINAL_PROCESSING
    44  		if err := windows.SetConsoleMode(handle, mode); err != nil {
    45  			return false
    46  		}
    47  		return true
    48  	}
    49  	return false
    50  }