from pathlib import Path
import re
re_heading1 = re.compile(r"^(# .*)$", re.MULTILINE)
re_chapter = re.compile(r"chapter([0-9]+)")

p = Path("/Users/mishig/Desktop/course/chapters")
files = [x for x in p.glob("**/*") if x.is_file() and (str(x).endswith(".md") or str(x).endswith(".mdx"))]

for file in files:
    if not re_chapter.findall(str(file)):
        continue
    chapter = int(re_chapter.findall(str(file))[0])
    with open(file, "r", encoding="utf-8-sig") as reader:
        content = reader.read()
    if chapter == 0 or "DocNotebookDropdown" in content:
        continue
    svelte = f"""<CourseFloatingBanner
    chapter={{{chapter}}}
    classNames="absolute z-10 right-0 top-0"
/>"""
    content = re_heading1.sub(rf"\1\n\n{svelte}", content, 1)
    with open(file, "w", encoding="utf-8") as writer:
        writer.write(content)