// Copyright (c) 2011, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

#include "vm/bootstrap_natives.h"

#include "vm/exceptions.h"
#include "vm/native_entry.h"
#include "vm/object.h"

namespace dart {

DEFINE_NATIVE_ENTRY(StringBase_createFromCodePoints, 1) {
  GET_NATIVE_ARGUMENT(Array, a, arguments->At(0));
  // TODO(srdjan): Check that parameterized type is an int.
  Zone* zone = Isolate::Current()->current_zone();
  intptr_t len = a.Length();

  // Unbox the array and determine the maximum element width.
  bool is_one_byte_string = true;
  bool is_two_byte_string = true;
  uint32_t* temp = reinterpret_cast<uint32_t*>(
      zone->Allocate(len * sizeof(uint32_t)));  // NOLINT
  Smi& element = Smi::Handle();
  for (intptr_t i = 0; i < len; i++) {
    const Object& index_object = Object::Handle(a.At(i));
    if (!index_object.IsSmi()) {
      GrowableArray<const Object*> args;
      args.Add(&index_object);
      Exceptions::ThrowByType(Exceptions::kIllegalArgument, args);
    }
    element ^= index_object.raw();
    intptr_t value = element.Value();
    if (value < 0) {
      GrowableArray<const Object*> args;
      Exceptions::ThrowByType(Exceptions::kIllegalArgument, args);
    } else if (value > 0xFFFF) {
      is_one_byte_string = false;
      is_two_byte_string = false;
    } else if (value > 0xFF) {
      is_one_byte_string = false;
    }
    temp[i] = value;
  }
  String& result = String::Handle();
  if (is_one_byte_string) {
    result ^= OneByteString::New(temp, len, Heap::kNew);
  } else if (is_two_byte_string) {
    result ^= TwoByteString::New(temp, len, Heap::kNew);
  } else {
    result ^= FourByteString::New(temp, len, Heap::kNew);
  }
  arguments->SetReturn(result);
}


DEFINE_NATIVE_ENTRY(String_hashCode, 1) {
  const String& receiver = String::CheckedHandle(arguments->At(0));
  intptr_t hash_val = receiver.Hash();
  ASSERT(hash_val > 0);
  ASSERT(Smi::IsValid(hash_val));
  const Smi& hash_smi = Smi::Handle(Smi::New(hash_val));
  arguments->SetReturn(hash_smi);
}


DEFINE_NATIVE_ENTRY(String_getLength, 1) {
  const String& receiver = String::CheckedHandle(arguments->At(0));
  arguments->SetReturn(Smi::Handle(Smi::New(receiver.Length())));
}


static int32_t StringValueAt(const String& str, const Integer& index) {
  if (index.IsSmi()) {
    Smi& smi = Smi::Handle();
    smi ^= index.raw();
    int32_t index = smi.Value();
    if ((index < 0) || (index >= str.Length())) {
      GrowableArray<const Object*> arguments;
      arguments.Add(&smi);
      Exceptions::ThrowByType(Exceptions::kIndexOutOfRange, arguments);
    }
    return str.CharAt(index);
  } else {
    // An index larger than Smi is always illegal.
    GrowableArray<const Object*> arguments;
    arguments.Add(&index);
    Exceptions::ThrowByType(Exceptions::kIndexOutOfRange, arguments);
    return 0;
  }
}


DEFINE_NATIVE_ENTRY(String_charAt, 2) {
  const String& receiver = String::CheckedHandle(arguments->At(0));
  GET_NATIVE_ARGUMENT(Integer, index, arguments->At(1));
  uint32_t value = StringValueAt(receiver, index);
  ASSERT(value <= 0x10FFFF);
  arguments->SetReturn(String::Handle(String::NewSymbol(&value, 1)));
}

DEFINE_NATIVE_ENTRY(String_charCodeAt, 2) {
  const String& receiver = String::CheckedHandle(arguments->At(0));
  GET_NATIVE_ARGUMENT(Integer, index, arguments->At(1));
  int32_t value = StringValueAt(receiver, index);
  ASSERT(value >= 0);
  ASSERT(value <= 0x10FFFF);
  arguments->SetReturn(Smi::Handle(Smi::New(value)));
}


DEFINE_NATIVE_ENTRY(String_concat, 2) {
  const String& receiver = String::CheckedHandle(arguments->At(0));
  GET_NATIVE_ARGUMENT(String, b, arguments->At(1));
  const String& result = String::Handle(String::Concat(receiver, b));
  arguments->SetReturn(result);
}


// TODO(hausner): Remove obsolete String_plus.
DECLARE_FLAG(bool, allow_string_plus);
DEFINE_NATIVE_ENTRY(String_plus, 2) {
  const String& receiver = String::CheckedHandle(arguments->At(0));
  GET_NATIVE_ARGUMENT(String, b, arguments->At(1));
  if (!FLAG_allow_string_plus) {
    // Throw a noSuchMethod exception if operator + is not supported.
    const String& func_name = String::Handle(String::New("+"));
    const Array& func_args = Array::Handle(Array::New(1));
    func_args.SetAt(0, b);
    GrowableArray<const Object*> dart_arguments(3);
    dart_arguments.Add(&receiver);
    dart_arguments.Add(&func_name);
    dart_arguments.Add(&func_args);
    Exceptions::ThrowByType(Exceptions::kNoSuchMethod, dart_arguments);
  }
  const String& result = String::Handle(String::Concat(receiver, b));
  arguments->SetReturn(result);
}


DEFINE_NATIVE_ENTRY(String_toLowerCase, 1) {
  const String& receiver = String::CheckedHandle(arguments->At(0));
  ASSERT(!receiver.IsNull());
  const String& result = String::Handle(String::ToLowerCase(receiver));
  arguments->SetReturn(result);
}


DEFINE_NATIVE_ENTRY(String_toUpperCase, 1) {
  const String& receiver = String::CheckedHandle(arguments->At(0));
  ASSERT(!receiver.IsNull());
  const String& result = String::Handle(String::ToUpperCase(receiver));
  arguments->SetReturn(result);
}


DEFINE_NATIVE_ENTRY(Strings_concatAll, 1) {
  GET_NATIVE_ARGUMENT(Array, strings, arguments->At(0));
  ASSERT(!strings.IsNull());
  // Check that the array contains strings.
  Instance& elem = Instance::Handle();
  for (intptr_t i = 0; i < strings.Length(); i++) {
    elem ^= strings.At(i);
    if (elem.IsNull()) {
      GrowableArray<const Object*> args;
      Exceptions::ThrowByType(Exceptions::kNullPointer, args);
    }
    if (!elem.IsString()) {
      GrowableArray<const Object*> args;
      args.Add(&elem);
      Exceptions::ThrowByType(Exceptions::kIllegalArgument, args);
    }
  }
  const String& result = String::Handle(String::ConcatAll(strings));
  arguments->SetReturn(result);
}

}  // namespace dart