/**
* @short Helps to prepare each pair.
*/
static void onion_dict_json_preorder(onion_block *block, const char *key, const void *value, int flags){
if (!onion_block_size(block)) // Error somewhere.
return;
char *s;
s=onion_c_quote_new(key);
if (s==NULL){
onion_block_clear(block);
return;
}
onion_block_add_str(block, s);
free(s);
onion_block_add_char(block, ':');
if (flags&OD_DICT){
onion_block *tmp;
tmp=onion_dict_to_json((onion_dict*)value);
if (!tmp){
onion_block_clear(block);
return;
}
onion_block_add_block(block, tmp);
onion_block_free(tmp);
}
else{
s=onion_c_quote_new(value);
if (s==NULL){
onion_block_clear(block);
return;
}
onion_block_add_str(block, s);
free(s);
}
onion_block_add_data(block, ", ",2);
}
/**
* @short One block read from in, prepare the output.
*
* Depending on the mode of the block it calls the appropiate handler: variable, tag or just write text.
*/
void write_block(parser_status *st, onion_block *b){
int mode=st->last_wmode;
//ONION_DEBUG("Write mode %d, code %s", mode, b->data);
switch(mode){
case TEXT:
{
int oldl;
if ( (oldl=onion_block_size(b)) ){
char *safe=onion_c_quote_new(onion_block_data(b));
function_add_code(st, " onion_response_write(res, %s, %d);\n", safe, oldl);
free(safe);
}
}
break;
case VARIABLE:
variable_write(st, b);
break;
case TAG:
tag_write(st, b);
break;
default:
ONION_ERROR("Unknown final mode %d", mode);
}
onion_block_clear(st->rawblock);
}
void t02_several_add_methods(){
INIT_TEST();
onion_block *block=onion_block_new();
FAIL_IF_EQUAL(block, NULL);
int i;
for (i=0;i<1024;i++){
onion_block_add_char(block, (char)i);
}
onion_block_clear(block);
onion_block_add_str(block, "first ");
for (i=0;i<1024;i++)
onion_block_add_str(block, "test ");
FAIL_IF_NOT_STRSTR(onion_block_data(block), "test");
for (i=0;i<1024;i++)
onion_block_add_data(block, "world", 4);
FAIL_IF_STRSTR(onion_block_data(block), "world");
FAIL_IF_NOT_STRSTR(onion_block_data(block), "worl");
int s=onion_block_size(block);
onion_block_add_block(block, block);
FAIL_IF_NOT_EQUAL(onion_block_size(block), s+s);
onion_block_free(block);
END_TEST();
}
/**
* @short One block read from in, prepare the output.
*
* Depending on the mode of the block it calls the appropiate handler: variable, tag or just write text.
*/
void write_block(parser_status *st, onion_block *b){
int mode=st->last_wmode;
//ONION_DEBUG("Write mode %d, code %s", mode, b->data);
switch(mode){
case TEXT:
{
int oldl;
if ( (oldl=onion_block_size(b)) ){
if (oldl>0){ // Not Just \0
int use_orig_line_numbers_bak=use_orig_line_numbers;
use_orig_line_numbers=0;
char *safe=onion_c_quote_new(onion_block_data(b));
function_add_code(st, " onion_response_write(res, ");
int pos=0;
int size=strlen(safe);
char *s=safe;
//ONION_DEBUG("------- pos %d, size %d",pos,size);
while (pos<size){
//ONION_DEBUG("pos %d, size %d",pos,size);
char *e=strstr(s, "\n");
if (!e)
break;
*e='\0';
if (pos==0)
function_add_code(st, "%s\n", s);
else
function_add_code(st, " %s\n", s);
pos+=(e-s)+1;
s=e+1;
}
if (pos==0)
function_add_code(st, "%s, %d);\n", s, oldl);
else
function_add_code(st, " %s, %d);\n", s, oldl);
use_orig_line_numbers=use_orig_line_numbers_bak;
free(safe);
}
}
}
break;
case VARIABLE:
variable_write(st, b);
break;
case TAG:
tag_write(st, b);
break;
default:
ONION_ERROR("Unknown final mode %d", mode);
}
onion_block_clear(st->rawblock);
}
/**
* @short Converts a dict to a json string
* @memberof onion_dict_t
*
* Given a dictionary and a buffer (with size), it writes a json dictionary to it.
*
* @returns an onion_block with the json data, or NULL on error
*/
block *onion_dict_to_json(onion_dict *dict){
onion_block *block=onion_block_new();
onion_block_add_char(block, '{');
if (dict && dict->root)
onion_dict_node_preorder(dict->root, (void*)onion_dict_json_preorder, block);
int s=onion_block_size(block);
if (s==0){ // Error.
onion_block_clear(block);
return NULL;
}
if (s!=1) // To remove a final ", "
onion_block_rewind(block, 2);
onion_block_add_char(block, '}');
return block;
}
/**
* @short Helps to prepare each pair.
*/
static void onion_dict_json_preorder(onion_block *block, const char *key, const void *value, int flags){
if (!onion_block_size(block)) // Error somewhere.
return;
onion_block_add_char(block,'\"');
onion_json_quote_add(block, key);
onion_block_add_data(block,"\":",2);
if (flags&OD_DICT){
onion_block *tmp;
tmp=onion_dict_to_json((onion_dict*)value);
if (!tmp){
onion_block_clear(block);
return;
}
onion_block_add_block(block, tmp);
onion_block_free(tmp);
}
else{
onion_block_add_char(block,'\"');
onion_json_quote_add(block, value);
onion_block_add_char(block,'\"');
}
onion_block_add_data(block, ", ",2);
}
onion_dict *onion_dict_from_json_(const char **_data){
const char *data=*_data;
ONION_DEBUG("Parse %s", *_data);
while (is_json_space(*data))
++data;
if (*data!='{')
return NULL;
++data;
while (is_json_space(*data))
++data;
;
onion_dict *ret=onion_dict_new();
onion_block *key=onion_block_new();
onion_block *value=onion_block_new();
while (*data!='}'){
// Get Key
ssize_t read_bytes=onion_json_unquote_add(key, data);
if (read_bytes<0)
goto error;
data+=read_bytes;
while (is_json_space(*data))
++data;
/// Get :
if (*data!=':'){ // Includes \0
ONION_DEBUG("Expected : got %c", *data);
goto error;
}
++data;
while (is_json_space(*data))
++data;
/// Get Value
if (*data=='{'){ // Includes \0
*_data=data;
onion_dict *sub=onion_dict_from_json_(_data);
if (!sub){
goto error;
}
onion_dict_add(ret, onion_block_data(key), sub, OD_DUP_KEY|OD_DICT|OD_FREE_VALUE);
data=*_data;
}
else if (is_json_digit(*data)){
while(is_json_digit(*data)){
onion_block_add_char(value, *data);
++data;
}
onion_dict_add(ret, onion_block_data(key), onion_block_data(value), OD_DUP_ALL);
}
else if (*data=='"'){ // parse string
ssize_t read_bytes=onion_json_unquote_add(value, data);
if (read_bytes<0)
goto error;
data+=read_bytes;
onion_dict_add(ret, onion_block_data(key), onion_block_data(value), OD_DUP_ALL);
onion_block_clear(value);
}
else { // Includes \0
ONION_DEBUG("Expected \" got %c", *data);
goto error;
}
onion_block_clear(key);
while (is_json_space(*data))
++data;
if (*data=='}'){
++data;
*_data=data;
onion_block_free(key);
onion_block_free(value);
return ret;
}
if (*data!=','){
ONION_DEBUG("Expected , got %c", *data);
goto error;
}
++data;
while (is_json_space(*data))
++data;
}
++data;
*_data=data;
onion_block_free(key);
onion_block_free(value);
return ret;
error:
onion_block_free(key);
onion_block_free(value);
onion_dict_free(ret);
return NULL;
}
onion_dict *onion_dict_from_json_(const char **_data){
const char *data=*_data;
ONION_DEBUG("Parse %s", *_data);
while (isspace(SAFETY_CAST(*data)))
++data;
if (*data!='{')
return NULL;
++data;
while (isspace(SAFETY_CAST(*data)))
++data;
;
onion_dict *ret=onion_dict_new();
onion_block *key=onion_block_new();
onion_block *value=onion_block_new();
while (*data!='}'){
// Get Key
if (*data!='"'){ // Includes \0
ONION_DEBUG("Expected \" got %c", *data);
goto error;
}
++data;
while (*data!='"'){
if (!*data){ // \0
ONION_DEBUG("Expected \" got eof");
goto error;
}
onion_block_add_char(key, *data);
++data;
}
++data;
while (isspace(SAFETY_CAST(*data)))
++data;
/// Get :
if (*data!=':'){ // Includes \0
ONION_DEBUG("Expected : got %c", *data);
goto error;
}
++data;
while (isspace(SAFETY_CAST(*data)))
++data;
/// Get Value
if (*data=='{'){ // Includes \0
*_data=data;
onion_dict *sub=onion_dict_from_json_(_data);
if (!sub){
goto error;
}
onion_dict_add(ret, onion_block_data(key), sub, OD_DUP_KEY|OD_DICT|OD_FREE_VALUE);
data=*_data;
}
else if (isdigit(SAFETY_CAST(*data))){
while(isdigit(SAFETY_CAST(*data))){
onion_block_add_char(value, *data);
++data;
}
onion_dict_add(ret, onion_block_data(key), onion_block_data(value), OD_DUP_ALL);
}
else if (*data=='"'){
++data;
while (*data!='"'){
if (!*data){ // \0
ONION_DEBUG("Expected \" got eof");
goto error;
}
onion_block_add_char(value, *data);
++data;
}
++data;
onion_dict_add(ret, onion_block_data(key), onion_block_data(value), OD_DUP_ALL);
onion_block_clear(value);
}
else { // Includes \0
ONION_DEBUG("Expected \" got %c", *data);
goto error;
}
onion_block_clear(key);
while (isspace(SAFETY_CAST(*data)))
++data;
if (*data=='}'){
++data;
*_data=data;
onion_block_free(key);
onion_block_free(value);
return ret;
}
if (*data!=','){
ONION_DEBUG("Expected , got %c", *data);
goto error;
}
++data;
while (isspace(SAFETY_CAST(*data)))
++data;
}
++data;
*_data=data;
onion_block_free(key);
onion_block_free(value);
return ret;
error:
onion_block_free(key);
onion_block_free(value);
onion_dict_free(ret);
return NULL;
}
