Core configuration as extension

Author: punyflash

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@westacks/vortex",
-  "version": "0.0.4",
+  "version": "0.0.5",
   "license": "MIT",
   "description": "Server-based routing for SPAs",
   "author": "Dmytro Morozov <puny.flash@gmail.com>",
@@ -52,6 +52,11 @@
       "import": "./dist/server.js",
       "require": "./dist/server.cjs",
       "types": "./dist/server.d.ts"
+    },
+    "./extensions": {
+      "import": "./dist/extensions.js",
+      "require": "./dist/extensions.cjs",
+      "types": "./dist/extensions/index.d.ts"
     }
   }
 }

src/extensions/core.ts

@@ -0,0 +1,62 @@
+import { getPage as page, setPage } from "../page";
+import type { VortexExtension, VortexConfig } from "../index";
+
+type CoreConfig = {
+    detect?: string
+    version?: string
+    component?: string
+    location?: string
+    only?: string
+}
+
+const core = ({
+    detect = 'x-vortex',
+    version = 'x-vortex-version',
+    component = 'x-vortex-component',
+    location = 'x-vortex-location',
+    only = 'x-vortex-only',
+}: CoreConfig): VortexExtension => ({ request, response }) => ({
+    request: request.use(function (request) {
+        if (!request.vortex) {
+            return request
+        }
+
+        const config = (typeof request.vortex === "object" ? request.vortex : {}) as VortexConfig
+
+        request.headers[detect] = true
+        request.headers[version] = page()?.version
+        request.headers[component] = page()?.component
+
+        if (config.only) {
+            request.headers[only] = config.only.join(",")
+        }
+
+        return request
+    }),
+    response: response.use(function (response) {
+        if (!response.config.vortex || !response.headers[detect]) {
+            return response
+        }
+
+        const config = typeof response.config.vortex === "object" ? response.config.vortex : {}
+
+        setPage(response.data)
+
+        if (config?.preserve || (window.history.state.url === response.data.url && config?.preserve !== false)) {
+            window.history.replaceState(response.data, "", response.data.url)
+        } else {
+            window.history.pushState(response.data, "", response.data.url)
+        }
+
+        return response
+    }, function (error) {
+        if (error.response?.headers[location]) {
+            window.location.href = error.response.headers[location];
+            return Promise.resolve(error.response)
+        }
+
+        return Promise.reject(error)
+    })
+})
+
+export default core

src/extensions/index.ts

@@ -0,0 +1 @@
+export { default as core } from "./core"

src/index.ts

@@ -1,3 +1,3 @@
 export * from "./client";
 export * from "./page";
-export { axios, extend, type VortexExtension } from "./router";
+export { axios, extend, type VortexExtension, type VortexConfig } from "./router";

src/router.ts

@@ -36,8 +36,9 @@ interface RouterDefaults<D = any> extends Omit<RouterRequestConfig<D>, 'headers'
     headers: HeadersDefaults;
 }
 
-type VortexConfig = {
+export type VortexConfig = {
     preserve?: boolean
+    only?: string[]
 }
 
 interface RouterRequestConfig<D = any> extends AxiosRequestConfig<D> {
@@ -99,48 +100,6 @@ export function createRouter() {
     window.addEventListener("popstate", popstate)
 
     axios.defaults.vortex = true
-
-    extend(({ request, response }) => ({
-        request: request.use(function (request) {
-            if (!request.vortex) {
-                return request
-            }
-
-            request.headers["x-vortex"] = true
-            request.headers["x-vortex-version"] = page()?.version
-            request.headers["x-vortex-component"] = page()?.component
-
-            return request
-        }),
-        response: response.use(function (response) {
-            if (!response.config.vortex || !response.headers["x-vortex"]) {
-                return response
-            }
-
-            const config = typeof response.config.vortex === "object" ? response.config.vortex : {}
-
-            setPage(response.data)
-
-            if (config?.preserve || (window.history.state.url === response.data.url && config?.preserve !== false)) {
-                window.history.replaceState(response.data, "", response.data.url)
-            } else {
-                window.history.pushState(response.data, "", response.data.url)
-            }
-
-            return response
-        }, function (error) {
-            if (error.response?.headers["x-vortex-location"]) {
-                window.location.href = error.response.headers["x-vortex-location"];
-                return Promise.resolve(error.response)
-            }
-
-            if (error.response?.headers["x-vortex"]) {
-                return Promise.resolve(error.response)
-            }
-
-            return Promise.reject(error)
-        })
-    }))
 }
 
 export function destroyRouter() {

vite.config.ts

@@ -11,6 +11,7 @@ export default defineConfig({
             entry: {
                 "index": resolve(__dirname, 'src/index.ts'),
                 "server": resolve(__dirname, 'src/server.ts'),
+                "extensions": resolve(__dirname, 'src/extensions/index.ts'),
             },
         },
         rollupOptions: {