whoami 0.4.1

Retrieve the current user and environment.
Documentation 
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

// Copyright © Jeron Lau 2017 - 2019.
// Dual-licensed under either the MIT License or the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at https://www.boost.org/LICENSE_1_0.txt)

use super::DesktopEnv;

pub fn username() -> String {
    extern "system" {
        fn GetUserNameW(a: *mut u16, b: *mut usize) -> i32;
    }

    let mut name = [0; 256];
    let mut size = 256;

    unsafe {
        GetUserNameW(&mut name[0], &mut size);
    }

    String::from_utf16_lossy(if size == 0 { &[] } else { &name[..size - 1] })
}

pub fn realname() -> String {
    username()
}

pub fn computer() -> String {
    hostname()
}

pub fn hostname() -> String {
    extern "system" {
        fn GetComputerNameW(a: *mut u16, b: *mut usize) -> i32;
    }

    let mut name = [0; 256];
    let mut size = 256;

    unsafe {
        GetComputerNameW(&mut name[0], &mut size);
    }

    String::from_utf16_lossy(&name[..size])
}

pub fn os() -> String {
    extern "system" {
        fn GetVersion() -> usize;
    }

    let bits = unsafe { GetVersion() } as u32;

    let mut out = "Windows ".to_string();

    let major: u8 = ((bits & 0b00000000_00000000_00000000_11111111) >> 0) as u8;
    let minor: u8 = ((bits & 0b00000000_00000000_11111111_00000000) >> 8) as u8;
    let build: u16 = ((bits & 0b11111111_11111111_00000000_00000000) >> 16) as u16;

    match major {
        5 => out.push_str("XP"),
        6 => match minor {
            0 => out.push_str("Vista"),
            1 => out.push_str("7"),
            2 => match build {
                9200 => out.push_str("10"),
                _ => out.push_str("8"),
            },
            _ => out.push_str("8"),
        },
        _ => out.push_str("Unknown"),
    }

    out
}

#[inline(always)]
pub fn env() -> DesktopEnv {
    DesktopEnv::Windows
}