import datetime
import os
import re

canonical = "https://docs.pyrogram.ml"

dirs = {
    "start": ("weekly", 0.9),
    "resources": ("weekly", 0.8),
    "pyrogram": ("weekly", 0.8),
    "functions": ("monthly", 0.7),
    "types": ("monthly", 0.7),
    "errors": ("weekly", 0.6)
}


def now():
    return datetime.datetime.today().strftime("%Y-%m-%d")


with open("sitemap.xml", "w") as f:
    f.write("<?xml version=\"1.0\" encoding=\"utf-8\"?>\n")
    f.write("<urlset xmlns=\"http://www.sitemaps.org/schemas/sitemap/0.9\">\n")

    urls = [(canonical, now(), "weekly", 1.0)]


    def search(path):
        try:
            for j in os.listdir(path):
                search("{}/{}".format(path, j))
        except NotADirectoryError:
            d = path.split("/")[0]
            path = "{}/{}".format(canonical, path.split(".")[0])
            path = re.sub("^(.+)/index$", "\g<1>", path)
            urls.append((path, now(), dirs[d][0], dirs[d][1]))


    for i in dirs.keys():
        search(i)

    for i in urls:
        f.write("    <url>\n")
        f.write("        <loc>{}</loc>\n".format(i[0]))
        f.write("        <lastmod>{}</lastmod>\n".format(i[1]))
        f.write("        <changefreq>{}</changefreq>\n".format(i[2]))
        f.write("        <priority>{}</priority>\n".format(i[3]))
        f.write("    </url>\n\n")

    f.write("</urlset>")