Optimized the variable renamer to rename more variables to 'a'.

[cycript.git] / Parser.hpp

diff --git a/Parser.hpp b/Parser.hpp
index 1e8825df6eaeff2b1615a7c43c316f7f5dec27da..cdeff501547b454cc71e26e8e7b283c8207c5c30 100644 (file)
--- a/Parser.hpp
+++ b/Parser.hpp
@@ -1,4 +1,4 @@
-/* Cycript - Remote Execution Server and Disassembler
+/* Cycript - Inlining/Optimizing JavaScript Compiler

  * Copyright (C) 2009  Jay Freeman (saurik)
 */
 
  * Copyright (C) 2009  Jay Freeman (saurik)
 */
 


@@ -47,11 +47,16 @@
 
 #include <string>
 #include <vector>
 
 #include <string>
 #include <vector>

+#include <map>
+#include <set>

 
 #include <cstdlib>
 
 #include "location.hh"
 #include "Pooling.hpp"
 
 #include <cstdlib>
 
 #include "location.hh"
 #include "Pooling.hpp"

+#include "Options.hpp"
+
+class CYContext;

 
 template <typename Type_>
 struct CYNext {
 
 template <typename Type_>
 struct CYNext {


@@ -73,13 +78,18 @@ struct CYNext {
 };
 
 struct CYThing {
 };
 
 struct CYThing {

+    virtual ~CYThing() {
+    }
+

     virtual void Output(struct CYOutput &out) const = 0;
 };
 
 struct CYOutput {
     std::ostream &out_;
     virtual void Output(struct CYOutput &out) const = 0;
 };
 
 struct CYOutput {
     std::ostream &out_;

+    CYOptions &options_;

     bool pretty_;
     unsigned indent_;
     bool pretty_;
     unsigned indent_;

+    bool right_;

 
     enum {
         NoMode,
 
     enum {
         NoMode,


@@ -89,10 +99,12 @@ struct CYOutput {
         Terminated
     } mode_;
 
         Terminated
     } mode_;
 

-    CYOutput(std::ostream &out) :
+    CYOutput(std::ostream &out, CYOptions &options) :

         out_(out),
         out_(out),

+        options_(options),

         pretty_(false),
         indent_(0),
         pretty_(false),
         indent_(0),

+        right_(false),

         mode_(NoMode)
     {
     }
         mode_(NoMode)
     {
     }


@@ -117,6 +129,9 @@ struct CYOutput {
 
 struct CYPropertyName {
     virtual void PropertyName(CYOutput &out) const = 0;
 
 struct CYPropertyName {
     virtual void PropertyName(CYOutput &out) const = 0;

+
+    virtual ~CYPropertyName() {
+    }

 };
 
 struct CYExpression;
 };
 
 struct CYExpression;


@@ -135,32 +150,21 @@ enum CYFlags {
     CYNoCall =       (1 << 3),
     CYNoRightHand =  (1 << 4),
     CYNoDangle =     (1 << 5),
     CYNoCall =       (1 << 3),
     CYNoRightHand =  (1 << 4),
     CYNoDangle =     (1 << 5),

+    CYNoInteger =    (1 << 6),

     CYNoBF =         (CYNoBrace | CYNoFunction),
 };
 
     CYNoBF =         (CYNoBrace | CYNoFunction),
 };
 

-struct CYContext {
-    apr_pool_t *pool_;
-
-    CYContext(apr_pool_t *pool) :
-        pool_(pool)
-    {
-    }
-
-    template <typename Type_>
-    void Replace(Type_ *&value) {
-        if (value != NULL)
-            while (Type_ *replace = value->Replace(*this))
-                value = replace;
-    }
-};
-

 struct CYStatement :
     CYNext<CYStatement>
 {
 struct CYStatement :
     CYNext<CYStatement>
 {

+    virtual ~CYStatement() {
+    }
+

     void Single(CYOutput &out, CYFlags flags) const;
     void Multiple(CYOutput &out, CYFlags flags = CYNoFlags) const;
 
     CYStatement *ReplaceAll(CYContext &context);
     void Single(CYOutput &out, CYFlags flags) const;
     void Multiple(CYOutput &out, CYFlags flags = CYNoFlags) const;
 
     CYStatement *ReplaceAll(CYContext &context);

+    virtual CYStatement *Collapse(CYContext &context);

 
     virtual CYStatement *Replace(CYContext &context) = 0;
 
 
     virtual CYStatement *Replace(CYContext &context) = 0;
 


@@ -198,6 +202,9 @@ struct CYStatements {
 };
 
 struct CYClassName {
 };
 
 struct CYClassName {

+    virtual ~CYClassName() {
+    }
+

     virtual CYExpression *ClassName(CYContext &context, bool object) = 0;
     virtual void ClassName(CYOutput &out, bool object) const = 0;
 };
     virtual CYExpression *ClassName(CYContext &context, bool object) = 0;
     virtual void ClassName(CYOutput &out, bool object) const = 0;
 };


@@ -214,10 +221,11 @@ struct CYWord :
     {
     }
 
     {
     }
 

-    const char *Value() const {
-        return word_;
+    void Set(const char *value) {
+        word_ = value;

     }
 
     }
 

+    virtual const char *Word() const;

     virtual void Output(CYOutput &out) const;
 
     virtual CYExpression *ClassName(CYContext &context, bool object);
     virtual void Output(CYOutput &out) const;
 
     virtual CYExpression *ClassName(CYContext &context, bool object);


@@ -226,16 +234,40 @@ struct CYWord :
 };
 
 _finline std::ostream &operator <<(std::ostream &lhs, const CYWord &rhs) {
 };
 
 _finline std::ostream &operator <<(std::ostream &lhs, const CYWord &rhs) {

-    return lhs << rhs.Value();
+    lhs << &rhs << '=';
+    return lhs << rhs.Word();

 }
 
 struct CYIdentifier :
 }
 
 struct CYIdentifier :

+    CYNext<CYIdentifier>,

     CYWord
 {
     CYWord
 {

+    CYIdentifier *replace_;
+    size_t offset_;
+

     CYIdentifier(const char *word) :
     CYIdentifier(const char *word) :

-        CYWord(word)
+        CYWord(word),
+        replace_(NULL),
+        offset_(0)
+    {
+    }
+
+    virtual const char *Word() const;
+    CYIdentifier *Replace(CYContext &context);
+};
+
+struct CYComment :
+    CYStatement
+{
+    const char *value_;
+
+    CYComment(const char *value) :
+        value_(value)

     {
     }
     {
     }

+
+    virtual CYStatement *Replace(CYContext &context);
+    virtual void Output(CYOutput &out, CYFlags flags) const;

 };
 
 struct CYLabel :
 };
 
 struct CYLabel :


@@ -254,10 +286,58 @@ struct CYLabel :
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 

+struct CYCStringLess :
+    std::binary_function<const char *, const char *, bool>
+{
+    _finline bool operator ()(const char *lhs, const char *rhs) const {
+        return strcmp(lhs, rhs) < 0;
+    }
+};
+
+struct CYIdentifierValueLess :
+    std::binary_function<CYIdentifier *, CYIdentifier *, bool>
+{
+    _finline bool operator ()(CYIdentifier *lhs, CYIdentifier *rhs) const {
+        return CYCStringLess()(lhs->Word(), rhs->Word());
+    }
+};
+
+enum CYIdentifierFlags {
+    CYIdentifierArgument,
+    CYIdentifierVariable,
+    CYIdentifierOther,
+    CYIdentifierMagic,
+};
+
+typedef std::set<const char *, CYCStringLess> CYCStringSet;
+typedef std::set<CYIdentifier *, CYIdentifierValueLess> CYIdentifierValueSet;
+typedef std::set<CYIdentifier *> CYIdentifierAddressSet;
+typedef std::vector<CYIdentifier *> CYIdentifierAddressVector;
+typedef std::map<CYIdentifier *, CYIdentifierFlags> CYIdentifierAddressFlagsMap;
+
+struct CYScope {
+    CYScope *parent_;
+    CYIdentifierAddressFlagsMap internal_;
+
+    CYIdentifierValueSet identifiers_;
+
+    CYScope() :
+        parent_(NULL)
+    {
+    }
+
+    void Declare(CYContext &context, CYIdentifier *identifier, CYIdentifierFlags flags);
+    virtual CYIdentifier *Lookup(CYContext &context, CYIdentifier *identifier);
+    void Merge(CYContext &context, CYIdentifier *identifier);
+    void Scope(CYContext &context, CYStatement *&statements);
+};
+

 struct CYProgram :
 struct CYProgram :

+    CYScope,

     CYThing
 {
     CYStatement *statements_;
     CYThing
 {
     CYStatement *statements_;

+    CYIdentifierAddressVector rename_;

 
     CYProgram(CYStatement *statements) :
         statements_(statements)
 
     CYProgram(CYStatement *statements) :
         statements_(statements)


@@ -265,18 +345,48 @@ struct CYProgram :
     }
 
     virtual void Replace(CYContext &context);
     }
 
     virtual void Replace(CYContext &context);

-

     virtual void Output(CYOutput &out) const;
 };
 
     virtual void Output(CYOutput &out) const;
 };
 

+struct CYContext :
+    CYScope
+{
+    apr_pool_t *pool_;
+    CYOptions &options_;
+    CYScope *scope_;
+    CYProgram *program_;
+
+    CYContext(apr_pool_t *pool, CYOptions &options) :
+        pool_(pool),
+        options_(options),
+        scope_(this),
+        program_(NULL)
+    {
+    }
+
+    template <typename Type_>
+    void Replace(Type_ *&value) {
+        for (;;) if (value == NULL)
+            break;
+        else {
+            Type_ *replace(value->Replace(*this));
+            if (replace != value)
+                value = replace;
+            else break;
+        }
+    }
+};
+

 struct CYBlock :
     CYStatement,
     CYThing
 {
     CYStatement *statements_;
 struct CYBlock :
     CYStatement,
     CYThing
 {
     CYStatement *statements_;

+    CYScope *scope_;

 
 

-    CYBlock(CYStatement *statements) :
-        statements_(statements)
+    CYBlock(CYStatement *statements, CYScope *scope = NULL) :
+        statements_(statements),
+        scope_(scope)

     {
     }
 
     {
     }
 


@@ -284,6 +394,14 @@ struct CYBlock :
         return statements_;
     }
 
         return statements_;
     }
 

+    void AddPrev(CYStatement *statement) {
+        CYStatement *last(statement);
+        while (last->next_ != NULL)
+            last = last->next_;
+        last->SetNext(statements_);
+        statements_ = statement;
+    }
+

     virtual CYStatement *Replace(CYContext &context);
 
     virtual void Output(CYOutput &out) const;
     virtual CYStatement *Replace(CYContext &context);
 
     virtual void Output(CYOutput &out) const;


@@ -310,8 +428,9 @@ class CYDriver {
     bool strict_;
 
     enum Condition {
     bool strict_;
 
     enum Condition {

-        RegExStart,
-        RegExRest
+        RegExpCondition,
+        XMLContentCondition,
+        XMLTagCondition,

     };
 
     std::string filename_;
     };
 
     std::string filename_;


@@ -335,19 +454,31 @@ class CYDriver {
     CYDriver(const std::string &filename);
     ~CYDriver();
 
     CYDriver(const std::string &filename);
     ~CYDriver();
 

+    Condition GetCondition();

     void SetCondition(Condition condition);
 
     void SetCondition(Condition condition);
 

+    void PushCondition(Condition condition);
+    void PopCondition();
+

     void Warning(const cy::location &location, const char *message);
 };
 
 struct CYForInitialiser {
     void Warning(const cy::location &location, const char *message);
 };
 
 struct CYForInitialiser {

+    virtual ~CYForInitialiser() {
+    }
+

     virtual void For(CYOutput &out) const = 0;
     virtual void For(CYOutput &out) const = 0;

+    virtual CYExpression *Replace(CYContext &context) = 0;

 };
 
 struct CYForInInitialiser {
 };
 
 struct CYForInInitialiser {

+    virtual ~CYForInInitialiser() {
+    }
+

     virtual void ForIn(CYOutput &out, CYFlags flags) const = 0;
     virtual const char *ForEachIn() const = 0;
     virtual CYExpression *ForEachIn(CYContext &out) = 0;
     virtual void ForIn(CYOutput &out, CYFlags flags) const = 0;
     virtual const char *ForEachIn() const = 0;
     virtual CYExpression *ForEachIn(CYContext &out) = 0;

+    virtual CYExpression *Replace(CYContext &context) = 0;

 };
 
 struct CYNumber;
 };
 
 struct CYNumber;


@@ -420,7 +551,7 @@ struct CYCompound :
 {
     CYExpression *expressions_;
 
 {
     CYExpression *expressions_;
 

-    CYCompound(CYExpression *expressions) :
+    CYCompound(CYExpression *expressions = NULL) :

         expressions_(expressions)
     {
     }
         expressions_(expressions)
     {
     }


@@ -451,6 +582,7 @@ struct CYFunctionParameter :
     {
     }
 
     {
     }
 

+    void Replace(CYContext &context);

     virtual void Output(CYOutput &out) const;
 };
 
     virtual void Output(CYOutput &out) const;
 };
 


@@ -479,7 +611,7 @@ struct CYForInComprehension :
     }
 
     virtual const char *Name() const {
     }
 
     virtual const char *Name() const {

-        return name_->Value();
+        return name_->Word();

     }
 
     virtual CYFunctionParameter *Parameter(CYContext &context) const;
     }
 
     virtual CYFunctionParameter *Parameter(CYContext &context) const;


@@ -500,7 +632,7 @@ struct CYForEachInComprehension :
     }
 
     virtual const char *Name() const {
     }
 
     virtual const char *Name() const {

-        return name_->Value();
+        return name_->Word();

     }
 
     virtual CYFunctionParameter *Parameter(CYContext &context) const;
     }
 
     virtual CYFunctionParameter *Parameter(CYContext &context) const;


@@ -615,7 +747,7 @@ struct CYString :
     }
 
     CYString(const CYWord *word) :
     }
 
     CYString(const CYWord *word) :

-        value_(word->Value()),
+        value_(word->Word()),

         size_(strlen(value_))
     {
     }
         size_(strlen(value_))
     {
     }


@@ -629,7 +761,7 @@ struct CYString :
     virtual CYNumber *Number(CYContext &context);
     virtual CYString *String(CYContext &context);
 
     virtual CYNumber *Number(CYContext &context);
     virtual CYString *String(CYContext &context);
 

-    virtual CYString *Concat(CYContext &out, CYString *rhs) const;
+    CYString *Concat(CYContext &out, CYString *rhs) const;

     virtual void Output(CYOutput &out, CYFlags flags) const;
     virtual void PropertyName(CYOutput &out) const;
 };
     virtual void Output(CYOutput &out, CYFlags flags) const;
     virtual void PropertyName(CYOutput &out) const;
 };


@@ -924,13 +1056,31 @@ struct CYArray :
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 

+struct CYProperty :
+    CYNext<CYProperty>,
+    CYThing
+{
+    CYPropertyName *name_;
+    CYExpression *value_;
+
+    CYProperty(CYPropertyName *name, CYExpression *value, CYProperty *next = NULL) :
+        CYNext<CYProperty>(next),
+        name_(name),
+        value_(value)
+    {
+    }
+
+    void Replace(CYContext &context);
+    virtual void Output(CYOutput &out) const;
+};
+

 struct CYDeclaration :
     CYForInInitialiser
 {
     CYIdentifier *identifier_;
     CYExpression *initialiser_;
 
 struct CYDeclaration :
     CYForInInitialiser
 {
     CYIdentifier *identifier_;
     CYExpression *initialiser_;
 

-    CYDeclaration(CYIdentifier *identifier, CYExpression *initialiser) :
+    CYDeclaration(CYIdentifier *identifier, CYExpression *initialiser = NULL) :

         identifier_(identifier),
         initialiser_(initialiser)
     {
         identifier_(identifier),
         initialiser_(initialiser)
     {


@@ -941,19 +1091,20 @@ struct CYDeclaration :
     virtual const char *ForEachIn() const;
     virtual CYExpression *ForEachIn(CYContext &out);
 
     virtual const char *ForEachIn() const;
     virtual CYExpression *ForEachIn(CYContext &out);
 

-    void Replace(CYContext &context);
+    virtual CYExpression *Replace(CYContext &context);
+    virtual CYAssignment *Assignment(CYContext &context);

 
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 
 struct CYDeclarations :
     CYNext<CYDeclarations>,
 
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 
 struct CYDeclarations :
     CYNext<CYDeclarations>,

-    CYForInitialiser,
-    CYThing
+    CYThing,
+    CYForInitialiser

 {
     CYDeclaration *declaration_;
 
 {
     CYDeclaration *declaration_;
 

-    CYDeclarations(CYDeclaration *declaration, CYDeclarations *next) :
+    CYDeclarations(CYDeclaration *declaration, CYDeclarations *next = NULL) :

         CYNext<CYDeclarations>(next),
         declaration_(declaration)
     {
         CYNext<CYDeclarations>(next),
         declaration_(declaration)
     {


@@ -961,7 +1112,8 @@ struct CYDeclarations :
 
     virtual void For(CYOutput &out) const;
 
 
     virtual void For(CYOutput &out) const;
 

-    void Replace(CYContext &context);
+    virtual CYCompound *Replace(CYContext &context);
+    CYProperty *Property(CYContext &context);

 
     virtual void Output(CYOutput &out) const;
     virtual void Output(CYOutput &out, CYFlags flags) const;
 
     virtual void Output(CYOutput &out) const;
     virtual void Output(CYOutput &out, CYFlags flags) const;


@@ -1053,24 +1205,6 @@ struct CYForEachIn :
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 

-struct CYProperty :
-    CYNext<CYProperty>,
-    CYThing
-{
-    CYPropertyName *name_;
-    CYExpression *value_;
-
-    CYProperty(CYPropertyName *name, CYExpression *value, CYProperty *next = NULL) :
-        CYNext<CYProperty>(next),
-        name_(name),
-        value_(value)
-    {
-    }
-
-    void Replace(CYContext &context);
-    virtual void Output(CYOutput &out) const;
-};
-

 struct CYObject :
     CYLiteral
 {
 struct CYObject :
     CYLiteral
 {


@@ -1085,22 +1219,6 @@ struct CYObject :
     void Output(CYOutput &out, CYFlags flags) const;
 };
 
     void Output(CYOutput &out, CYFlags flags) const;
 };
 

-struct CYCatch :
-    CYThing
-{
-    CYIdentifier *name_;
-    CYBlock code_;
-
-    CYCatch(CYIdentifier *name, CYStatement *statements) :
-        name_(name),
-        code_(statements)
-    {
-    }
-
-    void Replace(CYContext &context);
-    virtual void Output(CYOutput &out) const;
-};
-

 struct CYMember :
     CYExpression
 {
 struct CYMember :
     CYExpression
 {


@@ -1241,7 +1359,9 @@ struct CYWhile :
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 

-struct CYFunction {
+struct CYFunction :
+    CYScope
+{

     CYIdentifier *name_;
     CYFunctionParameter *parameters_;
     CYBlock code_;
     CYIdentifier *name_;
     CYFunctionParameter *parameters_;
     CYBlock code_;


@@ -1249,11 +1369,15 @@ struct CYFunction {
     CYFunction(CYIdentifier *name, CYFunctionParameter *parameters, CYStatement *statements) :
         name_(name),
         parameters_(parameters),
     CYFunction(CYIdentifier *name, CYFunctionParameter *parameters, CYStatement *statements) :
         name_(name),
         parameters_(parameters),

-        code_(statements)
+        code_(statements, this)

     {
     }
 
     {
     }
 

-    virtual void Replace_(CYContext &context);
+    virtual ~CYFunction() {
+    }
+
+    void Inject(CYContext &context);
+    virtual void Replace_(CYContext &context, bool outer);

     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 


@@ -1294,8 +1418,11 @@ struct CYExpress :
     CYExpress(CYExpression *expression) :
         expression_(expression)
     {
     CYExpress(CYExpression *expression) :
         expression_(expression)
     {

+        if (expression == NULL)
+            throw;

     }
 
     }
 

+    virtual CYStatement *Collapse(CYContext &context);

     virtual CYStatement *Replace(CYContext &context);
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
     virtual CYStatement *Replace(CYContext &context);
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };


@@ -1345,6 +1472,7 @@ struct CYReturn :
 struct CYEmpty :
     CYStatement
 {
 struct CYEmpty :
     CYStatement
 {

+    virtual CYStatement *Collapse(CYContext &context);

     virtual CYStatement *Replace(CYContext &context);
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
     virtual CYStatement *Replace(CYContext &context);
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };


@@ -1363,14 +1491,33 @@ struct CYFinally :
     virtual void Output(CYOutput &out) const;
 };
 
     virtual void Output(CYOutput &out) const;
 };
 

-struct CYTry :
+namespace cy {
+namespace Syntax {
+
+struct Catch :
+    CYThing
+{
+    CYIdentifier *name_;
+    CYBlock code_;
+
+    Catch(CYIdentifier *name, CYStatement *statements) :
+        name_(name),
+        code_(statements)
+    {
+    }
+
+    void Replace(CYContext &context);
+    virtual void Output(CYOutput &out) const;
+};
+
+struct Try :

     CYStatement
 {
     CYBlock code_;
     CYStatement
 {
     CYBlock code_;

-    CYCatch *catch_;
+    Catch *catch_;

     CYFinally *finally_;
 
     CYFinally *finally_;
 

-    CYTry(CYStatement *statements, CYCatch *_catch, CYFinally *finally) :
+    Try(CYStatement *statements, Catch *_catch, CYFinally *finally) :

         code_(statements),
         catch_(_catch),
         finally_(finally)
         code_(statements),
         catch_(_catch),
         finally_(finally)


@@ -1381,12 +1528,12 @@ struct CYTry :
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 

-struct CYThrow :
+struct Throw :

     CYStatement
 {
     CYExpression *value_;
 
     CYStatement
 {
     CYExpression *value_;
 

-    CYThrow(CYExpression *value) :
+    Throw(CYExpression *value) :

         value_(value)
     {
     }
         value_(value)
     {
     }


@@ -1395,6 +1542,8 @@ struct CYThrow :
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 
     virtual void Output(CYOutput &out, CYFlags flags) const;
 };
 

+} }
+

 struct CYWith :
     CYStatement
 {
 struct CYWith :
     CYStatement
 {