Annotation of parser3/src/classes/string.C, revision 1.127
1.24 paf 1: /** @file
2: Parser: @b string parser class.
3:
1.126 paf 4: Copyright (c) 2001-2003 ArtLebedev Group (http://www.artlebedev.com)
1.97 paf 5: Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
1.113 paf 6: */
1.24 paf 7:
1.127 ! paf 8: static const char* IDENT_STRING_C="$Date: 2003/07/24 11:31:20 $";
1.1 paf 9:
1.43 paf 10: #include "classes.h"
1.126 paf 11: #include "pa_vmethod_frame.h"
12:
1.1 paf 13: #include "pa_request.h"
14: #include "pa_vdouble.h"
15: #include "pa_vint.h"
1.17 paf 16: #include "pa_vtable.h"
1.25 paf 17: #include "pa_vbool.h"
1.27 paf 18: #include "pa_string.h"
1.53 parser 19: #include "pa_sql_connection.h"
1.71 parser 20: #include "pa_dictionary.h"
1.119 paf 21: #include "pa_vmethod_frame.h"
1.1 paf 22:
1.41 paf 23: // class
24:
1.126 paf 25: class MString: public Methoded {
1.41 paf 26: public:
1.126 paf 27: MString();
1.44 paf 28: public: // Methoded
1.53 parser 29: bool used_directly() { return true; }
1.41 paf 30: };
1.1 paf 31:
1.126 paf 32: // global variable
33:
34: DECLARE_CLASS_VAR(string, new MString, 0);
35:
36: // defines for statics
37:
38: #define MATCH_VAR_NAME "match"
39:
40: // statics
41:
42: static const String match_var_name(MATCH_VAR_NAME);
43:
1.1 paf 44: // methods
45:
1.126 paf 46: static void _length(Request& r, MethodParams&) {
47: double result=GET_SELF(r, VString).string().length();
48: r.write_no_lang(*new VDouble(result));
1.1 paf 49: }
50:
1.126 paf 51: static void _int(Request& r, MethodParams& params) {
52: const String& self_string=GET_SELF(r, VString).string();
1.72 parser 53: int converted;
1.126 paf 54: Value* default_code=params.count()>0?¶ms.as_junction(0, "default must be int")
55: :0; // (default)
1.84 parser 56: try {
1.126 paf 57: if(self_string.is_empty())
1.121 paf 58: throw Exception("parser.runtime",
1.126 paf 59: 0,
1.121 paf 60: "parameter is empty string, error converting");
1.126 paf 61: converted=self_string.as_int();
1.84 parser 62: } catch(...) { // convert problem
1.126 paf 63: if(default_code)
64: converted=r.process_to_value(*default_code).as_int();
1.84 parser 65: else
1.126 paf 66: rethrow; // we have a problem when no default
1.72 parser 67: }
1.126 paf 68: r.write_no_lang(*new VInt(converted));
1.1 paf 69: }
70:
1.126 paf 71: static void _double(Request& r, MethodParams& params) {
72: const String& self_string=GET_SELF(r, VString).string();
1.72 parser 73: double converted;
1.126 paf 74: Value* default_code=params.count()>0?¶ms.as_junction(0, "default must be double")
75: :0; // (default)
1.84 parser 76: try {
1.126 paf 77: if(self_string.is_empty())
1.121 paf 78: throw Exception("parser.runtime",
1.126 paf 79: 0,
1.121 paf 80: "parameter is empty string, error converting");
1.126 paf 81: converted=self_string.as_double();
1.84 parser 82: } catch(...) { // convert problem
1.126 paf 83: if(default_code)
84: converted=r.process_to_value(*default_code).as_double();
1.84 parser 85: else
1.126 paf 86: rethrow; // we have a problem when no default
1.72 parser 87: }
88:
1.126 paf 89: r.write_no_lang(*new VDouble(converted));
1.1 paf 90: }
91:
1.126 paf 92: /*not static*/void _string_format(Request& r, MethodParams& params) {
1.9 paf 93:
1.126 paf 94: Value& fmt_maybe_code=params[0];
1.95 paf 95: // for some time due to stupid {} in original design
1.99 paf 96: const String& fmt=r.process_to_string(fmt_maybe_code);
1.9 paf 97:
1.126 paf 98: const char* buf=format(r.get_self().as_double(), fmt.cstrm());
1.63 parser 99:
1.126 paf 100: r.write_no_lang(String(buf));
1.9 paf 101: }
1.11 paf 102:
1.126 paf 103: static void _left(Request& r, MethodParams& params) {
104: size_t n=(size_t)params.as_int(0, "n must be int", r);
1.15 paf 105:
1.126 paf 106: const String& string=GET_SELF(r, VString).string();
1.102 paf 107: r.write_assign_lang(string.mid(0, n));
1.15 paf 108: }
109:
1.126 paf 110: static void _right(Request& r, MethodParams& params) {
111: size_t n=(size_t)params.as_int(0, "n must be int", r);
1.15 paf 112:
1.126 paf 113: const String& string=GET_SELF(r, VString).string();
1.127 ! paf 114: size_t length=string.length();
! 115: r.write_assign_lang(n<length?string.mid(length-n, string.length()):string);
1.15 paf 116: }
117:
1.126 paf 118: static void _mid(Request& r, MethodParams& params) {
119: const String& string=GET_SELF(r, VString).string();
1.83 parser 120:
1.126 paf 121: size_t p=(size_t)max(0, params.as_int(0, "p must be int", r));
122: size_t n=params.count()>1?
123: (size_t)max(0, params.as_int(1, "n must be int", r)):string.length();
1.15 paf 124:
1.102 paf 125: r.write_assign_lang(string.mid(p, p+n));
1.15 paf 126: }
127:
1.126 paf 128: static void _pos(Request& r, MethodParams& params) {
129: Value& substr=params.as_no_junction(0, "substr must not be code");
1.16 paf 130:
1.126 paf 131: const String& string=GET_SELF(r, VString).string();
132: r.write_assign_lang(*new VInt((int)string.pos(substr.as_string())));
1.16 paf 133: }
134:
1.126 paf 135: static void split_list(Request& r,
136: MethodParams& params, int paramIndex,
137: const String& string,
138: ArrayString& result) {
139: Value& delim_value=params.as_no_junction(paramIndex, "delimiter must not be code");
1.23 paf 140:
1.126 paf 141: size_t pos_after=0;
142: string.split(result, pos_after, delim_value.as_string());
1.19 paf 143: }
144:
1.118 paf 145: #define SPLIT_LEFT 0x0001
146: #define SPLIT_RIGHT 0x0010
147: #define SPLIT_HORIZONTAL 0x0100
148: #define SPLIT_VERTICAL 0x1000
149:
1.126 paf 150: static int split_options(const String* options) {
1.118 paf 151: struct Split_option {
1.126 paf 152: const char* keyL;
153: const char* keyU;
1.118 paf 154: int setBit;
155: int checkBit;
156: } split_option[]={
157: {"l", "L", SPLIT_LEFT, SPLIT_RIGHT}, // 0xVHRL
158: {"r", "R", SPLIT_RIGHT, SPLIT_LEFT},
159: {"h", "H", SPLIT_HORIZONTAL, SPLIT_VERTICAL},
160: {"v", "V", SPLIT_VERTICAL, SPLIT_HORIZONTAL},
161: {0}
162: };
163:
164: int result=0;
1.126 paf 165: if(options) {
1.118 paf 166: for(Split_option *o=split_option; o->keyL; o++)
1.126 paf 167: if(options->pos(o->keyL)!=STRING_NOT_FOUND
168: || (o->keyU && options->pos(o->keyU)!=STRING_NOT_FOUND)) {
1.118 paf 169: if(result & o->checkBit)
170: throw Exception("parser.runtime",
171: options,
172: "conflicting split options");
173: result |= o->setBit;
174: }
175: }
176:
177: return result;
178: }
179:
1.126 paf 180: static Table& split_vertical(Request& r, ArrayString& pieces, bool right) {
1.61 parser 181:
1.126 paf 182: Table::columns_type columns(new ArrayString);
183: *columns+=new String("piece");
1.19 paf 184:
1.126 paf 185: Table& table=*new Table(columns, pieces.count());
1.118 paf 186: if(right) { // right
1.126 paf 187: for(int i=pieces.count(); --i>=0; ) {
188: Table::element_type row(new ArrayString);
189: *row+=pieces[i];
190: table+=row;
1.118 paf 191: }
192: } else { // left
1.126 paf 193: Array_iterator<const String*> i(pieces);
1.118 paf 194: while(i.has_next()) {
1.126 paf 195: Table::element_type row(new ArrayString);
196: *row+=i.next();
197: table+=row;
1.118 paf 198: }
1.61 parser 199: }
1.118 paf 200:
1.126 paf 201: return table;
1.19 paf 202: }
203:
1.126 paf 204: static Table& split_horizontal(Request& r, ArrayString& pieces, bool right) {
205: Table& table=*new Table(Table::columns_type(0) /* nameless */);
206: Table::element_type row(new ArrayString(pieces.count()));
1.118 paf 207: if(right) { // right
1.126 paf 208: for(size_t i=pieces.count(); --i>=0; )
209: *row+=pieces[i];
1.118 paf 210: } else { // left
1.126 paf 211: for(Array_iterator<const String*> i(pieces); i.has_next(); )
212: *row+=i.next();
1.118 paf 213: }
1.126 paf 214: table+=row;
1.118 paf 215:
1.126 paf 216: return table;
1.118 paf 217: }
218:
1.126 paf 219: static void split_with_options(Request& r, MethodParams& params,
1.118 paf 220: int bits) {
1.126 paf 221: const String& string=GET_SELF(r, VString).string();
1.19 paf 222:
1.126 paf 223: ArrayString pieces;
224: split_list(r, params, 0, string, pieces);
1.19 paf 225:
1.118 paf 226: if(!bits) {
1.126 paf 227: const String* options=0;
228: if(params.count()>1)
229: options=¶ms.as_string(1, "options must not be code");
230:
1.118 paf 231: bits=split_options(options);
232: }
1.21 paf 233:
1.118 paf 234: bool right=(bits & SPLIT_RIGHT) != 0;
235: bool horizontal=(bits & SPLIT_HORIZONTAL) !=0;
1.126 paf 236: Table& table=horizontal?split_horizontal(r, pieces, right)
237: :split_vertical(r, pieces, right);
1.17 paf 238:
1.126 paf 239: r.write_no_lang(*new VTable(&table));
1.118 paf 240: }
1.126 paf 241: static void _split(Request& r, MethodParams& params) {
242: split_with_options(r, params, 0 /* maybe-determine from param #2 */);
1.118 paf 243: }
1.126 paf 244: static void _lsplit(Request& r, MethodParams& params) {
245: split_with_options(r, params, SPLIT_LEFT);
1.118 paf 246: }
1.126 paf 247: static void _rsplit(Request& r, MethodParams& params) {
248: split_with_options(r, params, SPLIT_RIGHT);
1.17 paf 249: }
250:
1.126 paf 251: static void search_action(Table& table, Table::element_type row, int, int, int, int, void *) {
1.28 paf 252: if(row)
253: table+=row;
1.27 paf 254: }
255:
1.74 parser 256: #ifndef DOXYGEN
1.27 paf 257: struct Replace_action_info {
1.126 paf 258: Request* request;
259: const String* src; String* dest;
260: VTable* vtable;
261: Value* replacement_code;
1.27 paf 262: };
1.74 parser 263: #endif
1.105 paf 264: /// @todo they can do $global[$result] there, getting pointer to later-invalid local var, kill this
1.126 paf 265: static void replace_action(Table& table, ArrayString* row,
266: int prestart, int prefinish,
267: int poststart, int postfinish,
268: void *info) {
1.27 paf 269: Replace_action_info& ai=*static_cast<Replace_action_info *>(info);
1.31 paf 270: if(row) { // begin&middle
1.104 paf 271: // piece from last match['prestart'] to beginning of this match['prefinish']
272: if(prestart!=prefinish)
273: *ai.dest << ai.src->mid(prestart, prefinish);//ai.dest->APPEND_CONST("-");
1.51 parser 274: // store found parts in one-record VTable
1.126 paf 275: if(table.count()) // middle
1.32 paf 276: table.put(0, row);
277: else // begin
1.30 paf 278: table+=row;
279: { // execute 'replacement_code' in 'table' context
1.105 paf 280: ai.vtable->set_table(table);
1.30 paf 281:
1.105 paf 282: *ai.dest << ai.request->process_to_string(*ai.replacement_code);
1.29 paf 283: }
284: } else // end
1.104 paf 285: *ai.dest << ai.src->mid(poststart, postfinish);
1.27 paf 286: }
287:
1.50 parser 288: /// @todo use pcre:study somehow
1.126 paf 289: static void _match(Request& r, MethodParams& params) {
290: Value& regexp=params.as_no_junction(0, "regexp must not be code");
291:
292: const String* options=
293: params.count()>1?
294: ¶ms.as_no_junction(1, "options must not be code").as_string():0;
295:
296: Temp_lang temp_lang(r, String::L_PASS_APPENDED);
297: const String& src=GET_SELF(r, VString).string();
298: bool just_matched;
299: if(params.count()<3) { // search
300: Table* table=src.match(r.charsets.source(),
1.32 paf 301: regexp.as_string(), options,
1.64 parser 302: search_action, 0,
1.126 paf 303: just_matched);
304: Value* result;
305: if(table)
306: result=new VTable(table); // table of pre/match/post+substrings
1.64 parser 307: else
1.126 paf 308: result=new VBool(just_matched);
1.102 paf 309: r.write_assign_lang(*result);
1.27 paf 310: } else { // replace
1.126 paf 311: Value& replacement_code=params.as_junction(2, "replacement param must be code");
1.106 paf 312:
1.126 paf 313: String result;
314: VTable* vtable=new VTable;
315: Replace_action_info info={0};
316: info.request=&r;
317: info.src=&src;
318: info.dest=&result;
319: info.vtable=vtable;
320: info.replacement_code=&replacement_code;
1.105 paf 321: Temp_value_element temp_match_var(
1.119 paf 322: *replacement_code.get_junction()->method_frame,
1.126 paf 323: match_var_name, vtable);
324: src.match(r.charsets.source(),
1.99 paf 325: r.process_to_string(regexp), options,
1.126 paf 326: replace_action, &info,
327: just_matched);
1.102 paf 328: r.write_assign_lang(result);
1.27 paf 329: }
1.24 paf 330: }
331:
1.126 paf 332: static void change_case(Request& r, MethodParams& params,
1.49 parser 333: String::Change_case_kind kind) {
1.126 paf 334: const String& src=GET_SELF(r, VString).string();
1.49 parser 335:
1.126 paf 336: r.write_assign_lang(src.change_case(r.charsets.source(), kind));
1.49 parser 337: }
1.126 paf 338: static void _upper(Request& r, MethodParams& params) {
339: change_case(r, params, String::CC_UPPER);
1.49 parser 340: }
1.126 paf 341: static void _lower(Request& r, MethodParams& params) {
342: change_case(r, params, String::CC_LOWER);
1.49 parser 343: }
344:
1.65 parser 345: #ifndef DOXYGEN
1.126 paf 346: class String_sql_event_handlers: public SQL_Driver_query_event_handlers {
347: const String& statement_string; const char* statement_cstr;
348: bool got_column;
1.65 parser 349: public:
1.126 paf 350: bool got_cell;
351: String& result;
352: public:
353: String_sql_event_handlers(
354: const String& astatement_string, const char* astatement_cstr):
355: statement_string(astatement_string), statement_cstr(astatement_cstr),
356: got_column(false),
357: got_cell(false),
358: result(*new String) {}
1.65 parser 359:
1.126 paf 360: bool add_column(SQL_Error& error, const char* str, size_t /*length*/) {
1.124 paf 361: if(got_column) {
362: error=SQL_Error("parser.runtime",
1.126 paf 363: //statement_string,
1.65 parser 364: "result must contain exactly one column");
1.124 paf 365: return true;
366: }
1.65 parser 367: got_column=true;
1.124 paf 368: return false;
1.65 parser 369: }
1.124 paf 370: bool before_rows(SQL_Error& /*error*/ ) { /* ignore */ return false; }
371: bool add_row(SQL_Error& /*error*/) { /* ignore */ return false; }
1.126 paf 372: bool add_row_cell(SQL_Error& error, const char* str, size_t length) {
1.124 paf 373: if(got_cell) {
374: error=SQL_Error("parser.runtime",
1.126 paf 375: //statement_string,
1.65 parser 376: "result must not contain more then one row");
1.124 paf 377: return true;
378: }
1.65 parser 379:
1.124 paf 380: try {
381: got_cell=true;
1.126 paf 382: result.append_know_length(str, length, String::L_TAINTED);
1.124 paf 383: return false;
384: } catch(...) {
385: error=SQL_Error("exception occured in String_sql_event_handlers::add_row_cell");
386: return true;
387: }
1.65 parser 388: }
389: };
390: #endif
1.126 paf 391: extern String sql_limit_name;
392: extern String sql_offset_name;
393: extern String sql_default_name;
394: extern String sql_distinct_name;
395: const String* sql_result_string(Request& r, MethodParams& params,
396: HashStringValue*& options, Value*& default_code) {
397: Value& statement=params.as_junction(0, "statement must be code");
1.53 parser 398:
1.70 parser 399: ulong limit=0;
400: ulong offset=0;
1.81 parser 401: default_code=0;
1.126 paf 402: if(params.count()>1) {
403: Value& voptions=params.as_no_junction(1, "options must be hash, not code");
1.115 paf 404: if(!voptions.is_string())
1.126 paf 405: if(options=voptions.get_hash()) {
406: if(Value* vlimit=options->get(sql_limit_name))
1.99 paf 407: limit=(ulong)r.process_to_value(*vlimit).as_double();
1.126 paf 408: if(Value* voffset=options->get(sql_offset_name))
1.99 paf 409: offset=(ulong)r.process_to_value(*voffset).as_double();
1.126 paf 410: if(default_code=options->get(sql_default_name)) {
411: if(Junction* default_junction=default_code->get_junction())
1.119 paf 412: ;//default_junction->change_context(statement.get_junction());
1.110 paf 413: else
1.98 paf 414: throw Exception("parser.runtime",
1.126 paf 415: 0,
1.81 parser 416: "default option must be code");
417: }
1.73 parser 418: } else
1.98 paf 419: throw Exception("parser.runtime",
1.126 paf 420: 0,
1.73 parser 421: "options must be hash");
1.70 parser 422: } else
423: options=0;
424:
1.126 paf 425: Temp_lang temp_lang(r, String::L_SQL);
1.99 paf 426: const String& statement_string=r.process_to_string(statement);
1.126 paf 427: const char* statement_cstr=
428: statement_string.cstr(String::L_UNSPECIFIED, r.connection());
429: String_sql_event_handlers handlers(statement_string, statement_cstr);
430: r.connection()->query(
1.117 paf 431: statement_cstr, offset, limit,
432: handlers,
433: statement_string);
1.53 parser 434:
1.65 parser 435: if(!handlers.got_cell)
436: return 0; // no lines, caller should return second param[default value]
1.62 parser 437:
1.126 paf 438: return &handlers.result;
1.53 parser 439: }
440:
1.126 paf 441: static void _sql(Request& r, MethodParams& params) {
1.53 parser 442:
1.126 paf 443: HashStringValue* options;
444: Value* default_code;
445: const String* string=sql_result_string(r, params, options, default_code);
1.62 parser 446: if(!string) {
1.81 parser 447: if(default_code) {
1.99 paf 448: string=&r.process_to_string(*default_code);
1.68 parser 449: } else
1.98 paf 450: throw Exception("parser.runtime",
1.126 paf 451: 0,
1.81 parser 452: "produced no result, but no default option specified");
1.62 parser 453: }
1.100 paf 454:
1.102 paf 455: r.write_assign_lang(*string);
1.53 parser 456: }
457:
1.126 paf 458: static void _replace(Request& r, MethodParams& params) {
459: const String& src=GET_SELF(r, VString).string();
1.71 parser 460:
1.126 paf 461: Table* table=params.as_no_junction(0, "parameter must not be code").get_table();
1.71 parser 462: if(!table)
1.98 paf 463: throw Exception("parser.runtime",
1.126 paf 464: 0,
1.71 parser 465: "parameter must be table");
466:
467: Dictionary dict(*table);
1.126 paf 468: r.write_assign_lang(src.replace(dict));
1.71 parser 469: }
1.79 parser 470:
1.126 paf 471: static void _save(Request& r, MethodParams& params) {
472: const String& file_name=params.as_string(params.count()-1,
1.87 paf 473: "file name must be string");
1.79 parser 474:
1.126 paf 475: const String& src=GET_SELF(r, VString).string();
1.79 parser 476:
1.87 paf 477: bool do_append=false;
1.126 paf 478: if(params.count()>1) {
479: const String& mode=params.as_string(0, "mode must be string");
1.87 paf 480: if(mode=="append")
481: do_append=true;
482: else
1.98 paf 483: throw Exception("parser.runtime",
1.87 paf 484: &mode,
485: "unknown mode, must be 'append'");
486: }
487:
1.79 parser 488: // write
1.126 paf 489: const char* buf=src.cstr(String::L_UNSPECIFIED, r.connection(false/*no error if none*/));
1.94 paf 490: file_write(r.absolute(file_name),
1.89 paf 491: buf, strlen(buf), true, do_append);
1.79 parser 492: }
493:
1.126 paf 494: static void _normalize(Request& r, MethodParams&) {
495: const String& src=GET_SELF(r, VString).string();
496:
497: r.write_assign_lang(src);
1.109 paf 498: }
499:
1.41 paf 500: // constructor
501:
1.126 paf 502: MString::MString(): Methoded("string") {
1.1 paf 503: // ^string.length[]
1.41 paf 504: add_native_method("length", Method::CT_DYNAMIC, _length, 0, 0);
1.6 paf 505:
1.1 paf 506: // ^string.int[]
1.72 parser 507: // ^string.int(default)
508: add_native_method("int", Method::CT_DYNAMIC, _int, 0, 1);
1.1 paf 509: // ^string.double[]
1.72 parser 510: // ^string.double(default)
511: add_native_method("double", Method::CT_DYNAMIC, _double, 0, 1);
1.9 paf 512:
1.24 paf 513: // ^string.format{format}
1.41 paf 514: add_native_method("format", Method::CT_DYNAMIC, _string_format, 1, 1);
1.14 paf 515:
1.15 paf 516: // ^string.left(n)
1.41 paf 517: add_native_method("left", Method::CT_DYNAMIC, _left, 1, 1);
1.15 paf 518: // ^string.right(n)
1.41 paf 519: add_native_method("right", Method::CT_DYNAMIC, _right, 1, 1);
1.15 paf 520: // ^string.mid(p;n)
1.82 parser 521: add_native_method("mid", Method::CT_DYNAMIC, _mid, 1, 2);
1.16 paf 522:
523: // ^string.pos[substr]
1.41 paf 524: add_native_method("pos", Method::CT_DYNAMIC, _pos, 1, 1);
1.17 paf 525:
1.118 paf 526: // ^string.split[delim]
527: // ^string.split[delim][options]
528: add_native_method("split", Method::CT_DYNAMIC, _split, 1, 2);
529: // old names for backward compatibility
530: // ^string.lsplit[delim]
531: add_native_method("lsplit", Method::CT_DYNAMIC, _lsplit, 1, 1);
532: // ^string.rsplit[delim]
533: add_native_method("rsplit", Method::CT_DYNAMIC, _rsplit, 1, 1);
534:
1.32 paf 535: // ^string.match[regexp][options]
536: // ^string.match[regexp][options]{replacement-code}
1.41 paf 537: add_native_method("match", Method::CT_DYNAMIC, _match, 1, 3);
1.49 parser 538:
539: // ^string.toupper[]
540: add_native_method("upper", Method::CT_DYNAMIC, _upper, 0, 0);
541: // ^string.tolower[]
542: add_native_method("lower", Method::CT_DYNAMIC, _lower, 0, 0);
1.53 parser 543:
1.70 parser 544: // ^sql[query]
545: // ^sql[query][$.limit(1) $.offset(2) $.default[n/a]]
1.67 parser 546: add_native_method("sql", Method::CT_STATIC, _sql, 1, 2);
1.71 parser 547:
548: // ^string.replace[table]
549: add_native_method("replace", Method::CT_DYNAMIC, _replace, 1, 1);
1.79 parser 550:
551: // ^string.save[file]
1.87 paf 552: add_native_method("save", Method::CT_DYNAMIC, _save, 1, 2);
1.109 paf 553:
1.112 paf 554: // ^string.normalize[]
555: add_native_method("normalize", Method::CT_DYNAMIC, _normalize, 0, 0);
1.2 paf 556: }
E-mail: