|
|
| version 1.37, 2001/03/28 09:01:20 | version 1.55, 2001/04/05 13:27:11 |
|---|---|
| Line 1 | Line 1 |
| /** @file | /** @file |
| Parser: table parser class. | Parser: @b table parser class. |
| Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) | Copyright (c) 2001 ArtLebedev Group (http://www.artlebedev.com) |
| Line 14 | Line 14 |
| #include "_table.h" | #include "_table.h" |
| #include "pa_vtable.h" | #include "pa_vtable.h" |
| #include "pa_vint.h" | #include "pa_vint.h" |
| #include "pa_sql_connection.h" | |
| // global var | // global var |
| VStateless_class *table_class; | VStateless_class *table_class; |
| // methods | // methods |
| static void _set(Request& r, const String& method_name, Array *params) { | |
| static void set_or_load( | |
| Request& r, | |
| const String& method_name, Array *params, | |
| bool is_load) { | |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| // data is last parameter | // data is last parameter |
| Value *vdata_or_filename=static_cast<Value *>(params->get(params->size()-1)); | Value *vdata=static_cast<Value *>(params->get(params->size()-1)); |
| // forcing | // forcing {this body type} |
| // ^load[this file name type] | r.fail_if_junction_(false, *vdata, method_name, "body must be junction"); |
| // ^set{this body type} | |
| r.fail_if_junction_(is_load, *vdata_or_filename, | Temp_lang temp_lang(r, String::UL_PASS_APPENDED); |
| method_name, is_load?"file name must not be junction":"body must be junction"); | const String& data=r.process(*vdata).as_string(); |
| // data or file_name | size_t pos_after=0; |
| char *data; | // parse columns |
| if(is_load) { | Array *columns; |
| // forcing untaint language | if(params->size()==2) { |
| String lfile_name(pool); | columns=0; |
| lfile_name.append(vdata_or_filename->as_string(), String::UL_FILE_NAME, true); | |
| // loading text | |
| data=file_read_text(pool, r.absolute(lfile_name)); | |
| } else { | } else { |
| // suggesting untaint language | columns=new(pool) Array(pool); |
| Temp_lang temp_lang(r, String::UL_TABLE); | |
| data=r.process(*vdata_or_filename).as_string().cstr(); | Array head(pool); |
| data.split(head, &pos_after, "\n", 1, String::UL_CLEAN, 1); | |
| if(head.size()) | |
| head.get_string(0)->split(*columns, 0, "\t", 1, String::UL_CLEAN); | |
| } | } |
| Table& table=*new(pool) Table(pool, &method_name, columns); | |
| // parse cells | |
| Array rows(pool); | |
| data.split(rows, &pos_after, "\n", 1, String::UL_CLEAN); | |
| int size=rows.quick_size(); | |
| for(int i=0; i<size; i++) { | |
| Array& row=*new(pool) Array(pool); | |
| const String& string=*rows.quick_get_string(i); | |
| // remove empty lines | |
| if(!string.size()) | |
| continue; | |
| string.split(row, 0, "\t", 1, String::UL_CLEAN); | |
| table+=&row; | |
| } | |
| // replace any previous table value | |
| static_cast<VTable *>(r.self)->set_table(table); | |
| } | |
| static void _load(Request& r, const String& method_name, Array *params) { | |
| Pool& pool=r.pool(); | |
| // filename is last parameter | |
| Value *vfilename=static_cast<Value *>(params->get(params->size()-1)); | |
| // forcing [this file name type] | |
| r.fail_if_junction_(true, *vfilename, | |
| method_name, "file name must not be junction"); | |
| // loading text | |
| char *data=file_read_text(pool, r.absolute(vfilename->as_string())); | |
| // parse columns | // parse columns |
| Array *columns; | Array *columns; |
| #ifndef NO_STRING_ORIGIN | #ifndef NO_STRING_ORIGIN |
| Line 63 static void set_or_load( | Line 91 static void set_or_load( |
| if(char *row_chars=getrow(&data)) | if(char *row_chars=getrow(&data)) |
| do { | do { |
| String *name=new(pool) String(pool); | String *name=new(pool) String(pool); |
| name->APPEND(lsplit(&row_chars, '\t'), 0, file, line++); | name->APPEND_TAINTED(lsplit(&row_chars, '\t'), 0, file, line++); |
| *columns+=name; | *columns+=name; |
| } while(row_chars); | } while(row_chars); |
| } | } |
| Line 77 static void set_or_load( | Line 105 static void set_or_load( |
| Array *row=new(pool) Array(pool); | Array *row=new(pool) Array(pool); |
| while(char *cell_chars=lsplit(&row_chars, '\t')) { | while(char *cell_chars=lsplit(&row_chars, '\t')) { |
| String *cell=new(pool) String(pool); | String *cell=new(pool) String(pool); |
| cell->APPEND(cell_chars, 0, file, line); | cell->APPEND_TAINTED(cell_chars, 0, file, line); |
| *row+=cell; | *row+=cell; |
| } | } |
| line++; | line++; |
| Line 88 static void set_or_load( | Line 116 static void set_or_load( |
| static_cast<VTable *>(r.self)->set_table(table); | static_cast<VTable *>(r.self)->set_table(table); |
| } | } |
| static void _set(Request& r, const String& method_name, Array *params) { | |
| set_or_load(r, method_name, params, false); | |
| } | |
| static void _load(Request& r, const String& method_name, Array *params) { | |
| set_or_load(r, method_name, params, true); | |
| } | |
| static void _save(Request& r, const String& method_name, Array *params) { | static void _save(Request& r, const String& method_name, Array *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Value *vfile_name=static_cast<Value *>(params->get(params->size()-1)); | Value *vtable_name=static_cast<Value *>(params->get(params->size()-1)); |
| // forcing | // forcing this body type] |
| // ^save[this body type] | r.fail_if_junction_(true, *vtable_name, |
| r.fail_if_junction_(true, *vfile_name, | |
| method_name, "file name must not be junction"); | method_name, "file name must not be junction"); |
| // forcing untaint language | |
| String lfile_name(pool); | |
| lfile_name.append(vfile_name->as_string(), | |
| String::UL_FILE_NAME, true); | |
| Table& table=static_cast<VTable *>(r.self)->table(); | Table& table=static_cast<VTable *>(r.self)->table(); |
| String sdata(pool); | String sdata(pool); |
| Line 148 static void _save(Request& r, const Stri | Line 161 static void _save(Request& r, const Stri |
| } | } |
| // write | // write |
| file_write(pool, r.absolute(lfile_name), sdata.cstr(), sdata.size(), true); | file_write(pool, r.absolute(vtable_name->as_string()), |
| sdata.cstr(), sdata.size(), true); | |
| } | } |
| static void _count(Request& r, const String&, Array *) { | static void _count(Request& r, const String&method_name, Array *) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Value& value=*new(pool) VInt(pool, static_cast<VTable *>(r.self)->table().size()); | Value& value=*new(pool) VInt(pool, static_cast<VTable *>(r.self)->table().size()); |
| r.write_no_lang(value); | r.write_no_lang(value); |
| } | } |
| static void _line(Request& r, const String&, Array *) { | static void _line(Request& r, const String& method_name, Array *) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Value& value=*new(pool) VInt(pool, 1+static_cast<VTable *>(r.self)->table().current()); | Value& value=*new(pool) VInt(pool, 1+static_cast<VTable *>(r.self)->table().current()); |
| r.write_no_lang(value); | r.write_no_lang(value); |
| } | } |
| static void _offset(Request& r, const String&, Array *params) { | static void _offset(Request& r, const String& method_name, Array *params) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Table& table=static_cast<VTable *>(r.self)->table(); | Table& table=static_cast<VTable *>(r.self)->table(); |
| if(params->size()) | if(params->size()) |
| Line 174 static void _offset(Request& r, const St | Line 188 static void _offset(Request& r, const St |
| } | } |
| } | } |
| /// @test $a.menu{ $a[123] } | |
| static void _menu(Request& r, const String& method_name, Array *params) { | static void _menu(Request& r, const String& method_name, Array *params) { |
| Value& body_code=*static_cast<Value *>(params->get(0)); | Value& body_code=*static_cast<Value *>(params->get(0)); |
| // forcing ^menu{this param type} | // forcing ^menu{this param type} |
| Line 200 static void _menu(Request& r, const Stri | Line 215 static void _menu(Request& r, const Stri |
| table.set_current(saved_current); | table.set_current(saved_current); |
| } | } |
| static void _empty(Request& r, const String&, Array *params) { | static void _empty(Request& r, const String& method_name, Array *params) { |
| Table& table=static_cast<VTable *>(r.self)->table(); | Table& table=static_cast<VTable *>(r.self)->table(); |
| if(table.size()==0) { | if(table.size()==0) { |
| Value& value=r.process(*static_cast<Value *>(params->get(0))); | Value& value=r.process(*static_cast<Value *>(params->get(0))); |
| Line 227 static void store_column_item_to_hash(Ar | Line 242 static void store_column_item_to_hash(Ar |
| value=new(*ri.pool) VUnknown(*ri.pool); | value=new(*ri.pool) VUnknown(*ri.pool); |
| ri.hash->put(column_name, value); | ri.hash->put(column_name, value); |
| } | } |
| static void _record(Request& r, const String&, Array *params) { | static void _record(Request& r, const String& method_name, Array *params) { |
| Table& table=static_cast<VTable *>(r.self)->table(); | Table& table=static_cast<VTable *>(r.self)->table(); |
| if(const Array *columns=table.columns()) { | if(const Array *columns=table.columns()) { |
| Pool& pool=r.pool(); | Pool& pool=r.pool(); |
| Line 312 static void _sort(Request& r, const Stri | Line 327 static void _sort(Request& r, const Stri |
| table.set_current(0); | table.set_current(0); |
| } | } |
| static void _locate(Request& r, const String&, Array *params) { | static void _locate(Request& r, const String& method_name, Array *params) { |
| VTable& vtable=*static_cast<VTable *>(r.self); | VTable& vtable=*static_cast<VTable *>(r.self); |
| Table& table=vtable.table(); | Table& table=vtable.table(); |
| vtable.last_locate_was_successful=table.locate( | vtable.last_locate_was_successful=table.locate( |
| Line 336 static void _found(Request& r, const Str | Line 351 static void _found(Request& r, const Str |
| } | } |
| } | } |
| static void _flip(Request& r, const String& method_name, Array *params) { | |
| Pool& pool=r.pool(); | |
| VTable& vtable=*static_cast<VTable *>(r.self); | |
| Table& old_table=*vtable.get_table(); | |
| Table& new_table=*new(pool) Table(pool, &method_name, 0/*nameless*/); | |
| if(old_table.size()) | |
| if(int old_cols=old_table.at(0).size()) | |
| for(int column=0; column<old_cols; column++) { | |
| Array& new_row=*new(pool) Array(pool, old_table.size()); | |
| for(int i=0; i<old_table.size(); i++) { | |
| const Array& old_row=old_table.at(i); | |
| new_row+=column<old_row.size()?old_row.get(column):empty_string; | |
| } | |
| new_table+=&new_row; | |
| } | |
| vtable.set_table(new_table); | |
| } | |
| static void _append(Request& r, const String& method_name, Array *params) { | |
| Pool& pool=r.pool(); | |
| // data is last parameter | |
| Value *value=static_cast<Value *>(params->get(0)); | |
| // forcing {this body type} | |
| r.fail_if_junction_(false, *value, method_name, "body must be junction"); | |
| const String& string=r.process(*value).as_string(); | |
| // parse cells | |
| Array& row=*new(pool) Array(pool); | |
| string.split(row, 0, "\t", 1, String::UL_CLEAN); | |
| static_cast<VTable *>(r.self)->table()+=&row; | |
| } | |
| static void _join(Request& r, const String& method_name, Array *params) { | |
| Pool& pool=r.pool(); | |
| Value *value=static_cast<Value *>(params->get(0)); | |
| // forcing [this table ref type] | |
| r.fail_if_junction_(true, *value, method_name, "table ref must not be junction"); | |
| Table *maybe_src=value->get_table(); | |
| if(!maybe_src) | |
| PTHROW(0, 0, | |
| &method_name, | |
| "source is not a table"); | |
| Table& src=*maybe_src; | |
| Table& dest=static_cast<VTable *>(r.self)->table(); | |
| if(&src == &dest) | |
| PTHROW(0, 0, | |
| &method_name, | |
| "source and destination are same table"); | |
| if(const Array *dest_columns=dest.columns()) { // dest is named | |
| int saved_src_current=src.current(); | |
| for(int src_row=0; src_row<src.size(); src_row++) { | |
| src.set_current(src_row); | |
| Array& dest_row=*new(pool) Array(pool); | |
| for(int dest_column=0; dest_column<dest_columns->size(); dest_column++) | |
| dest_row+=src.item(*dest_columns->get_string(dest_column)); | |
| dest+=&dest_row; | |
| } | |
| src.set_current(saved_src_current); | |
| } else { // dest is nameless | |
| for(int src_row=0; src_row<src.size(); src_row++) | |
| dest+=&src.at(src_row); | |
| } | |
| } | |
| /// ^table:sql{query}[(count[;offset])] @test limit/offset | |
| static void _sql(Request& r, const String& method_name, Array *params) { | |
| Pool& pool=r.pool(); | |
| if(!r.connection) | |
| PTHROW(0, 0, | |
| &method_name, | |
| "without connect"); | |
| Value& statement=*static_cast<Value *>(params->get(0)); | |
| // forcing {this query param type} | |
| r.fail_if_junction_(false, statement, method_name, "statement must be junction"); | |
| ulong limit=0; | |
| if(params->size()>1) { | |
| Value& limit_code=*static_cast<Value *>(params->get(1)); | |
| // forcing (this limit param type) | |
| r.fail_if_junction_(false, limit_code, method_name, "limit must be expression"); | |
| limit=(uint)r.process(limit_code).as_double(); | |
| } | |
| ulong offset=0; | |
| if(params->size()>2) { | |
| Value& offset_code=*static_cast<Value *>(params->get(2)); | |
| // forcing (this limit param type) | |
| r.fail_if_junction_(false, offset_code, method_name, "offset must be expression"); | |
| offset=(ulong)r.process(offset_code).as_double(); | |
| } | |
| 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, limit, | |
| &sql_column_count, &sql_columns, | |
| &sql_row_count, &sql_rows); | |
| } | |
| PCATCH(e) { // connect/process 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()); | |
| Array& table_columns=*new(pool) Array(pool); | |
| for(unsigned int i=0; i<sql_column_count; i++) { | |
| String& table_column=*new(pool) String(pool); | |
| table_column.APPEND_TAINTED( | |
| (const char *)sql_columns[i].ptr, sql_columns[i].size, | |
| statement_cstr, 0); | |
| table_columns+=&table_column; | |
| } | |
| Table& table=*new(pool) Table(pool, &method_name, &table_columns); | |
| { | |
| for(unsigned long r=0; r<sql_row_count; r++) { | |
| SQL_Driver::Cell *sql_cells=sql_rows[r]; | |
| Array& table_row=*new(pool) Array(pool); | |
| for(unsigned int i=0; i<sql_column_count; i++) { | |
| String& table_cell=*new(pool) String(pool); | |
| table_cell.APPEND_TAINTED( | |
| (const char *)sql_cells[i].ptr, sql_cells[i].size, | |
| statement_cstr, r); | |
| table_row+=&table_cell; | |
| } | |
| table+=&table_row; | |
| } | |
| } | |
| // replace any previous table value | |
| static_cast<VTable *>(r.self)->set_table(table); | |
| } | |
| // initialize | // initialize |
| void initialize_table_class(Pool& pool, VStateless_class& vclass) { | void initialize_table_class(Pool& pool, VStateless_class& vclass) { |
| // ^table.set{data} | // ^table:set{data} |
| // ^table.set[nameless]{data} | // ^table:set[nameless]{data} |
| vclass.add_native_method("set", _set, 1, 2); | vclass.add_native_method("set", Method::CT_DYNAMIC, _set, 1, 2); |
| // ^table.load[file] | // ^table:load[file] |
| // ^table.load[nameless;file] | // ^table:load[nameless;file] |
| vclass.add_native_method("load", _load, 1, 2); | vclass.add_native_method("load", Method::CT_DYNAMIC, _load, 1, 2); |
| // ^table.save[file] | // ^table.save[file] |
| // ^table.save[nameless;file] | // ^table.save[nameless;file] |
| vclass.add_native_method("save", _save, 1, 2); | vclass.add_native_method("save", Method::CT_DYNAMIC, _save, 1, 2); |
| // ^table.count[] | // ^table.count[] |
| vclass.add_native_method("count", _count, 0, 0); | vclass.add_native_method("count", Method::CT_DYNAMIC, _count, 0, 0); |
| // ^table.line[] | // ^table.line[] |
| vclass.add_native_method("line", _line, 0, 0); | vclass.add_native_method("line", Method::CT_DYNAMIC, _line, 0, 0); |
| // ^table.offset[] | // ^table.offset[] |
| // ^table.offset[offset] | // ^table.offset[offset] |
| vclass.add_native_method("offset", _offset, 0, 1); | vclass.add_native_method("offset", Method::CT_DYNAMIC, _offset, 0, 1); |
| // ^table.menu{code} | // ^table.menu{code} |
| // ^table.menu{code}[delim] | // ^table.menu{code}[delim] |
| vclass.add_native_method("menu", _menu, 1, 2); | vclass.add_native_method("menu", Method::CT_DYNAMIC, _menu, 1, 2); |
| // ^table.empty{code-when-empty} | // ^table.empty{code-when-empty} |
| // ^table.empty{code-when-empty}{code-when-not} | // ^table.empty{code-when-empty}{code-when-not} |
| vclass.add_native_method("empty", _empty, 1, 2); | vclass.add_native_method("empty", Method::CT_DYNAMIC, _empty, 1, 2); |
| // ^table.record[] | // ^table.record[] |
| vclass.add_native_method("record", _record, 0, 0); | vclass.add_native_method("record", Method::CT_DYNAMIC, _record, 0, 0); |
| // ^table.sort{string-key-maker} ^table.sort{string-key-maker}[asc|desc] | // ^table.sort{string-key-maker} ^table.sort{string-key-maker}[asc|desc] |
| // ^table.sort(numeric-key-maker) ^table.sort(numeric-key-maker)[asc|desc] | // ^table.sort(numeric-key-maker) ^table.sort(numeric-key-maker)[asc|desc] |
| vclass.add_native_method("sort", _sort, 1, 2); | vclass.add_native_method("sort", Method::CT_DYNAMIC, _sort, 1, 2); |
| // ^table.locate[field;value] | // ^table.locate[field;value] |
| vclass.add_native_method("locate", _locate, 2, 2); | vclass.add_native_method("locate", Method::CT_DYNAMIC, _locate, 2, 2); |
| // ^table.found{when-found} | // ^table.found{when-found} |
| // ^table.found{when-found}{when-not-found} | // ^table.found{when-found}{when-not-found} |
| vclass.add_native_method("found", _found, 1, 2); | vclass.add_native_method("found", Method::CT_DYNAMIC, _found, 1, 2); |
| } | |
| // ^table.flip[] | |
| vclass.add_native_method("flip", Method::CT_DYNAMIC, _flip, 0, 0); | |
| // ^table.append{r{tab}e{tab}c{tab}o{tab}r{tab}d} | |
| vclass.add_native_method("append", Method::CT_DYNAMIC, _append, 1, 1); | |
| // ^table.join[table] | |
| vclass.add_native_method("join", Method::CT_DYNAMIC, _join, 1, 1); | |
| // ^table:sql[query][(count[;offset])] | |
| vclass.add_native_method("sql", Method::CT_DYNAMIC, _sql, 1, 3); | |
| } |