From 199e0efb45807324d5467a950e74b9d0f9c23252 Mon Sep 17 00:00:00 2001
From: Bronley Plumb <bronley@gmail.com>
Date: Thu, 28 Oct 2021 15:26:08 -0400
Subject: [PATCH] Add transpile fix for instantresume components (#465)

* Add transpile fix for instantresume components

* Fix bug
---
 src/astUtils/xml.ts       |  3 +++
 src/files/XmlFile.spec.ts | 37 +++++++++++++++++++++++++++++++++++++
 src/parser/SGParser.ts    |  2 +-
 src/parser/SGTypes.ts     | 13 ++++++++-----
 4 files changed, 49 insertions(+), 6 deletions(-)

diff --git a/src/astUtils/xml.ts b/src/astUtils/xml.ts
index bb501eec7..11c2df7fa 100644
--- a/src/astUtils/xml.ts
+++ b/src/astUtils/xml.ts
@@ -21,3 +21,6 @@ export function isSGFunction(tag: SGTag): tag is SGFunction {
 export function isSGNode(tag: SGTag): tag is SGNode {
     return tag?.constructor.name === 'SGNode';
 }
+export function isSGCustomization(tag: SGTag): tag is SGNode {
+    return isSGNode(tag) && tag.tag?.text?.toLowerCase() === 'customization';
+}
diff --git a/src/files/XmlFile.spec.ts b/src/files/XmlFile.spec.ts
index 261dc84c7..fee1427ed 100644
--- a/src/files/XmlFile.spec.ts
+++ b/src/files/XmlFile.spec.ts
@@ -11,6 +11,8 @@ import { BrsFile } from './BrsFile';
 import { XmlFile } from './XmlFile';
 import { standardizePath as s } from '../util';
 import { expectZeroDiagnostics, getTestTranspile, trim, trimMap } from '../testHelpers.spec';
+import { ProgramBuilder } from '../ProgramBuilder';
+import { LogLevel } from '../Logger';
 
 describe('XmlFile', () => {
     const tempDir = s`${process.cwd()}/.tmp`;
@@ -624,6 +626,41 @@ describe('XmlFile', () => {
     });
 
     describe('transpile', () => {
+        it('supports instantresume <customization> elements', async () => {
+            fsExtra.outputFileSync(`${rootDir}/manifest`, '');
+            fsExtra.outputFileSync(`${rootDir}/source/main.brs`, `sub main()\nend sub`);
+            fsExtra.outputFileSync(`${rootDir}/components/MainScene.xml`, trim`
+                <?xml version="1.0" encoding="utf-8" ?>
+                <component name="MainScene" extends="Scene">
+                    <customization resumehandler="customResume" />
+                    <customization suspendhandler="customSuspend" />
+                    <children>
+                        <Rectangle width="1920" height="1080" />
+                    </children>
+                </component>
+            `);
+            const builder = new ProgramBuilder();
+            await builder.run({
+                cwd: rootDir,
+                retainStagingFolder: true,
+                stagingFolderPath: stagingDir,
+                logLevel: LogLevel.off
+            });
+            expect(
+                fsExtra.readFileSync(`${stagingDir}/components/MainScene.xml`).toString()
+            ).to.eql(trim`
+                <?xml version="1.0" encoding="utf-8" ?>
+                <component name="MainScene" extends="Scene">
+                    <script type="text/brightscript" uri="pkg:/source/bslib.brs" />
+                    <children>
+                        <Rectangle width="1920" height="1080" />
+                    </children>
+                    <customization resumehandler="customResume" />
+                    <customization suspendhandler="customSuspend" />
+                </component>
+            `);
+        });
+
         it(`honors the 'needsTranspiled' flag when set in 'afterFileParse'`, () => {
             program.plugins.add({
                 name: 'test',
diff --git a/src/parser/SGParser.ts b/src/parser/SGParser.ts
index 6d6ebe112..d0b849608 100644
--- a/src/parser/SGParser.ts
+++ b/src/parser/SGParser.ts
@@ -189,7 +189,7 @@ function mapElement({ children }: ElementCstNode, diagnostics: Diagnostic[]): SG
     const content = children.content?.[0];
     switch (name.text) {
         case 'component':
-            const componentContent = mapElements(content, ['interface', 'script', 'children'], diagnostics);
+            const componentContent = mapElements(content, ['interface', 'script', 'children', 'customization'], diagnostics);
             return new SGComponent(name, attributes, componentContent, range);
         case 'interface':
             const interfaceContent = mapElements(content, ['field', 'function'], diagnostics);
diff --git a/src/parser/SGTypes.ts b/src/parser/SGTypes.ts
index 355448056..d0db864bb 100644
--- a/src/parser/SGTypes.ts
+++ b/src/parser/SGTypes.ts
@@ -1,7 +1,7 @@
 import { SourceNode } from 'source-map';
 import type { Range } from 'vscode-languageserver';
 import { createSGAttribute } from '../astUtils/creators';
-import { isSGChildren, isSGField, isSGFunction, isSGInterface, isSGScript } from '../astUtils/xml';
+import { isSGChildren, isSGCustomization, isSGField, isSGFunction, isSGInterface, isSGScript } from '../astUtils/xml';
 import type { FileReference, TranspileResult } from '../interfaces';
 import util from '../util';
 import type { TranspileState } from './TranspileState';
@@ -296,8 +296,6 @@ export class SGInterface extends SGTag {
                     this.fields.push(tag);
                 } else if (isSGFunction(tag)) {
                     this.functions.push(tag);
-                } else {
-                    throw new Error(`Unexpected tag ${tag.tag.text}`);
                 }
             }
         }
@@ -366,8 +364,8 @@ export class SGComponent extends SGTag {
                     this.scripts.push(tag);
                 } else if (isSGChildren(tag)) {
                     this.children = tag;
-                } else {
-                    throw new Error(`Unexpected tag ${tag.tag.text}`);
+                } else if (isSGCustomization(tag)) {
+                    this.customizations.push(tag);
                 }
             }
         }
@@ -379,6 +377,8 @@ export class SGComponent extends SGTag {
 
     public children: SGChildren;
 
+    public customizations: SGNode[] = [];
+
     get name() {
         return this.getAttributeValue('name');
     }
@@ -405,6 +405,9 @@ export class SGComponent extends SGTag {
         if (this.children) {
             body.push(this.children.transpile(state));
         }
+        if (this.customizations.length > 0) {
+            body.push(...this.customizations.map(node => node.transpile(state)));
+        }
         state.blockDepth--;
         body.push(state.indentText, '</', this.tag.text, '>\n');
         return body;