-(*
-Domtool 2 (http://hcoop.sf.net/)
-Copyright (C) 2007 Adam Chlipala
-
-This program is free software; you can redistribute it and/or
-modify it under the terms of the GNU General Public License
-as published by the Free Software Foundation; either version 2
-of the License, or (at your option) any later version.
-
-This program is distributed in the hope that it will be useful,
-but WITHOUT ANY WARRANTY; without even the implied warranty of
-MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-GNU General Public License for more details.
-
-You should have received a copy of the GNU General Public License
-along with this program; if not, write to the Free Software
-Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
-*)
-
-structure Quotas = struct
-
- fun homedir s = Posix.SysDB.Passwd.home (Posix.SysDB.getpwnam s)
- fun userVols s = [homedir s]
-
- fun used path =
- let
- val proc = Unix.execute ("/bin/sh", ["-c", "/usr/bin/fs listquota -path " ^ path])
- val inf = Unix.textInstreamOf proc
- in
- (case TextIO.inputLine inf of
- NONE => NONE
- | SOME line =>
- if String.isPrefix "fs: " line then
- NONE
- else
- case TextIO.inputLine inf of
- NONE => NONE
- | SOME line =>
- case String.tokens Char.isSpace line of
- _ :: _ :: n :: _ => Int.fromString n
- | _ => NONE)
- before ignore (Unix.reap proc)
- end
-
- fun getDiskUsage () =
- let
- fun explorer (level, path, acc) =
- let
- val dir = Posix.FileSys.opendir path
-
- fun loop acc =
- case Posix.FileSys.readdir dir of
- NONE => acc
- | SOME file =>
- let
- val acc =
- if level = 2 then
- file :: acc
- else
- explorer (level+1, OS.Path.joinDirFile {dir = path, file = file}, acc)
- in
- loop acc
- end
- in
- loop acc
- before Posix.FileSys.closedir dir
- end
-
- val users = explorer (0, "/afs/hcoop.net/user", [])
-
- fun count uname =
- foldl (fn (vol, total) => case used vol of
- NONE => total
- | SOME n => total + n) 0 (userVols uname)
-
- val users = ListMergeSort.sort (fn (s1, s2) => String.compare (s1, s2) = GREATER) users
- val all = List.map (fn user => {uname = user, kbs = count user}) users
- val all = ListMergeSort.sort (fn ({kbs = kb1, ...}, {kbs = kb2, ...}) => kb1 > kb2) all
- in
- app (fn {uname, kbs} => (print uname;
- print " ";
- print (Int.toString kbs);
- print "\n")) all
- end
-
-end
HCoop / hcoop / domtool2.git / commitdiff