closes #3

package.json

@@ -31,7 +31,7 @@
   },
   "scripts": {
     "lint": "prettier -c ./ && eslint src test --ext .cjs,.js",
-    "test": "npm run lint && node --import ./test/setup.js test/core.spec.js",
+    "test": "npm run lint && node --import ./test/setup.js test/core.spec.js && RIOT_COMPONENTS_FILE_EXTENSION=.html node --import ./test/setup.js test/custom-extension.spec.js",
     "prepublishOnly": "npm run test"
   },
   "version": "9.0.0"

src/index.js

@@ -2,10 +2,14 @@ import { compile } from '@riotjs/compiler'
 
 // Only module format is supported
 const format = 'module'
+// Allow the use of custom file extensions https://github.com/riot/register/issues/3
+const FILE_EXTENSION_REGEX = new RegExp(
+  `${process.env.RIOT_COMPONENTS_FILE_EXTENSION || '.riot'}$`,
+)
 
 export async function load(url, context, nextLoad) {
   // non riot files will be skipped
-  if (!/\.riot$/.test(url)) return nextLoad(url)
+  if (!FILE_EXTENSION_REGEX.test(url)) return nextLoad(url)
 
   // get the source code of the riot file
   const { source } = await nextLoad(url, {

test/custom-extension.spec.js

@@ -0,0 +1,6 @@
+import assert from 'node:assert/strict'
+
+const component = await import('./test.html')
+assert.ok(component)
+
+console.log('All tests passed :)')

test/test.html

@@ -0,0 +1,3 @@
+<test>
+  <p>hello</p>
+</test>