Slave dispatching working

[hcoop/domtool2.git] / src / slave.sml

diff --git a/src/slave.sml b/src/slave.sml
index 1521b73..b1303b8 100644 (file)
--- a/src/slave.sml
+++ b/src/slave.sml
@@ -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 () => ())
@@ -87,17 +88,61 @@ fun concatTo p fname =
                                visitDir path
                            else if p fname' then
                                shellF ([Config.cat, " ", path, " >>", fname],
-                                              fn cl => "Error concatenating: " ^ cl)
+                                    fn cl => "Error concatenating: " ^ cl)
                            else
                                ();
                            loop ()
                        end
            in
-               TextIO.closeOut (TextIO.openOut fname);
                loop ()
            end
     in
+       TextIO.closeOut (TextIO.openOut fname);
        visitDir Config.resultRoot
     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 Config.resultRoot;
+       TextIO.closeOut outf
+    end
+
 end