scripts/trunk/dwm-status.pl

#!/usr/bin/perl -w

# dwm-status.pl
#
# 05/26/07 23:08:31 CEST Dobrica Pavlinusic <dpavlin@rot13.org>

use strict;
use POSIX qw/strftime/;
use File::Slurp;
use Time::HiRes;
use Data::Dump qw/dump/;

my $dt = 3;
my $acpi_every = 10;

my $disk_blk_size = 512;

my $debug = shift @ARGV;
my $awesome = 0;
my $dzen = 1;

$|=1;

sub proc2hash {
        my $f = shift;
        return unless -f $f;
        open(my $fh, '<', $f) || die "can't open $f: $!";
        my $h;
        while(<$fh>) {
                chomp;
                my ( $key, $value ) = split(/:\s+/, $_, 2);
                $value =~ s/ m[AVW]h*$//;
                $h->{$key} = $value;
        }
        warn "$f ",dump( $h ) if ( $debug );
        return $h;
}

my $proc_acpi_battery;
sub find_proc_acpi {
        my ( $path, $check ) = @_;

        if ( -e "/proc/acpi/$path" ) {
                opendir(my $items, "/proc/acpi/$path") || die "can't open /proc/acpi/$path: $!";
                foreach my $item ( readdir( $items ) ) {
                        $check->( $item ) && last;
                }
        }
}

my $proc_acpi_thermal_zone_temperature;
find_proc_acpi('thermal_zone', sub {
        my $thm = shift;
        return if ( ! -e "/proc/acpi/thermal_zone/$thm/temperature" );
        warn "# using thermal zone $thm\n" if ( $debug );
        $proc_acpi_thermal_zone_temperature = "/proc/acpi/thermal_zone/$thm/temperature";
});

find_proc_acpi('battery', sub {
        my $bat = shift;
        return unless $bat =~ /\w+/;

        warn "# testing battery $bat\n" if ( $debug );

        if ( proc2hash( "/proc/acpi/battery/$bat/info" )->{present} eq 'yes' ) {
                $proc_acpi_battery = "/proc/acpi/battery/$bat";
                warn "using $proc_acpi_battery to monitor battery\n";
        }
});

my $unit_colors = {
        'b' => '#ff0000',
        'k' => '#ffff00',
        'M' => '#00ff00',
        'G' => '#0000ff',
};

sub unit {
        my ($v,$align) = @_;
        $align ||= '';

        warn "unit( $v )\n" if ($debug);

        my @units = qw/b k M G/;
        my $o = 0;

        while ( ( $v / 10000 ) >= 1 ) {
                $o++;
                $v /= 1024;
        }

        my $unit = $units[$o];

        my $out = '';
        $out .= '^fg(' . $unit_colors->{$unit} . ')';

        if ( $v >= 1 ) {
                $out .= sprintf("%${align}5d",$v);
        } elsif ( $v == 0 ) {
                $out = ' ' x 5;
        } else {
                $out .= sprintf("%${align}5.1f", $v);
        }

        $unit = '^fg(#444444)' . $unit if $dzen;
        $out =~ s/\s(\s*)$/$unit$1/ unless $out =~ m/\s{5}/;
        $out .= '^fg(#ffffff)' if $dzen;

        return $out;
}

my ( $lrx, $ltx ) = ( 0, 0 );
my ( $ld_r, $ld_w ) = ( 0, 0 );
my $bat = '';

my $i = 0;
my $sys_fs = '/sys/class/power_supply/BAT0';

while ( 1 ) {
        my $s = strftime("%Y-%m-%d %H:%M:%S", localtime());

        if ( $i % $acpi_every == 0 ) {
                my $sysfs_path = glob "$sys_fs/*_full";
                if ( $sysfs_path ) {

                        my $full = read_file( $sysfs_path );
                        $sysfs_path =~ s/_full/_now/;
                        my $now = read_file( $sysfs_path );
                        $bat = sprintf("%2d%%", $now * 100 / $full );

                } elsif ( $proc_acpi_battery ) {

                        my $state = proc2hash( "$proc_acpi_battery/state" );

                        if ( $state->{'present rate'} =~ m/^\d+$/ && $state->{'present rate'} != 0 ) {
                                my $info = proc2hash( "$proc_acpi_battery/info" );

                                my $pcnt = $state->{'remaining capacity'} / $info->{'design capacity'};

                                my $time = $state->{'remaining capacity'} / ( $state->{'present rate'} );
                                $time = ( $info->{'design capacity'} - $state->{'remaining capacity'} ) / $state->{'present rate'} if ( $state->{'charging state'} eq 'charging' );

                                warn "time = $time\n" if ($debug);

                                my $hh = int( $time );
                                my $mm = int( ( $time - $hh ) * 60 );
                                my $ss = ( $time * 3600 ) % 60;

                                $bat = sprintf("%s %2d%% %02d:%02d:%02d %3.1fW!",
                                        substr($state->{'charging state'},0,1),
                                        $pcnt * 100, $hh, $mm, $ss,
                                        $state->{'present rate'} / 1000
                                );
                        }
                }
        }
        $i++;

        my $load = read_file('/proc/loadavg');
        chomp( $load );
        $load =~ s!\s\d+/\d+.*!!;

        my $temp = '';
        if ( $proc_acpi_thermal_zone_temperature ) {
                $temp = read_file( $proc_acpi_thermal_zone_temperature );
                chomp( $temp );
                $temp =~ s!^.*:\s+!!;
        }

        my $net = read_file('/proc/net/dev');
        my ( $rx, $tx ) = ( 0,0 );

        foreach ( split(/\n/, $net) ) {
                s/^\s+//;
                s/:\s*/ /;
                my @n = split(/\s+/, $_, 17);
                #next unless ( $n[0] =~ s!(eth\d|ath\d):!! );
                next unless $n[1] =~ m{^\d+$};

                warn dump( @n ) if ($debug);
                $rx += $n[1];
                $tx += $n[9];
        }
        warn "rx: $rx tx: $tx\n" if ($debug);

        my $net_rx = ( $rx - $lrx ) / $dt;
        my $net_tx = ( $tx - $ltx ) / $dt;
        ( $lrx, $ltx ) = ( $rx, $tx );

        my $disk = read_file('/proc/diskstats');
        my ( $d_r, $d_w ) = ( 0,0 );

        foreach ( split(/\n/, $disk) ) {
                s/^\s+//;
                my @d = split(/\s+/, $_, 17);
                next unless ( $d[2] =~ m/^[sh]d\w$/ );

                warn dump( @d ) if ($debug);
                $d_r += $d[5] * $disk_blk_size;
                $d_w += $d[7] * $disk_blk_size;
        }
        warn "d_r: $d_r d_w: $d_w\n" if ($debug);

        my $d_read = ( $d_r - $ld_r ) / $dt;
        my $d_write = ( $d_w - $ld_w ) / $dt;
        ( $ld_r, $ld_w ) = ( $d_r, $d_w );

        my $out = join( $dzen ? ' ^fg(#333333)|^fg(#ffffff) ' : ' | ',
                $load,
                unit( $d_read ) . ' D ' . unit( $d_write, '-' ),
                unit( $net_rx ) . ' > ' . unit( $net_tx, '-' ),
                $bat, $temp,
                $s,
        );


        print "$out\n";
        if ( $awesome ) {
                open(my $fh, '|-', 'awesome-client') || die "can't pipe to awesome-client: $!";
                print $fh
                        $awesome == 3   ? qq{mytextbox.text="$out"\n}
                                        : "0 widget_tell mystatusbar dwm-status text $out\n"
                                        ;
                close($fh);
        }

        sleep $dt;
}

1	dpavlin	30	#!/usr/bin/perl -w
2
3			# dwm-status.pl
4			#
5			# 05/26/07 23:08:31 CEST Dobrica Pavlinusic <dpavlin@rot13.org>
6
7			use strict;
8			use POSIX qw/strftime/;
9			use File::Slurp;
10			use Time::HiRes;
11	dpavlin	59	use Data::Dump qw/dump/;
12	dpavlin	30
13			my $dt = 3;
14	dpavlin	31	my $acpi_every = 10;
15	dpavlin	30
16	dpavlin	35	my $disk_blk_size = 512;
17
18	dpavlin	31	my $debug = shift @ARGV;
19	dpavlin	134	my $awesome = 0;
20	dpavlin	135	my $dzen = 1;
21	dpavlin	30
22			$\|=1;
23
24			sub proc2hash {
25			my $f = shift;
26	dpavlin	39	return unless -f $f;
27	dpavlin	30	open(my $fh, '<', $f) \|\| die "can't open $f: $!";
28			my $h;
29			while(<$fh>) {
30			chomp;
31			my ( $key, $value ) = split(/:\s+/, $_, 2);
32	dpavlin	105	$value =~ s/ m[AVW]h*$//;
33	dpavlin	30	$h->{$key} = $value;
34			}
35	dpavlin	38	warn "$f ",dump( $h ) if ( $debug );
36	dpavlin	30	return $h;
37			}
38
39	dpavlin	39	my $proc_acpi_battery;
40			sub find_proc_acpi {
41			my ( $path, $check ) = @_;
42
43			if ( -e "/proc/acpi/$path" ) {
44			opendir(my $items, "/proc/acpi/$path") \|\| die "can't open /proc/acpi/$path: $!";
45			foreach my $item ( readdir( $items ) ) {
46			$check->( $item ) && last;
47			}
48			}
49			}
50
51			my $proc_acpi_thermal_zone_temperature;
52			find_proc_acpi('thermal_zone', sub {
53			my $thm = shift;
54			return if ( ! -e "/proc/acpi/thermal_zone/$thm/temperature" );
55			warn "# using thermal zone $thm\n" if ( $debug );
56			$proc_acpi_thermal_zone_temperature = "/proc/acpi/thermal_zone/$thm/temperature";
57			});
58
59			find_proc_acpi('battery', sub {
60			my $bat = shift;
61	dpavlin	56	return unless $bat =~ /\w+/;
62	dpavlin	39
63			warn "# testing battery $bat\n" if ( $debug );
64
65			if ( proc2hash( "/proc/acpi/battery/$bat/info" )->{present} eq 'yes' ) {
66			$proc_acpi_battery = "/proc/acpi/battery/$bat";
67			warn "using $proc_acpi_battery to monitor battery\n";
68			}
69			});
70
71	dpavlin	135	my $unit_colors = {
72			'b' => '#ff0000',
73			'k' => '#ffff00',
74			'M' => '#00ff00',
75			'G' => '#0000ff',
76			};
77
78	dpavlin	30	sub unit {
79	dpavlin	135	my ($v,$align) = @_;
80			$align \|\|= '';
81	dpavlin	31
82			warn "unit( $v )\n" if ($debug);
83
84	dpavlin	35	my @units = qw/b k M G/;
85	dpavlin	30	my $o = 0;
86
87	dpavlin	40	while ( ( $v / 10000 ) >= 1 ) {
88	dpavlin	30	$o++;
89			$v /= 1024;
90			}
91
92	dpavlin	135	my $unit = $units[$o];
93
94			my $out = '';
95			$out .= '^fg(' . $unit_colors->{$unit} . ')';
96
97	dpavlin	31	if ( $v >= 1 ) {
98	dpavlin	135	$out .= sprintf("%${align}5d",$v);
99	dpavlin	33	} elsif ( $v == 0 ) {
100	dpavlin	135	$out = ' ' x 5;
101	dpavlin	31	} else {
102	dpavlin	135	$out .= sprintf("%${align}5.1f", $v);
103	dpavlin	31	}
104	dpavlin	135
105			$unit = '^fg(#444444)' . $unit if $dzen;
106			$out =~ s/\s(\s*)$/$unit$1/ unless $out =~ m/\s{5}/;
107			$out .= '^fg(#ffffff)' if $dzen;
108
109			return $out;
110	dpavlin	30	}
111
112			my ( $lrx, $ltx ) = ( 0, 0 );
113	dpavlin	35	my ( $ld_r, $ld_w ) = ( 0, 0 );
114	dpavlin	80	my $bat = '';
115	dpavlin	30
116			my $i = 0;
117	dpavlin	80	my $sys_fs = '/sys/class/power_supply/BAT0';
118	dpavlin	30
119			while ( 1 ) {
120			my $s = strftime("%Y-%m-%d %H:%M:%S", localtime());
121
122	dpavlin	80	if ( $i % $acpi_every == 0 ) {
123	dpavlin	96	my $sysfs_path = glob "$sys_fs/*_full";
124			if ( $sysfs_path ) {
125	dpavlin	30
126	dpavlin	96	my $full = read_file( $sysfs_path );
127			$sysfs_path =~ s/_full/_now/;
128			my $now = read_file( $sysfs_path );
129	dpavlin	135	$bat = sprintf("%2d%%", $now * 100 / $full );
130	dpavlin	89
131			} elsif ( $proc_acpi_battery ) {
132
133	dpavlin	80	my $state = proc2hash( "$proc_acpi_battery/state" );
134	dpavlin	30
135	dpavlin	80	if ( $state->{'present rate'} =~ m/^\d+$/ && $state->{'present rate'} != 0 ) {
136			my $info = proc2hash( "$proc_acpi_battery/info" );
137	dpavlin	30
138	dpavlin	80	my $pcnt = $state->{'remaining capacity'} / $info->{'design capacity'};
139	dpavlin	30
140	dpavlin	80	my $time = $state->{'remaining capacity'} / ( $state->{'present rate'} );
141			$time = ( $info->{'design capacity'} - $state->{'remaining capacity'} ) / $state->{'present rate'} if ( $state->{'charging state'} eq 'charging' );
142	dpavlin	31
143	dpavlin	80	warn "time = $time\n" if ($debug);
144	dpavlin	30
145	dpavlin	80	my $hh = int( $time );
146			my $mm = int( ( $time - $hh ) * 60 );
147			my $ss = ( $time * 3600 ) % 60;
148	dpavlin	30
149	dpavlin	135	$bat = sprintf("%s %2d%% %02d:%02d:%02d %3.1fW!",
150	dpavlin	80	substr($state->{'charging state'},0,1),
151			$pcnt * 100, $hh, $mm, $ss,
152			$state->{'present rate'} / 1000
153			);
154	dpavlin	32	}
155	dpavlin	80	}
156	dpavlin	30	}
157			$i++;
158
159			my $load = read_file('/proc/loadavg');
160			chomp( $load );
161			$load =~ s!\s\d+/\d+.*!!;
162
163	dpavlin	39	my $temp = '';
164			if ( $proc_acpi_thermal_zone_temperature ) {
165			$temp = read_file( $proc_acpi_thermal_zone_temperature );
166			chomp( $temp );
167			$temp =~ s!^.*:\s+!!;
168			}
169	dpavlin	30
170			my $net = read_file('/proc/net/dev');
171			my ( $rx, $tx ) = ( 0,0 );
172
173			foreach ( split(/\n/, $net) ) {
174			s/^\s+//;
175	dpavlin	115	s/:\s*/ /;
176	dpavlin	30	my @n = split(/\s+/, $_, 17);
177	dpavlin	112	#next unless ( $n[0] =~ s!(eth\d\|ath\d):!! );
178			next unless $n[1] =~ m{^\d+$};
179	dpavlin	30
180			warn dump( @n ) if ($debug);
181	dpavlin	112	$rx += $n[1];
182	dpavlin	115	$tx += $n[9];
183	dpavlin	30	}
184			warn "rx: $rx tx: $tx\n" if ($debug);
185
186	dpavlin	35	my $net_rx = ( $rx - $lrx ) / $dt;
187			my $net_tx = ( $tx - $ltx ) / $dt;
188	dpavlin	30	( $lrx, $ltx ) = ( $rx, $tx );
189
190	dpavlin	35	my $disk = read_file('/proc/diskstats');
191			my ( $d_r, $d_w ) = ( 0,0 );
192	dpavlin	30
193	dpavlin	35	foreach ( split(/\n/, $disk) ) {
194			s/^\s+//;
195			my @d = split(/\s+/, $_, 17);
196			next unless ( $d[2] =~ m/^[sh]d\w$/ );
197
198			warn dump( @d ) if ($debug);
199			$d_r += $d[5] * $disk_blk_size;
200			$d_w += $d[7] * $disk_blk_size;
201			}
202			warn "d_r: $d_r d_w: $d_w\n" if ($debug);
203
204			my $d_read = ( $d_r - $ld_r ) / $dt;
205			my $d_write = ( $d_w - $ld_w ) / $dt;
206			( $ld_r, $ld_w ) = ( $d_r, $d_w );
207
208	dpavlin	135	my $out = join( $dzen ? ' ^fg(#333333)\|^fg(#ffffff) ' : ' \| ',
209	dpavlin	35	$load,
210	dpavlin	135	unit( $d_read ) . ' D ' . unit( $d_write, '-' ),
211			unit( $net_rx ) . ' > ' . unit( $net_tx, '-' ),
212	dpavlin	35	$bat, $temp,
213	dpavlin	134	$s,
214	dpavlin	35	);
215
216	dpavlin	135
217	dpavlin	108	print "$out\n";
218	dpavlin	67	if ( $awesome ) {
219			open(my $fh, '\|-', 'awesome-client') \|\| die "can't pipe to awesome-client: $!";
220	dpavlin	108	print $fh
221			$awesome == 3 ? qq{mytextbox.text="$out"\n}
222			: "0 widget_tell mystatusbar dwm-status text $out\n"
223			;
224	dpavlin	67	close($fh);
225			}
226	dpavlin	57
227	dpavlin	30	sleep $dt;
228			}
229