summaryrefslogtreecommitdiff
path: root/ext
diff options
context:
space:
mode:
Diffstat (limited to 'ext')
-rw-r--r--ext/psych/emitter.c30
1 files changed, 28 insertions, 2 deletions
diff --git a/ext/psych/emitter.c b/ext/psych/emitter.c
index b9cda96..14ed6d0 100644
--- a/ext/psych/emitter.c
+++ b/ext/psych/emitter.c
@@ -80,17 +80,43 @@ static VALUE start_document(VALUE self, VALUE version, VALUE tags, VALUE imp)
version_directive.minor = NUM2INT(minor);
}
+ yaml_tag_directive_t * head = NULL;
+ yaml_tag_directive_t * tail = NULL;
+
+ if(RTEST(tags)) {
+ int i = 0;
+
+ head = xcalloc(RARRAY_LEN(tags), sizeof(yaml_tag_directive_t));
+ tail = head;
+
+ for(i = 0; i < RARRAY_LEN(tags); i++) {
+ VALUE tuple = RARRAY_PTR(tags)[i];
+
+ if(RARRAY_LEN(tuple) < 2) {
+ xfree(head);
+ rb_raise(rb_eRuntimeError, "tag tuple must be of length 2");
+ }
+
+ tail->handle = StringValuePtr(RARRAY_PTR(tuple)[0]);
+ tail->prefix = StringValuePtr(RARRAY_PTR(tuple)[1]);
+
+ tail++;
+ }
+ }
+
yaml_event_t event;
yaml_document_start_event_initialize(
&event,
(RARRAY_LEN(version) > 0) ? &version_directive : NULL,
- NULL,
- NULL,
+ head,
+ tail,
imp == Qtrue ? 1 : 0
);
emit(emitter, &event);
+ if(head) xfree(head);
+
return self;
}