1 files changed, 131 insertions, 0 deletions
diff --git a/stream/upb_textprinter.c b/stream/upb_textprinter.c
new file mode 100644
index 0000000..0f0357a
--- /dev/null
+++ b/stream/upb_textprinter.c
@@ -0,0 +1,131 @@
+/*
+ * upb - a minimalist implementation of protocol buffers.
+ *
+ * Copyright (c) 2009 Joshua Haberman.  See LICENSE for details.
+ */
+
+#include "upb_textprinter.h"
+
+#include <inttypes.h>
+#include <malloc.h>
+#include "upb_def.h"
+#include "upb_string.h"
+
+struct _upb_textprinter {
+  upb_sink sink;
+  upb_bytesink *bytesink;
+  upb_string *str;
+  int indent_depth;
+  bool single_line;
+  upb_fielddef *f;
+};
+
+static void upb_textprinter_endfield(upb_textprinter *p)
+{
+  if(p->single_line)
+    upb_bytesink_put(p->bytesink, UPB_STRLIT(' '));
+  else
+    upb_bytesink_put(p->bytesink, UPB_STRLIT('\n'));
+}
+
+static bool upb_textprinter_putval(upb_textprinter *p, upb_value val) {
+  p->str = upb_string_tryrecycle(p->str);
+#define CASE(fmtstr, member) upb_string_printf(p->str, fmtstr, val.member); break;
+  switch(p->f->type) {
+    case UPB_TYPE(DOUBLE):
+      CASE("%0.f", _double);
+    case UPB_TYPE(FLOAT):
+      CASE("%0.f", _float)
+    case UPB_TYPE(INT64):
+    case UPB_TYPE(SFIXED64):
+    case UPB_TYPE(SINT64):
+      CASE("%" PRId64, int64)
+    case UPB_TYPE(UINT64):
+    case UPB_TYPE(FIXED64):
+      CASE("%" PRIu64, uint64)
+    case UPB_TYPE(INT32):
+    case UPB_TYPE(SFIXED32):
+    case UPB_TYPE(SINT32):
+      CASE("%" PRId32, int32)
+    case UPB_TYPE(UINT32):
+    case UPB_TYPE(FIXED32):
+    case UPB_TYPE(ENUM):
+      CASE("%" PRIu32, uint32);
+    case UPB_TYPE(BOOL):
+      CASE("%hhu", _bool);
+  }
+  upb_bytesink_put(p->bytesink, p->str);
+  upb_textprinter_endfield(p);
+  return upb_ok(upb_bytesink_status(p->bytesink));
+}
+
+static bool upb_textprinter_putstr(upb_textprinter *p, upb_string *str) {
+  upb_bytesink_put(p->bytesink, UPB_STRLIT("\""));
+  // TODO: escaping.
+  upb_bytesink_put(p->bytesink, str);
+  upb_bytesink_put(p->bytesink, UPB_STRLIT("\""));
+  upb_textprinter_endfield(p);
+  return upb_ok(upb_bytesink_status(p->bytesink));
+}
+
+static void upb_textprinter_indent(upb_textprinter *p)
+{
+  if(!p->single_line)
+    for(int i = 0; i < p->indent_depth; i++)
+      upb_bytesink_put(p->bytesink, UPB_STRLIT("  "));
+}
+
+static bool upb_textprinter_putdef(upb_textprinter *p, upb_fielddef *f)
+{
+  upb_textprinter_indent(p);
+  upb_bytesink_put(p->bytesink, f->name);
+  upb_bytesink_put(p->bytesink, UPB_STRLIT(":"));
+  p->f = f;
+  return upb_ok(upb_bytesink_status(p->bytesink));
+}
+
+static bool upb_textprinter_startmsg(upb_textprinter *p)
+{
+  upb_textprinter_indent(p);
+  upb_bytesink_put(p->bytesink, p->f->def->fqname);
+  upb_bytesink_put(p->bytesink, UPB_STRLIT(" {"));
+  if(!p->single_line) upb_bytesink_put(p->bytesink, UPB_STRLIT('\n'));
+  p->indent_depth++;
+  return upb_ok(upb_bytesink_status(p->bytesink));
+}
+
+static bool upb_textprinter_endmsg(upb_textprinter *p)
+{
+  p->indent_depth--;
+  upb_textprinter_indent(p);
+  upb_bytesink_put(p->bytesink, UPB_STRLIT("}"));
+  upb_textprinter_endfield(p);
+  return upb_ok(upb_bytesink_status(p->bytesink));
+}
+
+upb_sink_vtable upb_textprinter_vtbl = {
+  (upb_sink_putdef_fptr)upb_textprinter_putdef,
+  (upb_sink_putval_fptr)upb_textprinter_putval,
+  (upb_sink_putstr_fptr)upb_textprinter_putstr,
+  (upb_sink_startmsg_fptr)upb_textprinter_startmsg,
+  (upb_sink_endmsg_fptr)upb_textprinter_endmsg,
+};
+
+upb_textprinter *upb_textprinter_new() {
+  upb_textprinter *p = malloc(sizeof(*p));
+  upb_sink_init(&p->sink, &upb_textprinter_vtbl);
+  return p;
+}
+
+void upb_textprinter_free(upb_textprinter *p) {
+  free(p);
+}
+
+void upb_textprinter_reset(upb_textprinter *p, upb_bytesink *sink,
+                           bool single_line) {
+  p->bytesink = sink;
+  p->single_line = single_line;
+  p->indent_depth = 0;
+}
+
+upb_sink *upb_textprinter_sink(upb_textprinter *p) { return &p->sink; }