X-Git-Url: https://git.hcoop.net/hcoop/domtool2.git/blobdiff_plain/361a1e7f7b226811bdb8b4cc0266f34f01ac4bcd..b2293570ee621fcb91f62b4effde0fe9b9c32cbd:/src/main-client.sml

diff --git a/src/main-client.sml b/src/main-client.sml
index d6fe010..642e35f 100644
--- a/src/main-client.sml
+++ b/src/main-client.sml
@@ -16,9 +16,72 @@
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  *)
 
-(* Driver for server *)
+(* Driver for configuration requests *)
+
+val () = Domain.declareClient ()
+
+fun uid () =
+    case Posix.ProcEnv.getenv "DOMTOOL_USER" of
+	NONE => Posix.ProcEnv.getuid ()
+      | SOME user => Posix.SysDB.Passwd.uid (Posix.SysDB.getpwnam user)
+
+fun domtoolRoot () =
+    let
+	val dname = Posix.SysDB.Passwd.home (Posix.SysDB.getpwuid (uid ()))
+    in
+	OS.Path.joinDirFile {dir = dname,
+			     file = ".domtool"}
+    end
+
+fun libnameOpt () =
+    let
+	val libname = OS.Path.joinDirFile {dir = domtoolRoot (),
+					   file = "lib.dtl"}
+    in
+	if Posix.FileSys.access (libname, []) then
+	    SOME libname
+	else
+	    NONE
+    end
+
+val (tc, fake, args) = foldl (fn (arg, (tc, fake, args)) =>
+				 case arg of
+				     "-tc" => (true, fake, args)
+				   | "-fake" => (tc, true, args)
+				   | _ => (tc, fake, arg :: args))
+		       (false, false, []) (CommandLine.arguments ())
+
+val args = rev args
+
+val (doit, doitDir) =
+    if tc then
+	(fn fname =>
+	    let
+		val _ : string = Main.setupUser ()
+		val () = if fake then
+			     Domain.fakePrivileges ()
+			 else
+			     ()
+		val env = Main.basis ()
+		val env =
+		    case libnameOpt () of
+			NONE => env
+		      | SOME libname => #1 (Main.check env libname)
+	    in
+		ignore (Main.check env fname)
+	    end,
+	 Main.checkDir)
+    else
+	(fn fname => Main.request (fname, libnameOpt ()),
+	 Main.requestDir)
 
 val _ =
-    case CommandLine.arguments () of
-	[fname] => Main.request fname
+    case args of
+	[fname] =>
+	if Posix.FileSys.access (fname, []) then
+	    doit fname
+	else
+	    doit (OS.Path.joinDirFile {dir = domtoolRoot (),
+				       file = fname})
+      | [] => doitDir (domtoolRoot ())
       | _ => print "Invalid command-line arguments\n"