extension_helpers.cpp

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

extern "C" {
#include "../extension/libherb/include/ast/ast_nodes.h"
#include "../extension/libherb/include/herb.h"
#include "../extension/libherb/include/location/location.h"
#include "../extension/libherb/include/location/position.h"
#include "../extension/libherb/include/location/range.h"
#include "../extension/libherb/include/lexer/token.h"
#include "../extension/libherb/include/lib/hb_array.h"
#include "../extension/libherb/include/lib/hb_buffer.h"
#include "../extension/libherb/include/lib/hb_string.h"
}

#include "error_helpers.h"
#include "nodes.h"

char* CheckString(napi_env env, napi_value value) {
  size_t length;
  size_t copied;
  napi_valuetype type;

  napi_typeof(env, value, &type);
  if (type != napi_string) {
    napi_throw_type_error(env, nullptr, "String expected");
    return nullptr;
  }

  napi_get_value_string_utf8(env, value, nullptr, 0, &length);
  char* result = (char*) malloc(length + 1);
  if (!result) {
    napi_throw_error(env, nullptr, "Memory allocation failed");
    return nullptr;
  }

  napi_get_value_string_utf8(env, value, result, length + 1, &copied);
  return result;
}

napi_value CreateString(napi_env env, const char* str) {
  napi_value result;
  napi_create_string_utf8(env, str, NAPI_AUTO_LENGTH, &result);
  return result;
}

napi_value CreateStringFromHbString(napi_env env, hb_string_T string) {
  if (hb_string_is_null(string)) {
    napi_value null_value;
    napi_get_null(env, &null_value);
    return null_value;
  }

  napi_value result;
  napi_create_string_utf8(env, string.data, string.length, &result);
  return result;
}


napi_value CreatePosition(napi_env env, position_T position) {
  napi_value result;
  napi_create_object(env, &result);

  napi_value line, column;
  napi_create_uint32(env, (uint32_t)position.line, &line);
  napi_create_uint32(env, (uint32_t)position.column, &column);

  napi_set_named_property(env, result, "line", line);
  napi_set_named_property(env, result, "column", column);

  return result;
}

napi_value CreateLocation(napi_env env, location_T location) {
  napi_value result;
  napi_create_object(env, &result);

  napi_value start = CreatePosition(env, location.start);
  napi_value end = CreatePosition(env, location.end);

  napi_set_named_property(env, result, "start", start);
  napi_set_named_property(env, result, "end", end);

  return result;
}

napi_value CreateRange(napi_env env, range_T range) {
  napi_value result;
  napi_create_array(env, &result);

  napi_value from, to;
  napi_create_uint32(env, (uint32_t)range.from, &from);
  napi_create_uint32(env, (uint32_t)range.to, &to);

  napi_set_element(env, result, 0, from);
  napi_set_element(env, result, 1, to);

  return result;
}

napi_value CreateToken(napi_env env, token_T* token) {
  if (!token) {
    napi_value null_value;
    napi_get_null(env, &null_value);
    return null_value;
  }

  napi_value result;
  napi_create_object(env, &result);

  napi_value value = CreateStringFromHbString(env, token->value);
  napi_value range = CreateRange(env, token->range);
  napi_value location = CreateLocation(env, token->location);
  napi_value type = CreateStringFromHbString(env, token_type_to_string(token->type));

  napi_set_named_property(env, result, "value", value);
  napi_set_named_property(env, result, "range", range);
  napi_set_named_property(env, result, "location", location);
  napi_set_named_property(env, result, "type", type);

  return result;
}

napi_value CreateLexResult(napi_env env, hb_array_T* tokens, napi_value source) {
  napi_value result, tokens_array, errors_array, warnings_array;

  napi_create_object(env, &result);
  napi_create_array(env, &tokens_array);
  napi_create_array(env, &errors_array);
  napi_create_array(env, &warnings_array);

  if (tokens) {
    for (size_t i = 0; i < hb_array_size(tokens); i++) {
      token_T* token = (token_T*)hb_array_get(tokens, i);

      if (token) {
        napi_value token_obj = CreateToken(env, token);
        napi_set_element(env, tokens_array, i, token_obj);
      }
    }
  }

  napi_set_named_property(env, result, "tokens", tokens_array);
  napi_set_named_property(env, result, "source", source);
  napi_set_named_property(env, result, "warnings", warnings_array);
  napi_set_named_property(env, result, "errors", errors_array);

  return result;
}

napi_value CreateParseResult(napi_env env, AST_DOCUMENT_NODE_T* root, napi_value source, parser_options_T* options) {
  napi_value result, errors_array, warnings_array;

  napi_create_object(env, &result);
  napi_create_array(env, &errors_array);
  napi_create_array(env, &warnings_array);

  napi_value ast_value;

  if (root) {
    ast_value = NodeFromCStruct(env, (AST_NODE_T*)root);
  } else {
    napi_get_null(env, &ast_value);
  }

  napi_set_named_property(env, result, "value", ast_value);
  napi_set_named_property(env, result, "source", source);
  napi_set_named_property(env, result, "warnings", warnings_array);
  napi_set_named_property(env, result, "errors", errors_array);

  napi_value options_object;
  napi_create_object(env, &options_object);

  napi_value strict_value, track_whitespace_value, analyze_value, action_view_helpers_value, render_nodes_value, prism_program_value, prism_nodes_value, prism_nodes_deep_value;

  napi_get_boolean(env, options->strict, &strict_value);
  napi_get_boolean(env, options->track_whitespace, &track_whitespace_value);
  napi_get_boolean(env, options->analyze, &analyze_value);
  napi_get_boolean(env, options->action_view_helpers, &action_view_helpers_value);
  napi_get_boolean(env, options->render_nodes, &render_nodes_value);
  napi_get_boolean(env, options->prism_program, &prism_program_value);
  napi_get_boolean(env, options->prism_nodes, &prism_nodes_value);
  napi_get_boolean(env, options->prism_nodes_deep, &prism_nodes_deep_value);

  napi_set_named_property(env, options_object, "strict", strict_value);
  napi_set_named_property(env, options_object, "track_whitespace", track_whitespace_value);
  napi_set_named_property(env, options_object, "analyze", analyze_value);
  napi_set_named_property(env, options_object, "action_view_helpers", action_view_helpers_value);
  napi_set_named_property(env, options_object, "render_nodes", render_nodes_value);
  napi_set_named_property(env, options_object, "prism_program", prism_program_value);
  napi_set_named_property(env, options_object, "prism_nodes", prism_nodes_value);
  napi_set_named_property(env, options_object, "prism_nodes_deep", prism_nodes_deep_value);

  napi_set_named_property(env, result, "options", options_object);

  return result;
}