2014-04-08 10:18:10 -05:00
|
|
|
// Copyright 2013-2014 The Rust Project Developers. See the COPYRIGHT
|
|
|
|
// file at the top-level directory of this distribution and at
|
|
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
|
|
// option. This file may not be copied, modified, or distributed
|
|
|
|
// except according to those terms.
|
|
|
|
|
|
|
|
//! Windows console handling
|
|
|
|
|
2014-08-23 03:12:59 -05:00
|
|
|
// FIXME (#13400): this is only a tiny fraction of the Windows console api
|
2014-04-08 10:18:10 -05:00
|
|
|
|
|
|
|
extern crate libc;
|
|
|
|
|
|
|
|
use std::io::IoResult;
|
|
|
|
|
|
|
|
use attr;
|
|
|
|
use color;
|
|
|
|
use Terminal;
|
|
|
|
|
2014-04-14 09:20:24 -05:00
|
|
|
/// A Terminal implementation which uses the Win32 Console API.
|
2014-04-08 10:18:10 -05:00
|
|
|
pub struct WinConsole<T> {
|
|
|
|
buf: T,
|
2014-09-20 17:22:11 -05:00
|
|
|
def_foreground: color::Color,
|
|
|
|
def_background: color::Color,
|
2014-04-08 10:18:10 -05:00
|
|
|
foreground: color::Color,
|
|
|
|
background: color::Color,
|
|
|
|
}
|
|
|
|
|
2014-09-20 17:22:11 -05:00
|
|
|
#[allow(non_snake_case)]
|
|
|
|
#[repr(C)]
|
|
|
|
struct CONSOLE_SCREEN_BUFFER_INFO {
|
|
|
|
dwSize: [libc::c_short, ..2],
|
|
|
|
dwCursorPosition: [libc::c_short, ..2],
|
|
|
|
wAttributes: libc::WORD,
|
|
|
|
srWindow: [libc::c_short, ..4],
|
|
|
|
dwMaximumWindowSize: [libc::c_short, ..2],
|
|
|
|
}
|
|
|
|
|
2014-07-18 07:45:17 -05:00
|
|
|
#[allow(non_snake_case)]
|
2014-04-08 10:18:10 -05:00
|
|
|
#[link(name = "kernel32")]
|
|
|
|
extern "system" {
|
|
|
|
fn SetConsoleTextAttribute(handle: libc::HANDLE, attr: libc::WORD) -> libc::BOOL;
|
|
|
|
fn GetStdHandle(which: libc::DWORD) -> libc::HANDLE;
|
2014-09-20 17:22:11 -05:00
|
|
|
fn GetConsoleScreenBufferInfo(handle: libc::HANDLE,
|
|
|
|
info: *mut CONSOLE_SCREEN_BUFFER_INFO) -> libc::BOOL;
|
2014-04-08 10:18:10 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
fn color_to_bits(color: color::Color) -> u16 {
|
|
|
|
// magic numbers from mingw-w64's wincon.h
|
|
|
|
|
|
|
|
let bits = match color % 8 {
|
|
|
|
color::BLACK => 0,
|
|
|
|
color::BLUE => 0x1,
|
|
|
|
color::GREEN => 0x2,
|
|
|
|
color::RED => 0x4,
|
|
|
|
color::YELLOW => 0x2 | 0x4,
|
|
|
|
color::MAGENTA => 0x1 | 0x4,
|
|
|
|
color::CYAN => 0x1 | 0x2,
|
|
|
|
color::WHITE => 0x1 | 0x2 | 0x4,
|
|
|
|
_ => unreachable!()
|
|
|
|
};
|
|
|
|
|
|
|
|
if color >= 8 {
|
|
|
|
bits | 0x8
|
|
|
|
} else {
|
|
|
|
bits
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-09-20 17:22:11 -05:00
|
|
|
fn bits_to_color(bits: u16) -> color::Color {
|
|
|
|
let color = match bits & 0x7 {
|
|
|
|
0 => color::BLACK,
|
|
|
|
0x1 => color::BLUE,
|
|
|
|
0x2 => color::GREEN,
|
|
|
|
0x4 => color::RED,
|
|
|
|
0x6 => color::YELLOW,
|
|
|
|
0x5 => color::MAGENTA,
|
|
|
|
0x3 => color::CYAN,
|
|
|
|
0x7 => color::WHITE,
|
|
|
|
_ => unreachable!()
|
|
|
|
};
|
|
|
|
|
|
|
|
if bits >= 8 {
|
|
|
|
color | 0x8
|
|
|
|
} else {
|
|
|
|
color
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-04-08 10:18:10 -05:00
|
|
|
impl<T: Writer> WinConsole<T> {
|
|
|
|
fn apply(&mut self) {
|
2014-05-11 19:08:49 -05:00
|
|
|
let _unused = self.buf.flush();
|
2014-04-08 10:18:10 -05:00
|
|
|
let mut accum: libc::WORD = 0;
|
|
|
|
accum |= color_to_bits(self.foreground);
|
|
|
|
accum |= color_to_bits(self.background) << 4;
|
|
|
|
|
|
|
|
unsafe {
|
|
|
|
// Magic -11 means stdout, from
|
|
|
|
// http://msdn.microsoft.com/en-us/library/windows/desktop/ms683231%28v=vs.85%29.aspx
|
|
|
|
//
|
|
|
|
// You may be wondering, "but what about stderr?", and the answer
|
|
|
|
// to that is that setting terminal attributes on the stdout
|
|
|
|
// handle also sets them for stderr, since they go to the same
|
|
|
|
// terminal! Admittedly, this is fragile, since stderr could be
|
|
|
|
// redirected to a different console. This is good enough for
|
|
|
|
// rustc though. See #13400.
|
|
|
|
let out = GetStdHandle(-11);
|
|
|
|
SetConsoleTextAttribute(out, accum);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<T: Writer> Writer for WinConsole<T> {
|
|
|
|
fn write(&mut self, buf: &[u8]) -> IoResult<()> {
|
|
|
|
self.buf.write(buf)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn flush(&mut self) -> IoResult<()> {
|
|
|
|
self.buf.flush()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<T: Writer> Terminal<T> for WinConsole<T> {
|
|
|
|
fn new(out: T) -> Option<WinConsole<T>> {
|
2014-09-20 17:22:11 -05:00
|
|
|
let fg;
|
|
|
|
let bg;
|
|
|
|
unsafe {
|
|
|
|
let mut buffer_info = ::std::mem::uninitialized();
|
|
|
|
if GetConsoleScreenBufferInfo(GetStdHandle(-11), &mut buffer_info) != 0 {
|
|
|
|
fg = bits_to_color(buffer_info.wAttributes);
|
|
|
|
bg = bits_to_color(buffer_info.wAttributes >> 4);
|
|
|
|
} else {
|
|
|
|
fg = color::WHITE;
|
|
|
|
bg = color::BLACK;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Some(WinConsole { buf: out,
|
|
|
|
def_foreground: fg, def_background: bg,
|
|
|
|
foreground: fg, background: bg } )
|
2014-04-08 10:18:10 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
fn fg(&mut self, color: color::Color) -> IoResult<bool> {
|
|
|
|
self.foreground = color;
|
|
|
|
self.apply();
|
|
|
|
|
|
|
|
Ok(true)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn bg(&mut self, color: color::Color) -> IoResult<bool> {
|
|
|
|
self.background = color;
|
|
|
|
self.apply();
|
|
|
|
|
|
|
|
Ok(true)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn attr(&mut self, attr: attr::Attr) -> IoResult<bool> {
|
|
|
|
match attr {
|
|
|
|
attr::ForegroundColor(f) => {
|
|
|
|
self.foreground = f;
|
|
|
|
self.apply();
|
|
|
|
Ok(true)
|
|
|
|
},
|
|
|
|
attr::BackgroundColor(b) => {
|
|
|
|
self.background = b;
|
|
|
|
self.apply();
|
|
|
|
Ok(true)
|
|
|
|
},
|
|
|
|
_ => Ok(false)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn supports_attr(&self, attr: attr::Attr) -> bool {
|
|
|
|
// it claims support for underscore and reverse video, but I can't get
|
|
|
|
// it to do anything -cmr
|
|
|
|
match attr {
|
|
|
|
attr::ForegroundColor(_) | attr::BackgroundColor(_) => true,
|
|
|
|
_ => false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn reset(&mut self) -> IoResult<()> {
|
2014-09-20 17:22:11 -05:00
|
|
|
self.foreground = self.def_foreground;
|
|
|
|
self.background = self.def_background;
|
2014-04-08 10:18:10 -05:00
|
|
|
self.apply();
|
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn unwrap(self) -> T { self.buf }
|
|
|
|
|
|
|
|
fn get_ref<'a>(&'a self) -> &'a T { &self.buf }
|
|
|
|
|
|
|
|
fn get_mut<'a>(&'a mut self) -> &'a mut T { &mut self.buf }
|
|
|
|
}
|