Add SerdSink interface and hide implementations

7 files changed, 197 insertions, 77 deletions

diff --git a/src/n3.c b/src/n3.c
index 4c4b9eb9..439474dc 100644
--- a/src/n3.c
+++ b/src/n3.c
@@ -1091,9 +1091,7 @@ read_anon(SerdReader* reader, ReadContext ctx, bool subject, Ref* dest)
       return r_err(reader, SERD_ERR_BAD_SYNTAX, "`.' inside blank\n");
     }
     read_ws_star(reader);
-    if (reader->end_func) {
-      reader->end_func(reader->handle, deref(reader, *dest));
-    }
+    serd_sink_write_end(reader->sink, deref(reader, *dest));
     *ctx.flags = old_flags;
   }
   return (eat_byte_check(reader, ']') == ']') ? SERD_SUCCESS
@@ -1432,9 +1430,7 @@ read_base(SerdReader* reader, bool sparql, bool token)
 
   Ref uri = 0;
   TRY(st, read_IRIREF(reader, &uri));
-  if (reader->base_func) {
-    TRY(st, reader->base_func(reader->handle, deref(reader, uri)));
-  }
+  TRY(st, serd_sink_write_base(reader->sink, deref(reader, uri)));
   pop_node(reader, uri);
 
   read_ws_star(reader);
@@ -1472,10 +1468,8 @@ read_prefixID(SerdReader* reader, bool sparql, bool token)
   Ref uri = 0;
   TRY(st, read_IRIREF(reader, &uri));
 
-  if (reader->prefix_func) {
-    st = reader->prefix_func(
-      reader->handle, deref(reader, name), deref(reader, uri));
-  }
+  st = serd_sink_write_prefix(
+    reader->sink, deref(reader, name), deref(reader, uri));
 
   pop_node(reader, uri);
   pop_node(reader, name);
@@ -1483,7 +1477,6 @@ read_prefixID(SerdReader* reader, bool sparql, bool token)
     read_ws_star(reader);
     st = eat_byte_check(reader, '.') ? SERD_SUCCESS : SERD_ERR_BAD_SYNTAX;
   }
-
   return st;
 }
 
diff --git a/src/reader.c b/src/reader.c
index d5de67d2..805d41a7 100644
--- a/src/reader.c
+++ b/src/reader.c
@@ -132,14 +132,12 @@ emit_statement(SerdReader* reader, ReadContext ctx, Ref o)
     graph = reader->default_graph;
   }
 
-  const SerdStatus st = !reader->statement_func
-                          ? SERD_SUCCESS
-                          : reader->statement_func(reader->handle,
-                                                   *ctx.flags,
-                                                   graph,
-                                                   deref(reader, ctx.subject),
-                                                   deref(reader, ctx.predicate),
-                                                   deref(reader, o));
+  const SerdStatus st = serd_sink_write(reader->sink,
+                                        *ctx.flags,
+                                        deref(reader, ctx.subject),
+                                        deref(reader, ctx.predicate),
+                                        deref(reader, o),
+                                        graph);
 
   *ctx.flags &= SERD_ANON_CONT | SERD_LIST_CONT; // Preserve only cont flags
   return st;
@@ -166,26 +164,16 @@ serd_reader_read_document(SerdReader* reader)
 }
 
 SerdReader*
-serd_reader_new(SerdSyntax syntax,
-                void*      handle,
-                void (*free_handle)(void*),
-                SerdBaseFunc      base_func,
-                SerdPrefixFunc    prefix_func,
-                SerdStatementFunc statement_func,
-                SerdEndFunc       end_func)
+serd_reader_new(SerdSyntax syntax, const SerdSink* sink)
 {
-  SerdReader* me     = (SerdReader*)calloc(1, sizeof(SerdReader));
-  me->handle         = handle;
-  me->free_handle    = free_handle;
-  me->base_func      = base_func;
-  me->prefix_func    = prefix_func;
-  me->statement_func = statement_func;
-  me->end_func       = end_func;
-  me->default_graph  = NULL;
-  me->stack          = serd_stack_new(SERD_PAGE_SIZE);
-  me->syntax         = syntax;
-  me->next_id        = 1;
-  me->strict         = true;
+  SerdReader* me = (SerdReader*)calloc(1, sizeof(SerdReader));
+
+  me->sink          = sink;
+  me->default_graph = NULL;
+  me->stack         = serd_stack_new(SERD_PAGE_SIZE);
+  me->syntax        = syntax;
+  me->next_id       = 1;
+  me->strict        = true;
 
   me->rdf_first = push_node(me, SERD_URI, NS_RDF "first", 48);
   me->rdf_rest  = push_node(me, SERD_URI, NS_RDF "rest", 47);
@@ -226,18 +214,9 @@ serd_reader_free(SerdReader* reader)
 #endif
   free(reader->stack.buf);
   free(reader->bprefix);
-  if (reader->free_handle) {
-    reader->free_handle(reader->handle);
-  }
   free(reader);
 }
 
-void*
-serd_reader_handle(const SerdReader* reader)
-{
-  return reader->handle;
-}
-
 void
 serd_reader_add_blank_prefix(SerdReader* reader, const char* prefix)
 {
diff --git a/src/reader.h b/src/reader.h
index 8df16da3..c34f3a8b 100644
--- a/src/reader.h
+++ b/src/reader.h
@@ -57,27 +57,22 @@ typedef struct {
 } ReadContext;
 
 struct SerdReaderImpl {
-  void* handle;
-  void (*free_handle)(void* ptr);
-  SerdBaseFunc      base_func;
-  SerdPrefixFunc    prefix_func;
-  SerdStatementFunc statement_func;
-  SerdEndFunc       end_func;
-  SerdErrorFunc     error_func;
-  void*             error_handle;
-  Ref               rdf_first;
-  Ref               rdf_rest;
-  Ref               rdf_nil;
-  SerdNode*         default_graph;
-  SerdByteSource    source;
-  SerdStack         stack;
-  SerdSyntax        syntax;
-  unsigned          next_id;
-  uint8_t*          buf;
-  char*             bprefix;
-  size_t            bprefix_len;
-  bool              strict; ///< True iff strict parsing
-  bool              seen_genid;
+  const SerdSink* sink;
+  SerdErrorFunc   error_func;
+  void*           error_handle;
+  Ref             rdf_first;
+  Ref             rdf_rest;
+  Ref             rdf_nil;
+  SerdNode*       default_graph;
+  SerdByteSource  source;
+  SerdStack       stack;
+  SerdSyntax      syntax;
+  unsigned        next_id;
+  uint8_t*        buf;
+  char*           bprefix;
+  size_t          bprefix_len;
+  bool            strict; ///< True iff strict parsing
+  bool            seen_genid;
 #ifdef SERD_STACK_CHECK
   Ref*   allocs;   ///< Stack of push offsets
   size_t n_allocs; ///< Number of stack pushes
diff --git a/src/serdi.c b/src/serdi.c
index c54d82e6..e7b8d7ba 100644
--- a/src/serdi.c
+++ b/src/serdi.c
@@ -340,13 +340,7 @@ main(int argc, char** argv)
     output_syntax, writer_flags, env, (SerdWriteFunc)fwrite, out_fd);
 
   SerdReader* const reader =
-    serd_reader_new(input_syntax,
-                    writer,
-                    NULL,
-                    (SerdBaseFunc)serd_writer_set_base_uri,
-                    (SerdPrefixFunc)serd_writer_set_prefix,
-                    (SerdStatementFunc)serd_writer_write_statement,
-                    (SerdEndFunc)serd_writer_end_anon);
+    serd_reader_new(input_syntax, serd_writer_sink(writer));
 
   serd_reader_set_strict(reader, !lax);
   if (quiet) {
diff --git a/src/sink.c b/src/sink.c
new file mode 100644
index 00000000..b62af23a
--- /dev/null
+++ b/src/sink.c
@@ -0,0 +1,106 @@
+/*
+  Copyright 2011-2020 David Robillard <d@drobilla.net>
+
+  Permission to use, copy, modify, and/or distribute this software for any
+  purpose with or without fee is hereby granted, provided that the above
+  copyright notice and this permission notice appear in all copies.
+
+  THIS SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+  WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+  MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+  ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+  WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+  ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+  OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+*/
+
+#include "sink.h"
+
+#include "serd/serd.h"
+
+#include <stdlib.h>
+
+SerdSink*
+serd_sink_new(void* handle, SerdFreeFunc free_handle)
+{
+  SerdSink* sink = (SerdSink*)calloc(1, sizeof(SerdSink));
+
+  sink->handle      = handle;
+  sink->free_handle = free_handle;
+
+  return sink;
+}
+
+void
+serd_sink_free(SerdSink* sink)
+{
+  if (sink) {
+    if (sink->free_handle) {
+      sink->free_handle(sink->handle);
+    }
+
+    free(sink);
+  }
+}
+
+SerdStatus
+serd_sink_set_base_func(SerdSink* sink, SerdBaseFunc base_func)
+{
+  sink->base = base_func;
+  return SERD_SUCCESS;
+}
+
+SerdStatus
+serd_sink_set_prefix_func(SerdSink* sink, SerdPrefixFunc prefix_func)
+{
+  sink->prefix = prefix_func;
+  return SERD_SUCCESS;
+}
+
+SerdStatus
+serd_sink_set_statement_func(SerdSink* sink, SerdStatementFunc statement_func)
+{
+  sink->statement = statement_func;
+  return SERD_SUCCESS;
+}
+
+SerdStatus
+serd_sink_set_end_func(SerdSink* sink, SerdEndFunc end_func)
+{
+  sink->end = end_func;
+  return SERD_SUCCESS;
+}
+
+SerdStatus
+serd_sink_write_base(const SerdSink* sink, const SerdNode* uri)
+{
+  return sink->base ? sink->base(sink->handle, uri) : SERD_SUCCESS;
+}
+
+SerdStatus
+serd_sink_write_prefix(const SerdSink* sink,
+                       const SerdNode* name,
+                       const SerdNode* uri)
+{
+  return sink->prefix ? sink->prefix(sink->handle, name, uri) : SERD_SUCCESS;
+}
+
+SerdStatus
+serd_sink_write(const SerdSink*          sink,
+                const SerdStatementFlags flags,
+                const SerdNode*          subject,
+                const SerdNode*          predicate,
+                const SerdNode*          object,
+                const SerdNode*          graph)
+{
+  return sink->statement
+           ? sink->statement(
+               sink->handle, flags, graph, subject, predicate, object)
+           : SERD_SUCCESS;
+}
+
+SerdStatus
+serd_sink_write_end(const SerdSink* sink, const SerdNode* node)
+{
+  return sink->end ? sink->end(sink->handle, node) : SERD_SUCCESS;
+}
diff --git a/src/sink.h b/src/sink.h
new file mode 100644
index 00000000..e0f99a53
--- /dev/null
+++ b/src/sink.h
@@ -0,0 +1,34 @@
+/*
+  Copyright 2011-2020 David Robillard <d@drobilla.net>
+
+  Permission to use, copy, modify, and/or distribute this software for any
+  purpose with or without fee is hereby granted, provided that the above
+  copyright notice and this permission notice appear in all copies.
+
+  THIS SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+  WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+  MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+  ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+  WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+  ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+  OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+*/
+
+#ifndef SERD_SINK_H
+#define SERD_SINK_H
+
+#include "serd/serd.h"
+
+/**
+   An interface that receives a stream of RDF data.
+*/
+struct SerdSinkImpl {
+  void*             handle;
+  SerdFreeFunc      free_handle;
+  SerdBaseFunc      base;
+  SerdPrefixFunc    prefix;
+  SerdStatementFunc statement;
+  SerdEndFunc       end;
+};
+
+#endif // SERD_SINK_H
diff --git a/src/writer.c b/src/writer.c
index e23058f0..2b2d5ed8 100644
--- a/src/writer.c
+++ b/src/writer.c
@@ -18,6 +18,7 @@
 #include "env.h"
 #include "node.h"
 #include "serd_internal.h"
+#include "sink.h"
 #include "stack.h"
 #include "string_utils.h"
 #include "uri_utils.h"
@@ -92,6 +93,7 @@ static const SepRule rules[] = {{NULL, 0, 0, 0, 0},
                                 {"\n", 1, 0, 1, 0}};
 
 struct SerdWriterImpl {
+  SerdSink        iface;
   SerdSyntax      syntax;
   SerdWriterFlags flags;
   SerdEnv*        env;
@@ -113,6 +115,11 @@ struct SerdWriterImpl {
 
 typedef enum { WRITE_STRING, WRITE_LONG_STRING } TextContext;
 
+static SerdStatus
+serd_writer_set_prefix(SerdWriter*     writer,
+                       const SerdNode* name,
+                       const SerdNode* uri);
+
 static bool
 write_node(SerdWriter*        writer,
            const SerdNode*    node,
@@ -785,7 +792,7 @@ write_list_obj(SerdWriter*        writer,
   return false;
 }
 
-SerdStatus
+static SerdStatus
 serd_writer_write_statement(SerdWriter*        writer,
                             SerdStatementFlags flags,
                             const SerdNode*    graph,
@@ -915,7 +922,7 @@ serd_writer_write_statement(SerdWriter*        writer,
   return SERD_SUCCESS;
 }
 
-SerdStatus
+static SerdStatus
 serd_writer_end_anon(SerdWriter* writer, const SerdNode* node)
 {
   if (writer->syntax == SERD_NTRIPLES || writer->syntax == SERD_NQUADS) {
@@ -976,6 +983,12 @@ serd_writer_new(SerdSyntax      syntax,
   writer->byte_sink  = serd_byte_sink_new(
     ssink, stream, (flags & SERD_WRITE_BULK) ? SERD_PAGE_SIZE : 1);
 
+  writer->iface.handle    = writer;
+  writer->iface.base      = (SerdBaseFunc)serd_writer_set_base_uri;
+  writer->iface.prefix    = (SerdPrefixFunc)serd_writer_set_prefix;
+  writer->iface.statement = (SerdStatementFunc)serd_writer_write_statement;
+  writer->iface.end       = (SerdEndFunc)serd_writer_end_anon;
+
   return writer;
 }
 
@@ -1083,6 +1096,12 @@ serd_writer_free(SerdWriter* writer)
   free(writer);
 }
 
+const SerdSink*
+serd_writer_sink(SerdWriter* writer)
+{
+  return &writer->iface;
+}
+
 SerdEnv*
 serd_writer_env(SerdWriter* writer)
 {