rascal 0.3.4

A parser and compiler for Flash ActionScript 2 files into SWFs
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130

use crate::internal::as2::hir::scope::Scope;
use crate::internal::as2::hir::visitor::{MutVisitor, walk_expr, walk_program, walk_statement};
use crate::internal::as2::hir::{BinaryOperator, ConstantKind, Expr, ExprKind, Function};
use crate::internal::as2::resolver::identifiers_to_path;
use crate::internal::definitions::{FieldDefinition, ProgramDefinition};

struct VirtualPropertyOptimizer {
    definition: ProgramDefinition,
    scope: Scope,
}

struct FieldInfo<'a> {
    owner: &'a Expr,
    definition: &'a FieldDefinition,
}

impl VirtualPropertyOptimizer {
    pub fn get_field<'a>(&'a self, expr: &'a ExprKind) -> Option<FieldInfo<'a>> {
        let mut path = identifiers_to_path(expr).ok()?;
        if path.len() < 2 {
            return None;
        }
        let first = path.remove(0);
        let last = path.pop()?;
        let mut type_name = if let Some(first_variable) = self.scope.defined_variables.get(first)
            && let Some(type_name) = &first_variable.type_name
        {
            &type_name.value
        } else {
            return None;
        };

        path.reverse(); // turn [foo, bar, baz] into [baz, bar, foo] for easy iteration
        while let Some(segment) = path.pop() {
            if let Some(type_definition) = self.definition.get_field(type_name, segment) {
                if let Some(next_name) = &type_definition.type_name {
                    type_name = next_name;
                } else {
                    return None;
                }
            } else {
                return None;
            }
        }

        let ExprKind::Field(owner, _) = expr else {
            unreachable!()
        };

        Some(FieldInfo {
            owner,
            definition: self.definition.get_field(type_name, last)?,
        })
    }
}

impl MutVisitor for VirtualPropertyOptimizer {
    fn visit_expr(&mut self, expr: &mut Expr) {
        if let ExprKind::BinaryOperator(
            BinaryOperator::Assign | BinaryOperator::AddAssign,
            path,
            value,
        ) = &mut expr.value
        {
            if let Some(field) = self.get_field(&path.value)
                && field.definition.is_virtual
            {
                expr.value = ExprKind::Call {
                    name: Box::new(Expr::new(
                        expr.span,
                        ExprKind::Field(
                            Box::new(field.owner.clone()),
                            Box::new(Expr::new(
                                expr.span,
                                ExprKind::Constant(ConstantKind::String(format!(
                                    "__set__{}",
                                    field.definition.name.clone()
                                ))),
                            )),
                        ),
                    )),
                    args: vec![*value.clone()],
                };
            };
            return;
        }

        walk_expr(self, expr);

        if let Some(field) = self.get_field(&expr.value)
            && field.definition.is_virtual
        {
            expr.value = ExprKind::Call {
                name: Box::new(Expr::new(
                    expr.span,
                    ExprKind::Field(
                        Box::new(field.owner.clone()),
                        Box::new(Expr::new(
                            expr.span,
                            ExprKind::Constant(ConstantKind::String(format!(
                                "__get__{}",
                                field.definition.name.clone()
                            ))),
                        )),
                    ),
                )),
                args: vec![],
            };
        };
    }

    fn visit_function(&mut self, function: &mut Function) {
        let old_scope = std::mem::replace(&mut self.scope, function.scope.clone());

        for statement in &mut function.body {
            walk_statement(self, statement);
        }

        self.scope = old_scope;
    }
}

pub fn optimize_virtual_properties(program: &mut crate::Program) {
    let definition = ProgramDefinition::from(&*program);
    let mut promoter = VirtualPropertyOptimizer {
        definition,
        scope: Scope::default(),
    };
    walk_program(&mut promoter, program);
}