Add co-location support to CoffeeScript component class files (#373)

Add co-location support to CoffeeScript component class files

Author: Robert Jackson

lib/colocated-broccoli-plugin.js

@@ -82,6 +82,9 @@ module.exports = class ColocatedTemplateProcessor extends Plugin {
       } else if (fs.existsSync(path.join(this.inputPaths[0], backingClassPath + '.ts'))) {
         backingClassPath += '.ts';
         hasBackingClass = true;
+      } else if (fs.existsSync(path.join(this.inputPaths[0], backingClassPath + '.coffee'))) {
+        backingClassPath += '.coffee';
+        hasBackingClass = true;
       } else {
         backingClassPath += '.js';
         hasBackingClass = false;
@@ -101,6 +104,9 @@ module.exports = class ColocatedTemplateProcessor extends Plugin {
         hbsInvocationOptions
       )})`;
       let prefix = `import { hbs } from 'ember-cli-htmlbars';\nconst __COLOCATED_TEMPLATE__ = ${hbsInvocation};\n`;
+      if (backingClassPath.endsWith('.coffee')) {
+        prefix = `import { hbs } from 'ember-cli-htmlbars'\n__COLOCATED_TEMPLATE__ = ${hbsInvocation}\n`;
+      }
 
       logger.debug(
         `processing colocated template: ${filePath} (template-only: ${hasBackingClass})`

node-tests/colocated-broccoli-plugin-test.js

@@ -149,6 +149,48 @@ describe('ColocatedTemplateCompiler', function() {
     });
   });
 
+  it('works for coffeescript component class with template', async function() {
+    input.write({
+      'app-name-here': {
+        components: {
+          'foo.hbs': `{{yield}}`,
+          'foo.coffee': stripIndent`
+            import Component from '@ember/component'
+            export default class extends Component
+          `,
+        },
+        templates: {
+          'application.hbs': `{{outlet}}`,
+        },
+      },
+    });
+
+    let tree = new ColocatedTemplateCompiler(input.path(), {
+      precompile(template) {
+        return JSON.stringify({ template });
+      },
+    });
+
+    output = createBuilder(tree);
+    await output.build();
+
+    assert.deepStrictEqual(output.read(), {
+      'app-name-here': {
+        components: {
+          'foo.coffee': stripIndent`
+            import { hbs } from 'ember-cli-htmlbars'
+            __COLOCATED_TEMPLATE__ = hbs("{{yield}}", {"contents":"{{yield}}","moduleName":"app-name-here/components/foo.hbs","parseOptions":{"srcName":"app-name-here/components/foo.hbs"}})
+            import Component from '@ember/component'
+            export default class extends Component
+          `,
+        },
+        templates: {
+          'application.hbs': '{{outlet}}',
+        },
+      },
+    });
+  });
+
   it('works for scoped addon using template only component', async function() {
     input.write({
       '@scope-name': {