add assertShape to validate templateElement

packages/babel-types/scripts/utils/stringifyValidator.js

@@ -31,6 +31,19 @@ module.exports = function stringifyValidator(validator, nodePrefix) {
     return validator.type;
   }
 
+  if (validator.shapeOf) {
+    return (
+      "{" +
+      Object.keys(validator.shapeOf)
+        .map(
+          shapeKey =>
+            shapeKey + ":" + stringifyValidator(validator.shapeOf[shapeKey])
+        )
+        .join(", ") +
+      "}"
+    );
+  }
+
   return ["any"];
 };
 

packages/babel-types/src/definitions/es2015.js

@@ -1,5 +1,6 @@
 // @flow
 import defineType, {
+  assertShape,
   assertNodeType,
   assertValueType,
   chain,
@@ -537,7 +538,10 @@ defineType("TemplateElement", {
   builder: ["value", "tail"],
   fields: {
     value: {
-      // todo: flatten `raw` into main node
+      validate: assertShape({
+        raw: assertValueType("string"),
+        cooked: assertValueType("string"),
+      }),
     },
     tail: {
       validate: assertValueType("boolean"),

packages/babel-types/src/definitions/utils.js

@@ -161,6 +161,35 @@ export function assertValueType(type: string): Validator {
   return validate;
 }
 
+export function assertShape(shape: { [string]: Validator }): Validator {
+  function validate(node, key, val) {
+    const errors = [];
+    for (const property of Object.keys(shape)) {
+      try {
+        const validator = shape[property];
+        validator(node, property, val[property]);
+      } catch (error) {
+        if (error instanceof TypeError) {
+          errors.push(error.message);
+          continue;
+        }
+        throw error;
+      }
+    }
+    if (errors.length) {
+      throw new TypeError(
+        `Property ${key} of ${
+          node.type
+        } expected to have the following:\n${errors.join("\n")}`,
+      );
+    }
+  }
+
+  validate.shapeOf = shape;
+
+  return validate;
+}
+
 export function chain(...fns: Array<Validator>): Validator {
   function validate(...args) {
     for (const fn of fns) {