From 218b734eca1eec46fdb949992a7e5c90acb151ca Mon Sep 17 00:00:00 2001
From: Junyi <zjy220124@alibaba-inc.com>
Date: Thu, 22 Aug 2024 17:06:40 +0800
Subject: [PATCH] feat: add function to check route matching

---
 index.js | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/index.js b/index.js
index 8b61b0f..38a2b78 100644
--- a/index.js
+++ b/index.js
@@ -30,6 +30,17 @@ const getPath = (ctx, map) => {
   return path;
 };
 
+const isMatch = (ctx, match) => {
+  if (typeof match === 'string') {
+    return ctx.path.startsWith(match);
+  } else if (match instanceof RegExp) {
+    return ctx.path.match(match);
+  } else if (typeof match === 'function') {
+    return match(ctx);
+  }
+  return false;
+};
+
 module.exports = (options = {}) => {
   if (!options.host) {
     throw new Error('miss option host');
@@ -40,7 +51,7 @@ module.exports = (options = {}) => {
   const { protocol, host, hostname, port, pathname } = new URL(options.host);
   const request = protocol === 'http:' ? http.request : https.request;
   return async (ctx, next) => {
-    if (options.match && ctx.path.match(options.match)) {
+    if (isMatch(ctx, options.match)) {
       const opt = {
         hostname,
         port: port || (protocol === 'http:' ? 80 : 443),