import os
import shutil
import sys
import pymysql
from peewee import *
import yaml

outputDir = "output"
# Clean the output dir & create a new
shutil.rmtree(outputDir, True)
os.mkdir(outputDir)

# Connect to the MySQL database
db = pymysql.connect(
    host="localhost",
    db="spip",
    user="spip",
    password="password",
)

# Query the database to retrieve all data
cursor = db.cursor()
cursor.execute("SELECT * FROM spip_articles ORDER BY date DESC")

# Loop through the results and format data into Markdown files
# Columns:
#   2: titre
#   7: texte
#   9: date

if len(sys.argv) > 1:
    if int(sys.argv[1]) > 0:
        fetch = cursor.fetchmany(int(sys.argv[1]) + 1)
    else:
        fetch = cursor.fetchall()
else:
    fetch = cursor.fetchmany(3 + 1)

for row in fetch:
    frontmatter = {"title": row[2], "date": row[9]}
    content = f"---\n{yaml.dump(frontmatter)}---\n{row[2]}\n\n{row[7]}"
    path = f"{outputDir}/{row[2]}.md"
    with open(path, "w") as f:
        f.write(content)

# Close the database connection
db.close()