Merge pull request #96 from groupon/dbushong/feature/main/ts

Support .ts .mjs .tsx .jsx extensions for license headers
groupon · Oct 29, 2021 · 42ba254 · 42ba254
2 parents 53eed49 + a7ee4ad
commit 42ba254
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 3 deletions.
diff --git a/lib/license/index.js b/lib/license/index.js
@@ -74,7 +74,7 @@ function collectFiles(cwd, whitelist, optionalExclude) {
   );
 
   function scanDirectory(directory) {
-    return globAsync(`${directory}/**/*.{js,coffee}`, {
+    return globAsync(`${directory}/**/*.{js,jsx,mjs,ts,tsx,coffee}`, {
       ignore: exclude,
       nodir: true,
       cwd,
@@ -115,6 +115,10 @@ function addLicenseHeader(entry) {
 function addMissingLicenseHeaders(licenseText, files) {
   const licenseHeaders = {
     '.js': COMMENT_TYPES['.js'].getLicenseHeader(licenseText),
+    '.jsx': COMMENT_TYPES['.js'].getLicenseHeader(licenseText),
+    '.mjs': COMMENT_TYPES['.js'].getLicenseHeader(licenseText),
+    '.ts': COMMENT_TYPES['.js'].getLicenseHeader(licenseText),
+    '.tsx': COMMENT_TYPES['.js'].getLicenseHeader(licenseText),
     '.coffee': COMMENT_TYPES['.coffee'].getLicenseHeader(licenseText),
   };
   return files

diff --git a/test/fixtures/merge-commit b/test/fixtures/merge-commit
@@ -2,6 +2,6 @@
 set -e
 git init
 
-echo "console.log('do stuff');" > merge.js
-git add merge.js
+echo "import { mittens } from 'kittens';" > merge.ts
+git add merge.ts
 git commit -m 'Merge pull request #119 from theowner/some/branch'
diff --git a/test/license/index.test.js b/test/license/index.test.js
@@ -87,4 +87,25 @@ describe('addLicenseHeaders', () => {
       );
     });
   });
+
+  describe('with a ts file w/o a license header', () => {
+    const dirname = withFixture('merge-commit');
+    const filename = `${dirname}/merge.ts`;
+
+    it('adds the header', async () => {
+      fs.writeFileSync(`${dirname}/LICENSE`, 'SIMPLE');
+
+      const changedFiles = await addLicenseHeaders(dirname);
+      assert.deepStrictEqual(changedFiles, [filename]);
+      assert.strictEqual(
+        fs.readFileSync(filename, 'utf8'),
+        `\
+/*
+ * SIMPLE
+ */
+import { mittens } from 'kittens';
+`
+      );
+    });
+  });
 });