@marko/language-tools 2.5.5 → 2.5.7

package/dist/extractors/script/index.d.ts

@@ -1,6 +1,7 @@
 import type { TaglibLookup } from "@marko/babel-utils";
 import type TS from "typescript/lib/tsserverlibrary";
 import { type Parsed } from "../../parser";
+import type { Meta } from "../../util/project";
 /**
  * Iterate over the Marko CST and extract all the script content.
  */
@@ -14,5 +15,6 @@ export interface ExtractScriptOptions {
     lookup: TaglibLookup;
     scriptLang: ScriptLang;
     runtimeTypesCode?: string;
+    translator: Meta["config"]["translator"];
 }
 export declare function extractScript(opts: ExtractScriptOptions): import("../../util/extractor").Extracted;

package/dist/extractors/script/util/get-runtime-api.d.ts

@@ -0,0 +1,9 @@
+import { Parsed } from "../../../parser";
+export type RuntimeAPI = (typeof RuntimeAPI)[keyof typeof RuntimeAPI] | void;
+export declare const RuntimeAPI: {
+    readonly tags: "tags";
+    readonly class: "class";
+};
+export declare function getRuntimeAPI(translator: {
+    preferAPI?: string;
+} | undefined, parsed: Parsed): RuntimeAPI;

package/dist/index.js

@@ -42,6 +42,7 @@ __export(src_exports, {
   getLines: () => import_htmljs_parser2.getLines,
   getLocation: () => import_htmljs_parser2.getLocation,
   getPosition: () => import_htmljs_parser2.getPosition,
+  isControlFlowTag: () => isControlFlowTag,
   isDefinitionFile: () => isDefinitionFile,
   parse: () => parse
 });
@@ -1653,6 +1654,120 @@ function tryReaddirSync(dir) {
   }
 }
 
+// src/extractors/script/util/get-runtime-api.ts
+var RuntimeAPI = {
+  tags: "tags",
+  class: "class"
+};
+function getRuntimeAPI(translator, parsed) {
+  return detectAPIFromTranslator(translator) || detectAPIFromFileName(parsed.filename) || detectAPIFromProgram(parsed, parsed.program);
+}
+function detectAPIFromTranslator(translator) {
+  switch (translator == null ? void 0 : translator.preferAPI) {
+    case "class":
+      return RuntimeAPI.class;
+    case "tags":
+      return RuntimeAPI.tags;
+  }
+}
+function detectAPIFromFileName(filename) {
+  for (let end = filename.length, i = end; --i; ) {
+    switch (filename[i]) {
+      case "/":
+      case "\\":
+        if (filename.startsWith("tags", i + 1)) {
+          return RuntimeAPI.tags;
+        } else if (filename.startsWith("components", i + 1)) {
+          return;
+        }
+        end = i;
+        break;
+    }
+  }
+}
+function detectAPIFromProgram(parsed, program) {
+  if (program.comments) {
+    switch (parsed.read(program.comments[0].value).trim()) {
+      case "use tags":
+        return RuntimeAPI.tags;
+      case "use class":
+        return RuntimeAPI.class;
+    }
+  }
+  if (program.static) {
+    for (const stmt of program.static) {
+      switch (stmt.type) {
+        case 26 /* Class */:
+        case 27 /* Style */:
+          return RuntimeAPI.class;
+      }
+    }
+  }
+  return detectAPIFromBody(parsed, program.body);
+}
+function detectAPIFromBody(parsed, body) {
+  if (body) {
+    for (const child of body) {
+      const api = detectAPIFromChild(parsed, child);
+      if (api) return api;
+    }
+  }
+}
+function detectAPIFromChild(parsed, child) {
+  switch (child.type) {
+    case 23 /* Scriptlet */:
+      return RuntimeAPI.class;
+    case 1 /* Tag */:
+    case 16 /* AttrTag */:
+      return detectAPIFromTag(parsed, child);
+  }
+}
+function detectAPIFromTag(parsed, tag) {
+  var _a;
+  switch (tag.nameText) {
+    case "macro":
+    case "include-text":
+    case "include-html":
+    case "init-components":
+    case "await-reorderer":
+    case "while":
+    case "module-code":
+      return RuntimeAPI.class;
+    case "const":
+    case "debug":
+    case "define":
+    case "html-script":
+    case "html-style":
+    case "id":
+    case "let":
+    case "lifecycle":
+    case "log":
+    case "return":
+      return RuntimeAPI.tags;
+  }
+  if (tag.var) {
+    return RuntimeAPI.tags;
+  }
+  if (tag.attrs) {
+    for (const attr of tag.attrs) {
+      if (attr.type !== 15 /* AttrSpread */) {
+        if (((_a = attr.value) == null ? void 0 : _a.type) === 13 /* AttrValue */ && attr.value.bound) {
+          return RuntimeAPI.tags;
+        }
+        if (attr.args) {
+          return RuntimeAPI.class;
+        }
+        if (/^(?:key|no-update(?:-body)?(?:-if)?)$|:(scoped:no-update)$/.test(
+          parsed.read(attr.name)
+        )) {
+          return RuntimeAPI.class;
+        }
+      }
+    }
+  }
+  return detectAPIFromBody(parsed, tag.body);
+}
+
 // src/extractors/script/util/jsdoc-input-type.ts
 var MaybeInputTypedefReg = /@typedef\b[\s\S]*\bInput\b/;
 function getJSDocInputType(comment, ts) {
@@ -1758,7 +1873,7 @@ function getRuntimeOverrides(runtimeTypes, generics, applyGenerics) {
 }
 
 // src/extractors/script/util/script-parser.ts
-var import_parser6 = require("@babel/parser");
+var import_parser7 = require("@babel/parser");
 var plugins = [
   "exportDefaultFrom",
   "importAssertions",
@@ -1772,7 +1887,7 @@ var ScriptParser = class {
   statementAt(startIndex, src) {
     const pos = this.#parsed.positionAt(startIndex);
     try {
-      return (0, import_parser6.parse)(src, {
+      return (0, import_parser7.parse)(src, {
         plugins,
         startIndex,
         startLine: pos.line + 1,
@@ -1793,7 +1908,7 @@ var ScriptParser = class {
   expressionAt(startIndex, src) {
     const pos = this.#parsed.positionAt(startIndex);
     try {
-      return (0, import_parser6.parseExpression)(src, {
+      return (0, import_parser7.parseExpression)(src, {
         plugins,
         startIndex,
         startLine: pos.line + 1,
@@ -1829,7 +1944,7 @@ var REG_TAG_IMPORT = /(?<=(['"]))<([^'">]+)>(?=\1)/;
 var REG_INPUT_TYPE = /\s*(interface|type)\s+Input\b/y;
 var REG_OBJECT_PROPERTY = /^[_$a-z][_$a-z0-9]*$/i;
 var REG_COMMENT_PRAGMA = /\/\/(?:\s*@ts-|\/\s*<)/y;
-var REG_TAG_NAME_IDENTIFIER = /^[A-Z][a-zA-Z_$]+$/;
+var REG_TAG_NAME_IDENTIFIER = /^[A-Z][a-zA-Z0-9_$]+$/;
 var IF_TAG_ALTERNATES = /* @__PURE__ */ new WeakMap();
 var WROTE_COMMENT = /* @__PURE__ */ new WeakSet();
 var START_OF_FILE = { start: 0, end: 0 };
@@ -1849,11 +1964,14 @@ var ScriptExtractor = class {
   #scriptParser;
   #read;
   #lookup;
+  #tagIds = /* @__PURE__ */ new Map();
   #renderIds = /* @__PURE__ */ new Map();
   #scriptLang;
   #ts;
   #runtimeTypes;
+  #api;
   #mutationOffsets;
+  #tagId = 1;
   #renderId = 1;
   constructor(opts) {
     const { parsed, lookup, scriptLang } = opts;
@@ -1867,6 +1985,7 @@ var ScriptExtractor = class {
     this.#extractor = new Extractor(parsed);
     this.#scriptParser = new ScriptParser(parsed);
     this.#read = parsed.read.bind(parsed);
+    this.#api = getRuntimeAPI(opts.translator, parsed);
     this.#mutationOffsets = crawlProgramScope(this.#parsed, this.#scriptParser);
     this.#writeProgram(parsed.program);
   }
@@ -2025,8 +2144,8 @@ function ${templateName}() {
   ${varShared("noop")}({ input, component, state, out, $global, $signal });
 `);
     const body = this.#processBody(program);
-    if (body == null ? void 0 : body.renderBody) {
-      this.#writeChildren(program, body.renderBody);
+    if (body == null ? void 0 : body.content) {
+      this.#writeChildren(program, body.content);
     }
     const hoists = getHoists(program);
     if (hoists) {
@@ -2057,6 +2176,7 @@ function ${templateName}() {
       "ReturnWithScope"
     )}<${internalInput}, ${didReturn ? `typeof ${templateName + typeArgsStr} extends () => infer Return ? Return : never` : "void"}>)`;
     const templateOverrideClass = `${templateBaseClass}<{${this.#runtimeTypes ? getRuntimeOverrides(this.#runtimeTypes, typeParamsStr, typeArgsStr) : ""}
+  ${this.#api ? `api: "${this.#api}",` : ""}
   _${typeParamsStr ? `<${internalApply} = 1>(): ${internalApply} extends 0
     ? ${typeParamsStr}() => <${internalInputWithExtends}>${renderAndReturn}
     : () => <${internalInputWithExtends}, ${typeParamsStr.slice(
@@ -2163,9 +2283,9 @@ constructor(_?: Return) {}
                 this.#getRangeWithoutTrailingComma((_a = child.args) == null ? void 0 : _a.value) || this.#getAttrValue(child, ATTR_UNAMED2) || "undefined"
               ).write(") {\n");
               const ifBody = this.#processBody(child);
-              if (ifBody == null ? void 0 : ifBody.renderBody) {
+              if (ifBody == null ? void 0 : ifBody.content) {
                 const localBindings = getHoistSources(child);
-                this.#writeChildren(child, ifBody.renderBody, true);
+                this.#writeChildren(child, ifBody.content, true);
                 if (localBindings) {
                   this.#extractor.write("return {\nscope:");
                   this.#writeObjectKeys(localBindings);
@@ -2185,9 +2305,9 @@ constructor(_?: Return) {}
                     this.#extractor.write("\n} else if (undefined) {\n");
                   }
                   const alternateBody = this.#processBody(node);
-                  if (alternateBody == null ? void 0 : alternateBody.renderBody) {
+                  if (alternateBody == null ? void 0 : alternateBody.content) {
                     const localBindings = getHoistSources(node);
-                    this.#writeChildren(node, alternateBody.renderBody, true);
+                    this.#writeChildren(node, alternateBody.content, true);
                     if (localBindings) {
                       this.#extractor.write("return {\nscope:");
                       this.#writeObjectKeys(localBindings);
@@ -2228,12 +2348,12 @@ constructor(_?: Return) {}
               }
               this.#extractor.write("\n) => {\n");
               const body = this.#processBody(child);
-              if (body == null ? void 0 : body.renderBody) {
-                this.#writeChildren(child, body.renderBody);
+              if (body == null ? void 0 : body.content) {
+                this.#writeChildren(child, body.content);
               }
               this.#writeReturn(
                 void 0,
-                (body == null ? void 0 : body.renderBody) ? getHoistSources(child) : void 0
+                (body == null ? void 0 : body.content) ? getHoistSources(child) : void 0
               );
               if (renderId) {
                 this.#extractor.write("\n}));\n");
@@ -2248,8 +2368,8 @@ constructor(_?: Return) {}
                 this.#getRangeWithoutTrailingComma((_b = child.args) == null ? void 0 : _b.value) || "undefined"
               ).write("\n) {\n");
               const body = this.#processBody(child);
-              if (body == null ? void 0 : body.renderBody) {
-                this.#writeChildren(child, body.renderBody);
+              if (body == null ? void 0 : body.content) {
+                this.#writeChildren(child, body.content);
               }
               this.#extractor.write("\n}\n");
               break;
@@ -2309,51 +2429,53 @@ constructor(_?: Return) {}
   #writeTag(tag) {
     const tagName = tag.nameText;
     const renderId = this.#getRenderId(tag);
-    let nestedTagType;
+    const def = tagName ? this.#lookup.getTag(tagName) : void 0;
     if (renderId) {
       this.#extractor.write(
-        `${varShared("assertRendered")}(${varShared(
-          "rendered"
-        )}, ${renderId}, `
+        `${varShared("assertRendered")}(${varShared("rendered")},${renderId},`
       );
     }
-    if (tagName) {
-      const def = this.#lookup.getTag(tagName);
-      if (def) {
-        const importPath = resolveTagImport(this.#filename, def);
-        const isMarkoFile = importPath == null ? void 0 : importPath.endsWith(".marko");
-        if (isMarkoFile) {
-          nestedTagType = `import("${importPath}").Input`;
-        }
-        const renderer = (importPath == null ? void 0 : importPath.endsWith(".marko")) ? `renderTemplate(import("${importPath}"))` : def.html ? `renderNativeTag("${def.name}")` : "missingTag";
-        if (!def.html && REG_TAG_NAME_IDENTIFIER.test(tagName)) {
+    if (def == null ? void 0 : def.html) {
+      this.#extractor.write(`${varShared("renderNativeTag")}("`).copy(tag.name).write('")');
+    } else {
+      const importPath = def && resolveTagImport(this.#filename, def);
+      if (def && !importPath) {
+        this.#extractor.write(varShared("missingTag"));
+      } else {
+        const tagId = this.#ensureTagId(tag);
+        let isDynamic = true;
+        this.#extractor.write(
+          `(${varShared("assertTag")}(${varShared("tags")},${tagId},(
+`
+        );
+        if (tagName && REG_TAG_NAME_IDENTIFIER.test(tagName)) {
+          if (importPath) {
+            this.#extractor.write(
+              `${varShared("fallbackTemplate")}(${tagName},import("${importPath}"))`
+            );
+          } else {
+            this.#extractor.copy(tag.name);
+          }
+        } else if (importPath) {
+          isDynamic = !importPath.endsWith(".marko");
           this.#extractor.write(
-            `${varShared("renderPreferLocal")}(
-// @ts-expect-error We expect the compiler to error because we are checking if the tag is defined.
-(${varShared("error")}, `
-          ).copy(tag.name).write(`),
-${varShared(renderer)})`);
+            `${varShared("resolveTemplate")}(import("${importPath}"))`
+          );
         } else {
-          this.#extractor.write(varShared(renderer));
+          this.#writeDynamicTagName(tag);
         }
-      } else if (REG_TAG_NAME_IDENTIFIER.test(tagName)) {
-        nestedTagType = `Marko.Input<typeof ${this.#read(tag.name)}>`;
-        this.#extractor.write(`${varShared("renderDynamicTag")}(
-`).copy(tag.name).write("\n)");
-      } else {
-        this.#extractor.write(`${varShared("missingTag")}`);
+        this.#extractor.write(
+          `
+)),${varShared(isDynamic ? "renderDynamicTag" : "renderTemplate")}(${varShared("tags")}[${tagId}]))`
+        );
       }
-    } else {
-      this.#extractor.write(`${varShared("renderDynamicTag")}(`);
-      this.#writeDynamicTagName(tag);
-      this.#extractor.write(")");
     }
     if (tag.typeArgs) {
       this.#extractor.write(`<0>()`).copy(tag.typeArgs).write("()(");
     } else {
       this.#extractor.write("()()(");
     }
-    this.#writeTagInputObject(tag, nestedTagType);
+    this.#writeTagInputObject(tag);
     if (renderId) {
       this.#extractor.write(`)`);
     }
@@ -2537,7 +2659,7 @@ ${isMutatedVar(tag.parent, valueLiteral) ? `${varLocal("return")}.mutate.` : ""}
     );
     return hasAttrs;
   }
-  #writeAttrTags({ staticAttrTags, dynamicAttrTagParents }, inMerge, nestedTagType) {
+  #writeAttrTags({ staticAttrTags, dynamicAttrTagParents }, inMerge) {
     let wasMerge = false;
     if (dynamicAttrTagParents) {
       if (staticAttrTags) {
@@ -2553,7 +2675,7 @@ ${isMutatedVar(tag.parent, valueLiteral) ? `${varLocal("return")}.mutate.` : ""}
       }
     }
     if (staticAttrTags) {
-      this.#writeStaticAttrTags(staticAttrTags, inMerge, nestedTagType);
+      this.#writeStaticAttrTags(staticAttrTags, inMerge);
       if (dynamicAttrTagParents)
         this.#extractor.write(`}${SEP_COMMA_NEW_LINE}`);
     }
@@ -2562,7 +2684,7 @@ ${isMutatedVar(tag.parent, valueLiteral) ? `${varLocal("return")}.mutate.` : ""}
       if (wasMerge) this.#extractor.write(`)${SEP_COMMA_NEW_LINE}`);
     }
   }
-  #writeStaticAttrTags(staticAttrTags, wasMerge, nestedTagType) {
+  #writeStaticAttrTags(staticAttrTags, wasMerge) {
     if (!wasMerge) this.#extractor.write("...{");
     this.#extractor.write(
       `[${varShared("never")}](){
@@ -2583,36 +2705,40 @@ const attrTags = ${varShared(
     this.#extractor.write(SEP_COMMA_NEW_LINE);
     for (const nameText in staticAttrTags) {
       const attrTag = staticAttrTags[nameText];
-      const attrTagDef = this.#lookup.getTag(nameText);
       const isRepeated = attrTag.length > 1;
       const [firstAttrTag] = attrTag;
-      const name = (attrTagDef == null ? void 0 : attrTagDef.targetProperty) || nameText.slice(nameText.lastIndexOf(":") + 1);
+      const name = this.#getAttrTagName(firstAttrTag);
       this.#extractor.write(`["${name}"`);
       this.#writeTagNameComment(firstAttrTag);
       this.#extractor.write("]: ");
       if (isRepeated) {
-        this.#extractor.write(`${varShared("repeatedAttrTag")}(
-...
-`);
-        if (nestedTagType && this.#scriptLang === "js" /* js */) {
+        const tagId = this.#tagIds.get(firstAttrTag.owner);
+        if (tagId) {
+          let accessor = `["${name}"]`;
+          let curTag = firstAttrTag.parent;
+          while (curTag) {
+            if (curTag.type === 16 /* AttrTag */) {
+              accessor = `["${this.#getAttrTagName(curTag)}"]${accessor}`;
+            } else if (!isControlFlowTag(curTag)) {
+              break;
+            }
+            curTag = curTag.parent;
+          }
           this.#extractor.write(
-            `/** @satisfies {${nestedTagType}["${name}"][]} */
+            `${varShared("attrTags")}(${varShared("input")}(${varShared("tags")}[${tagId}])${accessor})([
 `
           );
-        }
-        this.#extractor.write(`([
+        } else {
+          this.#extractor.write(`${varShared("attrTags")}()([
 `);
+        }
       }
       for (const childNode of attrTag) {
         this.#writeTagInputObject(childNode);
         this.#extractor.write(SEP_COMMA_NEW_LINE);
       }
       if (isRepeated) {
-        this.#extractor.write("])");
-        if (nestedTagType && this.#scriptLang === "ts" /* ts */) {
-          this.#extractor.write(` satisfies ${nestedTagType}["${name}"][]`);
-        }
-        this.#extractor.write(`)${SEP_COMMA_NEW_LINE}`);
+        this.#extractor.write(`])${SEP_COMMA_NEW_LINE}`);
       }
     }
   }
@@ -2675,7 +2801,7 @@ const attrTags = ${varShared(
       this.#extractor.write(SEP_COMMA_NEW_LINE);
     }
   }
-  #writeTagInputObject(tag, nestedTagType) {
+  #writeTagInputObject(tag) {
     if (!tag.params) this.#writeComments(tag);
     let hasInput = false;
     this.#extractor.write("{\n");
@@ -2693,7 +2819,7 @@ const attrTags = ${varShared(
       hasInput = true;
     }
     const isScript = isTextOnlyScript(tag);
-    let hasRenderBody = false;
+    let hasBodyContent = false;
     let body;
     if (isScript) {
       this.#extractor.write("async value(){");
@@ -2706,14 +2832,35 @@ const attrTags = ${varShared(
       body = this.#processBody(tag);
       if (body) {
         hasInput = true;
-        this.#writeAttrTags(body, false, nestedTagType);
-        hasRenderBody = body.renderBody !== void 0;
+        this.#writeAttrTags(body, false);
+        hasBodyContent = body.content !== void 0;
       } else if (tag.close) {
-        hasRenderBody = true;
+        hasBodyContent = true;
       }
     }
-    if (tag.params || hasRenderBody) {
-      this.#extractor.write('["renderBody"');
+    if (tag.params || hasBodyContent) {
+      this.#extractor.write("[");
+      switch (this.#api) {
+        case RuntimeAPI.tags:
+          this.#extractor.write('"content"');
+          break;
+        case RuntimeAPI.class:
+          this.#extractor.write('"renderBody"');
+          break;
+        default: {
+          const tagId = this.#tagIds.get(
+            tag.type === 16 /* AttrTag */ ? tag.owner : tag
+          );
+          if (tagId) {
+            this.#extractor.write(
+              `${varShared("contentFor")}(${varShared("tags")}[${tagId}])`
+            );
+          } else {
+            this.#extractor.write(varShared("content"));
+          }
+          break;
+        }
+      }
       this.#writeTagNameComment(tag);
       this.#extractor.write("]: ");
       if (tag.params) {
@@ -2727,8 +2874,8 @@ const attrTags = ${varShared(
 `);
       }
       let didReturn = false;
-      if (body == null ? void 0 : body.renderBody) {
-        didReturn = this.#writeChildren(tag, body.renderBody);
+      if (body == null ? void 0 : body.content) {
+        didReturn = this.#writeChildren(tag, body.content);
       }
       if (!tag.params) {
         this.#extractor.write(`return () => {
@@ -2817,7 +2964,7 @@ const attrTags = ${varShared(
     const { body } = parent;
     if (!body) return;
     const last = body.length - 1;
-    let renderBody;
+    let content;
     let staticAttrTags;
     let dynamicAttrTagParents;
     let i = 0;
@@ -2906,48 +3053,48 @@ const attrTags = ${varShared(
             } else {
               dynamicAttrTagParents = [child];
             }
-          } else if (renderBody) {
-            renderBody.push(child);
+          } else if (content) {
+            content.push(child);
           } else {
-            renderBody = [child];
+            content = [child];
           }
           break;
         }
         case 17 /* Text */: {
           if (!this.#isEmptyText(child)) {
-            if (renderBody) {
-              renderBody.push(child);
+            if (content) {
+              content.push(child);
             } else {
-              renderBody = [child];
+              content = [child];
             }
           }
           break;
         }
         default:
-          if (renderBody) {
-            renderBody.push(child);
+          if (content) {
+            content.push(child);
           } else {
-            renderBody = [child];
+            content = [child];
           }
           break;
       }
     }
-    if (renderBody || staticAttrTags || dynamicAttrTagParents) {
-      return { renderBody, staticAttrTags, dynamicAttrTagParents };
+    if (content || staticAttrTags || dynamicAttrTagParents) {
+      return { content, staticAttrTags, dynamicAttrTagParents };
     }
   }
   #writeDynamicAttrTagBody(tag) {
     const body = this.#processBody(tag);
     if (body) {
-      if (body.renderBody) {
+      if (body.content) {
         this.#extractor.write("(() => {\n");
-        this.#writeChildren(tag, body.renderBody);
+        this.#writeChildren(tag, body.content);
         this.#extractor.write("return ");
       }
       this.#extractor.write("{\n");
       this.#writeAttrTags(body, true);
       this.#extractor.write("}");
-      if (body.renderBody) {
+      if (body.content) {
         this.#extractor.write(";\n})()");
       }
     } else {
@@ -3050,6 +3197,14 @@ const attrTags = ${varShared(
     }
     return renderId;
   }
+  #ensureTagId(tag) {
+    let tagId = this.#tagIds.get(tag);
+    if (!tagId) {
+      tagId = this.#tagId++;
+      this.#tagIds.set(tag, tagId);
+    }
+    return tagId;
+  }
   #getNamedAttrModifierIndex(attr) {
     const start = attr.name.start + 1;
     const end = attr.name.end - 1;
@@ -3058,6 +3213,11 @@ const attrTags = ${varShared(
     }
     return false;
   }
+  #getAttrTagName(tag) {
+    var _a;
+    const { nameText } = tag;
+    return ((_a = this.#lookup.getTag(nameText)) == null ? void 0 : _a.targetProperty) || nameText.slice(nameText.lastIndexOf(":") + 1);
+  }
   #testAtIndex(reg, index) {
     reg.lastIndex = index;
     return reg.test(this.#code);
@@ -3547,6 +3707,7 @@ var marko_default = {
         return extractScript({
           ts,
           parsed,
+          translator: getConfig(dir).translator,
           lookup: getTagLookup(dir),
           scriptLang: getScriptLang(
             fileName,
@@ -3697,6 +3858,7 @@ function isDefinitionFile(fileName) {
   getLines,
   getLocation,
   getPosition,
+  isControlFlowTag,
   isDefinitionFile,
   parse
 });