src/plugins/apache.sml

(* 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.
 *)

(* Apache HTTPD handling *)

structure Apache :> APACHE = struct

open Ast

val _ = Env.type_one "proxy_port"
	Env.int
	(fn n => n > 1024)

val _ = Env.type_one "proxy_target"
	Env.string
	(fn s =>
	    let
		fun default () = List.exists (fn s' => s = s') Config.Apache.proxyTargets
	    in
		case String.fields (fn ch => ch = #":") s of
		    ["http", "//localhost", rest] =>
		    (case String.fields (fn ch => ch = #"/") rest of
			 port :: _ =>
			 (case Int.fromString port of
			      NONE => default ()
			    | SOME n => n > 1024 orelse default ())
		       | _ => default ())
		  | _ => default ()
	    end)

val _ = Env.type_one "rewrite_arg"
	Env.string
	(CharVector.all Char.isAlphaNum)

fun validLocation s =
    size s > 0 andalso size s < 1000 andalso CharVector.all
						 (fn ch => Char.isAlphaNum ch
							   orelse ch = #"-"
							   orelse ch = #"_"
							   orelse ch = #"."
							   orelse ch = #"/") s

val _ = Env.type_one "location"
	Env.string
	validLocation

val dl = ErrorMsg.dummyLoc

val _ = Main.registerDefault ("WebNodes",
			      (TList (TBase "node", dl), dl),
			      (fn () => (EList (map (fn s => (EString s, dl)) Config.Apache.webNodes), dl)))

val _ = Main.registerDefault ("SSL",
			      (TBase "bool", dl),
			      (fn () => (EVar "false", dl)))

val _ = Main.registerDefault ("User",
			      (TBase "your_user", dl),
			      (fn () => (EString (Domain.getUser ()), dl)))

val _ = Main.registerDefault ("Group",
			      (TBase "your_group", dl),
			      (fn () => (EString (Domain.getUser ()), dl)))

val _ = Main.registerDefault ("DocumentRoot",
			      (TBase "your_path", dl),
			      (fn () => (EString ("/home/" ^ Domain.getUser () ^ "/public_html"), dl)))

val _ = Main.registerDefault ("ServerAdmin",
			      (TBase "email", dl),
			      (fn () => (EString (Domain.getUser () ^ "@" ^ Config.defaultDomain), dl)))


val redirect_code = fn (EVar "temp", _) => SOME "temp"
		     | (EVar "permanent", _) => SOME "permanent"
		     | (EVar "seeother", _) => SOME "seeother"
		     | (EVar "redir300", _) => SOME "300"
		     | (EVar "redir301", _) => SOME "301"
		     | (EVar "redir302", _) => SOME "302"
		     | (EVar "redir303", _) => SOME "303"
		     | (EVar "redir304", _) => SOME "304"
		     | (EVar "redir305", _) => SOME "305"
		     | (EVar "redir307", _) => SOME "307"
		     | _ => NONE

val flag = fn (EVar "redirect", _) => SOME "R"
	    | (EVar "forbidden", _) => SOME "F"
	    | (EVar "gone", _) => SOME "G"
	    | (EVar "last", _) => SOME "L"
	    | (EVar "chain", _) => SOME "C"
	    | (EVar "nosubreq", _) => SOME "NS"
	    | (EVar "nocase", _) => SOME "NC"
	    | (EVar "qsappend", _) => SOME "QSA"
	    | (EVar "noescape", _) => SOME "NE"
	    | (EVar "passthrough", _) => SOME "PT"
	    | (EApp ((EVar "mimeType", _), e), _) =>
	      Option.map (fn s => "T=" ^ s) (Env.string e)
	    | (EApp ((EVar "redirectWith", _), e), _) =>
	      Option.map (fn s => "R=" ^ s) (redirect_code e)
	    | (EApp ((EVar "skip", _), e), _) =>
	      Option.map (fn n => "S=" ^ Int.toString n) (Env.int e)
	    | (EApp ((EApp ((EVar "env", _), e1), _), e2), _) =>
	      (case Env.string e1 of
		   NONE => NONE
		 | SOME s1 => Option.map (fn s2 => "E=" ^ s1 ^ ":" ^ s2)
					 (Env.string e2))

	    | _ => NONE

val cond_flag = fn (EVar "cond_nocase", _) => SOME "NC"
	    | (EVar "ornext", _) => SOME "OR"
	    | _ => NONE


val vhostsChanged = ref false

val () = Slave.registerPreHandler
	     (fn () => vhostsChanged := false)

val () = Slave.registerFileHandler (fn fs =>
				       let
					   val spl = OS.Path.splitDirFile (#file fs)
				       in
					   if String.isSuffix ".vhost" (#file spl)
					      orelse String.isSuffix ".vhost_ssl" (#file spl) then
					       (vhostsChanged := true;
						case #action fs of
						    Slave.Delete =>
						    ignore (OS.Process.system (Config.rm
									       ^ " -rf "
									       ^ Config.Apache.confDir
									       ^ "/"
									       ^ #file spl))
						  | _ =>
						    ignore (OS.Process.system (Config.cp
									       ^ " "
									       ^ #file fs
									       ^ " "
									       ^ Config.Apache.confDir
									       ^ "/"
									       ^ #file spl)))
					   else
					       ()
				       end)

val () = Slave.registerPostHandler
	 (fn () =>
	     (if !vhostsChanged then
		  Slave.shellF ([Config.Apache.reload],
				fn cl => "Error reloading Apache with " ^ cl)
	      else
		  ()))

val vhostFiles : TextIO.outstream list ref = ref []
fun write s = app (fn file => TextIO.output (file, s)) (!vhostFiles)

val rewriteEnabled = ref false

val () = Env.containerV_one "vhost"
	 ("host", Env.string)
	 (fn (env, host) =>
	     let
		 val nodes = Env.env (Env.list Env.string) (env, "WebNodes")

		 val ssl = Env.env Env.bool (env, "SSL")
		 val user = Env.env Env.string (env, "User")
		 val group = Env.env Env.string (env, "Group")
		 val docroot = Env.env Env.string (env, "DocumentRoot")
		 val sadmin = Env.env Env.string (env, "ServerAdmin")

		 val fullHost = host ^ "." ^ Domain.currentDomain ()
		 val confFile = fullHost ^ (if ssl then ".vhost_ssl" else ".vhost")
	     in
		 rewriteEnabled := false;
		 vhostFiles := map (fn node =>
				       let
					   val file = Domain.domainFile {node = node,
									 name = confFile}
				       in
					   TextIO.output (file, "<VirtualHost ");
					   TextIO.output (file, Domain.nodeIp node);
					   TextIO.output (file, ":");
					   TextIO.output (file, if ssl then
								    "443"
								else
								    "80");
					   TextIO.output (file, ">\n");
					   file
				       end)
				   nodes;
		 write "\tSuexecUserGroup ";
		 write user;
		 write " ";
		 write group;
		 write "\n\tDocumentRoot ";
		 write docroot;
		 write "\n\tServerAdmin ";
		 write sadmin;
		 write "\n"
	     end,
	  fn () => (write "</VirtualHost>\n";
		    app TextIO.closeOut (!vhostFiles)))

val () = Env.container_one "location"
	 ("prefix", Env.string)
	 (fn prefix =>
	     (write "\t<Location ";
	      write prefix;
	      write ">\n"),
	  fn () => write "\t</Location>\n")

val () = Env.container_one "directory"
	 ("directory", Env.string)
	 (fn directory =>
	     (write "\t<Directory ";
	      write directory;
	      write ">\n"),
	  fn () => write "\t</Directory>\n")

fun checkRewrite () =
    if !rewriteEnabled then
	()
    else
	(write "\tRewriteEngine on\n";
	 rewriteEnabled := true)

val () = Env.action_three "localProxyRewrite"
	 ("from", Env.string, "to", Env.string, "port", Env.int)
	 (fn (from, to, port) =>
	     (checkRewrite ();
	      write "\tRewriteRule\t";
	      write from;
	      write "\thttp://localhost:";
	      write (Int.toString port);
	      write "/";
	      write to;
	      write " [P]\n"))

val () = Env.action_two "proxyPass"
	 ("from", Env.string, "to", Env.string)
	 (fn (from, to) =>
		 (write "\tProxyPass\t";
		  write from;
		  write "\t";
		  write to;
		  write "\n"))

val () = Env.action_two "proxyPassReverse"
	 ("from", Env.string, "to", Env.string)
	 (fn (from, to) =>
		 (write "\tProxyPassReverse\t";
		  write from;
		  write "\t";
		  write to;
		  write "\n"))

val () = Env.action_three "rewriteRule"
	 ("from", Env.string, "to", Env.string, "flags", Env.list flag)
	 (fn (from, to, flags) =>
	     (checkRewrite ();
	      write "\tRewriteRule\t";
	      write from;
	      write "\t";
	      write to;
	      case flags of
		  [] => ()
		| flag::rest => (write " [";
				 write flag;
				 app (fn flag => (write ",";
						  write flag)) rest;
				 write "]");
	      write "\n"))

val () = Env.action_three "rewriteCond"
	 ("test", Env.string, "pattern", Env.string, "flags", Env.list cond_flag)
	 (fn (from, to, flags) =>
	     (checkRewrite ();
	      write "\tRewriteCond\t";
	      write from;
	      write "\t";
	      write to;
	      case flags of
		  [] => ()
		| flag::rest => (write " [";
				 write flag;
				 app (fn flag => (write ",";
						  write flag)) rest;
				 write "]");
	      write "\n"))

val () = Env.action_two "alias"
	 ("from", Env.string, "to", Env.string)
	 (fn (from, to) =>
	     (write "\tAlias\t";
	      write from;
	      write " ";
	      write to;
	      write "\n"))

val () = Env.action_two "scriptAlias"
	 ("from", Env.string, "to", Env.string)
	 (fn (from, to) =>
	     (write "\tScriptAlias\t";
	      write from;
	      write " ";
	      write to;
	      write "\n"))

val () = Env.action_two "errorDocument"
	 ("code", Env.string, "handler", Env.string)
	 (fn (code, handler) =>
	     (write "\tErrorDocument\t";
	      write code;
	      write " ";
	      write handler;
	      write "\n"))


end
Commit	Line	Data
	1	(* HCoop Domtool (http://hcoop.sourceforge.net/)
	2	* Copyright (c) 2006, Adam Chlipala
	3	*
	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.
	8	*
	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.
	13	*
	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.
	17	*)
	18
	19	(* Apache HTTPD handling *)
	20
	21	structure Apache :> APACHE = struct
	22
	23	open Ast
	24
	25	val _ = Env.type_one "proxy_port"
	26	Env.int
	27	(fn n => n > 1024)
	28
	29	val _ = Env.type_one "proxy_target"
	30	Env.string
	31	(fn s =>
	32	let
	33	fun default () = List.exists (fn s' => s = s') Config.Apache.proxyTargets
	34	in
	35	case String.fields (fn ch => ch = #":") s of
	36	["http", "//localhost", rest] =>
	37	(case String.fields (fn ch => ch = #"/") rest of
	38	port :: _ =>
	39	(case Int.fromString port of
	40	NONE => default ()
	41	\| SOME n => n > 1024 orelse default ())
	42	\| _ => default ())
	43	\| _ => default ()
	44	end)
	45
	46	val _ = Env.type_one "rewrite_arg"
	47	Env.string
	48	(CharVector.all Char.isAlphaNum)
	49
	50	fun validLocation s =
	51	size s > 0 andalso size s < 1000 andalso CharVector.all
	52	(fn ch => Char.isAlphaNum ch
	53	orelse ch = #"-"
	54	orelse ch = #"_"
	55	orelse ch = #"."
	56	orelse ch = #"/") s
	57
	58	val _ = Env.type_one "location"
	59	Env.string
	60	validLocation
	61
	62	val dl = ErrorMsg.dummyLoc
	63
	64	val _ = Main.registerDefault ("WebNodes",
	65	(TList (TBase "node", dl), dl),
	66	(fn () => (EList (map (fn s => (EString s, dl)) Config.Apache.webNodes), dl)))
	67
	68	val _ = Main.registerDefault ("SSL",
	69	(TBase "bool", dl),
	70	(fn () => (EVar "false", dl)))
	71
	72	val _ = Main.registerDefault ("User",
	73	(TBase "your_user", dl),
	74	(fn () => (EString (Domain.getUser ()), dl)))
	75
	76	val _ = Main.registerDefault ("Group",
	77	(TBase "your_group", dl),
	78	(fn () => (EString (Domain.getUser ()), dl)))
	79
	80	val _ = Main.registerDefault ("DocumentRoot",
	81	(TBase "your_path", dl),
	82	(fn () => (EString ("/home/" ^ Domain.getUser () ^ "/public_html"), dl)))
	83
	84	val _ = Main.registerDefault ("ServerAdmin",
	85	(TBase "email", dl),
	86	(fn () => (EString (Domain.getUser () ^ "@" ^ Config.defaultDomain), dl)))
	87
	88
	89	val redirect_code = fn (EVar "temp", _) => SOME "temp"
	90	\| (EVar "permanent", _) => SOME "permanent"
	91	\| (EVar "seeother", _) => SOME "seeother"
	92	\| (EVar "redir300", _) => SOME "300"
	93	\| (EVar "redir301", _) => SOME "301"
	94	\| (EVar "redir302", _) => SOME "302"
	95	\| (EVar "redir303", _) => SOME "303"
	96	\| (EVar "redir304", _) => SOME "304"
	97	\| (EVar "redir305", _) => SOME "305"
	98	\| (EVar "redir307", _) => SOME "307"
	99	\| _ => NONE
	100
	101	val flag = fn (EVar "redirect", _) => SOME "R"
	102	\| (EVar "forbidden", _) => SOME "F"
	103	\| (EVar "gone", _) => SOME "G"
	104	\| (EVar "last", _) => SOME "L"
	105	\| (EVar "chain", _) => SOME "C"
	106	\| (EVar "nosubreq", _) => SOME "NS"
	107	\| (EVar "nocase", _) => SOME "NC"
	108	\| (EVar "qsappend", _) => SOME "QSA"
	109	\| (EVar "noescape", _) => SOME "NE"
	110	\| (EVar "passthrough", _) => SOME "PT"
	111	\| (EApp ((EVar "mimeType", _), e), _) =>
	112	Option.map (fn s => "T=" ^ s) (Env.string e)
	113	\| (EApp ((EVar "redirectWith", _), e), _) =>
	114	Option.map (fn s => "R=" ^ s) (redirect_code e)
	115	\| (EApp ((EVar "skip", _), e), _) =>
	116	Option.map (fn n => "S=" ^ Int.toString n) (Env.int e)
	117	\| (EApp ((EApp ((EVar "env", _), e1), _), e2), _) =>
	118	(case Env.string e1 of
	119	NONE => NONE
	120	\| SOME s1 => Option.map (fn s2 => "E=" ^ s1 ^ ":" ^ s2)
	121	(Env.string e2))
	122
	123	\| _ => NONE
	124
	125	val cond_flag = fn (EVar "cond_nocase", _) => SOME "NC"
	126	\| (EVar "ornext", _) => SOME "OR"
	127	\| _ => NONE
	128
	129
	130	val vhostsChanged = ref false
	131
	132	val () = Slave.registerPreHandler
	133	(fn () => vhostsChanged := false)
	134
	135	val () = Slave.registerFileHandler (fn fs =>
	136	let
	137	val spl = OS.Path.splitDirFile (#file fs)
	138	in
	139	if String.isSuffix ".vhost" (#file spl)
	140	orelse String.isSuffix ".vhost_ssl" (#file spl) then
	141	(vhostsChanged := true;
	142	case #action fs of
	143	Slave.Delete =>
	144	ignore (OS.Process.system (Config.rm
	145	^ " -rf "
	146	^ Config.Apache.confDir
	147	^ "/"
	148	^ #file spl))
	149	\| _ =>
	150	ignore (OS.Process.system (Config.cp
	151	^ " "
	152	^ #file fs
	153	^ " "
	154	^ Config.Apache.confDir
	155	^ "/"
	156	^ #file spl)))
	157	else
	158	()
	159	end)
	160
	161	val () = Slave.registerPostHandler
	162	(fn () =>
	163	(if !vhostsChanged then
	164	Slave.shellF ([Config.Apache.reload],
	165	fn cl => "Error reloading Apache with " ^ cl)
	166	else
	167	()))
	168
	169	val vhostFiles : TextIO.outstream list ref = ref []
	170	fun write s = app (fn file => TextIO.output (file, s)) (!vhostFiles)
	171
	172	val rewriteEnabled = ref false
	173
	174	val () = Env.containerV_one "vhost"
	175	("host", Env.string)
	176	(fn (env, host) =>
	177	let
	178	val nodes = Env.env (Env.list Env.string) (env, "WebNodes")
	179
	180	val ssl = Env.env Env.bool (env, "SSL")
	181	val user = Env.env Env.string (env, "User")
	182	val group = Env.env Env.string (env, "Group")
	183	val docroot = Env.env Env.string (env, "DocumentRoot")
	184	val sadmin = Env.env Env.string (env, "ServerAdmin")
	185
	186	val fullHost = host ^ "." ^ Domain.currentDomain ()
	187	val confFile = fullHost ^ (if ssl then ".vhost_ssl" else ".vhost")
	188	in
	189	rewriteEnabled := false;
	190	vhostFiles := map (fn node =>
	191	let
	192	val file = Domain.domainFile {node = node,
	193	name = confFile}
	194	in
	195	TextIO.output (file, "<VirtualHost ");
	196	TextIO.output (file, Domain.nodeIp node);
	197	TextIO.output (file, ":");
	198	TextIO.output (file, if ssl then
	199	"443"
	200	else
	201	"80");
	202	TextIO.output (file, ">\n");
	203	file
	204	end)
	205	nodes;
	206	write "\tSuexecUserGroup ";
	207	write user;
	208	write " ";
	209	write group;
	210	write "\n\tDocumentRoot ";
	211	write docroot;
	212	write "\n\tServerAdmin ";
	213	write sadmin;
	214	write "\n"
	215	end,
	216	fn () => (write "</VirtualHost>\n";
	217	app TextIO.closeOut (!vhostFiles)))
	218
	219	val () = Env.container_one "location"
	220	("prefix", Env.string)
	221	(fn prefix =>
	222	(write "\t<Location ";
	223	write prefix;
	224	write ">\n"),
	225	fn () => write "\t</Location>\n")
	226
	227	val () = Env.container_one "directory"
	228	("directory", Env.string)
	229	(fn directory =>
	230	(write "\t<Directory ";
	231	write directory;
	232	write ">\n"),
	233	fn () => write "\t</Directory>\n")
	234
	235	fun checkRewrite () =
	236	if !rewriteEnabled then
	237	()
	238	else
	239	(write "\tRewriteEngine on\n";
	240	rewriteEnabled := true)
	241
	242	val () = Env.action_three "localProxyRewrite"
	243	("from", Env.string, "to", Env.string, "port", Env.int)
	244	(fn (from, to, port) =>
	245	(checkRewrite ();
	246	write "\tRewriteRule\t";
	247	write from;
	248	write "\thttp://localhost:";
	249	write (Int.toString port);
	250	write "/";
	251	write to;
	252	write " [P]\n"))
	253
	254	val () = Env.action_two "proxyPass"
	255	("from", Env.string, "to", Env.string)
	256	(fn (from, to) =>
	257	(write "\tProxyPass\t";
	258	write from;
	259	write "\t";
	260	write to;
	261	write "\n"))
	262
	263	val () = Env.action_two "proxyPassReverse"
	264	("from", Env.string, "to", Env.string)
	265	(fn (from, to) =>
	266	(write "\tProxyPassReverse\t";
	267	write from;
	268	write "\t";
	269	write to;
	270	write "\n"))
	271
	272	val () = Env.action_three "rewriteRule"
	273	("from", Env.string, "to", Env.string, "flags", Env.list flag)
	274	(fn (from, to, flags) =>
	275	(checkRewrite ();
	276	write "\tRewriteRule\t";
	277	write from;
	278	write "\t";
	279	write to;
	280	case flags of
	281	[] => ()
	282	\| flag::rest => (write " [";
	283	write flag;
	284	app (fn flag => (write ",";
	285	write flag)) rest;
	286	write "]");
	287	write "\n"))
	288
	289	val () = Env.action_three "rewriteCond"
	290	("test", Env.string, "pattern", Env.string, "flags", Env.list cond_flag)
	291	(fn (from, to, flags) =>
	292	(checkRewrite ();
	293	write "\tRewriteCond\t";
	294	write from;
	295	write "\t";
	296	write to;
	297	case flags of
	298	[] => ()
	299	\| flag::rest => (write " [";
	300	write flag;
	301	app (fn flag => (write ",";
	302	write flag)) rest;
	303	write "]");
	304	write "\n"))
	305
	306	val () = Env.action_two "alias"
	307	("from", Env.string, "to", Env.string)
	308	(fn (from, to) =>
	309	(write "\tAlias\t";
	310	write from;
	311	write " ";
	312	write to;
	313	write "\n"))
	314
	315	val () = Env.action_two "scriptAlias"
	316	("from", Env.string, "to", Env.string)
	317	(fn (from, to) =>
	318	(write "\tScriptAlias\t";
	319	write from;
	320	write " ";
	321	write to;
	322	write "\n"))
	323
	324	val () = Env.action_two "errorDocument"
	325	("code", Env.string, "handler", Env.string)
	326	(fn (code, handler) =>
	327	(write "\tErrorDocument\t";
	328	write code;
	329	write " ";
	330	write handler;
	331	write "\n"))
	332
	333
	334	end