compiler: accept partials (#36418)

* compiler: accept partials instead of whole doc

* Extract compile() to make it testable.

* forgot compile.js file

package.json

@@ -16,7 +16,7 @@
   },
   "dependencies": {
     "@ampproject/animations": "0.2.2",
-    "@ampproject/bento-compiler": "0.0.10",
+    "@ampproject/bento-compiler": "0.0.11",
     "@ampproject/toolbox-cache-url": "2.8.0",
     "@ampproject/viewer-messaging": "1.1.2",
     "@ampproject/worker-dom": "0.32.1",

src/compiler/compile.js

@@ -0,0 +1,28 @@
+import * as compiler from '@ampproject/bento-compiler';
+
+import {getBuilders} from './builders';
+
+/**
+ * Returns the AST for an AMP Document with eligible components server-rendered.
+ *
+ * @param {import('./types').CompilerRequest} request
+ * @return {import('./types').CompilerResponse}
+ */
+export function compile(request) {
+  if (!request || (!request.document && !request.nodes)) {
+    throw new Error('Must provide either document or nodes');
+  }
+
+  // TODO(samouri): remove the defaults.
+  const versions = request.versions ?? [
+    {component: 'amp-layout', version: 'v0'},
+  ];
+
+  const {document, nodes} = request;
+  if (document) {
+    return {
+      document: compiler.renderAstDocument(document, getBuilders(versions)),
+    };
+  }
+  return {nodes: compiler.renderAstNodes(nodes, getBuilders(versions))};
+}

src/compiler/index.js

@@ -1,26 +1,4 @@
 import './polyfills';
-import * as compiler from '@ampproject/bento-compiler';
-
-import {getBuilders} from './builders';
-
-/**
- * Returns the AST for an AMP Document with eligible components server-rendered.
- *
- * @param {import('./types').CompilerRequest} request
- * @return {import('./types').CompilerResponse}
- */
-function compile(request) {
-  // TODO(samouri): remove the defaults.
-  const document = request.document ?? {
-    root: 0,
-    tree: [{tagid: 92, children: []}],
-    'quirks_mode': false,
-  };
-  const versions = request.versions ?? [
-    {component: 'amp-layout', version: 'v0'},
-  ];
-
-  return {document: compiler.renderAst(document, getBuilders(versions))};
-}
+import {compile} from './compile';
 
 globalThis['compile'] = compile;

src/compiler/types.d.ts

@@ -1,4 +1,4 @@
-import {TreeProto} from '@ampproject/bento-compiler';
+import {TreeProto, NodeProto} from '@ampproject/bento-compiler';
 
 /**
  * AMP Components must implement this "buildDom" function in order to be server-rendered.
@@ -18,8 +18,12 @@ export type BuildDom = (Element) => void;
  */
 export type Versions = Array<{component: string; version: string}>;
 
-export type CompilerRequest = {document: TreeProto; versions: Versions};
+export type CompilerRequest = {
+  document?: TreeProto;
+  nodes?: NodeProto[];
+  versions: Versions;
+};
 
-export type CompilerResponse = {document: TreeProto};
+export type CompilerResponse = {document: TreeProto} | {nodes: NodeProto[]};
 
 export type BuilderMap = {[tagName: string]: BuildDom};

test/unit/compiler/test-compile.js

@@ -0,0 +1,29 @@
+import {compile} from '#compiler/compile';
+
+describes.sandboxed('compile', {}, () => {
+  it('should throw if provided invalid input', () => {
+    const errorMsg = /Must provide either document or nodes/;
+
+    expect(() => compile()).throw(errorMsg);
+    expect(() => compile({})).throw(errorMsg);
+    expect(() => compile({unknown: {}})).throw(errorMsg);
+    expect(() => compile({versions: []})).throw(errorMsg);
+  });
+
+  it('should return compiled document', () => {
+    const document = {
+      root: 0,
+      tree: [{tagid: 92, children: []}],
+      'quirks_mode': false,
+    };
+    expect(compile({document})).to.deep.equal({document});
+  });
+
+  it('should return compiled nodes', () => {
+    const nodes = [
+      {tagid: 1, value: 'a', children: [], attributes: []},
+      {tagid: 7, value: 'b', children: [], attributes: []},
+    ];
+    expect(compile({nodes})).to.deep.equal({nodes});
+  });
+});