Add handling for boolean options

Fryuni · Mar 5, 2024 · 065ab38 · 065ab38
1 parent ff5024e
commit 065ab38
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/packages/sitemap-ext/foo.astro b/packages/sitemap-ext/foo.astro
diff --git a/packages/sitemap-ext/index.ts b/packages/sitemap-ext/index.ts
@@ -84,8 +84,8 @@ export default defineIntegration({
 
 				defineRouteConfig({
 					importName: 'sitemap-ext:config',
-					callbackHandler: (context, configCb: ConfigCallback) => {
-						configCb({
+					callbackHandler: (context, configCb: ConfigCallback | boolean) => {
+						const hooks: Parameters<ConfigCallback>[0] = {
 							removeFromSitemap(routeParams) {
 								for (const route of context.routeData) {
 									makeDecision(false, route, routeParams);
@@ -103,7 +103,17 @@ export default defineIntegration({
 									}
 								}
 							},
-						});
+						};
+
+						if (typeof configCb === 'boolean') {
+							if (configCb) {
+								hooks.addToSitemap();
+							} else {
+								hooks.removeFromSitemap();
+							}
+						} else {
+							configCb(hooks);
+						}
 					},
 				});