topchetoeu/j2s
1
0
Fork 0
Code Issues 6 Pull Requests Actions Packages Projects Releases 74 Wiki Activity

feat: add assign shorthands

Browse Source
This commit is contained in:
TopchetoEU 2024-09-14 14:33:09 +03:00
parent cb82f4cf32
commit 4e8b110fc4
Signed by: topchetoeu
GPG Key ID: 6531B8583E5F6ED4
1 changed files with 47 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
src/main/java/me/topchetoeu/jscript/compilation/values/ObjectNode.java
View File

@ -136,7 +136,7 @@ public class ObjectNode extends Node implements AssignTargetLike {
public final Node key; public final Node key;
public final Node value; public final Node value;
@Override public void compile(CompileResult target, boolean pollute, BreakpointType bp) { @Override public void compile(CompileResult target, boolean pollute) {
key.compile(target, true); key.compile(target, true);
if (value == null) target.add(Instruction.pushUndefined()); if (value == null) target.add(Instruction.pushUndefined());
@ -178,15 +178,7 @@ public class ObjectNode extends Node implements AssignTargetLike {
if (!var.isSuccess()) return var.chainError(); if (!var.isSuccess()) return var.chainError();
n += var.n; n += var.n;
if (!src.is(i + n, "=")) return ParseRes.res(new FieldMemberNode(loc, new StringNode(loc, var.result.name), var.result), n); return ParseRes.res(new FieldMemberNode(loc, new StringNode(loc, var.result.name), var.result), n);
var equalsLoc = src.loc(i + n);
n++;
var value = JavaScript.parseExpression(src, i + n, 2);
if (!value.isSuccess()) return value.chainError(src.loc(i + n), "Expected a shorthand initializer");
n += value.n;
return ParseRes.res(new FieldMemberNode(loc, new StringNode(loc, var.result.name), new AssignNode(equalsLoc, var.result, value.result)), n);
} }
public static ParseRes<FieldMemberNode> parseClass(Source src, int i) { public static ParseRes<FieldMemberNode> parseClass(Source src, int i) {
@ -214,9 +206,50 @@ public class ObjectNode extends Node implements AssignTargetLike {
return ParseRes.res(new FieldMemberNode(loc, name.result, value.result), n); return ParseRes.res(new FieldMemberNode(loc, name.result, value.result), n);
} }
} }
public static class AssignShorthandNode extends Node {
public final Node key;
public final AssignTarget target;
public final Node value;
@Override public void compile(CompileResult target, boolean pollute) {
throw new SyntaxException(loc(), "Unexpected assign shorthand in non-destructor context");
}
public AssignShorthandNode(Location loc, Node key, AssignTarget target, Node value) {
super(loc);
this.key = key;
this.target = target;
this.value = value;
}
public AssignTarget target() {
return new AssignNode(loc(), target, value);
}
public static ParseRes<AssignShorthandNode> parse(Source src, int i) {
var n = Parsing.skipEmpty(src, i);
var loc = src.loc(i + n);
var var = VariableNode.parse(src, i + n);
if (!var.isSuccess()) return var.chainError();
n += var.n;
n += Parsing.skipEmpty(src, i + n);
if (!src.is(i + n, "=")) return ParseRes.failed();
n++;
var value = JavaScript.parseExpression(src, i + n, 2);
if (!value.isSuccess()) return value.chainError(src.loc(i + n), "Expected a shorthand initializer");
n += value.n;
return ParseRes.res(new AssignShorthandNode(loc, new StringNode(loc, var.result.name), var.result, value.result), n);
}
}
public final List<Node> members; public final List<Node> members;
// TODO: Implement spreading into object
// private void compileRestObjBuilder(CompileResult target, int srcDupN) { // private void compileRestObjBuilder(CompileResult target, int srcDupN) {
// var subtarget = target.subtarget(); // var subtarget = target.subtarget();
// var src = subtarget.scope.defineTemp(); // var src = subtarget.scope.defineTemp();
@ -274,6 +307,7 @@ public class ObjectNode extends Node implements AssignTargetLike {
if (field.value instanceof AssignTargetLike target) newMembers.add(new Member<>(field.key, target.toAssignTarget())); if (field.value instanceof AssignTargetLike target) newMembers.add(new Member<>(field.key, target.toAssignTarget()));
else throw new SyntaxException(field.value.loc(), "Expected an assignable in deconstructor"); else throw new SyntaxException(field.value.loc(), "Expected an assignable in deconstructor");
} }
else if (el instanceof AssignShorthandNode shorthand) newMembers.add(new Member<>(shorthand.key, shorthand.target()));
else throw new SyntaxException(el.loc(), "Unexpected member in deconstructor"); else throw new SyntaxException(el.loc(), "Unexpected member in deconstructor");
} }
@ -335,7 +369,9 @@ public class ObjectNode extends Node implements AssignTargetLike {
ParseRes<Node> prop = ParseRes.first(src, i + n, ParseRes<Node> prop = ParseRes.first(src, i + n,
MethodMemberNode::parse, MethodMemberNode::parse,
PropertyMemberNode::parse, PropertyMemberNode::parse,
FieldMemberNode::parseObject FieldMemberNode::parseObject,
AssignShorthandNode::parse,
FieldMemberNode::parseShorthand
); );
if (!prop.isSuccess()) return prop.chainError(src.loc(i + n), "Expected a member in object literal"); if (!prop.isSuccess()) return prop.chainError(src.loc(i + n), "Expected a member in object literal");
n += prop.n; n += prop.n;