From c4df229db0a5771bc3d7b0ca30820be3982765c5 Mon Sep 17 00:00:00 2001
From: Gene Levitzky <levitzky@google.com>
Date: Thu, 12 Apr 2018 14:54:36 -0400
Subject: [PATCH] Adding TemplateValidator.

---
 ...templates.js => amp-ad-template-helper.js} |   2 +-
 .../amp-a4a/0.1/cryptographic-validator.js    |   3 +-
 extensions/amp-a4a/0.1/template-validator.js  | 106 ++++++++++++++++++
 .../amp-a4a/0.1/test/test-amp-ad-templates.js |  19 ++--
 .../0.1/amp-ad-network-adzerk-impl.js         |  15 +--
 src/utils/array.js                            |  13 +++
 test/functional/utils/test-array.js           |  16 +++
 7 files changed, 155 insertions(+), 19 deletions(-)
 rename extensions/amp-a4a/0.1/{amp-ad-templates.js => amp-ad-template-helper.js} (99%)
 create mode 100644 extensions/amp-a4a/0.1/template-validator.js

diff --git a/extensions/amp-a4a/0.1/amp-ad-templates.js b/extensions/amp-a4a/0.1/amp-ad-template-helper.js
similarity index 99%
rename from extensions/amp-a4a/0.1/amp-ad-templates.js
rename to extensions/amp-a4a/0.1/amp-ad-template-helper.js
index 0be9b3b1449dd..c41847a0c271c 100644
--- a/extensions/amp-a4a/0.1/amp-ad-templates.js
+++ b/extensions/amp-a4a/0.1/amp-ad-template-helper.js
@@ -34,7 +34,7 @@ const TEMPLATE_CORS_CONFIG = {
   credentials: 'omit',
 };
 
-export class AmpAdTemplates {
+export class AmpAdTemplateHelper {
 
   /**
    * @param {!Window} win
diff --git a/extensions/amp-a4a/0.1/cryptographic-validator.js b/extensions/amp-a4a/0.1/cryptographic-validator.js
index 27e2a4ad020e8..22911f7ff96cc 100644
--- a/extensions/amp-a4a/0.1/cryptographic-validator.js
+++ b/extensions/amp-a4a/0.1/cryptographic-validator.js
@@ -21,7 +21,6 @@ import {
 } from './amp-ad-type-defs';
 import {SignatureVerifier, VerificationStatus} from './signature-verifier';
 import {getAmpAdMetadata} from './amp-ad-utils';
-import {getDefaultBootstrapBaseUrl} from '../../../src/3p-frame';
 import {signingServerURLs} from '../../../ads/_a4a-config';
 import {user} from '../../../src/log';
 import {utf8Decode} from '../../../src/utils/bytes';
@@ -29,7 +28,7 @@ import {utf8Decode} from '../../../src/utils/bytes';
 export const SIGNATURE_VERIFIER_PROPERTY_NAME =
     'AMP_FAST_FETCH_SIGNATURE_VERIFIER_';
 
-const TAG = 'amp-ad-render';
+const TAG = 'amp-ad-cryptographic-validator';
 
 export class CryptographicValidator extends Validator {
   /** @param {!Window} win */
diff --git a/extensions/amp-a4a/0.1/template-validator.js b/extensions/amp-a4a/0.1/template-validator.js
new file mode 100644
index 0000000000000..f2fa2b1d46136
--- /dev/null
+++ b/extensions/amp-a4a/0.1/template-validator.js
@@ -0,0 +1,106 @@
+/**
+ * Copyright 2018 The AMP HTML Authors. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS-IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import {AmpAdTemplateHelper} from '../../amp-a4a/0.1/amp-ad-template-helper';
+import {Services} from '../../../src/services';
+import {Validator, ValidatorResult} from './amp-ad-type-defs';
+import {pushIfNotExist} from '../../../src/utils/array';
+import {tryParseJson} from '../../../src/json';
+import {utf8Decode} from '../../../src/utils/bytes';
+
+/** @const {string} */
+export const AMP_TEMPLATED_CREATIVE_HEADER_NAME = 'AMP-template-amp-creative';
+
+/**
+ * Validator for Template ads.
+ */
+export class TemplateValidator extends Validator {
+
+  constructor() {
+    super();
+
+    /** @private {?AmpAdTemplateHelper} */
+    this.ampAdTemplateHelper_ = null;
+  }
+
+  /**
+   * @param {string} templateString
+   * @param {!./amp-ad-type-defs.AmpTemplateCreativeDef} parsedResponseBody
+   * @return {!./amp-ad-type-defs.CreativeMetaDataDef}
+   * @private
+   */
+  getAmpAdMetadata_(templateString, parsedResponseBody) {
+    // TODO(levitzky) The following minification is for demo purposes only. Once
+    // launched this will either be performed server-side, or will be replaced
+    // by more sophisticated logic.
+    const minifiedCreative = templateString.replace(
+        /<script async.+?<\/script>/g, '');
+    const metadata = /** @type {!./amp-ad-type-defs.CreativeMetaDataDef} */ ({
+      minifiedCreative,
+      customElementExtensions: [],
+      extensions: [],
+    });
+    if (parsedResponseBody.analytics) {
+      pushIfNotExist(metadata['customElementExtensions'], 'amp-analytics');
+    }
+    pushIfNotExist(metadata['customElementExtensions'], 'amp-mustache');
+    return metadata;
+  }
+
+  /**
+   * @param {!./amp-ad-type-defs.CreativeMetaDataDef} metadata
+   * @param {!Window} win
+   * @private
+   */
+  processMetadata_(metadata, win) {
+    const extensions = Services.extensionsFor(win);
+    metadata.customElementExtensions.forEach(
+        extensionId => extensions./*OK*/preloadExtension(extensionId));
+    // TODO(levitzky) Add preload logic for fonts / images.
+  }
+
+  /** @override */
+  validate(context, unvalidatedBytes, headers) {
+    const creativeData = {};
+    const body = utf8Decode(/** @type {!ArrayBuffer} */ (unvalidatedBytes));
+    if (!headers ||
+        headers.get(AMP_TEMPLATED_CREATIVE_HEADER_NAME) !== 'amp-mustache') {
+      creativeData['creative'] = body;
+      return Promise.resolve(
+          /** @type {!./amp-ad-type-defs.ValidatorOutput} */ ({
+            creativeData,
+            adResponseType: 'template',
+            type: ValidatorResult.NON_AMP,
+          }));
+    }
+
+    const parsedResponseBody =
+        /** @type {!./amp-ad-type-defs.AmpTemplateCreativeDef} */ (
+        tryParseJson(body) || {});
+    this.ampAdTemplateHelper_ = this.ampAdTemplateHelper_ ||
+        new AmpAdTemplateHelper(context.win);
+    return this.ampAdTemplateHelper_
+        .fetch(parsedResponseBody.templateUrl)
+        .then(template => {
+          const creativeMetadata =
+              this.getAmpAdMetadata_(template, parsedResponseBody);
+          this.processMetadata_(creativeMetadata, context.win);
+          creativeData.templateData = parsedResponseBody;
+          creativeData.creativeMetadata = creativeMetadata;
+          return {creativeData, type: ValidatorResult.AMP};
+        });
+  }
+}
diff --git a/extensions/amp-a4a/0.1/test/test-amp-ad-templates.js b/extensions/amp-a4a/0.1/test/test-amp-ad-templates.js
index 02a4502095cb4..8c0c9e22989b7 100644
--- a/extensions/amp-a4a/0.1/test/test-amp-ad-templates.js
+++ b/extensions/amp-a4a/0.1/test/test-amp-ad-templates.js
@@ -14,12 +14,12 @@
  * limitations under the License.
  */
 
-import {AmpAdTemplates} from '../amp-ad-templates';
+import {AmpAdTemplateHelper} from '../amp-ad-template-helper';
 import {AmpMustache} from '../../../amp-mustache/0.1/amp-mustache';
 import {Xhr} from '../../../../src/service/xhr-impl';
 
 
-describes.fakeWin('amp-ad-templates', {amp: true}, env => {
+describes.fakeWin('AmpAdTemplateHelper', {amp: true}, env => {
 
   const cdnUrl = 'https://adserver-com.cdn.ampproject.org/ad/s/' +
       'adserver.com/amp_template_1';
@@ -27,14 +27,14 @@ describes.fakeWin('amp-ad-templates', {amp: true}, env => {
 
   let win, doc;
   let fetchTextMock;
-  let ampAdTemplates;
+  let ampAdTemplateHelper;
 
   beforeEach(() => {
     win = env.win;
     win.AMP_MODE = {localDev: false};
     doc = win.document;
     fetchTextMock = sandbox.stub(Xhr.prototype, 'fetchText');
-    ampAdTemplates = new AmpAdTemplates(win);
+    ampAdTemplateHelper = new AmpAdTemplateHelper(win);
   });
 
   it('should return a promise resolving to a string template', () => {
@@ -52,16 +52,17 @@ describes.fakeWin('amp-ad-templates', {amp: true}, env => {
               headers: {},
               text: () => template,
             }));
-    return ampAdTemplates.fetch(canonicalUrl)
+    return ampAdTemplateHelper.fetch(canonicalUrl)
         .then(fetchedTemplate => expect(fetchedTemplate).to.equal(template));
   });
 
   it('should use CDN url if one is supplied', () => {
-    expect(ampAdTemplates.getTemplateProxyUrl_(cdnUrl)).to.equal(cdnUrl);
+    expect(ampAdTemplateHelper.getTemplateProxyUrl_(cdnUrl)).to.equal(cdnUrl);
   });
 
   it('should convert canonical to CDN', () => {
-    expect(ampAdTemplates.getTemplateProxyUrl_(canonicalUrl)).to.equal(cdnUrl);
+    expect(ampAdTemplateHelper.getTemplateProxyUrl_(canonicalUrl))
+        .to.equal(cdnUrl);
   });
 
   it('should render a template with correct values', () => {
@@ -74,7 +75,7 @@ describes.fakeWin('amp-ad-templates', {amp: true}, env => {
     parentDiv./*OK*/innerHTML =
         '<template type="amp-mustache"><p>{{foo}}</p></template>';
     doc.body.appendChild(parentDiv);
-    return ampAdTemplates.render({foo: 'bar'}, parentDiv).then(result => {
+    return ampAdTemplateHelper.render({foo: 'bar'}, parentDiv).then(result => {
       expect(result).to.not.be.null;
       expect(result./*OK*/innerHTML).to.equal('bar');
     });
@@ -93,7 +94,7 @@ describes.fakeWin('amp-ad-templates', {amp: true}, env => {
     }, {
       'type': 'googleanalytics',
     }];
-    ampAdTemplates.insertAnalytics(parentDiv, analytics);
+    ampAdTemplateHelper.insertAnalytics(parentDiv, analytics);
     expect(parentDiv.childNodes.length).to.equal(3);
     expect(parentDiv.innerHTML).to.equal('<p>123</p>' +
         '<amp-analytics config="remoteUrl">' +
diff --git a/extensions/amp-ad-network-adzerk-impl/0.1/amp-ad-network-adzerk-impl.js b/extensions/amp-ad-network-adzerk-impl/0.1/amp-ad-network-adzerk-impl.js
index 8cf9b2b472cf8..58170f929f3b0 100644
--- a/extensions/amp-ad-network-adzerk-impl/0.1/amp-ad-network-adzerk-impl.js
+++ b/extensions/amp-ad-network-adzerk-impl/0.1/amp-ad-network-adzerk-impl.js
@@ -19,7 +19,7 @@ import {
   CreativeMetaDataDef,
   NO_CONTENT_RESPONSE,
 } from '../../amp-a4a/0.1/amp-a4a';
-import {AmpAdTemplates} from '../../amp-a4a/0.1/amp-ad-templates';
+import {AmpAdTemplateHelper} from '../../amp-a4a/0.1/amp-ad-template-helper';
 import {dev} from '../../../src/log';
 import {getMode} from '../../../src/mode';
 import {tryParseJson} from '../../../src/json';
@@ -37,8 +37,8 @@ export const AMP_TEMPLATED_CREATIVE_HEADER_NAME = 'AMP-template-amp-creative';
     }} */
 let AmpTemplateCreativeDef;
 
-/** @private {?AmpAdTemplates} */
-let ampAdTemplates;
+/** @private {?AmpAdTemplateHelper} */
+let ampAdTemplateHelper;
 
 /**
  * Fast Fetch implementation for AdZerk network that allows AMP creative
@@ -72,7 +72,8 @@ export class AmpAdNetworkAdzerkImpl extends AmpA4A {
     /** @private {?AmpTemplateCreativeDef} */
     this.ampCreativeJson_ = null;
 
-    ampAdTemplates = ampAdTemplates || new AmpAdTemplates(this.win);
+    ampAdTemplateHelper = ampAdTemplateHelper ||
+        new AmpAdTemplateHelper(this.win);
   }
 
   /**
@@ -114,7 +115,7 @@ export class AmpAdNetworkAdzerkImpl extends AmpA4A {
       this.ampCreativeJson_ = /** @type {!AmpTemplateCreativeDef} */
         (tryParseJson(body) || {});
       // TODO(keithwrightbos): macro value validation?  E.g. http invalid?
-      return ampAdTemplates
+      return ampAdTemplateHelper
           .fetch(this.ampCreativeJson_.templateUrl)
           .then(parsedTemplate => {
             return utf8Encode(this.parseMetadataFromCreative(parsedTemplate));
@@ -169,12 +170,12 @@ export class AmpAdNetworkAdzerkImpl extends AmpA4A {
   /** @override */
   onCreativeRender(unusedMetadata) {
     if (this.ampCreativeJson_ && this.ampCreativeJson_.data) {
-      ampAdTemplates.render(
+      ampAdTemplateHelper.render(
           this.ampCreativeJson_.data,
           this.iframe.contentWindow.document.body)
           .then(renderedElement => {
             if (this.ampCreativeJson_.analytics) {
-              ampAdTemplates.insertAnalytics(
+              ampAdTemplateHelper.insertAnalytics(
                   renderedElement, this.ampCreativeJson_.analytics);
             }
             this.iframe.contentWindow.document.body./*OK*/innerHTML =
diff --git a/src/utils/array.js b/src/utils/array.js
index c7070ba4b3b8f..14820c11ea3f2 100644
--- a/src/utils/array.js
+++ b/src/utils/array.js
@@ -77,3 +77,16 @@ export function fromIterator(iterator) {
   }
   return array;
 }
+
+/**
+ * Adds item to array if it is not already present.
+ *
+ * @param {Array<T>} array
+ * @param {T} item
+ * @template T
+ */
+export function pushIfNotExist(array, item) {
+  if (array.indexOf(item) < 0) {
+    array.push(item);
+  }
+}
diff --git a/test/functional/utils/test-array.js b/test/functional/utils/test-array.js
index 386bc776118e9..281dee4c484c7 100644
--- a/test/functional/utils/test-array.js
+++ b/test/functional/utils/test-array.js
@@ -18,6 +18,7 @@ import {
   filterSplice,
   findIndex,
   fromIterator,
+  pushIfNotExist,
 } from '../../../src/utils/array';
 
 describe('filterSplice', function() {
@@ -98,3 +99,18 @@ describe('fromIterator', function() {
     expect(fromIterator(iterator)).to.deep.equal([0, 2, 4]);
   });
 });
+
+describe('pushIfNotExist', () => {
+  it('should push element', () => {
+    const array = [1, 2, 3, 4];
+    pushIfNotExist(array, 5);
+    expect(array).to.deep.equal([1, 2, 3, 4, 5]);
+    pushIfNotExist(array, 2.3);
+    expect(array).to.deep.equal([1, 2, 3, 4, 5, 2.3]);
+  });
+  it('should not push element', () => {
+    const array = [1, 2, 3, 4];
+    pushIfNotExist(array, 1);
+    expect(array).to.deep.equal([1, 2, 3, 4]);
+  });
+});