Testing log directory backup
[hcoop/domtool2.git] / src / slave.sml
index b08d41d..7a507fa 100644 (file)
@@ -27,8 +27,9 @@ datatype file_action =
 
 type file_status = {action : file_action,
                    domain : string,
+                   dir : string,
                    file : string}
-                  
+
 val fileHandler = ref (fn _ : file_status => ())
 val preHandler = ref (fn () => ())
 val postHandler = ref (fn () => ())
@@ -55,7 +56,171 @@ fun registerPostHandler handler =
     end
 
 fun handleChanges fs = (!preHandler ();
-                       app (!fileHandler) fs;
+                       app (fn recd as {action, file, ...} =>
+                               (!fileHandler recd;
+                                if action = Delete andalso Posix.FileSys.access (file, []) then
+                                    OS.FileSys.remove file
+                                else
+                                    ())) fs;
                        !postHandler ())
 
+fun shell ss = OS.Process.isSuccess (OS.Process.system (String.concat ss))
+
+fun shellF (ss, msg) =
+    let
+       val s = String.concat ss
+    in
+       if OS.Process.isSuccess (OS.Process.system s) then
+           ()
+       else
+           ErrorMsg.error NONE (msg s)
+    end
+
+fun hostname () =
+    let
+       val inf = TextIO.openIn "/etc/hostname"
+    in
+       case TextIO.inputLine inf of
+           NONE => (TextIO.closeIn inf; raise Fail "No line in /etc/hostname")
+         | SOME line => (TextIO.closeIn inf; String.substring (line, 0, size line - 1))
+    end
+
+fun concatTo p fname =
+    let
+       fun visitDir dname =
+           let
+               val dir = Posix.FileSys.opendir dname
+
+               fun loop () =
+                   case Posix.FileSys.readdir dir of
+                       NONE => Posix.FileSys.closedir dir
+                     | SOME fname' =>
+                       let
+                           val path = OS.Path.joinDirFile {dir = dname, file = fname'}
+                       in
+                           if Posix.FileSys.ST.isDir (Posix.FileSys.stat path) then
+                               visitDir path
+                           else if p fname' then
+                               shellF ([Config.cat, " ", path, " >>", fname],
+                                    fn cl => "Error concatenating: " ^ cl)
+                           else
+                               ();
+                           loop ()
+                       end
+           in
+               loop ()
+           end
+    in
+       TextIO.closeOut (TextIO.openOut fname);
+       visitDir (OS.Path.joinDirFile {dir = Config.resultRoot, file = hostname ()})
+    end
+
+fun enumerateTo p sep fname =
+    let
+       val outf = TextIO.openOut fname
+
+       val first = ref true
+       val baseLen = length (String.fields (fn ch => ch = #"/") Config.resultRoot) + 1
+
+       fun visitDir dname =
+           let
+               val dir = Posix.FileSys.opendir dname
+
+               fun loop () =
+                   case Posix.FileSys.readdir dir of
+                       NONE => Posix.FileSys.closedir dir
+                     | SOME fname' =>
+                       let
+                           val path = OS.Path.joinDirFile {dir = dname, file = fname'}
+                       in
+                           if Posix.FileSys.ST.isDir (Posix.FileSys.stat path) then
+                               visitDir path
+                           else if p fname' then
+                               let
+                                   val toks = String.fields (fn ch => ch = #"/") dname
+                                   val toks = List.drop (toks, baseLen)
+                                   val dom = String.concatWith "." (rev toks)
+                               in
+                                   if !first then
+                                       first := false
+                                   else
+                                       TextIO.output (outf, sep);
+                                   TextIO.output (outf, dom)
+                               end
+                           else
+                               ();
+                           loop ()
+                       end
+           in
+               loop ()
+           end
+    in
+       visitDir (OS.Path.joinDirFile {dir = Config.resultRoot, file = hostname ()});
+       TextIO.closeOut outf
+    end
+
+fun readList fname =
+    let
+       val inf = TextIO.openIn fname
+
+       fun loop acc =
+           case TextIO.inputLine inf of
+               NONE => rev acc
+             | SOME line => loop (String.substring (line, 0, size line - 1) :: acc)
+    in
+       loop []
+       before TextIO.closeIn inf
+    end
+
+fun writeList (fname, ls) =
+    let
+       val outf = TextIO.openOut fname
+    in
+       app (fn s => (TextIO.output (outf, s);
+                     TextIO.output1 (outf, #"\n"))) ls;
+       TextIO.closeOut outf
+    end
+
+fun lineInFile fname line =
+    let
+       val inf = TextIO.openIn fname
+       val line' = line ^ "\n"
+
+       fun loop () =
+           case TextIO.inputLine inf of
+               NONE => false
+             | SOME line => line = line' orelse loop ()
+    in
+       loop ()
+       before TextIO.closeIn inf
+    end handle IO.Io _ => false
+
+fun inGroup {user, group} =
+    List.exists (fn x => x = user)
+               (Posix.SysDB.Group.members (Posix.SysDB.getgrnam group))
+               handle OS.SysErr _ => false
+
+fun mkDirAll dir = ignore (OS.Process.system ("mkdir -p " ^ dir))
+
+fun remove (ls, x) = List.filter (fn y => y <> x) ls
+fun removeDups ls = List.foldr (fn (x, ls) =>
+                                  if List.exists (fn y => y = x) ls then
+                                      ls
+                                  else
+                                      x :: ls) [] ls
+
+fun copyDirCreate {from, to} =
+    (mkDirAll to;
+     if Posix.FileSys.access (from, []) then
+        ignore (OS.Process.system ("cp -r " ^ from ^ " " ^ to))
+     else
+        ())
+
+fun moveDirCreate {from, to} =
+    (mkDirAll to;
+     if Posix.FileSys.access (from, []) then
+        ignore (OS.Process.system ("mv " ^ from ^ " " ^ to))
+     else
+        ())
+
 end