feat(core): handle PrefixUnaryExpression.MinusToken

astahmer · Feb 28, 2023 · 83bfd0a · 83bfd0a
1 parent f550ed7
commit 83bfd0a
Show file tree

Hide file tree

Showing 3 changed files with 100 additions and 0 deletions.
diff --git a/.changeset/hip-nails-watch.md b/.changeset/hip-nails-watch.md
@@ -0,0 +1,5 @@
+---
+"@box-extractor/core": patch
+---
+
+feat(core): handle PrefixUnaryExpression.MinusToken
diff --git a/packages/box-extractor/src/extractor/maybeBoxNode.ts b/packages/box-extractor/src/extractor/maybeBoxNode.ts
@@ -74,6 +74,17 @@ export function maybeBoxNode(node: Node, stack: Node[]): MaybeBoxNodeReturn {
         return cache(box.literal(node.getLiteralValue(), node, stack));
     }
 
+    if (Node.isPrefixUnaryExpression(node)) {
+        const operand = node.getOperand();
+        const operator = node.getOperatorToken();
+        const maybeBox = maybeBoxNode(operand, stack);
+        if (!maybeBox || !(maybeBox.isLiteral() && maybeBox.kind === "number")) return;
+
+        return cache(
+            operator === ts.SyntaxKind.MinusToken ? box.literal(-(maybeBox.value as number), node, stack) : maybeBox
+        );
+    }
+
     // <ColorBox bool={true} falsy={false} />
     if (Node.isTrueLiteral(node) || Node.isFalseLiteral(node)) {
         return cache(box.literal(node.getLiteralValue(), node, stack));

diff --git a/packages/box-extractor/tests/extract.test.ts b/packages/box-extractor/tests/extract.test.ts
@@ -6482,6 +6482,90 @@ it("extract JsxAttribute > JsxExpression  > NumericLiteral", () => {
     `);
 });
 
+it("extract JsxAttribute > JsxExpression > NumericLiteral > PrefixUnaryExpression", () => {
+    expect(
+        extractFromCode(
+            `
+            <ThreeBox zIndex={1} position={[
+                -1.2466866852487384, 0.3325255778835592, -0.6517939595349769,
+              ]} scale={+1.25} someProp={-2}></ThreeBox>
+        `,
+            { components: ["ThreeBox"] }
+        )
+    ).toMatchInlineSnapshot(`
+      [
+          [
+              "ThreeBox",
+              [
+                  ["zIndex", 1],
+                  ["position", [-1.2466866852487384, 0.3325255778835592, -0.6517939595349769]],
+                  ["scale", 1.25],
+                  ["someProp", -2],
+              ],
+              {
+                  zIndex: [
+                      {
+                          stack: ["JsxAttribute", "JsxExpression", "NumericLiteral"],
+                          type: "literal",
+                          node: "NumericLiteral",
+                          value: 1,
+                          kind: "number",
+                      },
+                  ],
+                  position: [
+                      {
+                          stack: ["JsxAttribute", "JsxExpression", "ArrayLiteralExpression"],
+                          type: "list",
+                          node: "ArrayLiteralExpression",
+                          value: [
+                              {
+                                  stack: ["JsxAttribute", "JsxExpression", "ArrayLiteralExpression"],
+                                  type: "literal",
+                                  node: "PrefixUnaryExpression",
+                                  value: -1.2466866852487384,
+                                  kind: "number",
+                              },
+                              {
+                                  stack: ["JsxAttribute", "JsxExpression", "ArrayLiteralExpression"],
+                                  type: "literal",
+                                  node: "NumericLiteral",
+                                  value: 0.3325255778835592,
+                                  kind: "number",
+                              },
+                              {
+                                  stack: ["JsxAttribute", "JsxExpression", "ArrayLiteralExpression"],
+                                  type: "literal",
+                                  node: "PrefixUnaryExpression",
+                                  value: -0.6517939595349769,
+                                  kind: "number",
+                              },
+                          ],
+                      },
+                  ],
+                  scale: [
+                      {
+                          stack: ["JsxAttribute", "JsxExpression", "PrefixUnaryExpression"],
+                          type: "literal",
+                          node: "NumericLiteral",
+                          value: 1.25,
+                          kind: "number",
+                      },
+                  ],
+                  someProp: [
+                      {
+                          stack: ["JsxAttribute", "JsxExpression", "PrefixUnaryExpression"],
+                          type: "literal",
+                          node: "PrefixUnaryExpression",
+                          value: -2,
+                          kind: "number",
+                      },
+                  ],
+              },
+          ],
+      ]
+    `);
+});
+
 it("extract JsxAttribute > JsxExpression > Identifier > NumericLiteral", () => {
     expect(
         extractFromCode(`