--- parser3/src/classes/table.C 2001/04/05 18:22:57 1.56 +++ parser3/src/classes/table.C 2001/04/06 13:41:29 1.59 @@ -5,16 +5,20 @@ Author: Alexander Petrosyan (http://design.ru/paf) - $Id: table.C,v 1.56 2001/04/05 18:22:57 paf Exp $ + $Id: table.C,v 1.59 2001/04/06 13:41:29 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 @@ -423,7 +427,7 @@ static void _join(Request& r, const Stri } } -/// ^table:sql{query}[(count[;offset])] @test limit/offset +/// ^table:sql{query}[(count[;offset])] static void _sql(Request& r, const String& method_name, Array *params) { Pool& pool=r.pool(); @@ -505,8 +509,89 @@ 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, Array *params) { + Pool& pool=r.pool(); + + Value& relative_path=*static_cast(params->get(0)); + // forcing [path] + r.fail_if_junction_(true, relative_path, method_name, "path must not be code"); + + const String *regexp; + pcre *regexp_code; + int ovecsize; + int *ovector; + if(params->size()>1) { + Value& regexp_value=*static_cast(params->get(1)); + // forcing [regexp] + r.fail_if_junction_(true, regexp_value, method_name, "regexp must not be code"); + regexp=®exp_value.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 +648,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); }