|
|
| version 1.35, 2001/04/04 10:50:33 | version 1.63, 2001/07/18 10:06:04 |
|---|---|
| Line 4 | Line 4 |
| Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) | Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) |
| Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf) | Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf) |
| $Id$ | |
| */ | */ |
| static const char *RCSId="$Id$"; | |
| #include "classes.h" | |
| #include "pa_request.h" | #include "pa_request.h" |
| #include "_string.h" | |
| #include "pa_vdouble.h" | #include "pa_vdouble.h" |
| #include "pa_vint.h" | #include "pa_vint.h" |
| #include "pa_vtable.h" | #include "pa_vtable.h" |
| #include "pa_vbool.h" | #include "pa_vbool.h" |
| #include "pa_string.h" | #include "pa_string.h" |
| #include "pa_sql_connection.h" | |
| // defines | |
| #define STRING_CLASS_NAME "string" | |
| // global var | // class |
| VStateless_class *string_class; | class MString : public Methoded { |
| public: | |
| MString(Pool& pool); | |
| public: // Methoded | |
| bool used_directly() { return true; } | |
| }; | |
| // methods | // methods |
| static void _length(Request& r, const String&, Array *) { | static void _length(Request& r, const String& method_name, MethodParams *) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Value& value=*new(pool) VDouble(pool, r.self->get_string()->size()); | Value& result=*new(pool) VDouble(pool, r.self->get_string()->size()); |
| r.write_no_lang(value); | result.set_name(method_name); |
| r.write_no_lang(result); | |
| } | } |
| static void _int(Request& r, const String&, Array *) { | static void _int(Request& r, const String& method_name, MethodParams *) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Value& value=*new(pool) VInt(pool, (int)r.self->as_double()); | Value& result=*new(pool) VInt(pool, r.self->as_int()); |
| r.write_no_lang(value); | result.set_name(method_name); |
| r.write_no_lang(result); | |
| } | } |
| static void _double(Request& r, const String&, Array *) { | static void _double(Request& r, const String& method_name, MethodParams *) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Value& value=*new(pool) VDouble(pool, r.self->as_double()); | Value& result=*new(pool) VDouble(pool, r.self->as_double()); |
| r.write_no_lang(value); | result.set_name(method_name); |
| r.write_no_lang(result); | |
| } | } |
| /// ^string.format{format} | /*not static*/void _string_format(Request& r, const String& method_name, MethodParams *params) { |
| /*not static*/void _string_format(Request& r, const String& method_name, Array *params) { | |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Value& fmt=*static_cast<Value *>(params->get(0)); | Value& fmt=params->as_junction(0, "fmt must be code"); |
| // forcing {this param type} | |
| r.fail_if_junction_(false, fmt, method_name, "fmt must be junction"); | |
| Temp_lang temp_lang(r, String::UL_PASS_APPENDED); | Temp_lang temp_lang(r, String::UL_PASS_APPENDED); |
| char *buf=format(pool, r.self->as_double(), r.process(fmt).as_string().cstr()); | char *buf=format(pool, r.self->as_double(), r.process(fmt).as_string().cstr()); |
| r.write_no_lang(String(pool, buf)); | String result(pool); |
| result.APPEND_CLEAN(buf, 0, | |
| method_name.origin().file, | |
| method_name.origin().line); | |
| r.write_no_lang(result); | |
| } | } |
| static void _left(Request& r, const String&, Array *params) { | static void _left(Request& r, const String&, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| size_t n=(size_t)r.process(*static_cast<Value *>(params->get(0))).as_double(); | size_t n=(size_t)r.process(params->get(0)).as_double(); |
| const String& string=*static_cast<VString *>(r.self)->get_string(); | const String& string=*static_cast<VString *>(r.self)->get_string(); |
| r.write_assign_lang(*new(pool) VString(string.mid(0, n))); | r.write_assign_lang(*new(pool) VString(string.mid(0, n))); |
| } | } |
| static void _right(Request& r, const String&, Array *params) { | static void _right(Request& r, const String&, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| size_t n=(size_t)r.process(*static_cast<Value *>(params->get(0))).as_double(); | size_t n=(size_t)r.process(params->get(0)).as_double(); |
| const String& string=*static_cast<VString *>(r.self)->get_string(); | const String& string=*static_cast<VString *>(r.self)->get_string(); |
| r.write_assign_lang(*new(pool) VString(string.mid(string.size()-n, string.size()))); | r.write_assign_lang(*new(pool) VString(string.mid(string.size()-n, string.size()))); |
| } | } |
| static void _mid(Request& r, const String&, Array *params) { | static void _mid(Request& r, const String&, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| size_t p=(size_t)r.process(*static_cast<Value *>(params->get(0))).as_double(); | size_t p=(size_t)r.process(params->get(0)).as_double(); |
| size_t n=(size_t)r.process(*static_cast<Value *>(params->get(1))).as_double(); | size_t n=(size_t)r.process(params->get(1)).as_double(); |
| const String& string=*static_cast<VString *>(r.self)->get_string(); | const String& string=*static_cast<VString *>(r.self)->get_string(); |
| r.write_assign_lang(*new(pool) VString(string.mid(p, p+n))); | r.write_assign_lang(*new(pool) VString(string.mid(p, p+n))); |
| } | } |
| static void _pos(Request& r, const String& method_name, Array *params) { | static void _pos(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Value& substr=*static_cast<Value *>(params->get(0)); | Value& substr=params->as_no_junction(0, "substr must not be code"); |
| // forcing [this param type] | |
| r.fail_if_junction_(true, substr, method_name, "substr must not be junction"); | |
| const String& string=*static_cast<VString *>(r.self)->get_string(); | const String& string=*static_cast<VString *>(r.self)->get_string(); |
| r.write_assign_lang(*new(pool) VInt(pool, string.pos(substr.as_string()))); | r.write_assign_lang(*new(pool) VInt(pool, string.pos(substr.as_string()))); |
| } | } |
| static void split_list(Request& r, const String& method_name, Array *params, | static void split_list(Request& r, const String& method_name, MethodParams *params, |
| const String& string, | const String& string, |
| Array& result) { | Array& result) { |
| Pool& pool=r.pool(); | Value& delim_value=params->as_no_junction(0, "delimiter must not be code"); |
| Value& delim_value=*static_cast<Value *>(params->get(0)); | |
| // forcing [this param type] | |
| r.fail_if_junction_(true, delim_value, method_name, "delimiter must not be junction"); | |
| string.split(result, 0, delim_value.as_string(), String::UL_CLEAN, -1); | string.split(result, 0, delim_value.as_string()); |
| } | } |
| static void _lsplit(Request& r, const String& method_name, Array *params) { | static void _lsplit(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| const String& string=*static_cast<VString *>(r.self)->get_string(); | const String& string=*static_cast<VString *>(r.self)->get_string(); |
| Array& row=*new(pool) Array(pool); | Array pieces(pool); |
| split_list(r, method_name, params, string, row); | split_list(r, method_name, params, string, pieces); |
| Table& table=*new(pool) Table(pool, &string, | Array& columns=*new(pool) Array(pool); |
| 0/*nameless*/, 1/*row preallocate(and only)*/); | columns+=new(pool) String(pool, "piece"); |
| table+=&row; | |
| Table& table=*new(pool) Table(pool, &string, | |
| &columns, pieces.size()); | |
| int size=pieces.quick_size(); | |
| for(int i=0; i<size; i++) { | |
| Array& row=*new(pool) Array(pool); | |
| row+=pieces.quick_get(i); | |
| table+=&row; | |
| } | |
| r.write_no_lang(*new(pool) VTable(pool, &table)); | r.write_no_lang(*new(pool) VTable(pool, &table)); |
| } | } |
| static void _rsplit(Request& r, const String& method_name, Array *params) { | static void _rsplit(Request& r, const String& method_name, MethodParams *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| const String& string=*static_cast<VString *>(r.self)->get_string(); | const String& string=*static_cast<VString *>(r.self)->get_string(); |
| Array list(pool); | Array pieces(pool); |
| split_list(r, method_name, params, string, list); | split_list(r, method_name, params, string, pieces); |
| Array& row=*new(pool) Array(pool); | Array& columns=*new(pool) Array(pool); |
| for(int i=list.size(); --i>=0; ) | columns+=new(pool) String(pool, "piece"); |
| row+=list.get(i); | |
| Table& table=*new(pool) Table(pool, &string, | Table& table=*new(pool) Table(pool, &string, |
| 0/*nameless*/, 1/*row preallocate(and only)*/); | &columns, pieces.size()); |
| table+=&row; | for(int i=pieces.size(); --i>=0; ) { |
| Array& row=*new(pool) Array(pool); | |
| row+=pieces.get(i); | |
| table+=&row; | |
| } | |
| r.write_no_lang(*new(pool) VTable(pool, &table)); | r.write_no_lang(*new(pool) VTable(pool, &table)); |
| } | } |
| Line 142 static void search_action(Table& table, | Line 159 static void search_action(Table& table, |
| table+=row; | table+=row; |
| } | } |
| /// used by string: _match / replace_action | |
| struct Replace_action_info { | struct Replace_action_info { |
| Request *request; const String *origin; | Request *request; const String *origin; |
| const String *src; String *dest; | const String *src; String *dest; |
| Line 154 static void replace_action(Table& table, | Line 172 static void replace_action(Table& table, |
| if(row) { // begin&middle | if(row) { // begin&middle |
| // piece from last match['start'] to beginning of this match['finish'] | // piece from last match['start'] to beginning of this match['finish'] |
| if(start!=finish) | if(start!=finish) |
| ai.dest->append(ai.src->mid(start, finish), String::UL_PASS_APPENDED);//ai.dest->APPEND_CONST("-"); | *ai.dest << ai.src->mid(start, finish);//ai.dest->APPEND_CONST("-"); |
| // store found parts in one-record Vtable | // store found parts in one-record VTable |
| if(table.size()) // middle | if(table.size()) // middle |
| table.put(0, row); | table.put(0, row); |
| else // begin | else // begin |
| Line 165 static void replace_action(Table& table, | Line 183 static void replace_action(Table& table, |
| vtable.set_name(*ai.origin); | vtable.set_name(*ai.origin); |
| Junction *junction=ai.replacement_code->get_junction(); | Junction *junction=ai.replacement_code->get_junction(); |
| junction->rcontext=junction->root=&vtable; | junction->rcontext=/*must be some way to get to |
| outside world junction->root=*/&vtable; | |
| Value& replaced=ai.request->process(*ai.replacement_code, ai.origin, false); | Value& replaced=ai.request->process(*ai.replacement_code, ai.origin, false); |
| /* | /* |
| ai.dest->APPEND_CONST("("); | ai.dest->APPEND_CONST("("); |
| ai.dest->append(*(String *)row->get(1/*match* /), String::UL_PASS_APPENDED); | *ai.dest << *(String *)row->get(1/*match* /); |
| ai.dest->APPEND_CONST(")"); | ai.dest->APPEND_CONST(")"); |
| */ | */ |
| ai.dest->append(replaced.as_string(), String::UL_PASS_APPENDED); | *ai.dest << replaced.as_string(); |
| } | } |
| ai.post_match=(String *)row->get(2/*post_match*/); | ai.post_match=(String *)row->get(2/*post_match*/); |
| } else // end | } else // end |
| ai.dest->append(*ai.post_match, String::UL_PASS_APPENDED); | *ai.dest << *ai.post_match; |
| } | } |
| /** search/replace | /// @todo use pcre:study somehow |
| ^string.match[regexp][options] | static void _match(Request& r, const String& method_name, MethodParams *params) { |
| ^string.match[regexp][options]{replacement-code} | |
| */ | |
| static void _match(Request& r, const String& method_name, Array *params) { | |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| const String& src=*static_cast<VString *>(r.self)->get_string(); | const String& src=*static_cast<VString *>(r.self)->get_string(); |
| Value& regexp=*static_cast<Value *>(params->get(0)); | Value& regexp=params->as_no_junction(0, "regexp must not be code"); |
| // forcing [this param type] | |
| r.fail_if_junction_(true, regexp, method_name, "regexp must not be junction"); | const String *options= |
| params->size()>1? | |
| const String *options=0; | ¶ms->as_no_junction(1, "options must not be code").as_string():0; |
| if(params->size()>1) { | |
| Value& value=*static_cast<Value *>(params->get(1)); | |
| // forcing {this param type} | |
| r.fail_if_junction_(true, value, method_name, "options must not be junction"); | |
| options=&value.as_string(); | |
| } | |
| Value *result; | Value *result; |
| Temp_lang temp_lang(r, String::UL_PASS_APPENDED); | Temp_lang temp_lang(r, String::UL_PASS_APPENDED); |
| Table *table; | Table *table; |
| if(params->size()<3) { // search | if(params->size()<3) { // search |
| if(src.match(&method_name, | bool matched=src.match(r.pcre_tables, |
| &method_name, | |
| regexp.as_string(), options, | regexp.as_string(), options, |
| &table, | &table, |
| search_action, 0)) { | search_action, 0); |
| // matched | // matched |
| if(table->columns()->size()==3 && // just matched[3=pre/match/post], no substrings | if(table->columns()->size()==3 && // just matched[3=pre/match/post], no substrings |
| table->size()==1) // just one row, not /g_lobal search | table->size()<=1) // just one row, not /g_lobal search |
| result=new(pool) VBool(pool, true); | result=new(pool) VBool(pool, matched); |
| else // table of pre/match/post+substrings | else // table of pre/match/post+substrings |
| result=new(pool) VTable(pool, table); | result=new(pool) VTable(pool, table); |
| } else // not matched [not global] | |
| result=new(pool) VBool(pool, false); | |
| } else { // replace | } else { // replace |
| Value& replacement_code=*static_cast<Value *>(params->get(2)); | Value& replacement_code=params->as_junction(2, "replacement code must be code"); |
| // forcing {this param type} | |
| r.fail_if_junction_(false, replacement_code, | |
| method_name, "replacement code must be junction"); | |
| String& dest=*new(pool) String(pool); | String& dest=*new(pool) String(pool); |
| Replace_action_info replace_action_info={ | Replace_action_info replace_action_info={ |
| Line 229 static void _match(Request& r, const Str | Line 235 static void _match(Request& r, const Str |
| &replacement_code, | &replacement_code, |
| &src | &src |
| }; | }; |
| src.match(&method_name, | src.match(r.pcre_tables, |
| &method_name, | |
| r.process(regexp).as_string(), options, | r.process(regexp).as_string(), options, |
| &table, | &table, |
| replace_action, &replace_action_info); | replace_action, &replace_action_info); |
| Line 239 static void _match(Request& r, const Str | Line 246 static void _match(Request& r, const Str |
| r.write_assign_lang(*result); | r.write_assign_lang(*result); |
| } | } |
| // initialize | static void change_case(Request& r, const String& method_name, MethodParams *params, |
| String::Change_case_kind kind) { | |
| Pool& pool=r.pool(); | |
| const String& src=*static_cast<VString *>(r.self)->get_string(); | |
| r.write_assign_lang(*new(pool) VString(src.change_case(pool, r.pcre_tables, | |
| kind))); | |
| } | |
| static void _upper(Request& r, const String& method_name, MethodParams *params) { | |
| change_case(r, method_name, params, String::CC_UPPER); | |
| } | |
| static void _lower(Request& r, const String& method_name, MethodParams *params) { | |
| change_case(r, method_name, params, String::CC_LOWER); | |
| } | |
| const String* sql_result_string(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->as_junction(0, "statement must be code"); | |
| ulong offset=(ulong)(params->size()>2?params->as_int(2, r):0); | |
| 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, 0, | |
| &sql_column_count, &sql_columns, | |
| &sql_row_count, &sql_rows); | |
| } | |
| PCATCH(e) { // query 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()); | |
| if(sql_column_count!=1) | |
| PTHROW(0, 0, | |
| &statement_string, | |
| "result must contain exactly one column"); | |
| if(!sql_row_count) | |
| return 0; // no lines, should return second param[default value] | |
| if(sql_row_count>1) | |
| PTHROW(0, 0, | |
| &statement_string, | |
| "result must not contain more then one row"); | |
| String *result=new(pool) String(pool); | |
| SQL_Driver::Cell& cell=sql_rows[0][0]; | |
| result->APPEND_TAINTED( | |
| (const char *)cell.ptr, cell.size, | |
| statement_cstr, 0); | |
| return result; | |
| } | |
| static void _sql(Request& r, const String& method_name, MethodParams *params) { | |
| Pool& pool=r.pool(); | |
| const String *string=sql_result_string(r, method_name, params); | |
| if(!string) { | |
| Value& default_code=params->as_junction(1, "default result must code"); | |
| Value& processed_code=r.process(default_code); | |
| string=processed_code.get_string(); | |
| if(!string) | |
| string=empty_string; | |
| } | |
| VString& result=*new(pool) VString(*string); | |
| result.set_name(method_name); | |
| r.write_assign_lang(result); | |
| } | |
| // constructor | |
| MString::MString(Pool& apool) : Methoded(apool) { | |
| set_name(*NEW String(pool(), STRING_CLASS_NAME)); | |
| void initialize_string_class(Pool& pool, VStateless_class& vclass) { | |
| // ^string.length[] | // ^string.length[] |
| vclass.add_native_method("length", Method::CT_DYNAMIC, _length, 0, 0); | add_native_method("length", Method::CT_DYNAMIC, _length, 0, 0); |
| // ^string.int[] | // ^string.int[] |
| vclass.add_native_method("int", Method::CT_DYNAMIC, _int, 0, 0); | add_native_method("int", Method::CT_DYNAMIC, _int, 0, 0); |
| // ^string.double[] | // ^string.double[] |
| vclass.add_native_method("double", Method::CT_DYNAMIC, _double, 0, 0); | add_native_method("double", Method::CT_DYNAMIC, _double, 0, 0); |
| // ^string.format{format} | // ^string.format{format} |
| vclass.add_native_method("format", Method::CT_DYNAMIC, _string_format, 1, 1); | add_native_method("format", Method::CT_DYNAMIC, _string_format, 1, 1); |
| // ^string.left(n) | // ^string.left(n) |
| vclass.add_native_method("left", Method::CT_DYNAMIC, _left, 1, 1); | add_native_method("left", Method::CT_DYNAMIC, _left, 1, 1); |
| // ^string.right(n) | // ^string.right(n) |
| vclass.add_native_method("right", Method::CT_DYNAMIC, _right, 1, 1); | add_native_method("right", Method::CT_DYNAMIC, _right, 1, 1); |
| // ^string.mid(p;n) | // ^string.mid(p;n) |
| vclass.add_native_method("mid", Method::CT_DYNAMIC, _mid, 2, 2); | add_native_method("mid", Method::CT_DYNAMIC, _mid, 2, 2); |
| // ^string.pos[substr] | // ^string.pos[substr] |
| vclass.add_native_method("pos", Method::CT_DYNAMIC, _pos, 1, 1); | add_native_method("pos", Method::CT_DYNAMIC, _pos, 1, 1); |
| // ^string.lsplit[delim] | // ^string.lsplit[delim] |
| vclass.add_native_method("lsplit", Method::CT_DYNAMIC, _lsplit, 1, 1); | add_native_method("lsplit", Method::CT_DYNAMIC, _lsplit, 1, 1); |
| // ^string.rsplit[delim] | // ^string.rsplit[delim] |
| vclass.add_native_method("rsplit", Method::CT_DYNAMIC, _rsplit, 1, 1); | add_native_method("rsplit", Method::CT_DYNAMIC, _rsplit, 1, 1); |
| // ^string.match[regexp][options] | // ^string.match[regexp][options] |
| // ^string.match[regexp][options]{replacement-code} | // ^string.match[regexp][options]{replacement-code} |
| vclass.add_native_method("match", Method::CT_DYNAMIC, _match, 1, 3); | add_native_method("match", Method::CT_DYNAMIC, _match, 1, 3); |
| // ^string.toupper[] | |
| add_native_method("upper", Method::CT_DYNAMIC, _upper, 0, 0); | |
| // ^string.tolower[] | |
| add_native_method("lower", Method::CT_DYNAMIC, _lower, 0, 0); | |
| // ^string:sql[query]{default} | |
| // ^string:sql[query]{default}(offset) | |
| add_native_method("sql", Method::CT_STATIC, _sql, 2, 3); | |
| } | } |
| // global variable | |
| Methoded *string_class; | |
| // creator | |
| Methoded *MString_create(Pool& pool) { | |
| return string_class=new(pool) MString(pool); | |
| } |