Webalizer

[hcoop/domtool2.git] / src / plugins / webalizer.sml

diff --git a/src/plugins/webalizer.sml b/src/plugins/webalizer.sml
new file mode 100644 (file)
index 0000000..f88f2ff
--- /dev/null
+++ b/src/plugins/webalizer.sml
@@ -0,0 +1,106 @@
+(* HCoop Domtool (http://hcoop.sourceforge.net/)
+ * Copyright (c) 2006, Adam Chlipala
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; either version 2
+ * of the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+ *)
+
+(* Webalizer statistics tool handling *)
+
+structure Webalizer :> WEBALIZER = struct
+
+val files = ref ([] : TextIO.outstream list)
+val write = ref (fn _ : string => ())
+
+val () = Apache.registerPre
+        (fn {nodes, id, hostname} =>
+            let
+                val fds = map (fn node => Domain.domainFile {node = node,
+                                                             name = id ^ ".wbl"}) nodes
+            in
+                files := fds;
+                write := (fn s => app (fn fd => TextIO.output (fd, s)) fds);
+                !write "LogFile\t";
+                !write Config.Apache.logDir;
+                !write "/";
+                !write hostname;
+                !write "/access.log\nOutputDir\t";
+                !write Config.Webalizer.outputDir;
+                !write "/";
+                !write id;
+                !write "\nHostname\t";
+                !write hostname;
+                !write "\nHideSite\t";
+                !write hostname;
+                !write "\nHideReferrer\t";
+                !write hostname;
+                !write "\n"
+            end)
+
+val () = Apache.registerPost
+        (fn () => app TextIO.closeOut (!files))
+
+val () = Apache.registerAliaser
+        (fn hostname =>
+            (!write "HideSite\t";
+             !write hostname;
+             !write "\nHideReferrer\t";
+             !write hostname;
+             !write "\n"))
+
+val () = Slave.registerFileHandler (fn fs =>
+                                      let
+                                          val spl = OS.Path.splitDirFile (#file fs)
+                                      in
+                                          case OS.Path.splitBaseExt (#file spl) of
+                                              {base, ext = SOME "wbl"} =>
+                                              (case #action fs of
+                                                   Slave.Delete =>
+                                                   (ignore (OS.Process.system (Config.rm
+                                                                               ^ " -f "
+                                                                               ^ Config.Webalizer.configDir
+                                                                               ^ "/"
+                                                                               ^ base
+                                                                               ^ ".conf"));
+                                                    ignore (OS.Process.system (Config.rm
+                                                                               ^ " -rf "
+                                                                               ^ Config.Webalizer.outputDir
+                                                                               ^ "/"
+                                                                               ^ base)))
+                                                 | _ =>
+                                                   (ignore (OS.Process.system (Config.cp
+                                                                               ^ " "
+                                                                               ^ #file fs
+                                                                               ^ " "
+                                                                               ^ Config.Webalizer.configDir
+                                                                               ^ "/"
+                                                                               ^ base
+                                                                               ^ ".conf"));
+                                                    let
+                                                        val dir = Config.Webalizer.outputDir ^ "/" ^ base
+                                                    in
+                                                        if Posix.FileSys.access (dir, []) then
+                                                            ()
+                                                        else
+                                                            Posix.FileSys.mkdir (dir, Posix.FileSys.S.flags
+                                                                                          [Posix.FileSys.S.iroth,
+                                                                                           Posix.FileSys.S.ixoth,
+                                                                                           Posix.FileSys.S.irwxu,
+                                                                                           Posix.FileSys.S.irgrp,
+                                                                                           Posix.FileSys.S.iwgrp])
+                                                    end))
+                                            | _ => ()
+                                      end)
+
+end