Diff for /parser3/src/classes/table.C between versions 1.37 and 1.55

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);
   }

Removed from v.1.37  
changed lines
  Added in v.1.55


E-mail: