bsc/vendor/github.com/robertkrimen/otto/cmpl_parse.go

657 lines
15 KiB
Go
Raw Normal View History

2015-02-16 15:28:33 +02:00
package otto
import (
"fmt"
"regexp"
"github.com/robertkrimen/otto/ast"
"github.com/robertkrimen/otto/file"
2015-02-16 15:28:33 +02:00
"github.com/robertkrimen/otto/token"
)
var trueLiteral = &_nodeLiteral{value: toValue_bool(true)}
var falseLiteral = &_nodeLiteral{value: toValue_bool(false)}
var nullLiteral = &_nodeLiteral{value: nullValue}
2015-02-16 15:28:33 +02:00
var emptyStatement = &_nodeEmptyStatement{}
func (cmpl *_compiler) parseExpression(in ast.Expression) _nodeExpression {
if in == nil {
2015-02-16 15:28:33 +02:00
return nil
}
switch in := in.(type) {
2015-02-16 15:28:33 +02:00
case *ast.ArrayLiteral:
out := &_nodeArrayLiteral{
value: make([]_nodeExpression, len(in.Value)),
2015-02-16 15:28:33 +02:00
}
for i, value := range in.Value {
out.value[i] = cmpl.parseExpression(value)
2015-02-16 15:28:33 +02:00
}
return out
2015-02-16 15:28:33 +02:00
case *ast.AssignExpression:
return &_nodeAssignExpression{
operator: in.Operator,
left: cmpl.parseExpression(in.Left),
right: cmpl.parseExpression(in.Right),
2015-02-16 15:28:33 +02:00
}
case *ast.BinaryExpression:
return &_nodeBinaryExpression{
operator: in.Operator,
left: cmpl.parseExpression(in.Left),
right: cmpl.parseExpression(in.Right),
comparison: in.Comparison,
2015-02-16 15:28:33 +02:00
}
case *ast.BooleanLiteral:
if in.Value {
2015-02-16 15:28:33 +02:00
return trueLiteral
}
return falseLiteral
case *ast.BracketExpression:
return &_nodeBracketExpression{
idx: in.Left.Idx0(),
left: cmpl.parseExpression(in.Left),
member: cmpl.parseExpression(in.Member),
2015-02-16 15:28:33 +02:00
}
case *ast.CallExpression:
out := &_nodeCallExpression{
callee: cmpl.parseExpression(in.Callee),
argumentList: make([]_nodeExpression, len(in.ArgumentList)),
2015-02-16 15:28:33 +02:00
}
for i, value := range in.ArgumentList {
out.argumentList[i] = cmpl.parseExpression(value)
2015-02-16 15:28:33 +02:00
}
return out
2015-02-16 15:28:33 +02:00
case *ast.ConditionalExpression:
return &_nodeConditionalExpression{
test: cmpl.parseExpression(in.Test),
consequent: cmpl.parseExpression(in.Consequent),
alternate: cmpl.parseExpression(in.Alternate),
2015-02-16 15:28:33 +02:00
}
case *ast.DotExpression:
return &_nodeDotExpression{
idx: in.Left.Idx0(),
left: cmpl.parseExpression(in.Left),
identifier: in.Identifier.Name,
2015-02-16 15:28:33 +02:00
}
2016-02-19 13:06:58 +02:00
case *ast.EmptyExpression:
return nil
2015-02-16 15:28:33 +02:00
case *ast.FunctionLiteral:
name := ""
if in.Name != nil {
name = in.Name.Name
2015-02-16 15:28:33 +02:00
}
out := &_nodeFunctionLiteral{
2015-02-16 15:28:33 +02:00
name: name,
body: cmpl.parseStatement(in.Body),
source: in.Source,
file: cmpl.file,
2015-02-16 15:28:33 +02:00
}
if in.ParameterList != nil {
list := in.ParameterList.List
out.parameterList = make([]string, len(list))
2015-02-16 15:28:33 +02:00
for i, value := range list {
out.parameterList[i] = value.Name
2015-02-16 15:28:33 +02:00
}
}
for _, value := range in.DeclarationList {
2015-02-16 15:28:33 +02:00
switch value := value.(type) {
case *ast.FunctionDeclaration:
out.functionList = append(out.functionList, cmpl.parseExpression(value.Function).(*_nodeFunctionLiteral))
2015-02-16 15:28:33 +02:00
case *ast.VariableDeclaration:
for _, value := range value.List {
out.varList = append(out.varList, value.Name)
2015-02-16 15:28:33 +02:00
}
default:
panic(fmt.Errorf("Here be dragons: parseProgram.declaration(%T)", value))
}
}
return out
2015-02-16 15:28:33 +02:00
case *ast.Identifier:
return &_nodeIdentifier{
idx: in.Idx,
name: in.Name,
2015-02-16 15:28:33 +02:00
}
case *ast.NewExpression:
out := &_nodeNewExpression{
callee: cmpl.parseExpression(in.Callee),
argumentList: make([]_nodeExpression, len(in.ArgumentList)),
2015-02-16 15:28:33 +02:00
}
for i, value := range in.ArgumentList {
out.argumentList[i] = cmpl.parseExpression(value)
2015-02-16 15:28:33 +02:00
}
return out
2015-02-16 15:28:33 +02:00
case *ast.NullLiteral:
return nullLiteral
case *ast.NumberLiteral:
return &_nodeLiteral{
value: toValue(in.Value),
2015-02-16 15:28:33 +02:00
}
case *ast.ObjectLiteral:
out := &_nodeObjectLiteral{
value: make([]_nodeProperty, len(in.Value)),
2015-02-16 15:28:33 +02:00
}
for i, value := range in.Value {
out.value[i] = _nodeProperty{
2015-02-16 15:28:33 +02:00
key: value.Key,
kind: value.Kind,
value: cmpl.parseExpression(value.Value),
2015-02-16 15:28:33 +02:00
}
}
return out
2015-02-16 15:28:33 +02:00
case *ast.RegExpLiteral:
return &_nodeRegExpLiteral{
flags: in.Flags,
pattern: in.Pattern,
2015-02-16 15:28:33 +02:00
}
case *ast.SequenceExpression:
out := &_nodeSequenceExpression{
sequence: make([]_nodeExpression, len(in.Sequence)),
2015-02-16 15:28:33 +02:00
}
for i, value := range in.Sequence {
out.sequence[i] = cmpl.parseExpression(value)
2015-02-16 15:28:33 +02:00
}
return out
2015-02-16 15:28:33 +02:00
case *ast.StringLiteral:
return &_nodeLiteral{
value: toValue_string(in.Value),
2015-02-16 15:28:33 +02:00
}
case *ast.ThisExpression:
return &_nodeThisExpression{}
case *ast.UnaryExpression:
return &_nodeUnaryExpression{
operator: in.Operator,
operand: cmpl.parseExpression(in.Operand),
postfix: in.Postfix,
2015-02-16 15:28:33 +02:00
}
case *ast.VariableExpression:
return &_nodeVariableExpression{
idx: in.Idx0(),
name: in.Name,
initializer: cmpl.parseExpression(in.Initializer),
2015-02-16 15:28:33 +02:00
}
}
panic(fmt.Errorf("Here be dragons: cmpl.parseExpression(%T)", in))
2015-02-16 15:28:33 +02:00
}
func (cmpl *_compiler) parseStatement(in ast.Statement) _nodeStatement {
if in == nil {
2015-02-16 15:28:33 +02:00
return nil
}
switch in := in.(type) {
2015-02-16 15:28:33 +02:00
case *ast.BlockStatement:
out := &_nodeBlockStatement{
list: make([]_nodeStatement, len(in.List)),
2015-02-16 15:28:33 +02:00
}
for i, value := range in.List {
out.list[i] = cmpl.parseStatement(value)
2015-02-16 15:28:33 +02:00
}
return out
2015-02-16 15:28:33 +02:00
case *ast.BranchStatement:
out := &_nodeBranchStatement{
branch: in.Token,
2015-02-16 15:28:33 +02:00
}
if in.Label != nil {
out.label = in.Label.Name
2015-02-16 15:28:33 +02:00
}
return out
2015-02-16 15:28:33 +02:00
case *ast.DebuggerStatement:
return &_nodeDebuggerStatement{}
case *ast.DoWhileStatement:
out := &_nodeDoWhileStatement{
test: cmpl.parseExpression(in.Test),
2015-02-16 15:28:33 +02:00
}
body := cmpl.parseStatement(in.Body)
2015-02-16 15:28:33 +02:00
if block, ok := body.(*_nodeBlockStatement); ok {
out.body = block.list
2015-02-16 15:28:33 +02:00
} else {
out.body = append(out.body, body)
2015-02-16 15:28:33 +02:00
}
return out
2015-02-16 15:28:33 +02:00
case *ast.EmptyStatement:
return emptyStatement
case *ast.ExpressionStatement:
return &_nodeExpressionStatement{
expression: cmpl.parseExpression(in.Expression),
2015-02-16 15:28:33 +02:00
}
case *ast.ForInStatement:
out := &_nodeForInStatement{
into: cmpl.parseExpression(in.Into),
source: cmpl.parseExpression(in.Source),
2015-02-16 15:28:33 +02:00
}
body := cmpl.parseStatement(in.Body)
2015-02-16 15:28:33 +02:00
if block, ok := body.(*_nodeBlockStatement); ok {
out.body = block.list
2015-02-16 15:28:33 +02:00
} else {
out.body = append(out.body, body)
2015-02-16 15:28:33 +02:00
}
return out
2015-02-16 15:28:33 +02:00
case *ast.ForStatement:
out := &_nodeForStatement{
initializer: cmpl.parseExpression(in.Initializer),
update: cmpl.parseExpression(in.Update),
test: cmpl.parseExpression(in.Test),
2015-02-16 15:28:33 +02:00
}
body := cmpl.parseStatement(in.Body)
2015-02-16 15:28:33 +02:00
if block, ok := body.(*_nodeBlockStatement); ok {
out.body = block.list
2015-02-16 15:28:33 +02:00
} else {
out.body = append(out.body, body)
2015-02-16 15:28:33 +02:00
}
return out
2015-02-16 15:28:33 +02:00
case *ast.FunctionStatement:
return emptyStatement
2015-02-16 15:28:33 +02:00
case *ast.IfStatement:
return &_nodeIfStatement{
test: cmpl.parseExpression(in.Test),
consequent: cmpl.parseStatement(in.Consequent),
alternate: cmpl.parseStatement(in.Alternate),
2015-02-16 15:28:33 +02:00
}
case *ast.LabelledStatement:
return &_nodeLabelledStatement{
label: in.Label.Name,
statement: cmpl.parseStatement(in.Statement),
2015-02-16 15:28:33 +02:00
}
case *ast.ReturnStatement:
return &_nodeReturnStatement{
argument: cmpl.parseExpression(in.Argument),
2015-02-16 15:28:33 +02:00
}
case *ast.SwitchStatement:
out := &_nodeSwitchStatement{
discriminant: cmpl.parseExpression(in.Discriminant),
default_: in.Default,
body: make([]*_nodeCaseStatement, len(in.Body)),
}
for i, clause := range in.Body {
out.body[i] = &_nodeCaseStatement{
test: cmpl.parseExpression(clause.Test),
consequent: make([]_nodeStatement, len(clause.Consequent)),
2015-02-16 15:28:33 +02:00
}
for j, value := range clause.Consequent {
out.body[i].consequent[j] = cmpl.parseStatement(value)
2015-02-16 15:28:33 +02:00
}
}
return out
2015-02-16 15:28:33 +02:00
case *ast.ThrowStatement:
return &_nodeThrowStatement{
argument: cmpl.parseExpression(in.Argument),
2015-02-16 15:28:33 +02:00
}
case *ast.TryStatement:
out := &_nodeTryStatement{
body: cmpl.parseStatement(in.Body),
finally: cmpl.parseStatement(in.Finally),
}
if in.Catch != nil {
out.catch = &_nodeCatchStatement{
parameter: in.Catch.Parameter.Name,
body: cmpl.parseStatement(in.Catch.Body),
2015-02-16 15:28:33 +02:00
}
}
return out
2015-02-16 15:28:33 +02:00
case *ast.VariableStatement:
out := &_nodeVariableStatement{
list: make([]_nodeExpression, len(in.List)),
2015-02-16 15:28:33 +02:00
}
for i, value := range in.List {
out.list[i] = cmpl.parseExpression(value)
2015-02-16 15:28:33 +02:00
}
return out
2015-02-16 15:28:33 +02:00
case *ast.WhileStatement:
out := &_nodeWhileStatement{
test: cmpl.parseExpression(in.Test),
2015-02-16 15:28:33 +02:00
}
body := cmpl.parseStatement(in.Body)
2015-02-16 15:28:33 +02:00
if block, ok := body.(*_nodeBlockStatement); ok {
out.body = block.list
2015-02-16 15:28:33 +02:00
} else {
out.body = append(out.body, body)
2015-02-16 15:28:33 +02:00
}
return out
2015-02-16 15:28:33 +02:00
case *ast.WithStatement:
return &_nodeWithStatement{
object: cmpl.parseExpression(in.Object),
body: cmpl.parseStatement(in.Body),
2015-02-16 15:28:33 +02:00
}
}
panic(fmt.Errorf("Here be dragons: cmpl.parseStatement(%T)", in))
2015-02-16 15:28:33 +02:00
}
func cmpl_parse(in *ast.Program) *_nodeProgram {
cmpl := _compiler{
program: in,
2015-02-16 15:28:33 +02:00
}
return cmpl.parse()
}
func (cmpl *_compiler) _parse(in *ast.Program) *_nodeProgram {
out := &_nodeProgram{
body: make([]_nodeStatement, len(in.Body)),
file: in.File,
2015-02-16 15:28:33 +02:00
}
for i, value := range in.Body {
out.body[i] = cmpl.parseStatement(value)
}
for _, value := range in.DeclarationList {
2015-02-16 15:28:33 +02:00
switch value := value.(type) {
case *ast.FunctionDeclaration:
out.functionList = append(out.functionList, cmpl.parseExpression(value.Function).(*_nodeFunctionLiteral))
2015-02-16 15:28:33 +02:00
case *ast.VariableDeclaration:
for _, value := range value.List {
out.varList = append(out.varList, value.Name)
2015-02-16 15:28:33 +02:00
}
default:
panic(fmt.Errorf("Here be dragons: cmpl.parseProgram.DeclarationList(%T)", value))
2015-02-16 15:28:33 +02:00
}
}
return out
2015-02-16 15:28:33 +02:00
}
type _nodeProgram struct {
body []_nodeStatement
varList []string
functionList []*_nodeFunctionLiteral
variableList []_nodeDeclaration
file *file.File
2015-02-16 15:28:33 +02:00
}
type _nodeDeclaration struct {
name string
definition _node
}
type _node interface {
}
type (
_nodeExpression interface {
_node
_expressionNode()
}
_nodeArrayLiteral struct {
value []_nodeExpression
}
_nodeAssignExpression struct {
operator token.Token
left _nodeExpression
right _nodeExpression
}
_nodeBinaryExpression struct {
operator token.Token
left _nodeExpression
right _nodeExpression
comparison bool
}
_nodeBracketExpression struct {
idx file.Idx
2015-02-16 15:28:33 +02:00
left _nodeExpression
member _nodeExpression
}
_nodeCallExpression struct {
callee _nodeExpression
argumentList []_nodeExpression
}
_nodeConditionalExpression struct {
test _nodeExpression
consequent _nodeExpression
alternate _nodeExpression
}
_nodeDotExpression struct {
idx file.Idx
2015-02-16 15:28:33 +02:00
left _nodeExpression
identifier string
}
_nodeFunctionLiteral struct {
name string
body _nodeStatement
source string
parameterList []string
varList []string
functionList []*_nodeFunctionLiteral
file *file.File
2015-02-16 15:28:33 +02:00
}
_nodeIdentifier struct {
idx file.Idx
2015-02-16 15:28:33 +02:00
name string
}
_nodeLiteral struct {
value Value
}
_nodeNewExpression struct {
callee _nodeExpression
argumentList []_nodeExpression
}
_nodeObjectLiteral struct {
value []_nodeProperty
}
_nodeProperty struct {
key string
kind string
value _nodeExpression
}
_nodeRegExpLiteral struct {
flags string
pattern string // Value?
regexp *regexp.Regexp
}
_nodeSequenceExpression struct {
sequence []_nodeExpression
}
_nodeThisExpression struct {
}
_nodeUnaryExpression struct {
operator token.Token
operand _nodeExpression
postfix bool
}
_nodeVariableExpression struct {
idx file.Idx
2015-02-16 15:28:33 +02:00
name string
initializer _nodeExpression
}
)
type (
_nodeStatement interface {
_node
_statementNode()
}
_nodeBlockStatement struct {
list []_nodeStatement
}
_nodeBranchStatement struct {
branch token.Token
label string
}
_nodeCaseStatement struct {
test _nodeExpression
consequent []_nodeStatement
}
_nodeCatchStatement struct {
parameter string
body _nodeStatement
}
_nodeDebuggerStatement struct {
}
_nodeDoWhileStatement struct {
test _nodeExpression
body []_nodeStatement
}
_nodeEmptyStatement struct {
}
_nodeExpressionStatement struct {
expression _nodeExpression
}
_nodeForInStatement struct {
into _nodeExpression
source _nodeExpression
body []_nodeStatement
}
_nodeForStatement struct {
initializer _nodeExpression
update _nodeExpression
test _nodeExpression
body []_nodeStatement
}
_nodeIfStatement struct {
test _nodeExpression
consequent _nodeStatement
alternate _nodeStatement
}
_nodeLabelledStatement struct {
label string
statement _nodeStatement
}
_nodeReturnStatement struct {
argument _nodeExpression
}
_nodeSwitchStatement struct {
discriminant _nodeExpression
default_ int
body []*_nodeCaseStatement
}
_nodeThrowStatement struct {
argument _nodeExpression
}
_nodeTryStatement struct {
body _nodeStatement
catch *_nodeCatchStatement
finally _nodeStatement
}
_nodeVariableStatement struct {
list []_nodeExpression
}
_nodeWhileStatement struct {
test _nodeExpression
body []_nodeStatement
}
_nodeWithStatement struct {
object _nodeExpression
body _nodeStatement
}
)
// _expressionNode
func (*_nodeArrayLiteral) _expressionNode() {}
func (*_nodeAssignExpression) _expressionNode() {}
func (*_nodeBinaryExpression) _expressionNode() {}
func (*_nodeBracketExpression) _expressionNode() {}
func (*_nodeCallExpression) _expressionNode() {}
func (*_nodeConditionalExpression) _expressionNode() {}
func (*_nodeDotExpression) _expressionNode() {}
func (*_nodeFunctionLiteral) _expressionNode() {}
func (*_nodeIdentifier) _expressionNode() {}
func (*_nodeLiteral) _expressionNode() {}
func (*_nodeNewExpression) _expressionNode() {}
func (*_nodeObjectLiteral) _expressionNode() {}
func (*_nodeRegExpLiteral) _expressionNode() {}
func (*_nodeSequenceExpression) _expressionNode() {}
func (*_nodeThisExpression) _expressionNode() {}
func (*_nodeUnaryExpression) _expressionNode() {}
func (*_nodeVariableExpression) _expressionNode() {}
// _statementNode
func (*_nodeBlockStatement) _statementNode() {}
func (*_nodeBranchStatement) _statementNode() {}
func (*_nodeCaseStatement) _statementNode() {}
func (*_nodeCatchStatement) _statementNode() {}
func (*_nodeDebuggerStatement) _statementNode() {}
func (*_nodeDoWhileStatement) _statementNode() {}
func (*_nodeEmptyStatement) _statementNode() {}
func (*_nodeExpressionStatement) _statementNode() {}
func (*_nodeForInStatement) _statementNode() {}
func (*_nodeForStatement) _statementNode() {}
func (*_nodeIfStatement) _statementNode() {}
func (*_nodeLabelledStatement) _statementNode() {}
func (*_nodeReturnStatement) _statementNode() {}
func (*_nodeSwitchStatement) _statementNode() {}
func (*_nodeThrowStatement) _statementNode() {}
func (*_nodeTryStatement) _statementNode() {}
func (*_nodeVariableStatement) _statementNode() {}
func (*_nodeWhileStatement) _statementNode() {}
func (*_nodeWithStatement) _statementNode() {}