#!/usr/bin/perl
#
# Copyright 2014 Pierre Mavro <deimos@deimos.fr>
# Copyright 2014 Vivien Didelot <vivien@didelot.org>
# Copyright 2014 Andreas Guldstrand <andreas.guldstrand@gmail.com>
#
# Licensed under the terms of the GNU GPL v3, or any later version.

use strict;
use warnings;
use utf8;
use Getopt::Long;

# default values
my $t_warn = 70;
my $t_crit = 90;
my $cpu_usage = -1;

sub help {
    print "Usage: cpu_usage [-w <warning>] [-c <critical>]\n";
    print "-w <percent>: warning threshold to become yellow\n";
    print "-c <percent>: critical threshold to become red\n";
    exit 0;
}

GetOptions("help|h" => \&help,
           "w=i"    => \$t_warn,
           "c=i"    => \$t_crit);

# Get CPU usage
$ENV{LC_ALL}="en_US"; # if mpstat is not run under en_US locale, things may break, so make sure it is
open (MPSTAT, 'mpstat 1 1 |') or die;
while (<MPSTAT>) {
    if (/^.*\s+(\d+\.\d+)\s+$/) {
        $cpu_usage = 100 - $1; # 100% - %idle
        last;
    }
}
close(MPSTAT);


$cpu_usage eq -1 and die 'Can\'t find CPU information';
# Print short_text, full_text
#printf "%5.2f%%\n", $cpu_usage;
#printf "%2d%%\n", $cpu_usage;
my $color = "";

# Print color, if needed
if ($cpu_usage >= 80) {
	$color = "#f2777a";
    #exit 33;
} elsif ($cpu_usage >= 60) {
	$color = "#f99157";
} elsif ($cpu_usage >= 40) {
	$color = "#ffcc66";
} elsif ($cpu_usage >= 20) {
	$color = "#99cc99";
} else {
	$color = "#66cccc";
}

my $pave = " ";

if ($cpu_usage <= 11) {
	$pave = " ";
} elsif ($cpu_usage <= 22 ){
	$pave = "▁";
} elsif ($cpu_usage <= 33){
	$pave = "▂";
} elsif ($cpu_usage <= 44){
	$pave = "▃";
} elsif ($cpu_usage <= 56){
	$pave = "▄";
} elsif ($cpu_usage <= 67){
	$pave = "▅";
} elsif ($cpu_usage <= 78){
	$pave = "▆";
} elsif ($cpu_usage <= 89){
	$pave = "▇";
} else {
	$pave = "█";
}

printf "<span foreground=\"$color\" background=\"#515151\">$pave</span>";

exit 0;