jspacker - ast.go

  package jspacker
  
  import (
  	"strconv"
  
  	"vimagination.zapto.org/javascript"
  )
  
  func namespaceImport(ns *javascript.Token, prefix string) javascript.ModuleItem {
  	return wrapConst([]javascript.LexicalBinding{
  		{
  			BindingIdentifier: ns,
  			Initializer: &javascript.AssignmentExpression{
  				ConditionalExpression: javascript.WrapConditional(&javascript.PrimaryExpression{
  					IdentifierReference: jToken(prefix),
  				}),
  			},
  		},
  	})
  }
  
  func wrapConst(lds []javascript.LexicalBinding) javascript.ModuleItem {
  	return wrapDeclaration(&javascript.Declaration{
  		LexicalDeclaration: &javascript.LexicalDeclaration{
  			LetOrConst:  javascript.Const,
  			BindingList: lds,
  		},
  	})
  }
  
  func wrapDeclaration(ed *javascript.Declaration) javascript.ModuleItem {
  	return javascript.ModuleItem{
  		StatementListItem: &javascript.StatementListItem{
  			Declaration: ed,
  		},
  	}
  }
  
  func wrapVariableStatement(v *javascript.VariableStatement) javascript.ModuleItem {
  	return javascript.ModuleItem{
  		StatementListItem: &javascript.StatementListItem{
  			Statement: &javascript.Statement{
  				VariableStatement: v,
  			},
  		},
  	}
  }
  
  func wrapFunctionDeclaration(f *javascript.FunctionDeclaration) javascript.ModuleItem {
  	return wrapDeclaration(&javascript.Declaration{
  		FunctionDeclaration: f,
  	})
  }
  
  func wrapClassDeclaration(c *javascript.ClassDeclaration) javascript.ModuleItem {
  	return wrapDeclaration(&javascript.Declaration{
  		ClassDeclaration: c,
  	})
  }
  
  func wrapDefaultAssignment(def *javascript.Token, a *javascript.AssignmentExpression) javascript.ModuleItem {
  	return wrapConst([]javascript.LexicalBinding{
  		{
  			BindingIdentifier: def,
  			Initializer:       a,
  		},
  	})
  }
  
  func wrapAssignmentExpression(ae javascript.AssignmentExpression) javascript.ModuleItem {
  	return javascript.ModuleItem{
  		StatementListItem: &javascript.StatementListItem{
  			Statement: &javascript.Statement{
  				ExpressionStatement: &javascript.Expression{
  					Expressions: []javascript.AssignmentExpression{ae},
  				},
  			},
  		},
  	}
  }
  
  func importMeta(prefix, url string) javascript.LexicalBinding {
  	return javascript.LexicalBinding{
  		BindingIdentifier: jToken(prefix + "import"),
  		Initializer: &javascript.AssignmentExpression{
  			ConditionalExpression: javascript.WrapConditional(&javascript.ObjectLiteral{
  				PropertyDefinitionList: []javascript.PropertyDefinition{
  					{
  						PropertyName: &javascript.PropertyName{
  							LiteralPropertyName: jToken("url"),
  						},
  						AssignmentExpression: &javascript.AssignmentExpression{
  							ConditionalExpression: javascript.WrapConditional(&javascript.AdditiveExpression{
  								AdditiveExpression: &javascript.AdditiveExpression{
  									MultiplicativeExpression: javascript.MultiplicativeExpression{
  										ExponentiationExpression: javascript.ExponentiationExpression{
  											UnaryExpression: javascript.UnaryExpression{
  												UpdateExpression: javascript.UpdateExpression{
  													LeftHandSideExpression: &javascript.LeftHandSideExpression{
  														NewExpression: &javascript.NewExpression{
  															MemberExpression: javascript.MemberExpression{
  																PrimaryExpression: &javascript.PrimaryExpression{
  																	IdentifierReference: jToken("o"),
  																},
  															},
  														},
  													},
  												},
  											},
  										},
  									},
  								},
  								AdditiveOperator: javascript.AdditiveAdd,
  								MultiplicativeExpression: javascript.MultiplicativeExpression{
  									ExponentiationExpression: javascript.ExponentiationExpression{
  										UnaryExpression: javascript.UnaryExpression{
  											UpdateExpression: javascript.UpdateExpression{
  												LeftHandSideExpression: &javascript.LeftHandSideExpression{
  													NewExpression: &javascript.NewExpression{
  														MemberExpression: javascript.MemberExpression{
  															PrimaryExpression: &javascript.PrimaryExpression{
  																Literal: jToken(strconv.Quote(url)),
  															},
  														},
  													},
  												},
  											},
  										},
  									},
  								},
  							}),
  						},
  					},
  				},
  			}),
  		},
  	}
  }
  
  func replaceImportCall(ce *javascript.CallExpression) {
  	ce.MemberExpression = &javascript.MemberExpression{
  		PrimaryExpression: &javascript.PrimaryExpression{
  			IdentifierReference: jToken("include"),
  		},
  	}
  	ce.Arguments = &javascript.Arguments{
  		ArgumentList: []javascript.Argument{
  			{
  				AssignmentExpression: *ce.ImportCall,
  			},
  		},
  	}
  	ce.ImportCall = nil
  }
  
  func locationOrigin() javascript.ModuleItem {
  	return wrapConst([]javascript.LexicalBinding{
  		{
  			BindingIdentifier: jToken("o"),
  			Initializer: &javascript.AssignmentExpression{
  				ConditionalExpression: javascript.WrapConditional(javascript.MemberExpression{
  					MemberExpression: &javascript.MemberExpression{
  						PrimaryExpression: &javascript.PrimaryExpression{
  							IdentifierReference: jToken("location"),
  						},
  					},
  					IdentifierName: jToken("origin"),
  				}),
  			},
  		},
  	})
  }
  
  func wrapArgument(arg string) javascript.Argument {
  	return javascript.Argument{
  		AssignmentExpression: javascript.AssignmentExpression{
  			ConditionalExpression: javascript.WrapConditional(&javascript.PrimaryExpression{
  				Literal: jToken(strconv.Quote(arg)),
  			}),
  		},
  	}
  }
  
  func wrapMemberIdentifier(id string, in *javascript.Token) javascript.MemberExpression {
  	return javascript.MemberExpression{
  		MemberExpression: &javascript.MemberExpression{
  			PrimaryExpression: &javascript.PrimaryExpression{
  				IdentifierReference: jToken(id),
  			},
  		},
  		IdentifierName: in,
  	}
  }
  
  func wrapIncludeCall(ident *javascript.Token, args []javascript.Argument) javascript.ModuleItem {
  	return wrapConst([]javascript.LexicalBinding{
  		{
  			BindingIdentifier: ident,
  			Initializer: awaitCall(&javascript.MemberExpression{
  				PrimaryExpression: &javascript.PrimaryExpression{
  					IdentifierReference: jToken("include"),
  				},
  			},
  				args,
  			),
  		},
  	})
  }
  
  func awaitCall(me *javascript.MemberExpression, args []javascript.Argument) *javascript.AssignmentExpression {
  	return &javascript.AssignmentExpression{
  		ConditionalExpression: javascript.WrapConditional(&javascript.UnaryExpression{
  			UnaryOperators: []javascript.UnaryOperatorComments{{UnaryOperator: javascript.UnaryAwait}},
  			UpdateExpression: javascript.UpdateExpression{
  				LeftHandSideExpression: &javascript.LeftHandSideExpression{
  					CallExpression: &javascript.CallExpression{
  						MemberExpression: me,
  						Arguments: &javascript.Arguments{
  							ArgumentList: args,
  						},
  					},
  				},
  			},
  		}),
  	}
  }
  
  func wrapIncludeAllCall(importObjectBindings []javascript.BindingElement, importURLsArrayE []javascript.ArrayElement) javascript.ModuleItem {
  	return wrapConst([]javascript.LexicalBinding{
  		{
  			ArrayBindingPattern: &javascript.ArrayBindingPattern{
  				BindingElementList: importObjectBindings,
  			},
  			Initializer: awaitCall(&javascript.MemberExpression{
  				MemberExpression: &javascript.MemberExpression{
  					PrimaryExpression: &javascript.PrimaryExpression{
  						IdentifierReference: jToken("Promise"),
  					},
  				},
  				IdentifierName: jToken("all"),
  			},
  				[]javascript.Argument{
  					{
  						AssignmentExpression: javascript.AssignmentExpression{
  							ConditionalExpression: javascript.WrapConditional(&javascript.CallExpression{
  								MemberExpression: &javascript.MemberExpression{
  									MemberExpression: &javascript.MemberExpression{
  										PrimaryExpression: &javascript.PrimaryExpression{
  											ArrayLiteral: &javascript.ArrayLiteral{
  												ElementList: importURLsArrayE,
  											},
  										},
  									},
  									IdentifierName: jToken("map"),
  								},
  								Arguments: &javascript.Arguments{
  									ArgumentList: []javascript.Argument{
  										{
  											AssignmentExpression: javascript.AssignmentExpression{
  												ConditionalExpression: javascript.WrapConditional(&javascript.PrimaryExpression{
  													IdentifierReference: jToken("include"),
  												}),
  											},
  										},
  									},
  								},
  							}),
  						},
  					},
  				},
  			),
  		},
  	})
  }
  
  func makeGetter(binding string, id *javascript.Token) javascript.PropertyDefinition {
  	return javascript.PropertyDefinition{
  		MethodDefinition: &javascript.MethodDefinition{
  			Type: javascript.MethodGetter,
  			ClassElementName: javascript.ClassElementName{
  				PropertyName: &javascript.PropertyName{
  					LiteralPropertyName: jToken(binding),
  				},
  			},
  			FunctionBody: javascript.Block{
  				StatementList: []javascript.StatementListItem{
  					{
  						Statement: &javascript.Statement{
  							Type: javascript.StatementReturn,
  							ExpressionStatement: &javascript.Expression{
  								Expressions: []javascript.AssignmentExpression{
  									{
  										ConditionalExpression: javascript.WrapConditional(&javascript.PrimaryExpression{
  											IdentifierReference: id,
  										}),
  									},
  								},
  							},
  						},
  					},
  				},
  			},
  		},
  	}
  }
  
  func wrapNameSpaceFields(id string, fields []javascript.PropertyDefinition) javascript.LexicalBinding {
  	return javascript.LexicalBinding{
  		BindingIdentifier: jToken(id),
  		Initializer: &javascript.AssignmentExpression{
  			ConditionalExpression: javascript.WrapConditional(&javascript.ObjectLiteral{
  				PropertyDefinitionList: fields,
  			}),
  		},
  	}
  }
  
  func wrapURLNameSpace(url, prefix string) javascript.ArrayElement {
  	return javascript.ArrayElement{
  		AssignmentExpression: javascript.AssignmentExpression{
  			ConditionalExpression: javascript.WrapConditional(&javascript.ArrayLiteral{
  				ElementList: []javascript.ArrayElement{
  					{
  						AssignmentExpression: javascript.AssignmentExpression{
  							ConditionalExpression: javascript.WrapConditional(&javascript.PrimaryExpression{
  								Literal: jToken(strconv.Quote(url)),
  							}),
  						},
  					},
  					{
  						AssignmentExpression: javascript.AssignmentExpression{
  							ConditionalExpression: javascript.WrapConditional(&javascript.PrimaryExpression{
  								IdentifierReference: jToken(prefix),
  							}),
  						},
  					},
  				},
  			}),
  		},
  	}
  }
  
  func wrapImports(imports []javascript.ArrayElement) javascript.ModuleItem {
  	return javascript.ModuleItem{
  		StatementListItem: &javascript.StatementListItem{
  			Statement: &javascript.Statement{
  				ExpressionStatement: &javascript.Expression{
  					Expressions: []javascript.AssignmentExpression{
  						{
  							ConditionalExpression: javascript.WrapConditional(&javascript.CallExpression{
  								MemberExpression: &javascript.MemberExpression{
  									MemberExpression: &javascript.MemberExpression{
  										PrimaryExpression: &javascript.PrimaryExpression{
  											IdentifierReference: jToken("Object"),
  										},
  									},
  									IdentifierName: jToken("defineProperty"),
  								},
  								Arguments: &javascript.Arguments{
  									ArgumentList: []javascript.Argument{
  										{
  											AssignmentExpression: javascript.AssignmentExpression{
  												ConditionalExpression: javascript.WrapConditional(&javascript.PrimaryExpression{
  													IdentifierReference: jToken("globalThis"),
  												}),
  											},
  										},
  										{
  											AssignmentExpression: javascript.AssignmentExpression{
  												ConditionalExpression: javascript.WrapConditional(&javascript.PrimaryExpression{
  													Literal: jToken("include"),
  												}),
  											},
  										},
  										{
  											AssignmentExpression: javascript.AssignmentExpression{
  												ConditionalExpression: javascript.WrapConditional(&javascript.ObjectLiteral{
  													PropertyDefinitionList: []javascript.PropertyDefinition{
  														{
  															PropertyName: &javascript.PropertyName{
  																LiteralPropertyName: jToken("value"),
  															},
  															AssignmentExpression: &javascript.AssignmentExpression{
  																ConditionalExpression: javascript.WrapConditional(&javascript.CallExpression{
  																	MemberExpression: &javascript.MemberExpression{
  																		PrimaryExpression: &javascript.PrimaryExpression{
  																			ParenthesizedExpression: &javascript.ParenthesizedExpression{
  																				Expressions: []javascript.AssignmentExpression{
  																					{
  																						ArrowFunction: &javascript.ArrowFunction{
  																							FormalParameters: &javascript.FormalParameters{},
  																							FunctionBody: &javascript.Block{
  																								StatementList: []javascript.StatementListItem{
  																									{
  																										Declaration: &javascript.Declaration{
  																											LexicalDeclaration: &javascript.LexicalDeclaration{
  																												LetOrConst: javascript.Const,
  																												BindingList: []javascript.LexicalBinding{
  																													{
  																														BindingIdentifier: jToken("imports"),
  																														Initializer: &javascript.AssignmentExpression{
  																															ConditionalExpression: javascript.WrapConditional(javascript.MemberExpression{
  																																MemberExpression: &javascript.MemberExpression{
  																																	PrimaryExpression: &javascript.PrimaryExpression{
  																																		IdentifierReference: jToken("Map"),
  																																	},
  																																},
  																																Arguments: &javascript.Arguments{
  																																	ArgumentList: []javascript.Argument{
  																																		{
  																																			AssignmentExpression: javascript.AssignmentExpression{
  																																				ConditionalExpression: javascript.WrapConditional(&javascript.ArrayLiteral{
  																																					ElementList: imports,
  																																				}),
  																																			},
  																																		},
  																																	},
  																																},
  																															}),
  																														},
  																													},
  																												},
  																											},
  																										},
  																									},
  																									{
  																										Statement: &javascript.Statement{
  																											Type: javascript.StatementReturn,
  																											ExpressionStatement: &javascript.Expression{
  																												Expressions: []javascript.AssignmentExpression{
  																													{
  																														ArrowFunction: &javascript.ArrowFunction{
  																															BindingIdentifier: jToken("url"),
  																															AssignmentExpression: &javascript.AssignmentExpression{
  																																ConditionalExpression: javascript.WrapConditional(&javascript.CallExpression{
  																																	MemberExpression: &javascript.MemberExpression{
  																																		MemberExpression: &javascript.MemberExpression{
  																																			PrimaryExpression: &javascript.PrimaryExpression{
  																																				IdentifierReference: jToken("Promise"),
  																																			},
  																																			IdentifierName: jToken("resolve"),
  																																		},
  																																	},
  																																	Arguments: &javascript.Arguments{
  																																		ArgumentList: []javascript.Argument{
  																																			{
  																																				AssignmentExpression: javascript.AssignmentExpression{
  																																					ConditionalExpression: &javascript.ConditionalExpression{
  																																						CoalesceExpression: &javascript.CoalesceExpression{
  																																							CoalesceExpressionHead: &javascript.CoalesceExpression{
  																																								BitwiseORExpression: javascript.WrapConditional(&javascript.CallExpression{
  																																									MemberExpression: &javascript.MemberExpression{
  																																										MemberExpression: &javascript.MemberExpression{
  																																											PrimaryExpression: &javascript.PrimaryExpression{
  																																												IdentifierReference: jToken("imports"),
  																																											},
  																																										},
  																																										IdentifierName: jToken("get"),
  																																									},
  																																									Arguments: &javascript.Arguments{
  																																										ArgumentList: []javascript.Argument{
  																																											{
  																																												AssignmentExpression: javascript.AssignmentExpression{
  																																													ConditionalExpression: javascript.WrapConditional(&javascript.PrimaryExpression{
  																																														IdentifierReference: jToken("url"),
  																																													}),
  																																												},
  																																											},
  																																										},
  																																									},
  																																								}).LogicalORExpression.LogicalANDExpression.BitwiseORExpression,
  																																							},
  																																							BitwiseORExpression: javascript.WrapConditional(&javascript.CallExpression{
  																																								ImportCall: &javascript.AssignmentExpression{
  																																									ConditionalExpression: javascript.WrapConditional(&javascript.PrimaryExpression{
  																																										IdentifierReference: jToken("url"),
  																																									}),
  																																								},
  																																							}).LogicalORExpression.LogicalANDExpression.BitwiseORExpression,
  																																						},
  																																					},
  																																				},
  																																			},
  																																		},
  																																	},
  																																}),
  																															},
  																														},
  																													},
  																												},
  																											},
  																										},
  																									},
  																								},
  																							},
  																						},
  																					},
  																				},
  																			},
  																		},
  																	},
  																	Arguments: &javascript.Arguments{},
  																}),
  															},
  														},
  													},
  												}),
  											},
  										},
  									},
  								},
  							}),
  						},
  					},
  				},
  			},
  		},
  	}
  }