/* call-seq: emitter.line_width = width
*
* Set the preferred line with to +width+.
*/
static VALUE set_line_width(VALUE self, VALUE width)
{
yaml_emitter_t * emitter;
Data_Get_Struct(self, yaml_emitter_t, emitter);
yaml_emitter_set_width(emitter, NUM2INT(width));
return width;
}
int
Pemitter (lua_State *L)
{
lyaml_emitter *emitter;
lua_newtable (L); /* object table */
/* Create a user datum to store the emitter. */
emitter = (lyaml_emitter *) lua_newuserdata (L, sizeof (*emitter));
emitter->error = 0;
/* Initialize the emitter. */
if (!yaml_emitter_initialize (&emitter->emitter))
{
if (!emitter->emitter.problem)
emitter->emitter.problem = "cannot initialize emitter";
return luaL_error (L, "%s", emitter->emitter.problem);
}
yaml_emitter_set_unicode (&emitter->emitter, 1);
yaml_emitter_set_width (&emitter->emitter, 2);
yaml_emitter_set_output (&emitter->emitter, &append_output, emitter);
/* Set it's metatable, and ensure it is garbage collected properly. */
luaL_newmetatable (L, "lyaml.emitter");
lua_pushcfunction (L, emitter_gc);
lua_setfield (L, -2, "__gc");
lua_setmetatable (L, -2);
/* Set the emit method of object as a closure over the user datum, and
return the whole object. */
lua_pushcclosure (L, emit, 1);
lua_setfield (L, -2, "emit");
/* Set up a separate thread to collect error messages; save the thread
in the returned table so that it's not garbage collected when the
function call stack for Pemitter is cleaned up. */
emitter->errL = lua_newthread (L);
luaL_buffinit (emitter->errL, &emitter->errbuff);
lua_setfield (L, -2, "errthread");
/* Create a thread for the YAML buffer. */
emitter->outputL = lua_newthread (L);
luaL_buffinit (emitter->outputL, &emitter->yamlbuff);
lua_setfield (L, -2, "outputthread");
return 1;
}
/* call-seq: Psych::Emitter.new(io, options = Psych::Emitter::OPTIONS)
*
* Create a new Psych::Emitter that writes to +io+.
*/
static VALUE initialize(int argc, VALUE *argv, VALUE self)
{
yaml_emitter_t * emitter;
VALUE io, options;
VALUE line_width;
VALUE indent;
VALUE canonical;
TypedData_Get_Struct(self, yaml_emitter_t, &psych_emitter_type, emitter);
if (rb_scan_args(argc, argv, "11", &io, &options) == 2) {
line_width = rb_funcall(options, id_line_width, 0);
indent = rb_funcall(options, id_indentation, 0);
canonical = rb_funcall(options, id_canonical, 0);
yaml_emitter_set_width(emitter, NUM2INT(line_width));
yaml_emitter_set_indent(emitter, NUM2INT(indent));
yaml_emitter_set_canonical(emitter, Qtrue == canonical ? 1 : 0);
}
yaml_emitter_set_output(emitter, writer, (void *)io);
return self;
}
