--- parser3/src/classes/table.C 2001/04/05 13:27:11 1.55 +++ parser3/src/classes/table.C 2001/04/15 13:12:18 1.61 @@ -5,31 +5,33 @@ Author: Alexander Petrosyan (http://design.ru/paf) - $Id: table.C,v 1.55 2001/04/05 13:27:11 paf Exp $ + $Id: table.C,v 1.61 2001/04/15 13:12:18 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_vint.h" #include "pa_sql_connection.h" +#include "pa_dir.h" // global var VStateless_class *table_class; // methods -static void _set(Request& r, const String& method_name, Array *params) { +static void _set(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); // data is last parameter - Value *vdata=static_cast(params->get(params->size()-1)); - // forcing {this body type} - r.fail_if_junction_(false, *vdata, method_name, "body must be junction"); + 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(); + const String& data=r.process(vdata).as_string(); size_t pos_after=0; // parse columns @@ -65,16 +67,14 @@ static void _set(Request& r, const Strin static_cast(r.self)->set_table(table); } -static void _load(Request& r, const String& method_name, Array *params) { +static void _load(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); // filename is last parameter - Value *vfilename=static_cast(params->get(params->size()-1)); - // forcing [this file name type] - r.fail_if_junction_(true, *vfilename, - method_name, "file name must not be junction"); + 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())); + char *data=file_read_text(pool, r.absolute(vfilename.as_string())); // parse columns Array *columns; @@ -116,12 +116,10 @@ static void _load(Request& r, const Stri static_cast(r.self)->set_table(table); } -static void _save(Request& r, const String& method_name, Array *params) { +static void _save(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); - Value *vtable_name=static_cast(params->get(params->size()-1)); - // forcing this body type] - r.fail_if_junction_(true, *vtable_name, - method_name, "file name must not be junction"); + Value& vtable_name=params->get_no_junction(params->size()-1, + "file name must not be code"); Table& table=static_cast(r.self)->table(); @@ -161,27 +159,27 @@ static void _save(Request& r, const Stri } // write - file_write(pool, r.absolute(vtable_name->as_string()), + file_write(pool, r.absolute(vtable_name.as_string()), sdata.cstr(), sdata.size(), true); } -static void _count(Request& r, const String&method_name, Array *) { +static void _count(Request& r, const String&, MethodParams *) { Pool& pool=r.pool(); Value& value=*new(pool) VInt(pool, static_cast(r.self)->table().size()); r.write_no_lang(value); } -static void _line(Request& r, const String& method_name, Array *) { +static void _line(Request& r, const String& method_name, MethodParams *) { Pool& pool=r.pool(); Value& value=*new(pool) VInt(pool, 1+static_cast(r.self)->table().current()); r.write_no_lang(value); } -static void _offset(Request& r, const String& method_name, Array *params) { +static void _offset(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); Table& table=static_cast(r.self)->table(); if(params->size()) - table.shift((int)r.process(*static_cast(params->get(0))).as_double()); + table.shift((int)r.process(params->get(0)).as_double()); else { Value& value=*new(pool) VInt(pool, table.current()); r.write_no_lang(value); @@ -189,13 +187,10 @@ static void _offset(Request& r, const St } /// @test $a.menu{ $a[123] } -static void _menu(Request& r, const String& method_name, Array *params) { - Value& body_code=*static_cast(params->get(0)); - // forcing ^menu{this param type} - r.fail_if_junction_(false, body_code, - method_name, "body must be junction"); +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?static_cast(params->get(1)):0; + Value *delim_code=params->size()==2?¶ms->get(1):0; Table& table=static_cast(r.self)->table(); bool need_delim=false; @@ -215,15 +210,12 @@ static void _menu(Request& r, const Stri table.set_current(saved_current); } -static void _empty(Request& r, const String& method_name, Array *params) { +static void _empty(Request& r, const String& method_name, MethodParams *params) { Table& table=static_cast(r.self)->table(); - if(table.size()==0) { - Value& value=r.process(*static_cast(params->get(0))); - r.write_pass_lang(value); - } else if(params->size()==2) { - Value& value=r.process(*static_cast(params->get(1))); - r.write_pass_lang(value); - } + if(table.size()==0) + r.write_pass_lang(params->get(0)); + else if(params->size()==2) + r.write_pass_lang(params->get(1)); } struct Record_info { @@ -242,7 +234,7 @@ static void store_column_item_to_hash(Ar value=new(*ri.pool) VUnknown(*ri.pool); ri.hash->put(column_name, value); } -static void _record(Request& r, const String& method_name, Array *params) { +static void _record(Request& r, const String& method_name, MethodParams *params) { Table& table=static_cast(r.self)->table(); if(const Array *columns=table.columns()) { Pool& pool=r.pool(); @@ -277,19 +269,12 @@ static int sort_cmp_double(const void *a else return 0; } -static void _sort(Request& r, const String& method_name, Array *params) { - Value& key_maker=*(Value *)params->get(0); - // forcing ^sort{this} ^sort(or this) param type - r.fail_if_junction_(false, key_maker, method_name, "key-maker must be junction"); - - bool reverse; - if(params->size()==2) { // ..[asc|desc] - Value& order=*(Value *)params->get(1); - // forcing ..[this param-type] - r.fail_if_junction_(true, order, method_name, "order must not be junction"); - reverse=order.as_string()=="desc"; - } else - reverse=false; +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(r.self)->table(); @@ -305,7 +290,7 @@ static void _sort(Request& r, const Stri for(i=0; i(r.self); Table& table=vtable.table(); vtable.last_locate_was_successful=table.locate( - static_cast(params->get(0))->as_string(), - static_cast(params->get(1))->as_string()); + params->get(0).as_string(), + params->get(1).as_string()); } -static void _found(Request& r, const String& method_name, Array *params) { - if(static_cast(r.self)->last_locate_was_successful) { - Value& then_code=*static_cast(params->get(0)); - // forcing ^found{this param type} - r.fail_if_junction_(false, then_code, - method_name, "found-parameter must be junction"); - r.write_pass_lang(r.process(then_code)); - } else if(params->size()==2) { - Value& else_code=*static_cast(params->get(1)); - // forcing ^found{this param type} - r.fail_if_junction_(false, else_code, - method_name, "not found-parameter must be junction"); - r.write_pass_lang(r.process(else_code)); - } +static void _found(Request& r, const String& method_name, MethodParams *params) { + if(static_cast(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 _flip(Request& r, const String& method_name, Array *params) { +static void _flip(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); VTable& vtable=*static_cast(r.self); @@ -371,14 +347,11 @@ static void _flip(Request& r, const Stri vtable.set_table(new_table); } -static void _append(Request& r, const String& method_name, Array *params) { +static void _append(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); // data is last parameter - Value *value=static_cast(params->get(0)); - // forcing {this body type} - r.fail_if_junction_(false, *value, method_name, "body must be junction"); - - const String& string=r.process(*value).as_string(); + const String& string= + r.process(params->get_junction(0, "body must be code")).as_string(); // parse cells Array& row=*new(pool) Array(pool); @@ -387,14 +360,10 @@ static void _append(Request& r, const St static_cast(r.self)->table()+=&row; } -static void _join(Request& r, const String& method_name, Array *params) { +static void _join(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); - Value *value=static_cast(params->get(0)); - // forcing [this table ref type] - r.fail_if_junction_(true, *value, method_name, "table ref must not be junction"); - - Table *maybe_src=value->get_table(); + Table *maybe_src=params->get_no_junction(0, "table ref must not be code").get_table(); if(!maybe_src) PTHROW(0, 0, &method_name, @@ -423,8 +392,8 @@ static void _join(Request& r, const Stri } } -/// ^table:sql{query}[(count[;offset])] @test limit/offset -static void _sql(Request& r, const String& method_name, Array *params) { +/// ^table:sql{query}[(count[;offset])] +static void _sql(Request& r, const String& method_name, MethodParams *params) { Pool& pool=r.pool(); if(!r.connection) @@ -432,23 +401,17 @@ static void _sql(Request& r, const Strin &method_name, "without connect"); - Value& statement=*static_cast(params->get(0)); - // forcing {this query param type} - r.fail_if_junction_(false, statement, method_name, "statement must be junction"); + Value& statement=params->get_junction(0, "statement must be code"); ulong limit=0; if(params->size()>1) { - Value& limit_code=*static_cast(params->get(1)); - // forcing (this limit param type) - r.fail_if_junction_(false, limit_code, method_name, "limit must be expression"); + 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=*static_cast(params->get(2)); - // forcing (this limit param type) - r.fail_if_junction_(false, offset_code, method_name, "offset must be expression"); + Value& offset_code=params->get_junction(2, "offset must be expression"); offset=(ulong)r.process(offset_code).as_double(); } @@ -505,8 +468,84 @@ static void _sql(Request& r, const Strin static_cast(r.self)->set_table(table); } -// initialize +/// ^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"); + + const String *regexp; + pcre *regexp_code; + int ovecsize; + int *ovector; + if(params->size()>1) { + regexp=¶ms->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, + pcre_tables); + + if(!regexp_code) + PTHROW(0, 0, + ®exp->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(r.self)->set_table(table); +} + +// initialize void initialize_table_class(Pool& pool, VStateless_class& vclass) { // ^table:set{data} // ^table:set[nameless]{data} @@ -563,4 +602,8 @@ void initialize_table_class(Pool& pool, // ^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); }