|
|
| version 1.26, 2001/04/03 14:38:58 | version 1.53, 2001/05/21 16:38:46 |
|---|---|
| Line 8 | Line 8 |
| $Id$ | $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_sql_connection.h" | |
| // global var | // defines |
| VStateless_class *string_class; | #define STRING_CLASS_NAME "string" |
| // class | |
| class MString : public Methoded { | |
| public: // VStateless_class | |
| Value *create_new_value(Pool& pool) { return new(pool) VString(pool); } | |
| 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->get_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()); |
| Line 53 static void _double(Request& r, const St | Line 67 static void _double(Request& r, const St |
| r.write_no_lang(String(pool, buf)); | r.write_no_lang(String(pool, buf)); |
| } | } |
| 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.piece(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.piece(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.piece(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->get_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->get_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(); |
| Line 118 static void _lsplit(Request& r, const St | Line 126 static void _lsplit(Request& r, const St |
| 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(); |
| Line 136 static void _rsplit(Request& r, const St | Line 144 static void _rsplit(Request& r, const St |
| r.write_no_lang(*new(pool) VTable(pool, &table)); | r.write_no_lang(*new(pool) VTable(pool, &table)); |
| } | } |
| /// ^string.match{regexp}[options] | static void search_action(Table& table, Array *row, int, int, void *) { |
| static void _match(Request& r, const String& method_name, Array *params) { | if(row) |
| table+=row; | |
| } | |
| /// used by string: _match / replace_action | |
| struct Replace_action_info { | |
| Request *request; const String *origin; | |
| const String *src; String *dest; | |
| Value *replacement_code; | |
| const String *post_match; | |
| }; | |
| static void replace_action(Table& table, Array *row, int start, int finish, | |
| void *info) { | |
| Replace_action_info& ai=*static_cast<Replace_action_info *>(info); | |
| if(row) { // begin&middle | |
| // piece from last match['start'] to beginning of this match['finish'] | |
| if(start!=finish) | |
| *ai.dest << ai.src->mid(start, finish);//ai.dest->APPEND_CONST("-"); | |
| // store found parts in one-record VTable | |
| if(table.size()) // middle | |
| table.put(0, row); | |
| else // begin | |
| table+=row; | |
| { // execute 'replacement_code' in 'table' context | |
| VTable& vtable=*new(table.pool()) VTable(table.pool(), &table); | |
| vtable.set_name(*ai.origin); | |
| Junction *junction=ai.replacement_code->get_junction(); | |
| 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); | |
| /* | |
| ai.dest->APPEND_CONST("("); | |
| *ai.dest << *(String *)row->get(1/*match* /); | |
| ai.dest->APPEND_CONST(")"); | |
| */ | |
| *ai.dest << replaced.as_string(); | |
| } | |
| ai.post_match=(String *)row->get(2/*post_match*/); | |
| } else // end | |
| *ai.dest << *ai.post_match; | |
| } | |
| /// @todo use pcre:study somehow | |
| static void _match(Request& r, const String& method_name, MethodParams *params) { | |
| Pool& pool=r.pool(); | |
| const String& src=*static_cast<VString *>(r.self)->get_string(); | |
| Value& regexp=params->get_no_junction(0, "regexp must not be code"); | |
| const String *options= | |
| params->size()>1? | |
| ¶ms->get_no_junction(1, "options must not be code").as_string():0; | |
| Value *result; | |
| Temp_lang temp_lang(r, String::UL_PASS_APPENDED); | |
| Table *table; | |
| if(params->size()<3) { // search | |
| if(src.match(r.pcre_tables, | |
| &method_name, | |
| regexp.as_string(), options, | |
| &table, | |
| search_action, 0)) { | |
| // matched | |
| if(table->columns()->size()==3 && // just matched[3=pre/match/post], no substrings | |
| table->size()==1) // just one row, not /g_lobal search | |
| result=new(pool) VBool(pool, true); | |
| else // table of pre/match/post+substrings | |
| result=new(pool) VTable(pool, table); | |
| } else // not matched [not global] | |
| result=new(pool) VBool(pool, false); | |
| } else { // replace | |
| Value& replacement_code=params->get_junction(2, "replacement code must be code"); | |
| String& dest=*new(pool) String(pool); | |
| Replace_action_info replace_action_info={ | |
| &r, &method_name, | |
| &src, &dest, | |
| &replacement_code, | |
| &src | |
| }; | |
| src.match(r.pcre_tables, | |
| &method_name, | |
| r.process(regexp).as_string(), options, | |
| &table, | |
| replace_action, &replace_action_info); | |
| result=new(pool) VString(dest); | |
| } | |
| result->set_name(method_name); | |
| r.write_assign_lang(*result); | |
| } | |
| static void change_case(Request& r, const String& method_name, MethodParams *params, | |
| String::Change_case_kind kind) { | |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| const String& string=*static_cast<VString *>(r.self)->get_string(); | 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); | |
| } | |
| 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& regexp=*static_cast<Value *>(params->get(0)); | Value& statement=params->get_junction(0, "statement must be code"); |
| // forcing {this param type} | |
| r.fail_if_junction_(false, regexp, method_name, "regexp must be junction"); | |
| const String *options=0; | ulong offset=0; |
| if(params->size()>1) { | if(params->size()>1) { |
| Value& value=*static_cast<Value *>(params->get(1)); | Value& offset_code=params->get_junction(1, "offset must be expression"); |
| // forcing {this param type} | offset=(ulong)r.process(offset_code).as_double(); |
| r.fail_if_junction_(true, value, method_name, "options must not be junction"); | |
| options=&value.as_string(); | |
| } | } |
| Temp_lang temp_lang(r, String::UL_PASS_APPENDED); | Temp_lang temp_lang(r, String::UL_SQL); |
| Table *table; | const String& statement_string=r.process(statement).as_string(); |
| Value *result; | const char *statement_cstr= |
| if(string.match(&method_name, | statement_string.cstr(String::UL_UNSPECIFIED, r.connection); |
| r.process(regexp).as_string(), options, | unsigned int sql_column_count; SQL_Driver::Cell *sql_columns; |
| &table)) { | unsigned long sql_row_count; SQL_Driver::Cell **sql_rows; |
| // matched | bool need_rethrow=false; Exception rethrow_me; |
| if(table->columns()->size()==3 && // just matched[3=pre/match/post], no substrings | PTRY { |
| table->size()==1) // just one row, not Global search | r.connection->query( |
| result=new(pool) VBool(pool, true); | statement_cstr, offset, 0, |
| else // table of match column+substring columns | &sql_column_count, &sql_columns, |
| result=new(pool) VTable(pool, table); | &sql_row_count, &sql_rows); |
| } else // not global & not matched | } |
| result=new(pool) VBool(pool, false); | 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 only one column"); | |
| if(sql_row_count!=1) | |
| PTHROW(0, 0, | |
| &statement_string, | |
| "result must contain only 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; | |
| } | |
| result->set_name(method_name); | static void _sql(Request& r, const String& method_name, MethodParams *params) { |
| r.write_no_lang(*result); | Pool& pool=r.pool(); |
| VString& result=*new(pool) VString(sql_result_string(r, method_name, params)); | |
| result.set_name(method_name); | |
| r.write_assign_lang(result); | |
| } | } |
| // initialize | // 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] |
| vclass.add_native_method("match", Method::CT_DYNAMIC, _match, 1, 2); | // ^string.match[regexp][options]{replacement-code} |
| 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] | |
| // ^string:sql[query](offset) | |
| add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 2); | |
| } | } |
| // global variable | |
| Methoded *string_class; | |
| // creator | |
| Methoded *MString_create(Pool& pool) { | |
| return string_class=new(pool) MString(pool); | |
| } |