diff options
Diffstat (limited to 'src/inserter.c')
-rw-r--r-- | src/inserter.c | 117 |
1 files changed, 117 insertions, 0 deletions
diff --git a/src/inserter.c b/src/inserter.c new file mode 100644 index 00000000..e2f95896 --- /dev/null +++ b/src/inserter.c @@ -0,0 +1,117 @@ +/* + Copyright 2011-2020 David Robillard <http://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 "model.h" +#include "statement.h" +#include "world.h" + +#include "serd/serd.h" + +#include <stdlib.h> + +typedef struct { + SerdEnv* env; + SerdModel* model; + SerdNode* default_graph; +} SerdInserterData; + +static const SerdNode* +manage_or_intern(SerdNodes* nodes, SerdNode* manage, const SerdNode* intern) +{ + return manage ? serd_nodes_manage(nodes, manage) + : serd_nodes_intern(nodes, intern); +} + +static SerdStatus +serd_inserter_on_base(SerdInserterData* data, const SerdNode* uri) +{ + return serd_env_set_base_uri(data->env, uri); +} + +static SerdStatus +serd_inserter_on_prefix(SerdInserterData* data, + const SerdNode* name, + const SerdNode* uri) +{ + return serd_env_set_prefix(data->env, name, uri); +} + +static SerdStatus +serd_inserter_on_statement(SerdInserterData* data, + const SerdStatementFlags flags, + const SerdStatement* statement) +{ + (void)flags; + + const SerdNode* const subject = serd_statement_subject(statement); + const SerdNode* const predicate = serd_statement_predicate(statement); + const SerdNode* const object = serd_statement_object(statement); + const SerdNode* const graph = serd_statement_graph(statement); + + // Attempt to expand all nodes to eliminate CURIEs + SerdNode* const s = serd_env_expand(data->env, subject); + SerdNode* const p = serd_env_expand(data->env, predicate); + SerdNode* const o = serd_env_expand(data->env, object); + SerdNode* const g = serd_env_expand(data->env, graph); + + SerdNodes* const nodes = data->model->world->nodes; + const SerdNode* model_graph = manage_or_intern(nodes, g, graph); + if (!model_graph) { + model_graph = serd_nodes_intern(nodes, data->default_graph); + } + + const SerdStatus st = serd_model_add_internal( + data->model, + (data->model->flags & SERD_STORE_CURSORS) ? statement->cursor + : NULL, + manage_or_intern(nodes, s, subject), + manage_or_intern(nodes, p, predicate), + manage_or_intern(nodes, o, object), + model_graph); + + return st > SERD_FAILURE ? st : SERD_SUCCESS; +} + +static void +free_data(void* handle) +{ + if (handle) { + SerdInserterData* data = (SerdInserterData*)handle; + + serd_node_free(data->default_graph); + free(data); + } +} + +SerdSink* +serd_inserter_new(SerdModel* model, SerdEnv* env, const SerdNode* default_graph) +{ + SerdInserterData* const data = + (SerdInserterData*)calloc(1, sizeof(SerdInserterData)); + + data->env = env; + data->model = model; + data->default_graph = serd_node_copy(default_graph); + + SerdSink* const sink = serd_sink_new(data, free_data); + + serd_sink_set_base_func(sink, (SerdBaseFunc)serd_inserter_on_base); + serd_sink_set_prefix_func(sink, (SerdPrefixFunc)serd_inserter_on_prefix); + serd_sink_set_statement_func(sink, + (SerdStatementFunc)serd_inserter_on_statement); + + return sink; +} |