use crate::{
gc::{Finalize, Trace},
JsString, JsSymbol, JsValue,
};
use std::{convert::TryFrom, fmt};
mod attribute;
pub use attribute::Attribute;
use gc::unsafe_empty_trace;
#[derive(Default, Debug, Clone, Trace, Finalize)]
pub struct PropertyDescriptor {
enumerable: Option<bool>,
configurable: Option<bool>,
kind: DescriptorKind,
}
#[derive(Debug, Clone, Trace, Finalize)]
pub enum DescriptorKind {
Data {
value: Option<JsValue>,
writable: Option<bool>,
},
Accessor {
get: Option<JsValue>,
set: Option<JsValue>,
},
Generic,
}
impl Default for DescriptorKind {
fn default() -> Self {
Self::Generic
}
}
impl PropertyDescriptor {
#[inline]
pub fn is_accessor_descriptor(&self) -> bool {
matches!(self.kind, DescriptorKind::Accessor { .. })
}
#[inline]
pub fn is_data_descriptor(&self) -> bool {
matches!(self.kind, DescriptorKind::Data { .. })
}
#[inline]
pub fn is_generic_descriptor(&self) -> bool {
matches!(self.kind, DescriptorKind::Generic)
}
#[inline]
pub fn is_empty(&self) -> bool {
self.is_generic_descriptor() && self.enumerable.is_none() && self.configurable.is_none()
}
#[inline]
pub fn enumerable(&self) -> Option<bool> {
self.enumerable
}
#[inline]
pub fn configurable(&self) -> Option<bool> {
self.configurable
}
#[inline]
pub fn writable(&self) -> Option<bool> {
match self.kind {
DescriptorKind::Data { writable, .. } => writable,
_ => None,
}
}
#[inline]
pub fn value(&self) -> Option<&JsValue> {
match &self.kind {
DescriptorKind::Data { value, .. } => value.as_ref(),
_ => None,
}
}
#[inline]
pub fn get(&self) -> Option<&JsValue> {
match &self.kind {
DescriptorKind::Accessor { get, .. } => get.as_ref(),
_ => None,
}
}
#[inline]
pub fn set(&self) -> Option<&JsValue> {
match &self.kind {
DescriptorKind::Accessor { set, .. } => set.as_ref(),
_ => None,
}
}
#[inline]
pub fn expect_enumerable(&self) -> bool {
if let Some(enumerable) = self.enumerable {
enumerable
} else {
panic!("[[enumerable]] field not in property descriptor")
}
}
#[inline]
pub fn expect_configurable(&self) -> bool {
if let Some(configurable) = self.configurable {
configurable
} else {
panic!("[[configurable]] field not in property descriptor")
}
}
#[inline]
pub fn expect_writable(&self) -> bool {
if let Some(writable) = self.writable() {
writable
} else {
panic!("[[writable]] field not in property descriptor")
}
}
#[inline]
pub fn expect_value(&self) -> &JsValue {
if let Some(value) = self.value() {
value
} else {
panic!("[[value]] field not in property descriptor")
}
}
#[inline]
pub fn expect_get(&self) -> &JsValue {
if let Some(get) = self.get() {
get
} else {
panic!("[[get]] field not in property descriptor")
}
}
#[inline]
pub fn expect_set(&self) -> &JsValue {
if let Some(set) = self.set() {
set
} else {
panic!("[[set]] field not in property descriptor")
}
}
#[inline]
pub fn kind(&self) -> &DescriptorKind {
&self.kind
}
#[inline]
pub fn builder() -> PropertyDescriptorBuilder {
PropertyDescriptorBuilder::new()
}
#[inline]
pub fn into_accessor_defaulted(mut self) -> Self {
self.kind = DescriptorKind::Accessor {
get: self.get().cloned(),
set: self.set().cloned(),
};
PropertyDescriptorBuilder { inner: self }
.complete_with_defaults()
.build()
}
pub fn into_data_defaulted(mut self) -> Self {
self.kind = DescriptorKind::Data {
value: self.value().cloned(),
writable: self.writable(),
};
PropertyDescriptorBuilder { inner: self }
.complete_with_defaults()
.build()
}
#[inline]
pub fn complete_property_descriptor(self) -> Self {
PropertyDescriptorBuilder { inner: self }
.complete_with_defaults()
.build()
}
#[inline]
pub fn fill_with(&mut self, desc: Self) {
match (&mut self.kind, &desc.kind) {
(
DescriptorKind::Data { value, writable },
DescriptorKind::Data {
value: desc_value,
writable: desc_writable,
},
) => {
if let Some(desc_value) = desc_value {
*value = Some(desc_value.clone())
}
if let Some(desc_writable) = desc_writable {
*writable = Some(*desc_writable)
}
}
(
DescriptorKind::Accessor { get, set },
DescriptorKind::Accessor {
get: desc_get,
set: desc_set,
},
) => {
if let Some(desc_get) = desc_get {
*get = Some(desc_get.clone())
}
if let Some(desc_set) = desc_set {
*set = Some(desc_set.clone())
}
}
(_, DescriptorKind::Generic) => {}
_ => panic!("Tried to fill a descriptor with an incompatible descriptor"),
}
if let Some(enumerable) = desc.enumerable {
self.enumerable = Some(enumerable)
}
if let Some(configurable) = desc.configurable {
self.configurable = Some(configurable)
}
}
}
#[derive(Default, Debug, Clone)]
pub struct PropertyDescriptorBuilder {
inner: PropertyDescriptor,
}
impl PropertyDescriptorBuilder {
pub fn new() -> Self {
Self::default()
}
pub fn value<V: Into<JsValue>>(mut self, value: V) -> Self {
match self.inner.kind {
DescriptorKind::Data {
value: ref mut v, ..
} => *v = Some(value.into()),
_ => {
self.inner.kind = DescriptorKind::Data {
value: Some(value.into()),
writable: None,
}
}
}
self
}
pub fn writable(mut self, writable: bool) -> Self {
match self.inner.kind {
DescriptorKind::Data {
writable: ref mut w,
..
} => *w = Some(writable),
_ => {
self.inner.kind = DescriptorKind::Data {
value: None,
writable: Some(writable),
}
}
}
self
}
pub fn get<V: Into<JsValue>>(mut self, get: V) -> Self {
match self.inner.kind {
DescriptorKind::Accessor { get: ref mut g, .. } => *g = Some(get.into()),
_ => {
self.inner.kind = DescriptorKind::Accessor {
get: Some(get.into()),
set: None,
}
}
}
self
}
pub fn set<V: Into<JsValue>>(mut self, set: V) -> Self {
match self.inner.kind {
DescriptorKind::Accessor { set: ref mut s, .. } => *s = Some(set.into()),
_ => {
self.inner.kind = DescriptorKind::Accessor {
set: Some(set.into()),
get: None,
}
}
}
self
}
pub fn maybe_enumerable(mut self, enumerable: Option<bool>) -> Self {
if let Some(enumerable) = enumerable {
self = self.enumerable(enumerable);
}
self
}
pub fn maybe_configurable(mut self, configurable: Option<bool>) -> Self {
if let Some(configurable) = configurable {
self = self.configurable(configurable);
}
self
}
pub fn maybe_value<V: Into<JsValue>>(mut self, value: Option<V>) -> Self {
if let Some(value) = value {
self = self.value(value);
}
self
}
pub fn maybe_writable(mut self, writable: Option<bool>) -> Self {
if let Some(writable) = writable {
self = self.writable(writable);
}
self
}
pub fn maybe_get<V: Into<JsValue>>(mut self, get: Option<V>) -> Self {
if let Some(get) = get {
self = self.get(get);
}
self
}
pub fn maybe_set<V: Into<JsValue>>(mut self, set: Option<V>) -> Self {
if let Some(set) = set {
self = self.set(set);
}
self
}
pub fn enumerable(mut self, enumerable: bool) -> Self {
self.inner.enumerable = Some(enumerable);
self
}
pub fn configurable(mut self, configurable: bool) -> Self {
self.inner.configurable = Some(configurable);
self
}
pub fn complete_with_defaults(mut self) -> Self {
match self.inner.kind {
DescriptorKind::Generic => {
self.inner.kind = DescriptorKind::Data {
value: Some(JsValue::undefined()),
writable: Some(false),
}
}
DescriptorKind::Data {
ref mut value,
ref mut writable,
} => {
if value.is_none() {
*value = Some(JsValue::undefined())
}
if writable.is_none() {
*writable = Some(false)
}
}
DescriptorKind::Accessor {
ref mut set,
ref mut get,
} => {
if set.is_none() {
*set = Some(JsValue::undefined())
}
if get.is_none() {
*get = Some(JsValue::undefined())
}
}
}
if self.inner.configurable.is_none() {
self.inner.configurable = Some(false);
}
if self.inner.enumerable.is_none() {
self.inner.enumerable = Some(false);
}
self
}
pub fn inner(&self) -> &PropertyDescriptor {
&self.inner
}
pub fn build(self) -> PropertyDescriptor {
self.inner
}
}
impl From<PropertyDescriptorBuilder> for PropertyDescriptor {
fn from(builder: PropertyDescriptorBuilder) -> Self {
builder.build()
}
}
#[derive(Trace, Finalize, PartialEq, Debug, Clone)]
pub enum PropertyKey {
String(JsString),
Symbol(JsSymbol),
Index(u32),
}
impl From<JsString> for PropertyKey {
#[inline]
fn from(string: JsString) -> PropertyKey {
if let Ok(index) = string.parse() {
PropertyKey::Index(index)
} else {
PropertyKey::String(string)
}
}
}
impl From<&str> for PropertyKey {
#[inline]
fn from(string: &str) -> PropertyKey {
if let Ok(index) = string.parse() {
PropertyKey::Index(index)
} else {
PropertyKey::String(string.into())
}
}
}
impl From<String> for PropertyKey {
#[inline]
fn from(string: String) -> PropertyKey {
if let Ok(index) = string.parse() {
PropertyKey::Index(index)
} else {
PropertyKey::String(string.into())
}
}
}
impl From<Box<str>> for PropertyKey {
#[inline]
fn from(string: Box<str>) -> PropertyKey {
if let Ok(index) = string.parse() {
PropertyKey::Index(index)
} else {
PropertyKey::String(string.into())
}
}
}
impl From<JsSymbol> for PropertyKey {
#[inline]
fn from(symbol: JsSymbol) -> PropertyKey {
PropertyKey::Symbol(symbol)
}
}
impl fmt::Display for PropertyKey {
#[inline]
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
PropertyKey::String(ref string) => string.fmt(f),
PropertyKey::Symbol(ref symbol) => symbol.fmt(f),
PropertyKey::Index(index) => index.fmt(f),
}
}
}
impl From<&PropertyKey> for JsValue {
#[inline]
fn from(property_key: &PropertyKey) -> JsValue {
match property_key {
PropertyKey::String(ref string) => string.clone().into(),
PropertyKey::Symbol(ref symbol) => symbol.clone().into(),
PropertyKey::Index(index) => {
if let Ok(integer) = i32::try_from(*index) {
JsValue::new(integer)
} else {
JsValue::new(*index)
}
}
}
}
}
impl From<PropertyKey> for JsValue {
#[inline]
fn from(property_key: PropertyKey) -> JsValue {
match property_key {
PropertyKey::String(ref string) => string.clone().into(),
PropertyKey::Symbol(ref symbol) => symbol.clone().into(),
PropertyKey::Index(index) => {
if let Ok(integer) = i32::try_from(index) {
JsValue::new(integer)
} else {
JsValue::new(index)
}
}
}
}
}
impl From<u8> for PropertyKey {
fn from(value: u8) -> Self {
PropertyKey::Index(value.into())
}
}
impl From<u16> for PropertyKey {
fn from(value: u16) -> Self {
PropertyKey::Index(value.into())
}
}
impl From<u32> for PropertyKey {
fn from(value: u32) -> Self {
PropertyKey::Index(value)
}
}
impl From<usize> for PropertyKey {
fn from(value: usize) -> Self {
if let Ok(index) = u32::try_from(value) {
PropertyKey::Index(index)
} else {
PropertyKey::String(JsString::from(value.to_string()))
}
}
}
impl From<i64> for PropertyKey {
fn from(value: i64) -> Self {
if let Ok(index) = u32::try_from(value) {
PropertyKey::Index(index)
} else {
PropertyKey::String(JsString::from(value.to_string()))
}
}
}
impl From<u64> for PropertyKey {
fn from(value: u64) -> Self {
if let Ok(index) = u32::try_from(value) {
PropertyKey::Index(index)
} else {
PropertyKey::String(JsString::from(value.to_string()))
}
}
}
impl From<isize> for PropertyKey {
fn from(value: isize) -> Self {
if let Ok(index) = u32::try_from(value) {
PropertyKey::Index(index)
} else {
PropertyKey::String(JsString::from(value.to_string()))
}
}
}
impl From<i32> for PropertyKey {
fn from(value: i32) -> Self {
if let Ok(index) = u32::try_from(value) {
PropertyKey::Index(index)
} else {
PropertyKey::String(JsString::from(value.to_string()))
}
}
}
impl From<f64> for PropertyKey {
fn from(value: f64) -> Self {
use num_traits::cast::FromPrimitive;
if let Some(index) = u32::from_f64(value) {
return PropertyKey::Index(index);
}
PropertyKey::String(ryu_js::Buffer::new().format(value).into())
}
}
impl PartialEq<&str> for PropertyKey {
fn eq(&self, other: &&str) -> bool {
match self {
PropertyKey::String(ref string) => string == other,
_ => false,
}
}
}
#[derive(Debug, Clone, Copy, Finalize)]
pub(crate) enum PropertyNameKind {
Key,
Value,
KeyAndValue,
}
unsafe impl Trace for PropertyNameKind {
unsafe_empty_trace!();
}