fastly_world_adapter.cpp

#include <algorithm>
#include <string_view>
#include <vector>

#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Winvalid-offsetof"
#include "js/Utility.h"
#include "jsapi.h"
#pragma clang diagnostic pop

#include "core/allocator.h"
#include "fastly_world.h"
#include "host_interface/fastly.h"

// Ensure that all the things we want to use the hostcall buffer for actually
// fit into the buffer.
#define HOSTCALL_BUFFER_LEN HEADER_MAX_LEN
static_assert(DICTIONARY_ENTRY_MAX_LEN < HOSTCALL_BUFFER_LEN);
static_assert(METHOD_MAX_LEN < HOSTCALL_BUFFER_LEN);
static_assert(URI_MAX_LEN < HOSTCALL_BUFFER_LEN);

#define LIST_ALLOC_SIZE 50

static bool convert_result(int res, fastly_error_t *err) {
  if (res == 0)
    return true;
  switch (res) {
  case 1:
    *err = FASTLY_ERROR_GENERIC_ERROR;
    break;
  case 2:
    *err = FASTLY_ERROR_INVALID_ARGUMENT;
    break;
  case 3:
    *err = FASTLY_ERROR_BAD_HANDLE;
    break;
  case 4:
    *err = FASTLY_ERROR_BUFFER_LEN;
    break;
  case 5:
    *err = FASTLY_ERROR_UNSUPPORTED;
    break;
  case 6:
    *err = FASTLY_ERROR_BAD_ALIGN;
    break;
  case 7:
    *err = FASTLY_ERROR_HTTP_INVALID;
    break;
  case 8:
    *err = FASTLY_ERROR_HTTP_USER;
    break;
  case 9:
    *err = FASTLY_ERROR_HTTP_INCOMPLETE;
    break;
  case 10:
    *err = FASTLY_ERROR_OPTIONAL_NONE;
    break;
  case 11:
    *err = FASTLY_ERROR_HTTP_HEAD_TOO_LARGE;
    break;
  case 12:
    *err = FASTLY_ERROR_HTTP_INVALID_STATUS;
    break;
  case 13:
    *err = FASTLY_ERROR_LIMIT_EXCEEDED;
    break;
  case 100:
    *err = FASTLY_ERROR_UNKNOWN_ERROR;
    break;
  default:
    *err = FASTLY_ERROR_UNKNOWN_ERROR;
  }
  return false;
}

fastly_http_version_t convert_http_version(uint32_t version) {
  switch (version) {
  case 0:
    return FASTLY_HTTP_VERSION_HTTP09;
  case 1:
    return FASTLY_HTTP_VERSION_HTTP10;
  case 2:
    return FASTLY_HTTP_VERSION_HTTP11;
  case 3:
    return FASTLY_HTTP_VERSION_H2;
  case 4:
  default:
    return FASTLY_HTTP_VERSION_H3;
  }
}

bool fastly_http_body_new(fastly_body_handle_t *ret, fastly_error_t *err) {
  return convert_result(fastly::body_new(ret), err);
}

bool fastly_http_body_append(fastly_body_handle_t src, fastly_body_handle_t dest,
                             fastly_error_t *err) {
  return convert_result(fastly::body_append(src, dest), err);
}

bool fastly_http_body_read(fastly_body_handle_t h, uint32_t chunk_size, fastly_world_list_u8_t *ret,
                           fastly_error_t *err) {
  ret->ptr = static_cast<uint8_t *>(cabi_malloc(chunk_size, 1));
  return convert_result(fastly::body_read(h, reinterpret_cast<char *>(ret->ptr),
                                          static_cast<size_t>(chunk_size), &ret->len),
                        err);
}

bool fastly_http_body_write(fastly_body_handle_t h, fastly_world_list_u8_t *buf,
                            fastly_body_write_end_t end, uint32_t *ret, fastly_error_t *err) {
  return convert_result(fastly::body_write(h, reinterpret_cast<char *>(buf->ptr), buf->len,
                                           end == FASTLY_BODY_WRITE_END_BACK
                                               ? fastly::BodyWriteEnd::BodyWriteEndBack
                                               : fastly::BodyWriteEnd::BodyWriteEndFront,
                                           reinterpret_cast<size_t *>(ret)),
                        err);
}

bool fastly_http_body_close(fastly_body_handle_t h, fastly_error_t *err) {
  return convert_result(fastly::body_close(h), err);
}

bool fastly_log_endpoint_get(fastly_world_string_t *name, fastly_log_endpoint_handle_t *ret,
                             fastly_error_t *err) {
  return convert_result(
      fastly::log_endpoint_get(reinterpret_cast<const char *>(name->ptr), name->len, ret), err);
}

bool fastly_log_write(fastly_log_endpoint_handle_t h, fastly_world_string_t *msg,
                      fastly_error_t *err) {
  size_t nwritten = 0;
  return convert_result(
      fastly::log_write(h, reinterpret_cast<const char *>(msg->ptr), msg->len, &nwritten), err);
}

bool fastly_http_req_body_downstream_get(fastly_request_t *ret, fastly_error_t *err) {
  return convert_result(fastly::req_body_downstream_get(&ret->f0, &ret->f1), err);
}

bool fastly_http_req_redirect_to_grip_proxy(fastly_world_string_t *backend, fastly_error_t *err) {
  return convert_result(fastly::req_redirect_to_grip_proxy(
                            reinterpret_cast<const char *>(backend->ptr), backend->len),
                        err);
}

int convert_tag(fastly_http_cache_override_tag_t tag) {
  int out_tag = 0;
  if ((tag & FASTLY_HTTP_CACHE_OVERRIDE_TAG_PASS) > 0) {
    out_tag |= CACHE_OVERRIDE_PASS;
  }
  if ((tag & FASTLY_HTTP_CACHE_OVERRIDE_TAG_TTL) > 0) {
    out_tag |= CACHE_OVERRIDE_TTL;
  }
  if ((tag & FASTLY_HTTP_CACHE_OVERRIDE_TAG_STALE_WHILE_REVALIDATE) > 0) {
    out_tag |= CACHE_OVERRIDE_STALE_WHILE_REVALIDATE;
  }
  if ((tag & FASTLY_HTTP_CACHE_OVERRIDE_TAG_PCI) > 0) {
    out_tag |= CACHE_OVERRIDE_PCI;
  }
  return out_tag;
}

bool fastly_http_req_cache_override_set(fastly_request_handle_t h,
                                        fastly_http_cache_override_tag_t tag, uint32_t *maybe_ttl,
                                        uint32_t *maybe_stale_while_revalidate,
                                        fastly_world_string_t *maybe_sk, fastly_error_t *err) {
  fastly_world_string_t sk_str;
  if (maybe_sk) {
    sk_str = *maybe_sk;
  } else {
    sk_str.len = 0;
    sk_str.ptr = NULL;
  }
  return convert_result(
      fastly::req_cache_override_v2_set(
          h, convert_tag(tag), maybe_ttl == NULL ? 0 : *maybe_ttl,
          maybe_stale_while_revalidate == NULL ? 0 : *maybe_stale_while_revalidate,
          reinterpret_cast<char *>(sk_str.ptr), sk_str.len),
      err);
}

bool fastly_http_req_downstream_client_ip_addr(fastly_world_list_u8_t *ret, fastly_error_t *err) {
  ret->ptr = static_cast<uint8_t *>(cabi_malloc(16, 1));
  return convert_result(
      fastly::req_downstream_client_ip_addr_get(reinterpret_cast<char *>(ret->ptr), &ret->len),
      err);
}

bool fastly_http_req_downstream_tls_cipher_openssl_name(fastly_world_string_t *ret,
                                                        fastly_error_t *err) {
  ret->ptr = static_cast<char *>(cabi_malloc(128, 1));
  return convert_result(fastly::req_downstream_tls_cipher_openssl_name(
                            reinterpret_cast<char *>(ret->ptr), 128, &ret->len),
                        err);
}

bool fastly_http_req_downstream_tls_protocol(fastly_world_string_t *ret, fastly_error_t *err) {
  ret->ptr = static_cast<char *>(cabi_malloc(32, 1));
  return convert_result(
      fastly::req_downstream_tls_protocol(reinterpret_cast<char *>(ret->ptr), 32, &ret->len), err);
}

bool fastly_http_req_downstream_tls_raw_client_certificate(fastly_world_list_u8_t *ret,
                                                           fastly_error_t *err) {
  auto default_size = 4096;
  ret->ptr = static_cast<uint8_t *>(cabi_malloc(default_size, 4));
  auto status = fastly::req_downstream_tls_raw_client_certificate(reinterpret_cast<char *>(ret->ptr), default_size, &ret->len);
  if (status == FASTLY_ERROR_BUFFER_LEN) {
    cabi_realloc(ret->ptr, default_size, 4, ret->len);
    status = fastly::req_downstream_tls_raw_client_certificate(reinterpret_cast<char *>(ret->ptr), ret->len, &ret->len);
  }
  return convert_result(status, err);
}

bool fastly_http_req_downstream_tls_ja3_md5(fastly_world_list_u8_t *ret, fastly_error_t *err) {
  ret->ptr = static_cast<uint8_t *>(cabi_malloc(16, 1));
  return convert_result(
      fastly::req_downstream_tls_ja3_md5(reinterpret_cast<char *>(ret->ptr), &ret->len), err);
}
bool fastly_http_req_downstream_tls_client_hello(fastly_world_list_u8_t *ret, fastly_error_t *err) {
  auto default_size = 512;
  ret->ptr = static_cast<uint8_t *>(cabi_malloc(default_size, 4));
  auto status = fastly::req_downstream_tls_client_hello(reinterpret_cast<char *>(ret->ptr), default_size, &ret->len);
  if (status == FASTLY_ERROR_BUFFER_LEN) {
    cabi_realloc(ret->ptr, default_size, 4, ret->len);
    status = fastly::req_downstream_tls_client_hello(reinterpret_cast<char *>(ret->ptr), ret->len, &ret->len);
  }
  return convert_result(status, err);
}

bool fastly_http_req_new(fastly_request_handle_t *ret, fastly_error_t *err) {
  return convert_result(fastly::req_new(ret), err);
}

struct Chunk {
  JS::UniqueChars buffer;
  size_t length;

  static Chunk make(std::string_view data) {
    Chunk res{JS::UniqueChars{static_cast<char *>(cabi_malloc(data.size(), 1))}, data.size()};
    std::copy(data.begin(), data.end(), res.buffer.get());
    return res;
  }
};

bool fastly_http_req_header_names_get(fastly_request_handle_t h, fastly_world_list_string_t *ret,
                                      fastly_error_t *err) {
  std::vector<Chunk> header_names;
  {
    JS::UniqueChars buf{static_cast<char *>(cabi_malloc(HOSTCALL_BUFFER_LEN, 1))};
    uint32_t cursor = 0;
    while (true) {
      size_t length = 0;
      int64_t ending_cursor = 0;
      auto res = fastly::req_header_names_get(h, buf.get(), HEADER_MAX_LEN, cursor, &ending_cursor,
                                              &length);
      if (!convert_result(res, err)) {
        return false;
      }

      if (length == 0) {
        break;
      }

      std::string_view result{buf.get(), length};
      while (!result.empty()) {
        auto end = result.find('\0');
        header_names.emplace_back(Chunk::make(result.substr(0, end)));
        if (end == result.npos) {
          break;
        }

        result = result.substr(end + 1);
      }

      if (ending_cursor < 0) {
        break;
      }

      cursor = ending_cursor;
    }
  }

  ret->len = header_names.size();
  ret->ptr = static_cast<fastly_world_string_t *>(cabi_malloc(
      header_names.size() * sizeof(fastly_world_string_t), alignof(fastly_world_string_t)));
  auto *next = ret->ptr;
  for (auto &chunk : header_names) {
    next->len = chunk.length;
    next->ptr = chunk.buffer.release();
    ++next;
  }

  return true;
}

bool fastly_http_req_header_values_get(fastly_request_handle_t h, fastly_world_string_t *name,
                                       fastly_world_option_list_string_t *ret,
                                       fastly_error_t *err) {

  std::vector<Chunk> header_values;

  {
    JS::UniqueChars buffer(static_cast<char *>(cabi_malloc(HEADER_MAX_LEN, 1)));
    uint32_t cursor = 0;
    while (true) {
      int64_t ending_cursor = 0;
      size_t length = 0;
      auto res = fastly::req_header_values_get(h, name->ptr, name->len, buffer.get(),
                                               HEADER_MAX_LEN, cursor, &ending_cursor, &length);
      if (!convert_result(res, err)) {
        return false;
      }

      if (length == 0) {
        break;
      }

      std::string_view result{buffer.get(), length};
      while (!result.empty()) {
        auto end = result.find('\0');
        header_values.emplace_back(Chunk::make(result.substr(0, end)));
        if (end == result.npos) {
          break;
        }

        result = result.substr(end + 1);
      }

      if (ending_cursor < 0) {
        break;
      }
    }
  }

  if (header_values.empty()) {
    ret->is_some = false;
  } else {
    ret->is_some = true;
    ret->val.len = header_values.size();
    ret->val.ptr = static_cast<fastly_world_string_t *>(cabi_malloc(
        header_values.size() * sizeof(fastly_world_string_t), alignof(fastly_world_string_t)));
    auto *next = ret->val.ptr;
    for (auto &chunk : header_values) {
      next->len = chunk.length;
      next->ptr = chunk.buffer.release();
      ++next;
    }
  }

  return true;
}

bool fastly_http_req_header_insert(fastly_request_handle_t h, fastly_world_string_t *name,
                                   fastly_world_string_t *value, fastly_error_t *err) {
  return convert_result(fastly::req_header_insert(h, name->ptr, name->len, value->ptr, value->len),
                        err);
}

bool fastly_http_req_header_append(fastly_request_handle_t h, fastly_world_string_t *name,
                                   fastly_world_string_t *value, fastly_error_t *err) {
  return convert_result(fastly::req_header_append(h, name->ptr, name->len, value->ptr, value->len),
                        err);
}

bool fastly_http_req_header_remove(fastly_request_handle_t h, fastly_world_string_t *name,
                                   fastly_error_t *err) {
  return convert_result(fastly::req_header_remove(h, name->ptr, name->len), err);
}

bool fastly_http_req_method_get(fastly_request_handle_t h, fastly_world_string_t *ret,
                                fastly_error_t *err) {
  ret->ptr = static_cast<char *>(cabi_malloc(METHOD_MAX_LEN, 1));
  return convert_result(
      fastly::req_method_get(h, reinterpret_cast<char *>(ret->ptr), METHOD_MAX_LEN, &ret->len),
      err);
}

bool fastly_http_req_method_set(fastly_request_handle_t h, fastly_world_string_t *method,
                                fastly_error_t *err) {
  return convert_result(
      fastly::req_method_set(h, reinterpret_cast<const char *>(method->ptr), method->len), err);
}

bool fastly_http_req_uri_get(fastly_request_handle_t h, fastly_world_string_t *ret,
                             fastly_error_t *err) {
  ret->ptr = static_cast<char *>(cabi_malloc(URI_MAX_LEN, 1));
  if (!convert_result(fastly::req_uri_get(h, ret->ptr, URI_MAX_LEN, &ret->len), err)) {
    cabi_free(ret->ptr);
    return false;
  }
  ret->ptr = static_cast<char *>(cabi_realloc(ret->ptr, URI_MAX_LEN, 1, ret->len));
  return true;
}

bool fastly_http_req_uri_set(fastly_request_handle_t h, fastly_world_string_t *uri,
                             fastly_error_t *err) {
  return convert_result(fastly::req_uri_set(h, uri->ptr, uri->len), err);
}

bool fastly_http_req_version_get(fastly_request_handle_t h, fastly_http_version_t *ret,
                                 fastly_error_t *err) {
  uint32_t fastly_http_version;
  if (!convert_result(fastly::req_version_get(h, &fastly_http_version), err)) {
    return false;
  }
  *ret = convert_http_version(fastly_http_version);
  return true;
}

bool fastly_http_req_send_async(fastly_request_handle_t h, fastly_body_handle_t b,
                                fastly_world_string_t *backend,
                                fastly_pending_request_handle_t *ret, fastly_error_t *err) {
  return convert_result(fastly::req_send_async(h, b, backend->ptr, backend->len, ret), err);
}

bool fastly_http_req_send_async_streaming(fastly_request_handle_t h, fastly_body_handle_t b,
                                          fastly_world_string_t *backend,
                                          fastly_pending_request_handle_t *ret,
                                          fastly_error_t *err) {
  return convert_result(fastly::req_send_async_streaming(h, b, backend->ptr, backend->len, ret),
                        err);
}

bool fastly_http_req_pending_req_wait(fastly_pending_request_handle_t h, fastly_response_t *ret,
                                      fastly_error_t *err) {
  return convert_result(fastly::req_pending_req_wait(h, &ret->f0, &ret->f1), err);
}

bool fastly_http_req_register_dynamic_backend(fastly_world_string_t *prefix,
                                              fastly_world_string_t *target,
                                              fastly_dynamic_backend_config_t *config,
                                              fastly_error_t *err) {
  uint32_t backend_config_mask = 0;

  if (config->host_override.is_some)
    backend_config_mask |= BACKEND_CONFIG_HOST_OVERRIDE;
  if (config->connect_timeout.is_some)
    backend_config_mask |= BACKEND_CONFIG_CONNECT_TIMEOUT;
  if (config->use_ssl.is_some)
    backend_config_mask |= BACKEND_CONFIG_USE_SSL;
  if (config->ssl_min_version.is_some)
    backend_config_mask |= BACKEND_CONFIG_SSL_MIN_VERSION;
  if (config->ssl_max_version.is_some)
    backend_config_mask |= BACKEND_CONFIG_SSL_MAX_VERSION;
  if (config->cert_hostname.is_some)
    backend_config_mask |= BACKEND_CONFIG_CERT_HOSTNAME;
  if (config->ca_cert.is_some)
    backend_config_mask |= BACKEND_CONFIG_CA_CERT;
  if (config->ciphers.is_some)
    backend_config_mask |= BACKEND_CONFIG_CIPHERS;
  if (config->sni_hostname.is_some)
    backend_config_mask |= BACKEND_CONFIG_SNI_HOSTNAME;
  fastly::DynamicBackendConfig backend_configuration{
      .host_override = config->host_override.val.ptr,
      .host_override_len = config->host_override.val.len,
      .connect_timeout_ms = config->connect_timeout.val,
      .first_byte_timeout_ms = config->first_byte_timeout.val,
      .between_bytes_timeout_ms = config->between_bytes_timeout.val,
      .ssl_min_version = config->ssl_min_version.val,
      .ssl_max_version = config->ssl_max_version.val,
      .cert_hostname = config->cert_hostname.val.ptr,
      .cert_hostname_len = config->cert_hostname.val.len,
      .ca_cert = config->ca_cert.val.ptr,
      .ca_cert_len = config->ca_cert.val.len,
      .ciphers = config->ciphers.val.ptr,
      .ciphers_len = config->ciphers.val.len,
      .sni_hostname = config->sni_hostname.val.ptr,
      .sni_hostname_len = config->sni_hostname.val.len,
  };
  return convert_result(fastly::req_register_dynamic_backend(prefix->ptr, prefix->len, target->ptr,
                                                             target->len, backend_config_mask,
                                                             &backend_configuration),
                        err);
}

bool fastly_http_resp_new(fastly_response_handle_t *ret, fastly_error_t *err) {
  return convert_result(fastly::resp_new(ret), err);
}

bool fastly_http_resp_header_names_get(fastly_response_handle_t h, fastly_world_list_string_t *ret,
                                       fastly_error_t *err) {
  fastly_world_string_t *strs = static_cast<fastly_world_string_t *>(
      cabi_malloc(LIST_ALLOC_SIZE * sizeof(fastly_world_string_t), 1));
  size_t str_max = LIST_ALLOC_SIZE;
  size_t str_cnt = 0;
  size_t nwritten;
  char *buf = static_cast<char *>(cabi_malloc(HOSTCALL_BUFFER_LEN, 1));
  uint32_t cursor = 0;
  int64_t next_cursor = 0;
  while (true) {
    if (!convert_result(
            fastly::resp_header_names_get(h, buf, HEADER_MAX_LEN, cursor, &next_cursor, &nwritten),
            err)) {
      cabi_free(buf);
      return false;
    }
    if (nwritten == 0) {
      break;
    }
    uint32_t offset = 0;
    for (size_t i = 0; i < nwritten; i++) {
      if (buf[i] != '\0')
        continue;
      if (str_cnt == str_max) {
        strs = static_cast<fastly_world_string_t *>(
            cabi_realloc(strs, str_max * sizeof(fastly_world_string_t), 1,
                         (str_max + LIST_ALLOC_SIZE) * sizeof(fastly_world_string_t)));
        str_max += LIST_ALLOC_SIZE;
      }
      strs[str_cnt].ptr = static_cast<char *>(cabi_malloc(i - offset + 1, 1));
      strs[str_cnt].len = i - offset;
      memcpy(strs[str_cnt].ptr, buf + offset, i - offset + 1);
      offset = i + 1;
      str_cnt++;
    }
    if (next_cursor < 0)
      break;
    cursor = (uint32_t)next_cursor;
  }
  cabi_free(buf);
  strs = static_cast<fastly_world_string_t *>(cabi_realloc(
      strs, str_max * sizeof(fastly_world_string_t), 1, str_cnt * sizeof(fastly_world_string_t)));
  ret->ptr = strs;
  ret->len = str_cnt;
  return true;
}

bool fastly_http_resp_header_values_get(fastly_response_handle_t h, fastly_world_string_t *name,
                                        fastly_world_option_list_string_t *ret,
                                        fastly_error_t *err) {
  size_t str_max = LIST_ALLOC_SIZE;
  fastly_world_string_t *strs =
      static_cast<fastly_world_string_t *>(cabi_malloc(str_max * sizeof(fastly_world_string_t), 1));
  size_t str_cnt = 0;
  size_t nwritten;
  char *buf = static_cast<char *>(cabi_malloc(HOSTCALL_BUFFER_LEN, 1));
  uint32_t cursor = 0;
  int64_t next_cursor = 0;
  while (true) {
    if (!convert_result(fastly::resp_header_values_get(h, name->ptr, name->len, buf, HEADER_MAX_LEN,
                                                       cursor, &next_cursor, &nwritten),
                        err)) {
      cabi_free(buf);
      return false;
    }
    if (nwritten == 0)
      break;
    uint32_t offset = 0;
    for (size_t i = 0; i < nwritten; i++) {
      if (buf[i] != '\0')
        continue;
      if (str_cnt == str_max) {
        strs = static_cast<fastly_world_string_t *>(
            cabi_realloc(strs, str_max * sizeof(fastly_world_string_t), 1,
                         (str_max + LIST_ALLOC_SIZE) * sizeof(fastly_world_string_t)));
        str_max += LIST_ALLOC_SIZE;
      }
      strs[str_cnt].ptr = static_cast<char *>(cabi_malloc(i - offset + 1, 1));
      strs[str_cnt].len = i - offset;
      memcpy(strs[str_cnt].ptr, buf + offset, i - offset + 1);
      offset = i + 1;
      str_cnt++;
    }
    if (next_cursor < 0)
      break;
    cursor = static_cast<uint32_t>(next_cursor);
  }
  cabi_free(buf);
  if (str_cnt > 0) {
    ret->is_some = true;
    ret->val.ptr = strs;
    ret->val.len = str_cnt;
    strs = static_cast<fastly_world_string_t *>(cabi_realloc(
        strs, str_max * sizeof(fastly_world_string_t), 1, str_cnt * sizeof(fastly_world_string_t)));
  } else {
    ret->is_some = false;
    cabi_free(strs);
  }
  return true;
}

bool fastly_http_resp_header_insert(fastly_response_handle_t h, fastly_world_string_t *name,
                                    fastly_world_string_t *value, fastly_error_t *err) {
  return convert_result(fastly::resp_header_insert(h, name->ptr, name->len, value->ptr, value->len),
                        err);
}

bool fastly_http_resp_header_append(fastly_response_handle_t h, fastly_world_string_t *name,
                                    fastly_world_string_t *value, fastly_error_t *err) {
  return convert_result(fastly::resp_header_append(h, name->ptr, name->len, value->ptr, value->len),
                        err);
}

bool fastly_http_resp_header_remove(fastly_response_handle_t h, fastly_world_string_t *name,
                                    fastly_error_t *err) {
  return convert_result(fastly::resp_header_remove(h, name->ptr, name->len), err);
}

bool fastly_http_resp_version_get(fastly_response_handle_t h, fastly_http_version_t *ret,
                                  fastly_error_t *err) {
  uint32_t fastly_http_version;
  if (!convert_result(fastly::resp_version_get(h, &fastly_http_version), err)) {
    return false;
  }
  *ret = convert_http_version(fastly_http_version);
  return true;
}

bool fastly_http_resp_send_downstream(fastly_response_handle_t h, fastly_body_handle_t b,
                                      bool streaming, fastly_error_t *err) {
  return convert_result(fastly::resp_send_downstream(h, b, streaming), err);
}

bool fastly_http_resp_status_get(fastly_response_handle_t h, fastly_http_status_t *ret,
                                 fastly_error_t *err) {
  return convert_result(fastly::resp_status_get(h, ret), err);
}

bool fastly_http_resp_status_set(fastly_response_handle_t h, fastly_http_status_t status,
                                 fastly_error_t *err) {
  return convert_result(fastly::resp_status_set(h, status), err);
}

bool fastly_dictionary_open(fastly_world_string_t *name, fastly_dictionary_handle_t *ret,
                            fastly_error_t *err) {
  return convert_result(fastly::dictionary_open(name->ptr, name->len, ret), err);
}

bool fastly_dictionary_get(fastly_dictionary_handle_t h, fastly_world_string_t *key,
                           fastly_world_option_string_t *ret, fastly_error_t *err) {
  ret->val.ptr = static_cast<char *>(cabi_malloc(DICTIONARY_ENTRY_MAX_LEN, 1));
  if (!convert_result(fastly::dictionary_get(h, key->ptr, key->len, ret->val.ptr,
                                             DICTIONARY_ENTRY_MAX_LEN, &ret->val.len),
                      err)) {
    if (*err == FASTLY_ERROR_OPTIONAL_NONE) {
      ret->is_some = false;
      return true;
    } else {
      cabi_free(ret->val.ptr);
      return false;
    }
  }
  ret->is_some = true;
  ret->val.ptr =
      static_cast<char *>(cabi_realloc(ret->val.ptr, DICTIONARY_ENTRY_MAX_LEN, 1, ret->val.len));
  return true;
}

bool fastly_secret_store_open(fastly_world_string_t *name, fastly_secret_store_handle_t *ret,
                              fastly_error_t *err) {
  return convert_result(fastly::secret_store_open(name->ptr, name->len, ret), err);
}

bool fastly_secret_store_get(fastly_secret_store_handle_t store, fastly_world_string_t *key,
                             fastly_world_option_secret_handle_t *ret, fastly_error_t *err) {
  ret->val = INVALID_HANDLE;
  bool ok = convert_result(fastly::secret_store_get(store, key->ptr, key->len, &ret->val), err);
  if ((!ok && *err == FASTLY_ERROR_OPTIONAL_NONE) || ret->val == INVALID_HANDLE) {
    ret->is_some = false;
    return true;
  }
  ret->is_some = true;
  return ok;
}

bool fastly_secret_store_plaintext(fastly_dictionary_handle_t h, fastly_world_option_string_t *ret,
                                   fastly_error_t *err) {
  ret->val.ptr = static_cast<char *>(JS_malloc(CONTEXT, DICTIONARY_ENTRY_MAX_LEN));
  if (!convert_result(
          fastly::secret_store_plaintext(h, ret->val.ptr, DICTIONARY_ENTRY_MAX_LEN, &ret->val.len),
          err)) {
    if (*err == FASTLY_ERROR_OPTIONAL_NONE) {
      ret->is_some = false;
      return true;
    } else {
      JS_free(CONTEXT, ret->val.ptr);
      return false;
    }
  }
  ret->is_some = true;
  ret->val.ptr = static_cast<char *>(
      JS_realloc(CONTEXT, ret->val.ptr, DICTIONARY_ENTRY_MAX_LEN, ret->val.len));
  return true;
}

bool fastly_geo_lookup(fastly_world_list_u8_t *addr_octets, fastly_world_string_t *ret,
                       fastly_error_t *err) {
  ret->ptr = static_cast<char *>(cabi_malloc(HOSTCALL_BUFFER_LEN, 1));
  if (!convert_result(fastly::geo_lookup(reinterpret_cast<char *>(addr_octets->ptr),
                                         addr_octets->len, ret->ptr, HOSTCALL_BUFFER_LEN,
                                         &ret->len),
                      err)) {
    cabi_free(ret->ptr);
    return false;
  }
  ret->ptr = static_cast<char *>(cabi_realloc(ret->ptr, HOSTCALL_BUFFER_LEN, 1, ret->len));
  return true;
}

bool fastly_object_store_open(fastly_world_string_t *name, fastly_object_store_handle_t *ret,
                              fastly_error_t *err) {
  return convert_result(fastly::object_store_open(name->ptr, name->len, ret), err);
}

bool fastly_object_store_lookup(fastly_object_store_handle_t store, fastly_world_string_t *key,
                                fastly_world_option_body_handle_t *ret, fastly_error_t *err) {
  ret->val = INVALID_HANDLE;
  bool ok = convert_result(fastly::object_store_get(store, key->ptr, key->len, &ret->val), err);
  if ((!ok && *err == FASTLY_ERROR_OPTIONAL_NONE) || ret->val == INVALID_HANDLE) {
    ret->is_some = false;
    return true;
  }
  ret->is_some = true;
  return ok;
}

bool fastly_object_store_insert(fastly_object_store_handle_t store, fastly_world_string_t *key,
                                fastly_body_handle_t body_handle, fastly_error_t *err) {
  return convert_result(fastly::object_store_insert(store, key->ptr, key->len, body_handle), err);
}

bool fastly_async_io_select(fastly_world_list_async_handle_t *hs, uint32_t timeout_ms,
                            fastly_world_option_u32_t *ret, fastly_error_t *err) {
  if (!convert_result(fastly::async_select(hs->ptr, hs->len, timeout_ms, &ret->val), err)) {
    if (*err == FASTLY_ERROR_OPTIONAL_NONE) {
      ret->is_some = false;
      return true;
    }
    return false;
  }

  // The result is only valid if the timeout didn't expire.
  ret->is_some = ret->val != UINT32_MAX;

  return true;
}
bool fastly_async_io_is_ready(fastly_async_handle_t handle, bool *ret, fastly_error_t *err) {
  uint32_t ret_int;
  if (!convert_result(fastly::async_is_ready(handle, &ret_int), err)) {
    return false;
  }
  *ret = (bool)ret_int;
  return true;
}

bool fastly_purge_surrogate_key(fastly_world_string_t *surrogate_key,
                                fastly_purge_options_mask_t options_mask,
                                fastly_world_option_string_t *ret, fastly_error_t *err) {
  fastly::PurgeOptions options{nullptr, 0, nullptr};

  // Currently this host-call has been implemented to support the `SimpleCache.delete(key)` method,
  // which uses hard-purging and not soft-purging.
  // TODO: Create a JS API for this hostcall which supports hard-purging and another which supports
  // soft-purging. E.G. `fastly.purgeSurrogateKey(key)` and `fastly.softPurgeSurrogateKey(key)`
  MOZ_ASSERT(!(options_mask & FASTLY_PURGE_OPTIONS_MASK_SOFT_PURGE));
  MOZ_ASSERT(!(options_mask & FASTLY_PURGE_OPTIONS_MASK_RET_BUF));

  ret->is_some = false;

  return convert_result(
      fastly::purge_surrogate_key(surrogate_key->ptr, surrogate_key->len, options_mask, &options),
      err);
}

#define FASTLY_CACHE_LOOKUP_OPTIONS_MASK_RESERVED (1 << 0)
#define FASTLY_CACHE_LOOKUP_OPTIONS_MASK_REQUEST_HEADERS (1 << 1)

bool fastly_cache_lookup(fastly_world_string_t *cache_key, fastly_cache_lookup_options_t *options,
                         fastly_cache_handle_t *ret, fastly_error_t *err) {
  // Currently this host-call has been implemented to support the `SimpleCache.get(key)` method,
  // which does not use any fields from `fastly_cache_lookup_options_t`.
  uint8_t options_mask = 0;
  return convert_result(
      fastly::cache_lookup(cache_key->ptr, cache_key->len, options_mask, options, ret), err);
}

#define FASTLY_CACHE_WRITE_OPTIONS_MASK_RESERVED (1 << 0)
#define FASTLY_CACHE_WRITE_OPTIONS_MASK_REQUEST_HEADERS (1 << 1)
#define FASTLY_CACHE_WRITE_OPTIONS_MASK_VARY_RULE (1 << 2)
#define FASTLY_CACHE_WRITE_OPTIONS_MASK_INITIAL_AGE_NS (1 << 3)
#define FASTLY_CACHE_WRITE_OPTIONS_MASK_STALE_WHILE_REVALIDATE_NS (1 << 4)
#define FASTLY_CACHE_WRITE_OPTIONS_MASK_SURROGATE_KEYS (1 << 5)
#define FASTLY_CACHE_WRITE_OPTIONS_MASK_LENGTH (1 << 6)
#define FASTLY_CACHE_WRITE_OPTIONS_MASK_USER_METADATA (1 << 7)
#define FASTLY_CACHE_WRITE_OPTIONS_MASK_SENSITIVE_DATA (1 << 8)

bool fastly_cache_insert(fastly_world_string_t *cache_key, fastly_cache_write_options_t *options,
                         fastly_body_handle_t *ret, fastly_error_t *err) {
  uint16_t options_mask = 0;
  fastly::CacheWriteOptions opts;
  std::memset(&opts, 0, sizeof(opts));
  opts.max_age_ns = options->max_age_ns;

  if (options->request_headers != INVALID_HANDLE && options->request_headers != 0) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_REQUEST_HEADERS;
    opts.request_headers = options->request_headers;
  }
  if (options->vary_rule.ptr != nullptr) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_VARY_RULE;
    opts.vary_rule_len = options->vary_rule.len;
    opts.vary_rule_ptr = reinterpret_cast<uint8_t *>(options->vary_rule.ptr);
  }
  if (options->initial_age_ns != 0) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_INITIAL_AGE_NS;
    opts.initial_age_ns = options->initial_age_ns;
  }
  if (options->stale_while_revalidate_ns != 0) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_STALE_WHILE_REVALIDATE_NS;
    opts.stale_while_revalidate_ns = options->stale_while_revalidate_ns;
  }
  if (options->surrogate_keys.ptr != nullptr) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_SURROGATE_KEYS;
    opts.surrogate_keys_len = options->surrogate_keys.len;
    opts.surrogate_keys_ptr = reinterpret_cast<uint8_t *>(options->surrogate_keys.ptr);
  }
  if (options->length != 0) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_LENGTH;
    opts.length = options->length;
  }
  if (options->user_metadata.ptr != nullptr) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_USER_METADATA;
    opts.user_metadata_len = options->user_metadata.len;
    opts.user_metadata_ptr = options->user_metadata.ptr;
  }
  if (options->sensitive_data) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_SENSITIVE_DATA;
  }
  return convert_result(
      fastly::cache_insert(cache_key->ptr, cache_key->len, options_mask, &opts, ret), err);
}
bool fastly_cache_get_body(fastly_cache_handle_t handle, fastly_cache_get_body_options_t *options,
                           fastly_body_handle_t *ret, fastly_error_t *err) {
  uint32_t options_mask = 0;
  bool ok = convert_result(fastly::cache_get_body(handle, options_mask, options, ret), err);
  if (!ok && *err == FASTLY_ERROR_OPTIONAL_NONE) {
    *ret = INVALID_HANDLE;
    return true;
  }
  return ok;
}
bool fastly_transaction_lookup(fastly_world_string_t *cache_key,
                               fastly_cache_lookup_options_t *options, fastly_cache_handle_t *ret,
                               fastly_error_t *err) {
  // Currently this host-call has been implemented to support the `SimpleCache.getOrSet` method,
  // which does not use any fields from `fastly_cache_lookup_options_t`.
  uint32_t options_mask = 0;
  return convert_result(
      fastly::cache_transaction_lookup(cache_key->ptr, cache_key->len, options_mask, options, ret),
      err);
}
bool fastly_transaction_insert_and_stream_back(fastly_cache_handle_t handle,
                                               fastly_cache_write_options_t *options,
                                               fastly_world_tuple2_body_handle_cache_handle_t *ret,
                                               fastly_error_t *err) {
  uint16_t options_mask = 0;
  fastly::CacheWriteOptions opts;
  std::memset(&opts, 0, sizeof(opts));
  opts.max_age_ns = options->max_age_ns;

  MOZ_ASSERT(options->request_headers == INVALID_HANDLE || options->request_headers == 0);

  if (options->vary_rule.ptr != nullptr) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_VARY_RULE;
    opts.vary_rule_len = options->vary_rule.len;
    opts.vary_rule_ptr = reinterpret_cast<uint8_t *>(options->vary_rule.ptr);
  }
  if (options->initial_age_ns != 0) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_INITIAL_AGE_NS;
    opts.initial_age_ns = options->initial_age_ns;
  }
  if (options->stale_while_revalidate_ns != 0) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_STALE_WHILE_REVALIDATE_NS;
    opts.stale_while_revalidate_ns = options->stale_while_revalidate_ns;
  }
  if (options->surrogate_keys.ptr != nullptr) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_SURROGATE_KEYS;
    opts.surrogate_keys_len = options->surrogate_keys.len;
    opts.surrogate_keys_ptr = reinterpret_cast<uint8_t *>(options->surrogate_keys.ptr);
  }
  if (options->length != 0) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_LENGTH;
    opts.length = options->length;
  }
  if (options->user_metadata.ptr != nullptr) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_USER_METADATA;
    opts.user_metadata_len = options->user_metadata.len;
    opts.user_metadata_ptr = options->user_metadata.ptr;
  }
  if (options->sensitive_data) {
    options_mask |= FASTLY_CACHE_WRITE_OPTIONS_MASK_SENSITIVE_DATA;
  }
  return convert_result(fastly::cache_transaction_insert_and_stream_back(handle, options_mask,
                                                                         &opts, &ret->f0, &ret->f1),
                        err);
}

/// Cancel an obligation to provide an object to the cache.
///
/// Useful if there is an error before streaming is possible, e.g. if a backend is unreachable.
bool fastly_transaction_cancel(fastly_cache_handle_t handle, fastly_error_t *err) {
  return convert_result(fastly::cache_transaction_cancel(handle), err);
}

bool fastly_cache_get_state(fastly_cache_handle_t handle, fastly_cache_lookup_state_t *ret,
                            fastly_error_t *err) {
  return convert_result(fastly::cache_get_state(handle, ret), err);
}