X-Git-Url: https://git.hcoop.net/hcoop/domtool2.git/blobdiff_plain/d612d62cd04b713bb1057fd2e666365704aaf3d6..7db53a0b3693ddd01e6a36fc5bfb4ba56b4656eb:/src/slave.sml

diff --git a/src/slave.sml b/src/slave.sml
index b08d41d..b0148f1 100644
--- 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 () => ())
@@ -58,4 +59,121 @@ fun handleChanges fs = (!preHandler ();
 			app (!fileHandler) 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
+
 end