Merge branch 'dev' of feistymeow.org:feisty_meow into dev
[feisty_meow.git] / scripts / files / summing_dir.pl
1 #!/usr/bin/perl
2
3 ###############################################################################
4 #                                                                             #
5 #  Name   : summing_dir                                                       #
6 #  Author : Chris Koeritz                                                     #
7 #  Rights : Copyright (C) 2000-$now by Author                                 #
8 #                                                                             #
9 #  Purpose:                                                                   #
10 #                                                                             #
11 #    Provides a somewhat stylized directory lister.                           #
12 #                                                                             #
13 ###############################################################################
14 #  This program is free software; you can redistribute it and/or modify it    #
15 #  under the terms of the GNU General Public License as published by the Free #
16 #  Software Foundation; either version 2 of the License or (at your option)   #
17 #  any later version.  See: "http://www.gruntose.com/Info/GNU/GPL.html" for a #
18 #  version of the License.  Please send any updates to "fred@gruntose.com".   #
19 ###############################################################################
20
21 require "filename_helper.pl";
22
23 use Env qw($TMP $color_add $TERM);
24
25 local($chewed_line) = "";
26 local(@arg_list);
27 local($print_list) = "";
28
29 # if there were no parameters, we make the default listing for the current
30 # directory.
31 if ($#ARGV < 0) {
32   $print_list = "Current Directory";
33   @arg_list = ( "." );
34 } else {
35   local(@munged_list) = &patch_name_for_pc(&remove_trailing_slashes(@ARGV));
36   $print_list = "@munged_list";
37   @arg_list = &glob_list(@ARGV);
38 }
39
40 foreach $dir (@arg_list) {
41   if ($dir eq "-al") { next; }  # skip ls directives.
42   if ($dir eq "-l") { next; }  # skip ls directives.
43   $chewed_line = $chewed_line . " \"$dir\"";
44 }
45
46 if ("$chewed_line" eq "") {
47   print "No files matched that path specification.\n";
48   exit 0;
49 }
50
51 # show the header, now that we know there's something to print.
52 print "[" . $print_list . "]\n\n";
53
54 local($temp_file)=`mktemp "$TMP/zz_frdsumdir.XXXXXX"`;
55 chop($temp_file);
56
57 # drop the main payload, the list of directory info, but also save that
58 # info to a file for analysis.
59 system("ls -HhlF $color_add $chewed_line");
60 system("ls -HhlF $color_add $chewed_line > $temp_file");
61   # the color_add variable, if defined, will have flags for setting the
62   # directory listing color scheme.
63
64 local($lengths) = 0;
65
66 # open the file and process the lines to get file lengths.
67 open(DIRLIST, "<$temp_file");
68 # we only want to match ls -al style output lines, and only want to keep the size.
69 $pattern="^[^ ]+ +[^ ]+ +[^ ]+ +[^ ]+ +([0-9.]+[KMG]?).*\$";
70 foreach $file_line (<DIRLIST>) {
71   if ($file_line =~ /$pattern/) {
72     (local $munged = $file_line) =~ s/$pattern/\1/;
73     if ($munged =~ /K$/) {
74       chop $munged;
75       $munged *= 1024.0;
76       #print "K munged is now $munged\n";
77     }
78     if ($munged =~ /M$/) {
79       chop $munged;
80       $munged *= 1024.0 * 1024.0;
81       #print "M munged is now $munged\n";
82     }
83     if ($munged =~ /G$/) {
84       chop $munged;
85       $munged *= 1024.0 * 1024.0 * 1024.0;
86       #print "G munged is now $munged\n";
87     }
88     $lengths += $munged;
89   }
90 }
91 close(DIRLIST);
92 unlink($temp_file);  # clean up.
93
94 local($total)=int($lengths);
95 local($kbytes)=int($total / 102.4) / 10;
96 local($mbytes)=int($kbytes / 102.4) / 10;
97 local($gbytes)=int($mbytes / 102.4) / 10;
98
99 print "\n";
100 # print a fancy listing showing bytes at least, but only showing mb and gb if appropriate.
101 print "These files occupy $total bytes ($kbytes KB";
102 if ($mbytes ne 0) {
103  print " / $mbytes MB";
104 }
105 if ($gbytes ne 0) {
106  print " / $gbytes GB";
107 }
108 print ").\n";
109
110 print "Overall Drive Usage (megs):\n";
111
112 system("df -m $chewed_line >$temp_file");
113
114 # now eat the file again, but this time to get drive space info.
115 open(DIRLIST, "<$temp_file");
116 local($space_info) = "";
117 local($did_title) = 0;  # true if we have printed the title by now.
118 foreach $file_line (<DIRLIST>) {
119   ($space_info = $file_line) =~ s/[^ ]* *([^ ]*) *([^ ]*) *([^ ]*) *([^ ]*) *.*$/\1     \2      \3      \4/;
120   if (!$did_title) {
121     # if the title hasn't been printed yet, we take some of the info out of
122     # the line and use it for the right sense of the last column.
123     print "Total        Used    Free    ";
124     $space_info =~ s/[^ ]*      *[^     ]*      *[^     ]*      *([^    ]*)/\1/;
125     print "$space_info";
126     $did_title = 1;
127   } else {
128   }
129 }
130 close(DIRLIST);
131 unlink($temp_file);  # clean up.
132
133 print "$space_info\n";
134
135 exit 0;
136