refactor(heading): validate level (#28)

Co-authored-by: Pooya Parsa <[email protected]>
unjs · Nov 26, 2024 · 6f38211 · 6f38211
1 parent 6f95ba4
commit 6f38211
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/render.ts b/src/render.ts
@@ -13,8 +13,16 @@
  *
  * @group render_utils
  */
-export function heading(text: string, level: number): string {
-  return `\n${"#".repeat(level || 1)} ${text}\n`;
+export function heading(
+  text: string,
+  level: 1 | 2 | 3 | 4 | 5 | 6 | (number & {}),
+): string {
+  if (!level || level < 1) {
+    level = 1;
+  } else if (level > 6) {
+    level = 6;
+  }
+  return `\n${"#".repeat(level)} ${text}\n`;
 }
 
 /**