#include <winevt_c.h>

/* clang-format off */
/*
 * Document-class: Winevt::EventLog::Bookmark
 *
 * Bookmark for querying/subscribing Windows EventLog progress.
 *
 * @example
 *  require 'winevt'
 *
 *  @query = Winevt::EventLog::Query.new("Application", "*[System[(Level <= 3) and TimeCreated[timediff(@SystemTime) <= 86400000]]]")
 *  @bookmark = Winevt::EventLog::Bookmark.new
 *  @query.each do |xml|
 *    @bookmark.update(@query)
 *  end
 *
 *  puts @bookmark.render
 */
/* clang-format pn */

static void bookmark_free(void* ptr);

static const rb_data_type_t rb_winevt_bookmark_type = { "winevt/bookmark",
                                                        {
                                                          0,
                                                          bookmark_free,
                                                          0,
                                                        },
                                                        NULL,
                                                        NULL,
                                                        RUBY_TYPED_FREE_IMMEDIATELY };

static void
bookmark_free(void* ptr)
{
  struct WinevtBookmark* winevtBookmark = (struct WinevtBookmark*)ptr;
  if (winevtBookmark->bookmark)
    EvtClose(winevtBookmark->bookmark);

  xfree(ptr);
}

static VALUE
rb_winevt_bookmark_alloc(VALUE klass)
{
  VALUE obj;
  struct WinevtBookmark* winevtBookmark;
  obj = TypedData_Make_Struct(
    klass, struct WinevtBookmark, &rb_winevt_bookmark_type, winevtBookmark);
  return obj;
}

/*
 * Initalize Bookmark class. Receive XML string or nil.
 *
 * @overload initailize(options={})
 *   @option options [String] XML rendered Bookmark string.
 * @return [Bookmark]
 *
 */
static VALUE
rb_winevt_bookmark_initialize(int argc, VALUE* argv, VALUE self)
{
  PWSTR bookmarkXml;
  VALUE wbookmarkXmlBuf;
  DWORD len;
  struct WinevtBookmark* winevtBookmark;

  TypedData_Get_Struct(
    self, struct WinevtBookmark, &rb_winevt_bookmark_type, winevtBookmark);

  if (argc == 0) {
    winevtBookmark->bookmark = EvtCreateBookmark(NULL);
  } else if (argc == 1) {
    VALUE rb_bookmarkXml;
    rb_scan_args(argc, argv, "10", &rb_bookmarkXml);
    Check_Type(rb_bookmarkXml, T_STRING);

    // bookmarkXml : To wide char
    len = MultiByteToWideChar(
      CP_UTF8, 0, RSTRING_PTR(rb_bookmarkXml), RSTRING_LEN(rb_bookmarkXml), NULL, 0);
    bookmarkXml = ALLOCV_N(WCHAR, wbookmarkXmlBuf, len + 1);
    MultiByteToWideChar(CP_UTF8,
                        0,
                        RSTRING_PTR(rb_bookmarkXml),
                        RSTRING_LEN(rb_bookmarkXml),
                        bookmarkXml,
                        len);
    bookmarkXml[len] = L'\0';
    winevtBookmark->bookmark = EvtCreateBookmark(bookmarkXml);
    ALLOCV_END(wbookmarkXmlBuf);
  }

  return Qnil;
}

/*
 * This method updates bookmark and returns Bookmark instance.
 *
 * @param event [Query]
 * @return [Bookmark]
 */
static VALUE
rb_winevt_bookmark_update(VALUE self, VALUE event)
{
  struct WinevtQuery* winevtQuery;
  struct WinevtBookmark* winevtBookmark;

  winevtQuery = EventQuery(event);

  TypedData_Get_Struct(
    self, struct WinevtBookmark, &rb_winevt_bookmark_type, winevtBookmark);

  for (int i = 0; i < winevtQuery->count; i++) {
    if (!EvtUpdateBookmark(winevtBookmark->bookmark, winevtQuery->hEvents[i]))
      return Qfalse;
  }
  return Qtrue;
}

/*
 * This method renders bookmark class content.
 *
 * @return [String]
 */
static VALUE
rb_winevt_bookmark_render(VALUE self)
{
  struct WinevtBookmark* winevtBookmark;

  TypedData_Get_Struct(
    self, struct WinevtBookmark, &rb_winevt_bookmark_type, winevtBookmark);

  return render_to_rb_str(winevtBookmark->bookmark, EvtRenderBookmark);
}

void
Init_winevt_bookmark(VALUE rb_cEventLog)
{
  rb_cBookmark = rb_define_class_under(rb_cEventLog, "Bookmark", rb_cObject);

  rb_define_alloc_func(rb_cBookmark, rb_winevt_bookmark_alloc);
  rb_define_method(rb_cBookmark, "initialize", rb_winevt_bookmark_initialize, -1);
  rb_define_method(rb_cBookmark, "update", rb_winevt_bookmark_update, 1);
  rb_define_method(rb_cBookmark, "render", rb_winevt_bookmark_render, 0);
}