// Copyright 2019-2020 David Robillard <d@drobilla.net>
// SPDX-License-Identifier: ISC

#undef NDEBUG

#include "serd/buffer.h"
#include "serd/env.h"
#include "serd/node.h"
#include "serd/sink.h"
#include "serd/statement.h"
#include "serd/string_view.h"
#include "serd/syntax.h"
#include "serd/world.h"
#include "serd/writer.h"

#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define NS_RDF "http://www.w3.org/1999/02/22-rdf-syntax-ns#"

static void
check_output(SerdWriter* writer, SerdBuffer* buffer, const char* expected)
{
  serd_writer_finish(writer);
  serd_buffer_close(buffer);

  const char* output = (const char*)buffer->buf;

  assert(!strcmp(output, expected));

  buffer->len = 0;
}

static int
test(void)
{
  SerdBuffer buffer = {NULL, 0};
  SerdWorld* world  = serd_world_new();
  SerdEnv*   env    = serd_env_new(serd_empty_string());

  SerdNode* b1 = serd_new_blank(serd_string("b1"));
  SerdNode* l1 = serd_new_blank(serd_string("l1"));
  SerdNode* l2 = serd_new_blank(serd_string("l2"));
  SerdNode* s1 = serd_new_string(serd_string("s1"));
  SerdNode* s2 = serd_new_string(serd_string("s2"));

  SerdNode* rdf_first = serd_new_uri(serd_string(NS_RDF "first"));
  SerdNode* rdf_value = serd_new_uri(serd_string(NS_RDF "value"));
  SerdNode* rdf_rest  = serd_new_uri(serd_string(NS_RDF "rest"));
  SerdNode* rdf_nil   = serd_new_uri(serd_string(NS_RDF "nil"));

  serd_env_set_prefix(env, serd_string("rdf"), serd_string(NS_RDF));

  SerdWriter* writer =
    serd_writer_new(world, SERD_TURTLE, 0, env, serd_buffer_write, &buffer);

  const SerdSink* sink = serd_writer_sink(writer);

  // Simple lone list
  serd_sink_write(sink, SERD_TERSE_S | SERD_LIST_S, l1, rdf_first, s1, NULL);
  serd_sink_write(sink, 0, l1, rdf_rest, l2, NULL);
  serd_sink_write(sink, 0, l2, rdf_first, s2, NULL);
  serd_sink_write(sink, 0, l2, rdf_rest, rdf_nil, NULL);
  check_output(writer, &buffer, "( \"s1\" \"s2\" ) .\n");

  // Nested terse lists
  serd_sink_write(sink,
                  SERD_TERSE_S | SERD_LIST_S | SERD_TERSE_O | SERD_LIST_O,
                  l1,
                  rdf_first,
                  l2,
                  NULL);
  serd_sink_write(sink, 0, l2, rdf_first, s1, NULL);
  serd_sink_write(sink, 0, l1, rdf_rest, rdf_nil, NULL);
  serd_sink_write(sink, 0, l2, rdf_rest, rdf_nil, NULL);
  check_output(writer, &buffer, "( ( \"s1\" ) ) .\n");

  // List as object
  serd_sink_write(
    sink, SERD_EMPTY_S | SERD_LIST_O | SERD_TERSE_O, b1, rdf_value, l1, NULL);
  serd_sink_write(sink, 0, l1, rdf_first, s1, NULL);
  serd_sink_write(sink, 0, l1, rdf_rest, l2, NULL);
  serd_sink_write(sink, 0, l2, rdf_first, s2, NULL);
  serd_sink_write(sink, 0, l2, rdf_rest, rdf_nil, NULL);
  check_output(writer, &buffer, "[] rdf:value ( \"s1\" \"s2\" ) .\n");

  serd_buffer_close(&buffer);
  serd_writer_free(writer);
  serd_node_free(rdf_nil);
  serd_node_free(rdf_rest);
  serd_node_free(rdf_value);
  serd_node_free(rdf_first);
  serd_node_free(s2);
  serd_node_free(s1);
  serd_node_free(l2);
  serd_node_free(l1);
  serd_node_free(b1);
  serd_env_free(env);
  serd_world_free(world);
  free(buffer.buf);

  return 0;
}

int
main(void)
{
  return test();
}