1 (* HCoop
Domtool (http
://hcoop
.sourceforge
.net
/)
2 * Copyright (c
) 2006, Adam Chlipala
4 * This program is free software
; you can redistribute it
and/or
5 * modify it under the terms
of the GNU General Public License
6 * as published by the Free Software Foundation
; either version
2
7 * of the License
, or (at your option
) any later version
.
9 * This program is distributed
in the hope that it will be useful
,
10 * but WITHOUT ANY WARRANTY
; without even the implied warranty
of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE
. See the
12 * GNU General Public License for more details
.
14 * You should have received a copy
of the GNU General Public License
15 * along
with this program
; if not
, write to the Free Software
16 * Foundation
, Inc
., 51 Franklin Street
, Fifth Floor
, Boston
, MA
02110-1301, USA
.
19 (* Code for receiving
and executing configuration files
*)
21 structure Slave
:> SLAVE
= struct
23 datatype file_action
=
28 type file_status
= {action
: file_action
,
33 val fileHandler
= ref (fn _
: file_status
=> ())
34 val preHandler
= ref (fn () => ())
35 val postHandler
= ref (fn () => ())
37 fun registerFileHandler handler
=
39 val old
= !fileHandler
41 fileHandler
:= (fn x
=> (handler x
; old x
))
44 fun registerPreHandler handler
=
48 preHandler
:= (fn () => (handler (); old ()))
51 fun registerPostHandler handler
=
53 val old
= !postHandler
55 postHandler
:= (fn () => (handler (); old ()))
58 fun handleChanges fs
= (!preHandler ();
59 app (fn recd
as {action
, file
, ...} =>
61 if action
= Delete
andalso Posix
.FileSys
.access (file
, []) then
62 OS
.FileSys
.remove file
67 fun shell ss
= OS
.Process
.isSuccess (OS
.Process
.system (String.concat ss
))
69 fun shellF (ss
, msg
) =
71 val s
= String.concat ss
73 if OS
.Process
.isSuccess (OS
.Process
.system s
) then
76 ErrorMsg
.error
NONE (msg s
)
81 val inf
= TextIO.openIn
"/etc/hostname"
83 case TextIO.inputLine inf
of
84 NONE
=> (TextIO.closeIn inf
; raise Fail
"No line in /etc/hostname")
85 | SOME line
=> (TextIO.closeIn inf
; String.substring (line
, 0, size line
- 1))
88 fun concatTo p fname
=
92 val dir
= Posix
.FileSys
.opendir dname
95 case Posix
.FileSys
.readdir dir
of
96 NONE
=> Posix
.FileSys
.closedir dir
99 val path
= OS
.Path
.joinDirFile
{dir
= dname
, file
= fname
'}
101 if Posix
.FileSys
.ST
.isDir (Posix
.FileSys
.stat path
) then
103 else if p fname
' then
104 shellF ([Config
.cat
, " ", path
, " >>", fname
],
105 fn cl
=> "Error concatenating: " ^ cl
)
114 TextIO.closeOut (TextIO.openOut fname
);
115 visitDir (OS
.Path
.joinDirFile
{dir
= Config
.resultRoot
, file
= hostname ()})
118 fun enumerateTo p sep fname
=
120 val outf
= TextIO.openOut fname
123 val baseLen
= length (String.fields (fn ch
=> ch
= #
"/") Config
.resultRoot
) + 1
127 val dir
= Posix
.FileSys
.opendir dname
130 case Posix
.FileSys
.readdir dir
of
131 NONE
=> Posix
.FileSys
.closedir dir
134 val path
= OS
.Path
.joinDirFile
{dir
= dname
, file
= fname
'}
136 if Posix
.FileSys
.ST
.isDir (Posix
.FileSys
.stat path
) then
138 else if p fname
' then
140 val toks
= String.fields (fn ch
=> ch
= #
"/") dname
141 val toks
= List.drop (toks
, baseLen
)
142 val dom
= String.concatWith
"." (rev toks
)
147 TextIO.output (outf
, sep
);
148 TextIO.output (outf
, dom
)
158 visitDir (OS
.Path
.joinDirFile
{dir
= Config
.resultRoot
, file
= hostname ()});
164 val inf
= TextIO.openIn fname
167 case TextIO.inputLine inf
of
169 | SOME line
=> loop (String.substring (line
, 0, size line
- 1) :: acc
)
172 before TextIO.closeIn inf
175 fun writeList (fname
, ls
) =
177 val outf
= TextIO.openOut fname
179 app (fn s
=> (TextIO.output (outf
, s
);
180 TextIO.output1 (outf
, #
"\n"))) ls
;
184 fun lineInFile fname line
=
186 val inf
= TextIO.openIn fname
187 val line
' = line ^
"\n"
190 case TextIO.inputLine inf
of
192 | SOME line
=> line
= line
' orelse loop ()
195 before TextIO.closeIn inf
196 end handle IO
.Io _
=> false