chore: move benchmark data creation to python script

Author: MeanderingProgrammer

CHANGELOG.md

@@ -6,6 +6,7 @@
 
 - margin for paragraphs [#200](https://github.com/MeanderingProgrammer/render-markdown.nvim/issues/200)
   [d20d19f](https://github.com/MeanderingProgrammer/render-markdown.nvim/commit/d20d19fa54965f6eb94558c0b84fe9a942169fb4)
+- `on.attach` buffer callback [8b4149b](https://github.com/MeanderingProgrammer/render-markdown.nvim/commit/8b4149b122cfbf58b79a552ae89b3df2ddc39786)
 
 ### Bug Fixes
 

justfile

@@ -6,7 +6,7 @@ test:
   just busted "tests"
 
 bench:
-  just gen-medium
+  just generate
   just busted "benches"
 
 [private]
@@ -62,36 +62,5 @@ update:
 cat-log:
   cat ~/.local/state/nvim/render-markdown.log
 
-gen-medium:
-  just gen-headings "1000" "temp/medium.md"
-  just gen-table "50" "100" "temp/medium-table.md"
-
-gen-large:
-  just gen-headings "100000" "temp/large.md"
-
-[private]
-gen-headings lines path:
-  {{path_exists(path)}} || just gen-headings-content {{lines}} > {{path}}
-
-[private]
-gen-headings-content lines:
-  #!/usr/bin/env python
-  for i in range({{lines}}):
-    level = "#" * ((i % 6) + 1)
-    print(f"{level} Title {i}\n")
-
-[private]
-gen-table tables lines path:
-  {{path_exists(path)}} || just gen-table-contents {{tables}} {{lines}} > {{path}}
-
-[private]
-gen-table-contents tables lines:
-  #!/usr/bin/env python
-  for i in range({{tables}}):
-    print(f"# Table {i}")
-    print()
-    print(f"| `Column 1`     | **Column 2**     | *Column 3*     |")
-    print(f"| -------------- | :--------------- | -------------: |")
-    for j in range({{lines}}):
-      print(f"| Row {j:<4} Col 1 | `Row {j:<4} Col 2` | Row {j:<4} Col 3 |")
-    print()
+generate:
+  python scripts/generate.py

scripts/generate.py

@@ -0,0 +1,73 @@
+import abc
+from argparse import ArgumentParser
+from dataclasses import dataclass
+from pathlib import Path
+from typing import override
+
+
+@dataclass(frozen=True)
+class Generator(abc.ABC):
+    @abc.abstractmethod
+    def name(self, size: str) -> str:
+        pass
+
+    @abc.abstractmethod
+    def create(self, n: int) -> list[list[str]]:
+        pass
+
+
+@dataclass(frozen=True)
+class Heading(Generator):
+    @override
+    def name(self, size: str) -> str:
+        return f"{size}.md"
+
+    @override
+    def create(self, n: int) -> list[list[str]]:
+        sections: list[list[str]] = []
+        for i in range(10 * n):
+            sections.append([f"{'#' * ((i % 6) + 1)} Title {i}"])
+        return sections
+
+
+@dataclass(frozen=True)
+class Table(Generator):
+    @override
+    def name(self, size: str) -> str:
+        return f"{size}-table.md"
+
+    @override
+    def create(self, n: int) -> list[list[str]]:
+        sections: list[list[str]] = []
+        for i in range(n // 2):
+            sections.append([f"# Table {i}"])
+            sections.append(self.table(n))
+        return sections
+
+    def table(self, n: int) -> list[str]:
+        rows: list[str] = []
+        rows.append(f"| `Column 1`     | **Column 2**     | *Column 3*     |")
+        rows.append(f"| -------------- | :--------------- | -------------: |")
+        for i in range(n):
+            rows.append(f"| Row {i:<4} Col 1 | `Row {i:<4} Col 2` | Row {i:<4} Col 3 |")
+        return rows
+
+
+def main(force: bool) -> None:
+    sizes: list[str] = ["small", "medium", "large"]
+    generators: list[Generator] = [Heading(), Table()]
+    for i, size in enumerate(sizes):
+        n: int = 10 ** (i + 1)
+        for generator in generators:
+            path = Path("temp").joinpath(generator.name(size))
+            if not path.exists() or force:
+                sections = generator.create(n)
+                content = "\n\n".join(["\n".join(section) for section in sections])
+                path.write_text(content)
+
+
+if __name__ == "__main__":
+    parser = ArgumentParser(description="Generate sample data for benchmarking")
+    parser.add_argument("--force", action="store_true")
+    args = parser.parse_args()
+    main(args.force)