You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
1.9 KiB
82 lines
1.9 KiB
// Based on ssh/terminal: |
|
// Copyright 2011 The Go Authors. All rights reserved. |
|
// Use of this source code is governed by a BSD-style |
|
// license that can be found in the LICENSE file. |
|
|
|
// +build windows,!appengine |
|
|
|
package logrus |
|
|
|
import ( |
|
"bytes" |
|
"errors" |
|
"io" |
|
"os" |
|
"os/exec" |
|
"strconv" |
|
"strings" |
|
"syscall" |
|
"unsafe" |
|
) |
|
|
|
var kernel32 = syscall.NewLazyDLL("kernel32.dll") |
|
|
|
var ( |
|
procGetConsoleMode = kernel32.NewProc("GetConsoleMode") |
|
procSetConsoleMode = kernel32.NewProc("SetConsoleMode") |
|
) |
|
|
|
const ( |
|
enableProcessedOutput = 0x0001 |
|
enableWrapAtEolOutput = 0x0002 |
|
enableVirtualTerminalProcessing = 0x0004 |
|
) |
|
|
|
func getVersion() (float64, error) { |
|
stdout, stderr := &bytes.Buffer{}, &bytes.Buffer{} |
|
cmd := exec.Command("cmd", "ver") |
|
cmd.Stdout = stdout |
|
cmd.Stderr = stderr |
|
err := cmd.Run() |
|
if err != nil { |
|
return -1, err |
|
} |
|
|
|
// The output should be like "Microsoft Windows [Version XX.X.XXXXXX]" |
|
version := strings.Replace(stdout.String(), "\n", "", -1) |
|
version = strings.Replace(version, "\r\n", "", -1) |
|
|
|
x1 := strings.Index(version, "[Version") |
|
|
|
if x1 == -1 || strings.Index(version, "]") == -1 { |
|
return -1, errors.New("Can't determine Windows version") |
|
} |
|
|
|
return strconv.ParseFloat(version[x1+9:x1+13], 64) |
|
} |
|
|
|
func init() { |
|
ver, err := getVersion() |
|
if err != nil { |
|
return |
|
} |
|
|
|
// Activate Virtual Processing for Windows CMD |
|
// Info: https://msdn.microsoft.com/en-us/library/windows/desktop/ms686033(v=vs.85).aspx |
|
if ver >= 10 { |
|
handle := syscall.Handle(os.Stderr.Fd()) |
|
procSetConsoleMode.Call(uintptr(handle), enableProcessedOutput|enableWrapAtEolOutput|enableVirtualTerminalProcessing) |
|
} |
|
} |
|
|
|
// IsTerminal returns true if stderr's file descriptor is a terminal. |
|
func IsTerminal(f io.Writer) bool { |
|
switch v := f.(type) { |
|
case *os.File: |
|
var st uint32 |
|
r, _, e := syscall.Syscall(procGetConsoleMode.Addr(), 2, uintptr(v.Fd()), uintptr(unsafe.Pointer(&st)), 0) |
|
return r != 0 && e == 0 |
|
default: |
|
return false |
|
} |
|
}
|
|
|