--- parser3/src/classes/Attic/root.C	2001/04/03 08:23:06	1.55
+++ parser3/src/classes/Attic/root.C	2001/04/05 13:19:40	1.60
@@ -5,7 +5,7 @@
 
 	Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf)
 
-	$Id: root.C,v 1.55 2001/04/03 08:23:06 paf Exp $
+	$Id: root.C,v 1.60 2001/04/05 13:19:40 paf Exp $
 */
 
 #include "pa_config_includes.h"
@@ -15,6 +15,7 @@
 #include "pa_request.h"
 #include "_root.h"
 #include "pa_vint.h"
+#include "pa_sql_connection.h"
 
 static void _if(Request& r, const String& method_name, Array *params) {
 	Value& condition_code=*static_cast<Value *>(params->get(0));
@@ -41,11 +42,13 @@ static void _if(Request& r, const String
 }
 
 static void _untaint(Request& r, const String& method_name, Array *params) {
+	Pool& pool=r.pool();
+
 	const String& lang_name=r.process(*static_cast<Value *>(params->get(0))).as_string();
 	String::Untaint_lang lang=static_cast<String::Untaint_lang>(
 		untaint_lang_name2enum->get_int(lang_name));
 	if(!lang)
-		RTHROW(0, 0,
+		PTHROW(0, 0,
 			&lang_name,
 			"invalid untaint language");
 
@@ -61,6 +64,8 @@ static void _untaint(Request& r, const S
 }
 
 static void _taint(Request& r, const String& method_name, Array *params) {
+	Pool& pool=r.pool();
+
 	String::Untaint_lang lang;
 	if(params->size()==1)
 		lang=String::UL_TAINTED; // mark as simply 'tainted'. useful in table:set
@@ -69,21 +74,19 @@ static void _taint(Request& r, const Str
 		lang=static_cast<String::Untaint_lang>(
 			untaint_lang_name2enum->get_int(lang_name));
 		if(!lang)
-			RTHROW(0, 0,
+			PTHROW(0, 0,
 			&lang_name,
 			"invalid taint language");
 	}
 
 	{
-		Value *vbody=static_cast<Value *>(params->get(params->size()-1));
-		// forcing {this param type}
-		r.fail_if_junction_(true, *vbody, method_name, "body must not be junction");
+		Value& vbody=*static_cast<Value *>(params->get(params->size()-1));
+		// forcing [this param type]
+		r.fail_if_junction_(true, vbody, method_name, "body must not be junction");
 		
-		// set temporarily as-is language
-		Temp_lang temp_lang(r, String::UL_PASS_APPENDED);
 		String result(r.pool());
 		result.append(
-			r.process(*vbody).as_string(),  // process marking tainted with that lang
+			vbody.as_string(),  // process marking tainted with that lang
 			lang, true);  // force result language to specified
 		r.write_pass_lang(result);
 	}
@@ -132,6 +135,8 @@ static void _rem(Request& r, const Strin
 }
 
 static void _while(Request& r, const String& method_name, Array *params) {
+	Pool& pool=r.pool();
+
 	Value& vcondition=*static_cast<Value *>(params->get(0));
 	// forcing ^while(this param type){}
 	r.fail_if_junction_(false, vcondition, 
@@ -146,7 +151,7 @@ static void _while(Request& r, const Str
 	int endless_loop_count=0;
 	while(true) {
 		if(++endless_loop_count>=1973) // endless loop?
-			RTHROW(0, 0,
+			PTHROW(0, 0,
 				&method_name,
 				"endless loop detected");
 
@@ -190,7 +195,7 @@ static void _for(Request& r, const Strin
 	int endless_loop_count=0;
 	for(int i=from; i<=to; i++) {
 		if(++endless_loop_count>=2001) // endless loop?
-			RTHROW(0, 0,
+			PTHROW(0, 0,
 				&method_name,
 				"endless loop detected");
 		vint->set_int(i);
@@ -269,6 +274,62 @@ static void _sign(Request& r, const Stri
 	double_one_op(r, method_name, params,	&sign);
 }
 
+/// ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s}
+/**
+	@test make params not Array but something with useful method for extracting,
+	with typecast and junction/not test
+*/
+static void _connect(Request& r, const String& method_name, Array *params) {
+	Pool& pool=r.pool();
+
+	Value& url=*static_cast<Value *>(params->get(0));
+	r.fail_if_junction_(true, url, 
+		method_name, "url must not be junction");
+
+	Value& body_code=*static_cast<Value *>(params->get(1));
+	r.fail_if_junction_(false, body_code, 
+		method_name, "body must be junction");
+
+	// connect
+	SQL_Connection& connection=SQL_driver_manager->get_connection(
+		url.as_string(), r.protocol2library);
+
+	Exception rethrow_me;
+	// remember/set current connection
+	SQL_Connection *saved_connection=r.connection;
+	r.connection=&connection;
+	// execute body
+	bool body_failed=false;  
+	PTRY
+		r.write_assign_lang(r.process(body_code));
+	PCATCH(e) { // connect/process problem
+		rethrow_me=e;  body_failed=true; 
+	}
+	PEND_CATCH
+
+	bool finalizer_failed=false;
+	PTRY
+		// FINALLY
+		if(body_failed)
+			connection.rollback();
+		else
+			connection.commit();
+	PCATCH(e) { // commit/rollback problem
+		rethrow_me=e;  finalizer_failed=true; 
+	}
+	PEND_CATCH
+
+	// close connection [cache it]
+	connection.close();
+	// recall current connection from remembered
+	r.connection=saved_connection;
+
+	if(body_failed || finalizer_failed) // were there an exception for us to rethrow?
+		PTHROW(rethrow_me.type(), rethrow_me.code(),
+			rethrow_me.problem_source(),
+			rethrow_me.comment());
+}
+
 // initialize
 
 void initialize_root_class(Pool& pool, VStateless_class& vclass) {
@@ -318,4 +379,11 @@ void initialize_root_class(Pool& pool, V
 
 	// ^sign(expr)
 	vclass.add_native_method("sign", Method::CT_ANY, _sign, 1, 1);
+
+	
+	// connect
+
+	// ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s}
+	vclass.add_native_method("connect", Method::CT_ANY, _connect, 2, 2);
+
 }