Support for cleanUrls completed

lib/index.js

@@ -200,6 +200,51 @@ const applicableForCleanUrl = (decodedPath, cleanUrls) => {
 	return matches;
 };
 
+const getPossiblePaths = (relativePath, extension) => [
+	path.join(relativePath, `index${extension}`),
+	relativePath.endsWith('/') ? relativePath.replace(/\/$/g, extension) : (relativePath + extension)
+];
+
+const findRelated = async (current, relativePath, stat, extension = '.html') => {
+	const possible = getPossiblePaths(relativePath, extension);
+
+	let stats = null;
+
+	for (let index = 0; index < possible.length; index++) {
+		const related = possible[index];
+		const absolutePath = path.join(current, related);
+
+		try {
+			stats = await stat(absolutePath);
+		} catch (err) {
+			if (err.code !== 'ENOENT') {
+				throw err;
+			}
+		}
+
+		if (stats) {
+			return {
+				stats,
+				absolutePath
+			};
+		}
+	}
+
+	if (extension === '.htm') {
+		return null;
+	}
+
+	// At this point, no `.html` files have been found, so we
+	// need to check for the existance of `.htm` ones.
+	const relatedHTM = findRelated(current, relativePath, stat, '.htm');
+
+	if (relatedHTM) {
+		return relatedHTM;
+	}
+
+	return null;
+};
+
 module.exports = async (request, response, config = {}, methods = {}) => {
 	const cwd = process.cwd();
 	const current = config.public ? path.join(cwd, config.public) : cwd;
@@ -218,8 +263,8 @@ module.exports = async (request, response, config = {}, methods = {}) => {
 	}
 
 	const relativePath = applyRewrites(decodedPath, config.rewrites);
-	const absolutePath = path.join(current, relativePath);
 
+	let absolutePath = path.join(current, relativePath);
 	let stats = null;
 
 	try {
@@ -233,11 +278,30 @@ module.exports = async (request, response, config = {}, methods = {}) => {
 		}
 	}
 
-	if (!stats) {
-		response.statusCode = 404;
-		response.end('Not Found');
+	if (!stats || stats.isDirectory()) {
+		if (cleanUrl) {
+			try {
+				const related = await findRelated(current, relativePath, handlers.stat);
 
-		return;
+				if (related) {
+					({stats, absolutePath} = related);
+				}
+			} catch (err) {
+				if (err.code !== 'ENOENT') {
+					response.statusCode = 500;
+					response.end(err.message);
+
+					return;
+				}
+			}
+		}
+
+		if (!stats) {
+			response.statusCode = 404;
+			response.end('Not Found');
+
+			return;
+		}
 	}
 
 	const headers = await getHeaders(handlers, config.headers, relativePath, stats);