Annotation of parser3/src/classes/string.C, revision 1.57
1.24 paf 1: /** @file
2: Parser: @b string parser class.
3:
1.4 paf 4: Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com)
1.24 paf 5:
1.5 paf 6: Author: Alexander Petrosyan <paf@design.ru> (http://design.ru/paf)
1.4 paf 7:
1.57 ! parser 8: $Id: string.C,v 1.56 2001/06/28 07:23:59 parser Exp $
1.1 paf 9: */
1.57 ! parser 10: static char *RCSId="$Id$";
1.1 paf 11:
1.43 paf 12: #include "classes.h"
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.1 paf 20:
1.41 paf 21: // defines
1.1 paf 22:
1.41 paf 23: #define STRING_CLASS_NAME "string"
24:
25: // class
26:
27: class MString : public Methoded {
28: public:
29: MString(Pool& pool);
1.44 paf 30: public: // Methoded
1.53 parser 31: bool used_directly() { return true; }
1.41 paf 32: };
1.1 paf 33:
34: // methods
35:
1.47 paf 36: static void _length(Request& r, const String& method_name, MethodParams *) {
1.1 paf 37: Pool& pool=r.pool();
1.47 paf 38: Value& result=*new(pool) VDouble(pool, r.self->get_string()->size());
39: result.set_name(method_name);
40: r.write_no_lang(result);
1.1 paf 41: }
42:
1.47 paf 43: static void _int(Request& r, const String& method_name, MethodParams *) {
1.1 paf 44: Pool& pool=r.pool();
1.48 parser 45: Value& result=*new(pool) VInt(pool, r.self->as_int());
1.47 paf 46: result.set_name(method_name);
47: r.write_no_lang(result);
1.1 paf 48: }
49:
1.47 paf 50: static void _double(Request& r, const String& method_name, MethodParams *) {
1.1 paf 51: Pool& pool=r.pool();
1.47 paf 52: Value& result=*new(pool) VDouble(pool, r.self->as_double());
53: result.set_name(method_name);
54: r.write_no_lang(result);
1.1 paf 55: }
56:
1.38 paf 57: /*not static*/void _string_format(Request& r, const String& method_name, MethodParams *params) {
1.9 paf 58: Pool& pool=r.pool();
59:
1.38 paf 60: Value& fmt=params->get_junction(0, "fmt must be code");
1.9 paf 61:
1.24 paf 62: Temp_lang temp_lang(r, String::UL_PASS_APPENDED);
63: char *buf=format(pool, r.self->as_double(), r.process(fmt).as_string().cstr());
1.9 paf 64:
1.12 paf 65: r.write_no_lang(String(pool, buf));
1.9 paf 66: }
1.11 paf 67:
1.38 paf 68: static void _left(Request& r, const String&, MethodParams *params) {
1.15 paf 69: Pool& pool=r.pool();
70:
1.38 paf 71: size_t n=(size_t)r.process(params->get(0)).as_double();
1.15 paf 72:
73: const String& string=*static_cast<VString *>(r.self)->get_string();
1.35 paf 74: r.write_assign_lang(*new(pool) VString(string.mid(0, n)));
1.15 paf 75: }
76:
1.38 paf 77: static void _right(Request& r, const String&, MethodParams *params) {
1.15 paf 78: Pool& pool=r.pool();
79:
1.38 paf 80: size_t n=(size_t)r.process(params->get(0)).as_double();
1.15 paf 81:
82: const String& string=*static_cast<VString *>(r.self)->get_string();
1.35 paf 83: r.write_assign_lang(*new(pool) VString(string.mid(string.size()-n, string.size())));
1.15 paf 84: }
85:
1.38 paf 86: static void _mid(Request& r, const String&, MethodParams *params) {
1.15 paf 87: Pool& pool=r.pool();
88:
1.38 paf 89: size_t p=(size_t)r.process(params->get(0)).as_double();
90: size_t n=(size_t)r.process(params->get(1)).as_double();
1.15 paf 91:
92: const String& string=*static_cast<VString *>(r.self)->get_string();
1.35 paf 93: r.write_assign_lang(*new(pool) VString(string.mid(p, p+n)));
1.15 paf 94: }
95:
1.38 paf 96: static void _pos(Request& r, const String& method_name, MethodParams *params) {
1.16 paf 97: Pool& pool=r.pool();
98:
1.38 paf 99: Value& substr=params->get_no_junction(0, "substr must not be code");
1.16 paf 100:
101: const String& string=*static_cast<VString *>(r.self)->get_string();
102: r.write_assign_lang(*new(pool) VInt(pool, string.pos(substr.as_string())));
103: }
104:
1.38 paf 105: static void split_list(Request& r, const String& method_name, MethodParams *params,
1.23 paf 106: const String& string,
107: Array& result) {
1.38 paf 108: Value& delim_value=params->get_no_junction(0, "delimiter must not be code");
1.23 paf 109:
1.45 paf 110: string.split(result, 0, delim_value.as_string());
1.19 paf 111: }
112:
1.38 paf 113: static void _lsplit(Request& r, const String& method_name, MethodParams *params) {
1.19 paf 114: Pool& pool=r.pool();
115: const String& string=*static_cast<VString *>(r.self)->get_string();
116:
1.21 paf 117: Array& row=*new(pool) Array(pool);
118: split_list(r, method_name, params, string, row);
1.19 paf 119:
1.21 paf 120: Table& table=*new(pool) Table(pool, &string,
121: 0/*nameless*/, 1/*row preallocate(and only)*/);
122: table+=&row;
1.19 paf 123:
124: r.write_no_lang(*new(pool) VTable(pool, &table));
125: }
126:
1.38 paf 127: static void _rsplit(Request& r, const String& method_name, MethodParams *params) {
1.19 paf 128: Pool& pool=r.pool();
129: const String& string=*static_cast<VString *>(r.self)->get_string();
130:
131: Array list(pool);
132: split_list(r, method_name, params, string, list);
133:
1.21 paf 134: Array& row=*new(pool) Array(pool);
135: for(int i=list.size(); --i>=0; )
136: row+=list.get(i);
137:
138: Table& table=*new(pool) Table(pool, &string,
139: 0/*nameless*/, 1/*row preallocate(and only)*/);
140: table+=&row;
1.17 paf 141:
1.19 paf 142: r.write_no_lang(*new(pool) VTable(pool, &table));
1.17 paf 143: }
144:
1.33 paf 145: static void search_action(Table& table, Array *row, int, int, void *) {
1.28 paf 146: if(row)
147: table+=row;
1.27 paf 148: }
149:
1.40 paf 150: /// used by string: _match / replace_action
1.27 paf 151: struct Replace_action_info {
1.30 paf 152: Request *request; const String *origin;
1.31 paf 153: const String *src; String *dest;
1.27 paf 154: Value *replacement_code;
1.29 paf 155: const String *post_match;
1.27 paf 156: };
1.33 paf 157: static void replace_action(Table& table, Array *row, int start, int finish,
1.31 paf 158: void *info) {
1.27 paf 159: Replace_action_info& ai=*static_cast<Replace_action_info *>(info);
1.31 paf 160: if(row) { // begin&middle
161: // piece from last match['start'] to beginning of this match['finish']
162: if(start!=finish)
1.37 paf 163: *ai.dest << ai.src->mid(start, finish);//ai.dest->APPEND_CONST("-");
1.51 parser 164: // store found parts in one-record VTable
1.32 paf 165: if(table.size()) // middle
166: table.put(0, row);
167: else // begin
1.30 paf 168: table+=row;
169: { // execute 'replacement_code' in 'table' context
170: VTable& vtable=*new(table.pool()) VTable(table.pool(), &table);
171: vtable.set_name(*ai.origin);
172:
173: Junction *junction=ai.replacement_code->get_junction();
1.52 parser 174: junction->rcontext=/*must be some way to get to
175: outside world junction->root=*/&vtable;
1.30 paf 176: Value& replaced=ai.request->process(*ai.replacement_code, ai.origin, false);
177:
1.34 paf 178: /*
1.31 paf 179: ai.dest->APPEND_CONST("(");
1.37 paf 180: *ai.dest << *(String *)row->get(1/*match* /);
1.31 paf 181: ai.dest->APPEND_CONST(")");
1.34 paf 182: */
1.37 paf 183: *ai.dest << replaced.as_string();
1.29 paf 184: }
185: ai.post_match=(String *)row->get(2/*post_match*/);
186: } else // end
1.37 paf 187: *ai.dest << *ai.post_match;
1.27 paf 188: }
189:
1.50 parser 190: /// @todo use pcre:study somehow
1.38 paf 191: static void _match(Request& r, const String& method_name, MethodParams *params) {
1.24 paf 192: Pool& pool=r.pool();
1.28 paf 193: const String& src=*static_cast<VString *>(r.self)->get_string();
1.24 paf 194:
1.38 paf 195: Value& regexp=params->get_no_junction(0, "regexp must not be code");
196:
197: const String *options=
198: params->size()>1?
199: ¶ms->get_no_junction(1, "options must not be code").as_string():0;
1.24 paf 200:
1.27 paf 201: Value *result;
1.24 paf 202: Temp_lang temp_lang(r, String::UL_PASS_APPENDED);
1.25 paf 203: Table *table;
1.27 paf 204: if(params->size()<3) { // search
1.39 paf 205: if(src.match(r.pcre_tables,
206: &method_name,
1.32 paf 207: regexp.as_string(), options,
1.27 paf 208: &table,
1.33 paf 209: search_action, 0)) {
1.27 paf 210: // matched
211: if(table->columns()->size()==3 && // just matched[3=pre/match/post], no substrings
212: table->size()==1) // just one row, not /g_lobal search
213: result=new(pool) VBool(pool, true);
214: else // table of pre/match/post+substrings
215: result=new(pool) VTable(pool, table);
216: } else // not matched [not global]
217: result=new(pool) VBool(pool, false);
218: } else { // replace
1.38 paf 219: Value& replacement_code=params->get_junction(2, "replacement code must be code");
1.24 paf 220:
1.28 paf 221: String& dest=*new(pool) String(pool);
1.27 paf 222: Replace_action_info replace_action_info={
1.30 paf 223: &r, &method_name,
1.31 paf 224: &src, &dest,
1.28 paf 225: &replacement_code,
1.29 paf 226: &src
1.27 paf 227: };
1.39 paf 228: src.match(r.pcre_tables,
229: &method_name,
1.27 paf 230: r.process(regexp).as_string(), options,
231: &table,
1.33 paf 232: replace_action, &replace_action_info);
1.28 paf 233: result=new(pool) VString(dest);
1.27 paf 234: }
1.26 paf 235: result->set_name(method_name);
1.34 paf 236: r.write_assign_lang(*result);
1.24 paf 237: }
238:
1.49 parser 239: static void change_case(Request& r, const String& method_name, MethodParams *params,
240: String::Change_case_kind kind) {
241: Pool& pool=r.pool();
242: const String& src=*static_cast<VString *>(r.self)->get_string();
243:
244: r.write_assign_lang(*new(pool) VString(src.change_case(pool, r.pcre_tables,
245: kind)));
246: }
247: static void _upper(Request& r, const String& method_name, MethodParams *params) {
248: change_case(r, method_name, params, String::CC_UPPER);
249: }
250: static void _lower(Request& r, const String& method_name, MethodParams *params) {
251: change_case(r, method_name, params, String::CC_LOWER);
252: }
253:
1.53 parser 254: String& sql_result_string(Request& r, const String& method_name, MethodParams *params) {
255: Pool& pool=r.pool();
256:
257: if(!r.connection)
258: PTHROW(0, 0,
259: &method_name,
260: "without connect");
261:
262: Value& statement=params->get_junction(0, "statement must be code");
263:
264: ulong offset=0;
265: if(params->size()>1) {
266: Value& offset_code=params->get_junction(1, "offset must be expression");
267: offset=(ulong)r.process(offset_code).as_double();
268: }
269:
270: Temp_lang temp_lang(r, String::UL_SQL);
271: const String& statement_string=r.process(statement).as_string();
272: const char *statement_cstr=
273: statement_string.cstr(String::UL_UNSPECIFIED, r.connection);
274: unsigned int sql_column_count; SQL_Driver::Cell *sql_columns;
275: unsigned long sql_row_count; SQL_Driver::Cell **sql_rows;
276: bool need_rethrow=false; Exception rethrow_me;
277: PTRY {
278: r.connection->query(
279: statement_cstr, offset, 0,
280: &sql_column_count, &sql_columns,
281: &sql_row_count, &sql_rows);
282: }
283: PCATCH(e) { // query problem
284: rethrow_me=e; need_rethrow=true;
285: }
286: PEND_CATCH
287: if(need_rethrow)
288: PTHROW(rethrow_me.type(), rethrow_me.code(),
289: &statement_string, // setting more specific source [were url]
290: rethrow_me.comment());
291:
292: if(sql_column_count!=1)
293: PTHROW(0, 0,
294: &statement_string,
1.56 parser 295: "result must contain exactly one column");
1.53 parser 296:
297: if(sql_row_count!=1)
298: PTHROW(0, 0,
299: &statement_string,
1.56 parser 300: "result must contain exactly one row");
1.53 parser 301:
302: String& result=*new(pool) String(pool);
303: SQL_Driver::Cell& cell=sql_rows[0][0];
304: result.APPEND_TAINTED(
305: (const char *)cell.ptr, cell.size,
306: statement_cstr, 0);
307:
308: return result;
309: }
310:
311: static void _sql(Request& r, const String& method_name, MethodParams *params) {
312: Pool& pool=r.pool();
313:
314: VString& result=*new(pool) VString(sql_result_string(r, method_name, params));
315: result.set_name(method_name);
316: r.write_assign_lang(result);
317: }
318:
1.41 paf 319: // constructor
320:
321: MString::MString(Pool& apool) : Methoded(apool) {
322: set_name(*NEW String(pool(), STRING_CLASS_NAME));
323:
1.9 paf 324:
1.1 paf 325: // ^string.length[]
1.41 paf 326: add_native_method("length", Method::CT_DYNAMIC, _length, 0, 0);
1.6 paf 327:
1.1 paf 328: // ^string.int[]
1.41 paf 329: add_native_method("int", Method::CT_DYNAMIC, _int, 0, 0);
1.6 paf 330:
1.1 paf 331: // ^string.double[]
1.41 paf 332: add_native_method("double", Method::CT_DYNAMIC, _double, 0, 0);
1.9 paf 333:
1.24 paf 334: // ^string.format{format}
1.41 paf 335: add_native_method("format", Method::CT_DYNAMIC, _string_format, 1, 1);
1.14 paf 336:
1.15 paf 337: // ^string.left(n)
1.41 paf 338: add_native_method("left", Method::CT_DYNAMIC, _left, 1, 1);
1.15 paf 339: // ^string.right(n)
1.41 paf 340: add_native_method("right", Method::CT_DYNAMIC, _right, 1, 1);
1.15 paf 341: // ^string.mid(p;n)
1.41 paf 342: add_native_method("mid", Method::CT_DYNAMIC, _mid, 2, 2);
1.16 paf 343:
344: // ^string.pos[substr]
1.41 paf 345: add_native_method("pos", Method::CT_DYNAMIC, _pos, 1, 1);
1.17 paf 346:
347: // ^string.lsplit[delim]
1.41 paf 348: add_native_method("lsplit", Method::CT_DYNAMIC, _lsplit, 1, 1);
1.19 paf 349: // ^string.rsplit[delim]
1.41 paf 350: add_native_method("rsplit", Method::CT_DYNAMIC, _rsplit, 1, 1);
1.24 paf 351:
1.32 paf 352: // ^string.match[regexp][options]
353: // ^string.match[regexp][options]{replacement-code}
1.41 paf 354: add_native_method("match", Method::CT_DYNAMIC, _match, 1, 3);
1.49 parser 355:
356: // ^string.toupper[]
357: add_native_method("upper", Method::CT_DYNAMIC, _upper, 0, 0);
358: // ^string.tolower[]
359: add_native_method("lower", Method::CT_DYNAMIC, _lower, 0, 0);
1.53 parser 360:
361: // ^string:sql[query]
362: // ^string:sql[query](offset)
1.54 parser 363: add_native_method("sql", Method::CT_STATIC, _sql, 1, 2);
1.2 paf 364: }
1.1 paf 365:
1.41 paf 366: // global variable
367:
368: Methoded *string_class;
369:
370: // creator
371:
372: Methoded *MString_create(Pool& pool) {
373: return string_class=new(pool) MString(pool);
374: }
E-mail: