chore: run license script on pre-commit for staged files

lefthook.yml

@@ -0,0 +1,6 @@
+pre-commit:
+  parallel: true
+  commands:
+    generate-license:
+      run: pnpm run license "--files={staged_files}"
+      stage_fixed: true

package.json

@@ -37,6 +37,7 @@
     "@changesets/cli": "2.27.9",
     "@ls-lint/ls-lint": "2.3.0-beta.1",
     "@swc/core": "1.7.36",
+    "lefthook": "1.9.3",
     "rimraf": "5.0.10",
     "syncpack": "13.0.0",
     "taze": "0.17.2",

scripts/license.mjs

@@ -12,7 +12,7 @@
  * governing permissions and limitations under the License.
  */
 
-import { fs, glob, path } from 'zx';
+import { fs, glob, path, argv } from 'zx';
 
 const HEADER = `Copyright ${new Date().getFullYear()} Hypergiant Galactic Systems Inc. All rights reserved.
 This file is licensed to you under the Apache License, Version 2.0 (the "License");
@@ -46,28 +46,46 @@ const COMMENT_STYLES = {
   '.mdx': HTML_COMMENT_STYLE,
 };
 
-const files = await glob(['**/*.{js,ts,tsx,mjs,mdx,md,css}'], {
-  ignore: [
-    '**/node_modules/**',
-    '**/dist/**',
-    '**/README.md',
-    '**/LICENSE.md',
-    '**/CHANGELOG.md',
-    '**/.github/**/*.md',
-  ],
-});
+function getFormattedHeader(fileExtension) {
+  const style = COMMENT_STYLES[fileExtension];
+  return `${style.start}${HEADER.split('\n').join(`\n${style.middle}`)}${style.end}`.replace(
+    /\s+\n/g,
+    '\n',
+  );
+}
+
+const filesToParse = argv.files?.split(' ');
+const files = await glob(
+  filesToParse && filesToParse.length > 0
+    ? filesToParse
+    : ['**/*.{js,ts,tsx,mjs,mdx,md,css}'],
+  {
+    ignore: [
+      '**/node_modules/**',
+      '**/dist/**',
+      '**/README.md',
+      '**/LICENSE.md',
+      '**/CHANGELOG.md',
+      '**/.github/**/*.md',
+      '**/*.yml',
+    ],
+  },
+);
 
 for (const file of files) {
-  const style = COMMENT_STYLES[path.extname(file)];
+  const header = getFormattedHeader(path.extname(file));
+
   let contents = fs.readFileSync(file, 'utf8');
-  let header =
-    style.start + HEADER.split('\n').join(`\n${style.middle}`) + style.end;
+  if (!/Copyright \d+ Hypergiant/.test(contents)) {
+    const interpreterDirective = contents.match(/^#!.*$/m)?.[0];
 
-  header = header.replace(/\s+\n/g, '\n');
+    if (interpreterDirective) {
+      contents = contents.replace(interpreterDirective, '').trimStart();
+      contents = `${interpreterDirective}\n\n${header}\n${contents}`;
+    } else {
+      contents = `${header}\n${contents}`;
+    }
 
-  if (!/Copyright \d+ Hypergiant/.test(contents)) {
-    // TODO: check for hashbang for things like zx scripts
-    contents = `${header}\n${contents}`;
     fs.writeFileSync(file, contents);
   }
 }