Password prompts for MySQL adduser
[hcoop/domtool2.git] / src / main.sml
index 0f65362..ec98e4e 100644 (file)
@@ -116,10 +116,13 @@ fun checkDir dname =
        val (_, files) = Order.order (SOME b) files
     in
        if !ErrorMsg.anyErrors then
-           false
+           raise ErrorMsg.Error
        else
            (foldl (fn (fname, G) => check' G fname) b files;
-            !ErrorMsg.anyErrors)
+            if !ErrorMsg.anyErrors then
+                raise ErrorMsg.Error
+            else
+                ())
     end
 
 fun reduce fname =
@@ -152,6 +155,15 @@ fun eval fname =
            Eval.exec (Defaults.eInit ()) body'
       | NONE => raise ErrorMsg.Error
 
+fun eval' fname =
+    case reduce fname of
+       (SOME body') =>
+       if !ErrorMsg.anyErrors then
+           raise ErrorMsg.Error
+       else
+           ignore (Eval.exec' (Defaults.eInit ()) body')
+      | NONE => raise ErrorMsg.Error
+
 val dispatcher =
     Config.dispatcher ^ ":" ^ Int.toString Config.dispatcherPort
 
@@ -207,7 +219,9 @@ fun request fname =
 
 fun requestDir dname =
     let
-       val (user, bio) = requestBio (fn () => ignore (checkDir dname))
+       val _ = ErrorMsg.reset ()
+
+       val (user, bio) = requestBio (fn () => checkDir dname)
 
        val b = basis ()
 
@@ -246,15 +260,18 @@ fun requestDir dname =
                                before TextIO.closeIn inf
                            end) files
     in
-       Msg.send (bio, MsgMultiConfig codes);
-       case Msg.recv bio of
-           NONE => print "Server closed connection unexpectedly.\n"
-         | SOME m =>
-           case m of
-               MsgOk => print "Configuration succeeded.\n"
-             | MsgError s => print ("Configuration failed: " ^ s ^ "\n")
-             | _ => print "Unexpected server reply.\n";
-       OpenSSL.close bio
+       if !ErrorMsg.anyErrors then
+           ()
+       else
+           (Msg.send (bio, MsgMultiConfig codes);
+            case Msg.recv bio of
+                NONE => print "Server closed connection unexpectedly.\n"
+              | SOME m =>
+                case m of
+                    MsgOk => print "Configuration succeeded.\n"
+                  | MsgError s => print ("Configuration failed: " ^ s ^ "\n")
+                  | _ => print "Unexpected server reply.\n";
+            OpenSSL.close bio)
     end
     handle ErrorMsg.Error => ()
 
@@ -324,6 +341,21 @@ fun requestWhoHas perm =
        before OpenSSL.close bio
     end
 
+fun requestRegen () =
+    let
+       val (_, bio) = requestBio (fn () => ())
+    in
+       Msg.send (bio, MsgRegenerate);
+       case Msg.recv bio of
+           NONE => print "Server closed connection unexpectedly.\n"
+         | SOME m =>
+           case m of
+               MsgOk => print "Regeneration succeeded.\n"
+             | MsgError s => print ("Regeneration failed: " ^ s ^ "\n")
+             | _ => print "Unexpected server reply.\n";
+       OpenSSL.close bio
+    end
+
 fun requestRmdom dom =
     let
        val (_, bio) = requestBio (fn () => ())
@@ -339,6 +371,131 @@ fun requestRmdom dom =
        OpenSSL.close bio
     end
 
+fun requestRmuser user =
+    let
+       val (_, bio) = requestBio (fn () => ())
+    in
+       Msg.send (bio, MsgRmuser user);
+       case Msg.recv bio of
+           NONE => print "Server closed connection unexpectedly.\n"
+         | SOME m =>
+           case m of
+               MsgOk => print "Removal succeeded.\n"
+             | MsgError s => print ("Removal failed: " ^ s ^ "\n")
+             | _ => print "Unexpected server reply.\n";
+       OpenSSL.close bio
+    end
+
+fun requestDbUser dbtype =
+    let
+       val (_, bio) = requestBio (fn () => ())
+    in
+       Msg.send (bio, MsgCreateDbUser dbtype);
+       case Msg.recv bio of
+           NONE => print "Server closed connection unexpectedly.\n"
+         | SOME m =>
+           case m of
+               MsgOk => print "Your user has been created.\n"
+             | MsgError s => print ("Creation failed: " ^ s ^ "\n")
+             | _ => print "Unexpected server reply.\n";
+       OpenSSL.close bio
+    end
+
+fun requestDbTable p =
+    let
+       val (user, bio) = requestBio (fn () => ())
+    in
+       Msg.send (bio, MsgCreateDbTable p);
+       case Msg.recv bio of
+           NONE => print "Server closed connection unexpectedly.\n"
+         | SOME m =>
+           case m of
+               MsgOk => print ("Your database " ^ user ^ "_" ^ #dbname p ^ " has been created.\n")
+             | MsgError s => print ("Creation failed: " ^ s ^ "\n")
+             | _ => print "Unexpected server reply.\n";
+       OpenSSL.close bio
+    end
+
+fun regenerate context =
+    let
+       val b = basis ()
+       val () = Tycheck.disallowExterns ()
+
+       val () = Domain.resetGlobal ()
+
+       fun contactNode (node, ip) =
+           if node = Config.defaultNode then
+               Domain.resetLocal ()
+           else let
+                   val bio = OpenSSL.connect (context,
+                                              ip
+                                              ^ ":"
+                                              ^ Int.toString Config.slavePort)
+               in
+                   Msg.send (bio, MsgRegenerate);
+                   case Msg.recv bio of
+                       NONE => print "Slave closed connection unexpectedly\n"
+                     | SOME m =>
+                       case m of
+                           MsgOk => print ("Slave " ^ node ^ " pre-regeneration finished\n")
+                         | MsgError s => print ("Slave " ^ node
+                                                ^ " returned error: " ^
+                                                s ^ "\n")
+                         | _ => print ("Slave " ^ node
+                                       ^ " returned unexpected command\n");
+                   OpenSSL.close bio
+               end             
+
+       fun doUser user =
+           let
+               val _ = Domain.setUser user
+               val _ = ErrorMsg.reset ()
+
+               val dname = Config.domtoolDir user
+
+               val dir = Posix.FileSys.opendir dname
+
+               fun loop files =
+                   case Posix.FileSys.readdir dir of
+                       NONE => (Posix.FileSys.closedir dir;
+                                files)
+                     | SOME fname =>
+                       if notTmp fname then
+                           loop (OS.Path.joinDirFile {dir = dname,
+                                                      file = fname}
+                                 :: files)
+                       else
+                           loop files
+
+               val files = loop []
+               val (_, files) = Order.order (SOME b) files
+           in
+               if !ErrorMsg.anyErrors then
+                   print ("User " ^ user ^ "'s configuration has errors!\n")
+               else
+                   app eval' files
+           end
+               handle IO.Io _ => ()
+                    | OS.SysErr (s, _) => print ("System error processing user " ^ user ^ ": " ^ s ^ "\n")
+    in
+       app contactNode Config.nodeIps;
+       Env.pre ();
+       app doUser (Acl.users ());
+       Env.post ()
+    end
+
+fun rmuser user =
+    let
+       val doms = Acl.class {user = user, class = "domain"}
+       val doms = List.filter (fn dom =>
+                                  case Acl.whoHas {class = "domain", value = dom} of
+                                      [_] => true
+                                    | _ => false) (StringSet.listItems doms)
+    in
+       Acl.rmuser user;
+       Domain.rmdom doms
+    end
+
 fun service () =
     let
        val () = Acl.read Config.aclFile
@@ -373,10 +530,12 @@ fun service () =
                                in
                                    TextIO.output (outf, code);
                                    TextIO.closeOut outf;
-                                   eval outname
+                                   eval' outname
                                end
                        in
-                           (app doOne codes;
+                           (Env.pre ();
+                            app doOne codes;
+                            Env.post ();
                             Msg.send (bio, MsgOk))
                            handle ErrorMsg.Error =>
                                   (print "Compilation error\n";
@@ -481,12 +640,15 @@ fun service () =
                                handle OpenSSL.OpenSSL _ => ();
                                loop ())
 
-                             | MsgRmdom dom =>
+                             | MsgRmdom doms =>
                                if Acl.query {user = user, class = "priv", value = "all"}
-                                  orelse Acl.query {user = user, class = "domain", value = dom} then
-                                   ((Domain.rmdom dom;
+                                  orelse List.all (fn dom => Acl.query {user = user, class = "domain", value = dom}) doms then
+                                   ((Domain.rmdom doms;
+                                     app (fn dom =>
+                                             Acl.revokeFromAll {class = "domain", value = dom}) doms;
+                                     Acl.write Config.aclFile;
                                      Msg.send (bio, MsgOk);
-                                     print ("Removed domain " ^ dom ^ ".\n"))
+                                     print ("Removed domains" ^ foldl (fn (d, s) => s ^ " " ^ d) "" doms ^ ".\n"))
                                     handle OpenSSL.OpenSSL s =>
                                            (print "OpenSSL error\n";
                                             Msg.send (bio,
@@ -503,7 +665,123 @@ fun service () =
                                      ignore (OpenSSL.readChar bio);
                                      OpenSSL.close bio)
                                     handle OpenSSL.OpenSSL _ => ();
-                                    loop ())                           
+                                    loop ())
+
+                             | MsgRegenerate =>
+                               if Acl.query {user = user, class = "priv", value = "regen"}
+                                  orelse Acl.query {user = user, class = "priv", value = "all"} then
+                                   ((regenerate context;
+                                     Msg.send (bio, MsgOk);
+                                     print "Regenerated all configuration.\n")
+                                    handle OpenSSL.OpenSSL s =>
+                                           (print "OpenSSL error\n";
+                                            Msg.send (bio,
+                                                      MsgError
+                                                          ("Error during regeneration: "
+                                                           ^ s)));
+                                   (ignore (OpenSSL.readChar bio);
+                                    OpenSSL.close bio)
+                                   handle OpenSSL.OpenSSL _ => ();
+                                   loop ())
+                               else
+                                   ((Msg.send (bio, MsgError "Not authorized to regeneration");
+                                     print "Unauthorized user asked to regenerate!\n";
+                                     ignore (OpenSSL.readChar bio);
+                                     OpenSSL.close bio)
+                                    handle OpenSSL.OpenSSL _ => ();
+                                    loop ())
+
+                             | MsgRmuser user' =>
+                               if Acl.query {user = user, class = "priv", value = "all"} then
+                                   ((rmuser user';
+                                     Acl.write Config.aclFile;
+                                     Msg.send (bio, MsgOk);
+                                     print ("Removed user " ^ user' ^ ".\n"))
+                                    handle OpenSSL.OpenSSL s =>
+                                           (print "OpenSSL error\n";
+                                            Msg.send (bio,
+                                                      MsgError
+                                                          ("Error during revocation: "
+                                                           ^ s)));
+                                   (ignore (OpenSSL.readChar bio);
+                                    OpenSSL.close bio)
+                                   handle OpenSSL.OpenSSL _ => ();
+                                   loop ())
+                               else
+                                   ((Msg.send (bio, MsgError "Not authorized to remove users");
+                                     print "Unauthorized user asked to remove a user!\n";
+                                     ignore (OpenSSL.readChar bio);
+                                     OpenSSL.close bio)
+                                    handle OpenSSL.OpenSSL _ => ();
+                                    loop ())
+
+                             | MsgCreateDbUser {dbtype, passwd} =>
+                               (case Dbms.lookup dbtype of
+                                    NONE => ((Msg.send (bio, MsgError ("Unknown database type " ^ dbtype));
+                                              print ("Database user creation request with unknown datatype type " ^ dbtype);
+                                              ignore (OpenSSL.readChar bio))
+                                             handle OpenSSL.OpenSSL _ => ();
+                                             OpenSSL.close bio
+                                             handle OpenSSL.OpenSSL _ => ();
+                                             loop ())
+                                  | SOME handler =>
+                                    case #adduser handler {user = user, passwd = passwd} of
+                                        NONE => ((Msg.send (bio, MsgOk);
+                                                  print ("Added " ^ dbtype ^ " user " ^ user ^ ".\n"))
+                                                 handle OpenSSL.OpenSSL s =>
+                                                        (print "OpenSSL error\n";
+                                                         Msg.send (bio,
+                                                                   MsgError
+                                                                       ("Error during creation: "
+                                                                        ^ s)));
+                                                 (ignore (OpenSSL.readChar bio);
+                                                  OpenSSL.close bio)
+                                                 handle OpenSSL.OpenSSL _ => ();
+                                                 loop ())
+                                      | SOME msg => ((Msg.send (bio, MsgError ("Error adding user: " ^ msg));
+                                                      print ("Error adding a " ^ dbtype ^ " user " ^ user ^ ": " ^ msg ^ "\n");
+                                                      ignore (OpenSSL.readChar bio);
+                                                      OpenSSL.close bio)
+                                                     handle OpenSSL.OpenSSL _ => ();
+                                                     loop ()))
+
+                             | MsgCreateDbTable {dbtype, dbname} =>
+                               if Dbms.validDbname dbname then
+                                   (case Dbms.lookup dbtype of
+                                        NONE => ((Msg.send (bio, MsgError ("Unknown database type " ^ dbtype));
+                                                  print ("Database creation request with unknown datatype type " ^ dbtype);
+                                                  ignore (OpenSSL.readChar bio))
+                                                 handle OpenSSL.OpenSSL _ => ();
+                                                 OpenSSL.close bio
+                                                 handle OpenSSL.OpenSSL _ => ();
+                                                 loop ())
+                                      | SOME handler =>
+                                        case #createdb handler {user = user, dbname = dbname} of
+                                            NONE => ((Msg.send (bio, MsgOk);
+                                                      print ("Created database " ^ user ^ "_" ^ dbname ^ ".\n"))
+                                                     handle OpenSSL.OpenSSL s =>
+                                                            (print "OpenSSL error\n";
+                                                             Msg.send (bio,
+                                                                       MsgError
+                                                                           ("Error during creation: "
+                                                                            ^ s)));
+                                                     (ignore (OpenSSL.readChar bio);
+                                                      OpenSSL.close bio)
+                                                     handle OpenSSL.OpenSSL _ => ();
+                                                     loop ())
+                                          | SOME msg => ((Msg.send (bio, MsgError ("Error creating database: " ^ msg));
+                                                          print ("Error creating database " ^ user ^ "_" ^ dbname ^ ": " ^ msg ^ "\n");
+                                                          ignore (OpenSSL.readChar bio);
+                                                          OpenSSL.close bio)
+                                                         handle OpenSSL.OpenSSL _ => ();
+                                                         loop ()))
+                               else
+                                   ((Msg.send (bio, MsgError ("Invalid database name " ^ dbname));
+                                     print ("Invalid database name " ^ user ^ "_" ^ dbname ^ "\n");
+                                     ignore (OpenSSL.readChar bio);
+                                     OpenSSL.close bio)
+                                    handle OpenSSL.OpenSSL _ => ();
+                                    loop ())
 
                              | _ =>
                                (Msg.send (bio, MsgError "Unexpected command")
@@ -561,6 +839,8 @@ fun slave () =
                                        MsgFile file => loop' (file :: files)
                                      | MsgDoFiles => (Slave.handleChanges files;
                                                       Msg.send (bio, MsgOk))
+                                     | MsgRegenerate => (Domain.resetLocal ();
+                                                         Msg.send (bio, MsgOk))
                                      | _ => (print "Dispatcher sent unexpected command\n";
                                              Msg.send (bio, MsgError "Unexpected command"))
                        in
@@ -584,7 +864,7 @@ fun slave () =
        OpenSSL.shutdown sock
     end
 
-fun autodocBasis outdir =
+fun listBasis () =
     let
        val dir = Posix.FileSys.opendir Config.libRoot
 
@@ -599,10 +879,11 @@ fun autodocBasis outdir =
                          :: files)
                else
                    loop files
-
-       val files = loop []
     in
-       Autodoc.autodoc {outdir = outdir, infiles = files}
+       loop []
     end
 
+fun autodocBasis outdir =
+    Autodoc.autodoc {outdir = outdir, infiles = listBasis ()}
+
 end