--- parser3/src/classes/table.C	2001/03/12 20:36:52	1.6
+++ parser3/src/classes/table.C	2001/04/20 14:18:30	1.63
@@ -1,43 +1,80 @@
-/*
-	Parser
+/** @file
+	Parser: @b table parser class.
+
 	Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com)
+
 	Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf)
 
-	$Id: table.C,v 1.6 2001/03/12 20:36:52 paf Exp $
+	$Id: table.C,v 1.63 2001/04/20 14:18:30 paf Exp $
 */
 
+#include "pa_config_includes.h"
+
+#include "pcre.h"
+
+#include "pa_common.h"
 #include "pa_request.h"
 #include "_table.h"
 #include "pa_vtable.h"
-#include "pa_common.h"
 #include "pa_vint.h"
+#include "pa_sql_connection.h"
+#include "pa_dir.h"
 
 // global var
 
-VClass *table_class;
+VStateless_class *table_class;
 
 // methods
+static void _set(Request& r, const String& method_name, MethodParams *params) {
+	Pool& pool=r.pool();
+	// data is last parameter
+	Value& vdata=params->get_junction(params->size()-1, "body must be code");
 
+	Temp_lang temp_lang(r, String::UL_PASS_APPENDED);
+	const String& data=r.process(vdata).as_string();
 
-// TODO: проверить ^set в ^menu & co
+	size_t pos_after=0;
+	// parse columns
+	Array *columns;
+	if(params->size()==2) {
+		columns=0;
+	} else {
+		columns=new(pool) Array(pool);
 
-static void set_or_load(
-						Request& r, 
-						const String& method_name, Array *params, 
-						bool is_load) {
+		Array head(pool);
+		data.split(head, &pos_after, "\n", 1, String::UL_CLEAN, 1);
+		if(head.size())
+			head.get_string(0)->split(*columns, 0, "\t", 1, String::UL_CLEAN);
+	}
+
+	Table& table=*new(pool) Table(pool, &method_name, columns);
+	// parse cells
+	Array rows(pool);
+	data.split(rows, &pos_after, "\n", 1, String::UL_CLEAN);
+	int size=rows.quick_size();
+	for(int i=0; i<size; i++) {
+		Array& row=*new(pool) Array(pool);
+		const String& string=*rows.quick_get_string(i);
+		// remove empty lines
+		if(!string.size())
+			continue;
+
+		string.split(row, 0, "\t", 1, String::UL_CLEAN);
+		table+=&row;
+	}
+
+	// replace any previous table value
+	static_cast<VTable *>(r.self)->set_table(table);
+}
+
+static void _load(Request& r, const String& method_name, MethodParams *params) {
 	Pool& pool=r.pool();
-	// data is last parameter
-	Value *vdata=static_cast<Value *>(params->get(params->size()-1));
-	// forcing
-	//	[this param type] 
-	//	[this param type] 
-	r.fail_if_junction_(true, *vdata, 
-		method_name, "body must not be a junction");
-
-	// data or file_name
-	char *data_or_filename=vdata->as_string().cstr();
-	// data
-	char *data=is_load?file_read(pool, r.absolute(data_or_filename)):data_or_filename;
+	// filename is last parameter
+	Value& vfilename=params->get_no_junction(params->size()-1, 
+		"file name must not be code");
+
+	// loading text
+	char *data=file_read_text(pool, r.absolute(vfilename.as_string()));
 
 	// parse columns
 	Array *columns;
@@ -54,19 +91,21 @@ static void set_or_load(
 		if(char *row_chars=getrow(&data)) 
 			do {
 				String *name=new(pool) String(pool);
-				name->APPEND(lsplit(&row_chars, '\t'), 0, file, line++);
+				name->APPEND_TAINTED(lsplit(&row_chars, '\t'), 0, file, line++);
 				*columns+=name;
 			} while(row_chars);
 	}
 
 	// parse cells
-	Table& table=*new(pool) Table(pool, method_name, columns);
+	Table& table=*new(pool) Table(pool, &method_name, columns);
 	char *row_chars;
 	while(row_chars=getrow(&data)) {
+		if(!*row_chars) // remove empty lines
+			continue;
 		Array *row=new(pool) Array(pool);
 		while(char *cell_chars=lsplit(&row_chars, '\t')) {
 			String *cell=new(pool) String(pool);
-			cell->APPEND(cell_chars, 0, file, line);
+			cell->APPEND_TAINTED(cell_chars, 0, file, line);
 			*row+=cell;
 		}
 		line++;
@@ -74,58 +113,497 @@ static void set_or_load(
 	};
 
 	// replace any previous table value
-	r.self->as_vtable().set_table(table);
+	static_cast<VTable *>(r.self)->set_table(table);
+}
+
+static void _save(Request& r, const String& method_name, MethodParams *params) {
+	Pool& pool=r.pool();
+	Value& vtable_name=params->get_no_junction(params->size()-1, 
+		"file name must not be code");
+
+	Table& table=static_cast<VTable *>(r.self)->table();
+
+	String sdata(pool);
+	if(params->size()==1) { // not nameless=named output
+		// write out names line
+		if(table.columns()) { // named table
+			for(int column=0; column<table.columns()->size(); column++) {
+				if(column)
+					sdata.APPEND_CONST("\t");
+				sdata.append(*static_cast<String *>(table.columns()->quick_get(column)), 
+					String::UL_TABLE);
+			}
+		} else { // nameless table
+			int lsize=table.size()?static_cast<Array *>(table.get(0))->size():0;
+			if(lsize)
+				for(int column=0; column<lsize; column++) {
+					char *cindex_tab=(char *)malloc(MAX_NUMBER);
+					snprintf(cindex_tab, MAX_NUMBER, "%d\t", column);
+					sdata.APPEND_CONST(cindex_tab);
+				}
+			else
+				sdata.APPEND_CONST("empty nameless table");
+		}
+		sdata.APPEND_CONST("\n");
+	}
+	// data lines
+	for(int index=0; index<table.size(); index++) {
+		Array *row=static_cast<Array *>(table.quick_get(index));
+		for(int column=0; column<row->size(); column++) {
+			if(column)
+				sdata.APPEND_CONST("\t");
+			sdata.append(*static_cast<String *>(row->quick_get(column)), 
+				String::UL_TABLE);
+		}
+		sdata.APPEND_CONST("\n");
+	}
+
+	// write
+	file_write(pool, r.absolute(vtable_name.as_string()), 
+		sdata.cstr(), sdata.size(), true);
+}
+
+static void _count(Request& r, const String&, MethodParams *) {
+	Pool& pool=r.pool();
+	Value& value=*new(pool) VInt(pool, static_cast<VTable *>(r.self)->table().size());
+	r.write_no_lang(value);
+}
+
+static void _line(Request& r, const String& method_name, MethodParams *) {
+	Pool& pool=r.pool();
+	Value& value=*new(pool) VInt(pool, 1+static_cast<VTable *>(r.self)->table().current());
+	r.write_no_lang(value);
+}
+
+static void _offset(Request& r, const String& method_name, MethodParams *params) {
+	Pool& pool=r.pool();
+	Table& table=static_cast<VTable *>(r.self)->table();
+	if(params->size())
+		table.shift((int)r.process(params->get(0)).as_double());
+	else {
+		Value& value=*new(pool) VInt(pool, table.current());
+		r.write_no_lang(value);
+	}
+}
+
+/// @test $a.menu{ $a[123] }
+static void _menu(Request& r, const String& method_name, MethodParams *params) {
+	Value& body_code=params->get_junction(0, "body must be code");
+	
+	Value *delim_code=params->size()==2?&params->get(1):0;
+
+	Table& table=static_cast<VTable *>(r.self)->table();
+	bool need_delim=false;
+	int saved_current=table.current();
+	for(int row=0; row<table.size(); row++) {
+		table.set_current(row);
+
+		Value& processed_body=r.process(body_code);
+		if(delim_code) { // delimiter set?
+			const String *string=processed_body.get_string();
+			if(need_delim && string && string->size()) // need delim & iteration produced string?
+				r.write_pass_lang(r.process(*delim_code));
+			need_delim=true;
+		}
+		r.write_pass_lang(processed_body);
+	}
+	table.set_current(saved_current);
+}
+
+static void _empty(Request& r, const String& method_name, MethodParams *params) {
+	Table& table=static_cast<VTable *>(r.self)->table();
+	if(table.size()==0)
+		r.write_pass_lang(r.process(params->get(0)));
+	else if(params->size()>1)
+		r.write_pass_lang(r.process(params->get(1)));
+}
+
+struct Record_info {
+	Pool *pool;
+	Table *table;
+	Hash *hash;
+};
+static void store_column_item_to_hash(Array::Item *item, void *info) {
+	Record_info& ri=*static_cast<Record_info *>(info);
+	String& column_name=*static_cast<String *>(item);
+	const String *column_item=ri.table->item(column_name);
+	Value *value;
+	if(column_item)
+		value=new(*ri.pool) VString(*column_item);
+	else
+		value=new(*ri.pool) VUnknown(*ri.pool);
+	ri.hash->put(column_name, value);
+}
+static void _record(Request& r, const String& method_name, MethodParams *params) {
+	Table& table=static_cast<VTable *>(r.self)->table();
+	if(const Array *columns=table.columns()) {
+		Pool& pool=r.pool();
+		Value& value=*new(pool) VHash(pool);
+		Record_info record_info={&pool, &table, value.get_hash()};
+		columns->for_each(store_column_item_to_hash, &record_info);
+		
+		r.write_no_lang(value);
+	}
+}
+
+struct Seq_item {
+	Array *row;
+	union {
+		char *c_str;
+		double d;
+	} value;
+};
+static int sort_cmp_string(const void *a, const void *b) {
+	return strcmp(
+		static_cast<const Seq_item *>(a)->value.c_str, 
+		static_cast<const Seq_item *>(b)->value.c_str
+	);
+}
+static int sort_cmp_double(const void *a, const void *b) {
+	double va=static_cast<const Seq_item *>(a)->value.d;
+	double vb=static_cast<const Seq_item *>(b)->value.d;
+	if(va<vb)
+		return -1;
+	else if(va>vb)
+		return +1;
+	else 
+		return 0;
 }
+static void _sort(Request& r, const String& method_name, MethodParams *params) {
+	Value& key_maker=params->get_junction(0, "key-maker must be code");
 
+	bool reverse=params->size()==2/*..[asc|desc]*/?
+		reverse=params->get_no_junction(1, "order must not be code").as_string()=="desc":
+		false;
+
+	Table& table=static_cast<VTable *>(r.self)->table();
+
+	// anything to sort?
+	if(!table.size())
+		return;
+
+	Seq_item *seq=(Seq_item *)malloc(sizeof(Seq_item)*table.size());
+	int i;
+
+	// calculate key values
+	bool key_values_are_strings=true;
+	for(i=0; i<table.size(); i++) {
+		table.set_current(i);
+		// calculate key value
+		seq[i].row=(MethodParams *)table.get(i);
+		Value& value=*r.process(key_maker).as_expr_result(true/*return string as-is*/);
+		if(i==0) // determining key values type by first one
+			key_values_are_strings=value.is_string();
+
+		if(key_values_are_strings)
+			seq[i].value.c_str=value.as_string().cstr();
+		else
+			seq[i].value.d=value.as_double();
+	}
+	// sort keys
+	_qsort(seq, table.size(), sizeof(Seq_item), 
+		key_values_are_strings?sort_cmp_string:sort_cmp_double);
+
+	// reorder table as they require in 'seq'
+	for(i=0; i<table.size(); i++)
+		table.put(i, seq[reverse?table.size()-1-i:i].row);
+
+	// reset 'current'
+	table.set_current(0);
+}
 
-static void _set(Request& r, const String& method_name, Array *params) {
-	set_or_load(r, method_name, params, false);
+static void _locate(Request& r, const String& method_name, MethodParams *params) {
+	VTable& vtable=*static_cast<VTable *>(r.self);
+	Table& table=vtable.table();
+	vtable.last_locate_was_successful=table.locate(
+		params->get(0).as_string(),
+		params->get(1).as_string());
 }
 
-static void _load(Request& r, const String& method_name, Array *params) {
-	set_or_load(r, method_name, params, true);
+static void _found(Request& r, const String& method_name, MethodParams *params) {
+	if(static_cast<VTable *>(r.self)->last_locate_was_successful)
+		r.write_pass_lang(r.process(params->get_junction(0, "found-parameter must be code")));
+	else if(params->size()==2)
+		r.write_pass_lang(r.process(params->get_junction(0, "else-parameter must be code")));
 }
 
-static void _count(Request& r, const String&, Array *) {
+static void _flip(Request& r, const String& method_name, MethodParams *params) {
 	Pool& pool=r.pool();
-	Value& value=*new(pool) VInt(pool, r.self->as_vtable().table().size());
-	r.wcontext->write(value, String::Untaint_lang::NO /*always object, not string*/);
+	VTable& vtable=*static_cast<VTable *>(r.self);
+
+	Table& old_table=*vtable.get_table();
+	Table& new_table=*new(pool) Table(pool, &method_name, 0/*nameless*/);
+	if(old_table.size())
+		if(int old_cols=old_table.at(0).size()) 
+			for(int column=0; column<old_cols; column++) {
+				Array& new_row=*new(pool) Array(pool, old_table.size());
+				for(int i=0; i<old_table.size(); i++) {
+					const Array& old_row=old_table.at(i);
+					new_row+=column<old_row.size()?old_row.get(column):empty_string;
+				}
+				new_table+=&new_row;
+			}
+
+	vtable.set_table(new_table);
 }
 
-static void _line(Request& r, const String&, Array *) {
+static void _append(Request& r, const String& method_name, MethodParams *params) {
 	Pool& pool=r.pool();
-	Value& value=*new(pool) VInt(pool, 1+r.self->as_vtable().table().get_current());
-	r.wcontext->write(value, String::Untaint_lang::NO /*always object, not string*/);
+	// data is last parameter
+	const String& string=
+		r.process(params->get_junction(0, "body must be code")).as_string();
+
+	// parse cells
+	Array& row=*new(pool) Array(pool);
+	string.split(row, 0, "\t", 1, String::UL_CLEAN);
+
+	static_cast<VTable *>(r.self)->table()+=&row;
 }
 
-static void _offset(Request& r, const String&, Array *params) {
+static void _join(Request& r, const String& method_name, MethodParams *params) {
 	Pool& pool=r.pool();
-	Table& table=r.self->as_vtable().table();
-	if(params->size()) {
-		if(int size=table.size()) {
-			int offset=static_cast<int>(
-				r.process(*static_cast<Value *>(params->get(0))).get_double());
-			table.set_current((table.get_current()+offset+size)%size);
+
+	Table *maybe_src=params->get_no_junction(0, "table ref must not be code").get_table();
+	if(!maybe_src)
+		PTHROW(0, 0,
+			&method_name,
+			"source is not a table");
+
+	Table& src=*maybe_src;
+	Table& dest=static_cast<VTable *>(r.self)->table();
+	if(&src == &dest)
+		PTHROW(0, 0,
+			&method_name,
+			"source and destination are same table");
+
+	if(const Array *dest_columns=dest.columns()) { // dest is named
+		int saved_src_current=src.current();
+		for(int src_row=0; src_row<src.size(); src_row++) {
+			src.set_current(src_row);
+			Array& dest_row=*new(pool) Array(pool);
+			for(int dest_column=0; dest_column<dest_columns->size(); dest_column++) 
+				dest_row+=src.item(*dest_columns->get_string(dest_column));
+			dest+=&dest_row;
+		}
+		src.set_current(saved_src_current);
+	} else { // dest is nameless
+		for(int src_row=0; src_row<src.size(); src_row++)
+			dest+=&src.at(src_row);
+	}
+}
+
+/// ^table:sql{query}[(count[;offset])]
+static void _sql(Request& r, const String& method_name, MethodParams *params) {
+	Pool& pool=r.pool();
+
+	if(!r.connection)
+		PTHROW(0, 0,
+			&method_name,
+			"without connect");
+
+	Value& statement=params->get_junction(0, "statement must be code");
+
+	ulong limit=0;
+	if(params->size()>1) {
+		Value& limit_code=params->get_junction(1, "limit must be expression");
+		limit=(uint)r.process(limit_code).as_double();
+	}
+
+	ulong offset=0;
+	if(params->size()>2) {
+		Value& offset_code=params->get_junction(2, "offset must be expression");
+		offset=(ulong)r.process(offset_code).as_double();
+	}
+
+	Temp_lang temp_lang(r, String::UL_SQL);
+	const String& statement_string=r.process(statement).as_string();
+	const char *statement_cstr=
+		statement_string.cstr(String::UL_UNSPECIFIED, r.connection);
+	unsigned int sql_column_count; SQL_Driver::Cell *sql_columns;
+	unsigned long sql_row_count; SQL_Driver::Cell **sql_rows;
+	bool need_rethrow=false; Exception rethrow_me;
+	PTRY {
+		r.connection->query(
+			statement_cstr, offset, limit,
+			&sql_column_count, &sql_columns,
+			&sql_row_count, &sql_rows);
+	}
+	PCATCH(e) { // connect/process problem
+		rethrow_me=e;  need_rethrow=true;
+	}
+	PEND_CATCH
+	if(need_rethrow)
+		PTHROW(rethrow_me.type(), rethrow_me.code(),
+			&statement_string, // setting more specific source [were url]
+			rethrow_me.comment());
+	
+	Array& table_columns=*new(pool) Array(pool);
+	for(unsigned int i=0; i<sql_column_count; i++) {
+		String& table_column=*new(pool) String(pool);
+		table_column.APPEND_TAINTED(
+			(const char *)sql_columns[i].ptr, sql_columns[i].size,
+			statement_cstr, 0);
+		table_columns+=&table_column;
+	}
+
+	Table& table=*new(pool) Table(pool, &method_name, &table_columns);
+
+	{
+		for(unsigned long r=0; r<sql_row_count; r++) {
+			SQL_Driver::Cell *sql_cells=sql_rows[r];
+			Array& table_row=*new(pool) Array(pool);
+			
+			for(unsigned int i=0; i<sql_column_count; i++) {
+				String& table_cell=*new(pool) String(pool);
+				table_cell.APPEND_TAINTED(
+					(const char *)sql_cells[i].ptr, sql_cells[i].size,
+					statement_cstr, r);
+				table_row+=&table_cell;
+			}
+			table+=&table_row;
 		}
-	} else {
-		Value& value=*new(pool) VInt(pool, table.get_current());
-		r.wcontext->write(value, String::Untaint_lang::NO /*always object, not string*/);
 	}
+
+	// replace any previous table value
+	static_cast<VTable *>(r.self)->set_table(table);
 }
 
-void initialize_table_class(Pool& pool, VClass& vclass) {
-	// ^table.set[data]  ^table.set[nameless;data]
-	vclass.add_native_method("set", _set, 1, 2);
+/// ^table:dir[path]
+/// ^table:dir[path][regexp]
+static void _dir(Request& r, const String& method_name, MethodParams *params) {
+	Pool& pool=r.pool();
+
+	Value& relative_path=params->get_no_junction(0, "path must not be code");
 
-	// ^table.load[file]  ^table.load[nameless;file]
-	vclass.add_native_method("load", _load, 1, 2);
+	const String *regexp;
+	pcre *regexp_code;
+	int ovecsize;
+	int *ovector;
+	if(params->size()>1) {
+		regexp=&params->get_no_junction(1, "regexp must not be code").as_string();
+
+		const char *pattern=regexp->cstr(String::UL_AS_IS);
+		const char *errptr;
+		int erroffset;
+		regexp_code=pcre_compile(pattern, PCRE_EXTRA | PCRE_DOTALL, 
+			&errptr, &erroffset,
+			r.pcre_tables);
+
+		if(!regexp_code)
+			PTHROW(0, 0,
+				&regexp->mid(erroffset, regexp->size()),
+				"regular expression syntax error - %s", errptr);
+
+		ovector=(int *)malloc(sizeof(int)*(ovecsize=(1/*match*/)*3));
+	} else 
+		regexp_code=0;
+
+
+	const char* absolute_path_cstr=r.absolute(relative_path.as_string())
+		.cstr(String::UL_FILE_NAME);
+
+	Array& columns=*new(pool) Array(pool);
+	columns+=new(pool) String(pool, "name");	
+	Table& table=*new(pool) Table(pool, &method_name, &columns);
+
+	LOAD_DIR(absolute_path_cstr, 
+		size_t file_name_size=strlen(ffblk.ff_name);
+		bool suits=true;
+		if(regexp_code) {
+			int exec_result=pcre_exec(regexp_code, 0,
+				ffblk.ff_name, file_name_size, 0,
+				0, ovector, ovecsize);
+			
+			if(exec_result==PCRE_ERROR_NOMATCH)
+				suits=false;
+			else if(exec_result<0) {
+				(*pcre_free)(regexp_code);
+				PTHROW(0, 0,
+					regexp,
+					"regular expression execute (%d)", 
+						exec_result);
+			}
+		}
+
+		if(suits) {
+			char *file_name_cstr=(char *)r.malloc(file_name_size);
+			memcpy(file_name_cstr, ffblk.ff_name, file_name_size);
+			String &file_name=*new(pool) String(pool);
+			file_name.APPEND_TAINTED(file_name_cstr, file_name_size, 
+				method_name.origin().file, method_name.origin().line);
+		
+			Array& row=*new(pool) Array(pool);
+			row+=&file_name;
+			table+=&row;
+		}
+	);
+
+	if(regexp_code)
+		(*pcre_free)(regexp_code);
+
+	// replace any previous table value
+	static_cast<VTable *>(r.self)->set_table(table);
+}
+
+// initialize
+void initialize_table_class(Pool& pool, VStateless_class& vclass) {
+	// ^table:set{data}
+	// ^table:set[nameless]{data}
+	vclass.add_native_method("set", Method::CT_DYNAMIC, _set, 1, 2);
+
+	// ^table:load[file]  
+	// ^table:load[nameless;file]
+	vclass.add_native_method("load", Method::CT_DYNAMIC, _load, 1, 2);
+
+	// ^table.save[file]  
+	// ^table.save[nameless;file]
+	vclass.add_native_method("save", Method::CT_DYNAMIC, _save, 1, 2);
 
 	// ^table.count[]
-	vclass.add_native_method("count", _count, 0, 0);
+	vclass.add_native_method("count", Method::CT_DYNAMIC, _count, 0, 0);
 
 	// ^table.line[]
-	vclass.add_native_method("line", _line, 0, 0);
+	vclass.add_native_method("line", Method::CT_DYNAMIC, _line, 0, 0);
+
+	// ^table.offset[]  
+	// ^table.offset[offset]
+	vclass.add_native_method("offset", Method::CT_DYNAMIC, _offset, 0, 1);
+
+	// ^table.menu{code}  
+	// ^table.menu{code}[delim]
+	vclass.add_native_method("menu", Method::CT_DYNAMIC, _menu, 1, 2);
+
+	// ^table.empty{code-when-empty}  
+	// ^table.empty{code-when-empty}{code-when-not}
+	vclass.add_native_method("empty", Method::CT_DYNAMIC, _empty, 1, 2);
+
+	// ^table.record[]
+	vclass.add_native_method("record", Method::CT_DYNAMIC, _record, 0, 0);
+
+	// ^table.sort{string-key-maker} ^table.sort{string-key-maker}[asc|desc]
+	// ^table.sort(numeric-key-maker) ^table.sort(numeric-key-maker)[asc|desc]
+	vclass.add_native_method("sort", Method::CT_DYNAMIC, _sort, 1, 2);
+
+	// ^table.locate[field;value]
+	vclass.add_native_method("locate", Method::CT_DYNAMIC, _locate, 2, 2);
+	// ^table.found{when-found}
+	// ^table.found{when-found}{when-not-found}
+	vclass.add_native_method("found", Method::CT_DYNAMIC, _found, 1, 2);
 
-	// ^table.offset[]  ^table.offset[offset]
-	vclass.add_native_method("offset", _offset, 0, 1);
-}	
+	// ^table.flip[]
+	vclass.add_native_method("flip", Method::CT_DYNAMIC, _flip, 0, 0);
+
+	// ^table.append{r{tab}e{tab}c{tab}o{tab}r{tab}d}
+	vclass.add_native_method("append", Method::CT_DYNAMIC, _append, 1, 1);
+
+	// ^table.join[table]
+	vclass.add_native_method("join", Method::CT_DYNAMIC, _join, 1, 1);
+
+
+	// ^table:sql[query][(count[;offset])]
+	vclass.add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 3);
+
+	// ^table:dir[path]
+	// ^table:dir[path][regexp]
+	vclass.add_native_method("dir", Method::CT_DYNAMIC, _dir, 1, 2);
+}