@ -8,6 +8,7 @@ use gc_derive::{Finalize, Trace};
use serde_json ::{ map ::Map , Number as JSONNumber , Value as JSONValue } ; use serde_json ::{ map ::Map , Number as JSONNumber , Value as JSONValue } ;
use std ::{ use std ::{
any ::Any , any ::Any ,
collections ::HashSet ,
f64 ::NAN , f64 ::NAN ,
fmt ::{ self , Display } , fmt ::{ self , Display } ,
ops ::{ Add , BitAnd , BitOr , BitXor , Deref , DerefMut , Div , Mul , Not , Rem , Shl , Shr , Sub } , ops ::{ Add , BitAnd , BitOr , BitXor , Deref , DerefMut , Div , Mul , Not , Rem , Shl , Shr , Sub } ,
@ -631,9 +632,199 @@ impl Default for ValueData {
} }
} }
/// A helper macro for printing objects
/// Can be used to print both properties and internal slots
/// All of the overloads take:
/// - The object to be printed
/// - The function with which to print
/// - The indentation for the current level (for nested objects)
/// - A HashSet with the addresses of the already printed objects for the current branch
/// (used to avoid infinite loops when there are cyclic deps)
macro_rules! print_obj_value {
( all of $obj :expr , $display_fn :ident , $indent :expr , $encounters :expr ) = > {
{
let mut internals = print_obj_value ! ( internals of $obj , $display_fn , $indent , $encounters ) ;
let mut props = print_obj_value ! ( props of $obj , $display_fn , $indent , $encounters , true ) ;
props . reserve ( internals . len ( ) ) ;
props . append ( & mut internals ) ;
props
}
} ;
( internals of $obj :expr , $display_fn :ident , $indent :expr , $encounters :expr ) = > {
print_obj_value ! ( impl internal_slots , $obj , | ( key , val ) | {
format! (
"{}{}: {}" ,
String ::from_utf8 ( vec! [ b' ' ; $indent ] )
. expect ( "Could not create indentation string" ) ,
key ,
$display_fn ( & val , $encounters , $indent . wrapping_add ( 4 ) , true )
)
} )
} ;
( props of $obj :expr , $display_fn :ident , $indent :expr , $encounters :expr , $print_internals :expr ) = > {
print_obj_value ! ( impl properties , $obj , | ( key , val ) | {
let v = & val
. value
. as_ref ( )
. expect ( "Could not get the property's value" ) ;
format! (
"{}{}: {}" ,
String ::from_utf8 ( vec! [ b' ' ; $indent ] )
. expect ( "Could not create indentation string" ) ,
key ,
$display_fn ( v , $encounters , $indent . wrapping_add ( 4 ) , $print_internals )
)
} )
} ;
// A private overload of the macro
// DO NOT use directly
( impl $field :ident , $v :expr , $f :expr ) = > {
$v
. borrow ( )
. $field
. iter ( )
. map ( $f )
. collect ::< Vec < String > > ( )
} ;
}
pub ( crate ) fn log_string_from ( x : & ValueData , print_internals : bool ) -> String {
match x {
// We don't want to print private (compiler) or prototype properties
ValueData ::Object ( ref v ) = > {
// Can use the private "type" field of an Object to match on
// which type of Object it represents for special printing
match v . borrow ( ) . kind {
ObjectKind ::String = > from_value (
v . borrow ( )
. internal_slots
. get ( "StringData" )
. expect ( "Cannot get primitive value from String" )
. clone ( ) ,
)
. expect ( "Cannot clone primitive value from String" ) ,
ObjectKind ::Boolean = > {
let bool_data = v . borrow ( ) . get_internal_slot ( "BooleanData" ) . to_string ( ) ;
format! ( "Boolean {{ {} }}" , bool_data )
}
ObjectKind ::Array = > {
let len : i32 = from_value (
v . borrow ( )
. properties
. get ( "length" )
. unwrap ( )
. value
. clone ( )
. expect ( "Could not borrow value" ) ,
)
. expect ( "Could not convert JS value to i32" ) ;
if len = = 0 {
return String ::from ( "[]" ) ;
}
let arr = ( 0 .. len )
. map ( | i | {
// Introduce recursive call to stringify any objects
// which are part of the Array
log_string_from (
& v . borrow ( )
. properties
. get ( & i . to_string ( ) )
. unwrap ( )
. value
. clone ( )
. expect ( "Could not borrow value" ) ,
print_internals ,
)
} )
. collect ::< Vec < String > > ( )
. join ( ", " ) ;
format! ( "[ {} ]" , arr )
}
_ = > display_obj ( & x , print_internals ) ,
}
}
ValueData ::Symbol ( ref sym ) = > {
let desc : Value = sym . borrow ( ) . get_internal_slot ( "Description" ) ;
match * desc {
ValueData ::String ( ref st ) = > format! ( "Symbol(\"{}\")" , st . to_string ( ) ) ,
_ = > String ::from ( "Symbol()" ) ,
}
}
_ = > format! ( "{}" , x ) ,
}
}
/// A helper function for specifically printing object values
fn display_obj ( v : & ValueData , print_internals : bool ) -> String {
// A simple helper for getting the address of a value
// TODO: Find a more general place for this, as it can be used in other situations as well
fn address_of < T > ( t : & T ) -> usize {
let my_ptr : * const T = t ;
my_ptr as usize
}
// We keep track of which objects we have encountered by keeping their
// in-memory address in this set
let mut encounters = HashSet ::new ( ) ;
fn display_obj_internal (
data : & ValueData ,
encounters : & mut HashSet < usize > ,
indent : usize ,
print_internals : bool ,
) -> String {
match * data {
ValueData ::Object ( ref v ) = > {
// The in-memory address of the current object
let addr = address_of ( v . borrow ( ) . deref ( ) ) ;
// We need not continue if this object has already been
// printed up the current chain
if encounters . contains ( & addr ) {
return String ::from ( "[Cycle]" ) ;
}
// Mark the current object as encountered
encounters . insert ( addr ) ;
let result = if print_internals {
print_obj_value ! ( all of v , display_obj_internal , indent , encounters ) . join ( ",\n" )
} else {
print_obj_value ! ( props of v , display_obj_internal , indent , encounters , print_internals )
. join ( ",\n" )
} ;
// If the current object is referenced in a different branch,
// it will not cause an infinte printing loop, so it is safe to be printed again
encounters . remove ( & addr ) ;
let closing_indent = String ::from_utf8 ( vec! [ b' ' ; indent . wrapping_sub ( 4 ) ] )
. expect ( "Could not create the closing brace's indentation string" ) ;
format! ( "{{\n{}\n{}}}" , result , closing_indent )
}
// Every other type of data is printed as is
_ = > format! ( "{}" , data ) ,
}
}
display_obj_internal ( v , & mut encounters , 4 , print_internals )
}
impl Display for ValueData { impl Display for ValueData {
fn fmt ( & self , f : & mut fmt ::Formatter ) -> fmt ::Result { fn fmt ( & self , f : & mut fmt ::Formatter ) -> fmt ::Result {
match * self { match self {
ValueData ::Null = > write! ( f , "null" ) , ValueData ::Null = > write! ( f , "null" ) ,
ValueData ::Undefined = > write! ( f , "undefined" ) , ValueData ::Undefined = > write! ( f , "undefined" ) ,
ValueData ::Boolean ( v ) = > write! ( f , "{}" , v ) , ValueData ::Boolean ( v ) = > write! ( f , "{}" , v ) ,
@ -653,7 +844,7 @@ impl Display for ValueData {
_ = > v . to_string ( ) , _ = > v . to_string ( ) ,
} }
) , ) ,
ValueData ::Object ( _ ) = > write! ( f , "{{} }" ) , ValueData ::Object ( _ ) = > write! ( f , "{}" , log_string_from ( self , true ) ) ,
ValueData ::Integer ( v ) = > write! ( f , "{}" , v ) , ValueData ::Integer ( v ) = > write! ( f , "{}" , v ) ,
ValueData ::Function ( ref v ) = > match * v . borrow ( ) { ValueData ::Function ( ref v ) = > match * v . borrow ( ) {
Function ::NativeFunc ( _ ) = > write! ( f , "function() {{ [native code] }}" ) , Function ::NativeFunc ( _ ) = > write! ( f , "function() {{ [native code] }}" ) ,
Iovoslav Iovchev
5 years ago
committed by
GitHub