core: Add keyboard shortcut to dump every variable (ctrl+alt+V, same as Flash Player)

This commit is contained in:
Nathan Adams 2020-05-08 07:58:12 +02:00 committed by Mike Welsh
parent c787e28fcd
commit edcd1e6d65
3 changed files with 398 additions and 1 deletions

View File

@ -25,6 +25,7 @@ mod test_utils;
pub mod listeners;
mod activation;
pub mod debug;
mod fscommand;
pub mod function;
pub mod globals;

362
core/src/avm1/debug.rs Normal file
View File

@ -0,0 +1,362 @@
use crate::avm1::{Avm1, Object, ObjectPtr, TObject, Value};
use crate::context::UpdateContext;
#[allow(dead_code)]
pub struct VariableDumper<'a> {
objects: Vec<*const ObjectPtr>,
depth: u32,
output: String,
indent: &'a str,
}
impl<'a> VariableDumper<'a> {
pub fn new(indent: &'a str) -> Self {
Self {
objects: Vec::new(),
depth: 0,
output: String::new(),
indent,
}
}
#[allow(dead_code)]
pub fn dump<'gc>(
value: &Value<'gc>,
indent: &str,
avm: &mut Avm1<'gc>,
context: &mut UpdateContext<'_, 'gc, '_>,
) -> String {
let mut dumper = VariableDumper::new(indent);
dumper.print_value(value, avm, context);
dumper.output
}
pub fn output(&self) -> &str {
&self.output
}
fn object_id(&mut self, object: &Object) -> (usize, bool) {
let ptr = object.as_ptr();
for (i, other) in self.objects.iter().enumerate() {
if *other == ptr {
return (i, false);
}
}
let id = self.objects.len();
self.objects.push(ptr);
(id, true)
}
fn indent(&mut self) {
for _ in 0..self.depth {
self.output.push_str(self.indent);
}
}
pub fn print_string(&mut self, string: &str) {
self.output.push_str("\"");
for c in string.chars() {
if c == '"' {
self.output.push_str("\\\"");
} else if c == '\\' {
self.output.push_str("\\\\");
} else if c == '\n' {
self.output.push_str("\\n");
} else if c == '\r' {
self.output.push_str("\\r");
} else if c == '\t' {
self.output.push_str("\\t");
} else if c == '\u{0008}' {
self.output.push_str("\\b");
} else if c == '\u{000C}' {
self.output.push_str("\\f");
} else {
self.output.push(c);
}
}
self.output.push_str("\"");
}
pub fn print_object<'gc>(
&mut self,
object: &Object<'gc>,
avm: &mut Avm1<'gc>,
context: &mut UpdateContext<'_, 'gc, '_>,
) {
let (id, new) = self.object_id(object);
self.output.push_str("[object #");
self.output.push_str(&id.to_string());
self.output.push_str("]");
if new {
self.print_properties(object, avm, context);
}
}
pub fn print_property<'gc>(
&mut self,
object: &Object<'gc>,
key: &str,
avm: &mut Avm1<'gc>,
context: &mut UpdateContext<'_, 'gc, '_>,
) {
match object
.get(&key, avm, context)
.and_then(|v| v.resolve(avm, context))
{
Ok(value) => {
self.print_value(&value, avm, context);
}
Err(e) => {
self.output.push_str("Error: \"");
self.output.push_str(&e.to_string());
self.output.push_str("\"");
}
}
}
pub fn print_properties<'gc>(
&mut self,
object: &Object<'gc>,
avm: &mut Avm1<'gc>,
context: &mut UpdateContext<'_, 'gc, '_>,
) {
let keys = object.get_keys(avm);
if keys.is_empty() {
self.output.push_str(" {}");
} else {
self.output.push_str(" {\n");
self.depth += 1;
for key in keys.into_iter() {
self.indent();
self.output.push_str(&key);
self.output.push_str(": ");
self.print_property(object, &key, avm, context);
self.output.push_str("\n");
}
self.depth -= 1;
self.indent();
self.output.push_str("}");
}
}
pub fn print_value<'gc>(
&mut self,
value: &Value<'gc>,
avm: &mut Avm1<'gc>,
context: &mut UpdateContext<'_, 'gc, '_>,
) {
match value {
Value::Undefined => self.output.push_str("undefined"),
Value::Null => self.output.push_str("null"),
Value::Bool(value) => self.output.push_str(&value.to_string()),
Value::Number(value) => self.output.push_str(&value.to_string()),
Value::String(value) => {
self.print_string(value);
}
Value::Object(object) => {
self.print_object(object, avm, context);
}
}
}
pub fn print_variables<'gc>(
&mut self,
header: &str,
name: &str,
object: &Object<'gc>,
avm: &mut Avm1<'gc>,
context: &mut UpdateContext<'_, 'gc, '_>,
) {
let keys = object.get_keys(avm);
if keys.is_empty() {
return;
}
self.output.push_str(header);
self.output.push_str("\n");
self.depth += 1;
for key in keys.into_iter() {
self.output.push_str(&format!("{}.{}", name, key));
self.output.push_str(" = ");
self.print_property(object, &key, avm, context);
self.output.push_str("\n");
}
self.depth -= 1;
self.output.push_str("\n");
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::avm1::function::Executable;
use crate::avm1::return_value::ReturnValue;
use crate::avm1::test_utils::with_avm;
use crate::avm1::{Error, ScriptObject};
use enumset::EnumSet;
fn throw_error<'gc>(
_avm: &mut Avm1<'gc>,
_context: &mut UpdateContext<'_, 'gc, '_>,
_this: Object<'gc>,
_args: &[Value<'gc>],
) -> Result<ReturnValue<'gc>, Error> {
Err("Error message!".into())
}
#[test]
fn dump_undefined() -> Result<(), Error> {
with_avm(19, |avm, context, _root| -> Result<(), Error> {
assert_eq!(
VariableDumper::dump(&Value::Undefined, " ", avm, context),
"undefined"
);
Ok(())
})
}
#[test]
fn dump_null() -> Result<(), Error> {
with_avm(19, |avm, context, _root| -> Result<(), Error> {
assert_eq!(
VariableDumper::dump(&Value::Null, " ", avm, context),
"null"
);
Ok(())
})
}
#[test]
fn dump_bool() -> Result<(), Error> {
with_avm(19, |avm, context, _root| -> Result<(), Error> {
assert_eq!(
VariableDumper::dump(&Value::Bool(true), " ", avm, context),
"true"
);
assert_eq!(
VariableDumper::dump(&Value::Bool(false), " ", avm, context),
"false"
);
Ok(())
})
}
#[test]
fn dump_number() -> Result<(), Error> {
with_avm(19, |avm, context, _root| -> Result<(), Error> {
assert_eq!(
VariableDumper::dump(&Value::Number(1000.0), " ", avm, context),
"1000"
);
assert_eq!(
VariableDumper::dump(&Value::Number(-0.05), " ", avm, context),
"-0.05"
);
Ok(())
})
}
#[test]
fn dump_string() -> Result<(), Error> {
with_avm(19, |avm, context, _root| -> Result<(), Error> {
assert_eq!(
VariableDumper::dump(&Value::String("".to_string()), " ", avm, context),
"\"\""
);
assert_eq!(
VariableDumper::dump(&Value::String("HELLO WORLD".to_string()), " ", avm, context),
"\"HELLO WORLD\""
);
assert_eq!(
VariableDumper::dump(
&Value::String(
"Escape \"this\" string\nplease! \u{0008}\u{000C}\n\r\t\"\\".to_string()
),
" ",
avm,
context
),
"\"Escape \\\"this\\\" string\\nplease! \\b\\f\\n\\r\\t\\\"\\\\\""
);
Ok(())
})
}
#[test]
fn dump_empty_object() -> Result<(), Error> {
with_avm(19, |avm, context, _root| -> Result<(), Error> {
let object = ScriptObject::object(context.gc_context, None);
assert_eq!(
VariableDumper::dump(&object.into(), " ", avm, context),
"[object #0] {}"
);
Ok(())
})
}
#[test]
fn dump_object() -> Result<(), Error> {
with_avm(19, |avm, context, _root| -> Result<(), Error> {
let object = ScriptObject::object(context.gc_context, None);
let child = ScriptObject::object(context.gc_context, None);
object.set("self", object.into(), avm, context)?;
object.set("test", Value::String("value".to_string()), avm, context)?;
object.set("child", child.into(), avm, context)?;
child.set("parent", object.into(), avm, context)?;
child.set("age", Value::Number(6.0), avm, context)?;
assert_eq!(
VariableDumper::dump(&object.into(), " ", avm, context),
"[object #0] {\n child: [object #1] {\n age: 6\n parent: [object #0]\n }\n test: \"value\"\n self: [object #0]\n}",
);
Ok(())
})
}
#[test]
fn dump_object_with_error() -> Result<(), Error> {
with_avm(19, |avm, context, _root| -> Result<(), Error> {
let object = ScriptObject::object(context.gc_context, None);
object.add_property(
context.gc_context,
"broken_value",
Executable::Native(throw_error),
None,
EnumSet::empty(),
);
assert_eq!(
VariableDumper::dump(&object.into(), " ", avm, context),
"[object #0] {\n broken_value: Error: \"Error message!\"\n}"
);
Ok(())
})
}
#[test]
fn dump_variables() -> Result<(), Error> {
with_avm(19, |avm, context, _root| -> Result<(), Error> {
let object = ScriptObject::object(context.gc_context, None);
let child = ScriptObject::object(context.gc_context, None);
object.set("self", object.into(), avm, context)?;
object.set("test", Value::String("value".to_string()), avm, context)?;
object.set("child", child.into(), avm, context)?;
child.set("parent", object.into(), avm, context)?;
child.set("age", Value::Number(6.0), avm, context)?;
let mut dumper = VariableDumper::new(" ");
dumper.print_variables("Variables:", "object", &object.into(), avm, context);
assert_eq!(
dumper.output,
"Variables:\nobject.child = [object #0] {\n age: 6\n parent: [object #1] {\n child: [object #0]\n test: \"value\"\n self: [object #1]\n }\n }\nobject.test = \"value\"\nobject.self = [object #1]\n\n"
);
Ok(())
})
}
}

View File

@ -1,3 +1,4 @@
use crate::avm1::debug::VariableDumper;
use crate::avm1::listeners::SystemListener;
use crate::avm1::{Activation, Avm1, TObject, Value};
use crate::backend::input::{InputBackend, MouseCursor};
@ -6,7 +7,9 @@ use crate::backend::{
};
use crate::context::{ActionQueue, ActionType, RenderContext, UpdateContext};
use crate::display_object::{MorphShape, MovieClip};
use crate::events::{ButtonEvent, ButtonEventResult, ButtonKeyCode, ClipEvent, PlayerEvent};
use crate::events::{
ButtonEvent, ButtonEventResult, ButtonKeyCode, ClipEvent, KeyCode, PlayerEvent,
};
use crate::library::Library;
use crate::loader::LoadManager;
use crate::prelude::*;
@ -333,6 +336,37 @@ impl Player {
pub fn handle_event(&mut self, event: PlayerEvent) {
let mut needs_render = self.needs_render;
if let PlayerEvent::KeyDown {
key_code: KeyCode::V,
} = event
{
if self.input.is_key_down(KeyCode::Control) && self.input.is_key_down(KeyCode::Alt) {
self.mutate_with_update_context(|avm, context| {
let mut dumper = VariableDumper::new(" ");
dumper.print_variables(
"Global Variables:",
"_global",
&avm.global_object_cell(),
avm,
context,
);
let levels = context.levels.clone();
for (level, display_object) in levels {
if let Ok(object) = display_object.object().as_object() {
dumper.print_variables(
&format!("Level #{}:", level),
&format!("_level{}", level),
&object,
avm,
context,
);
}
}
log::info!("Variable dump:\n{}", dumper.output());
});
}
}
// Update mouse position from mouse events.
if let PlayerEvent::MouseMove { x, y }
| PlayerEvent::MouseDown { x, y }