import os
from os import walk
import os.path
from md2gemini import md2gemini

def merge_lines(input_text):
    lines = input_text.splitlines()
    result = []
    last_line = None

    for line in lines:
        stripped = line.rstrip()

        if last_line is not None:
            if (last_line == '' or last_line == '*') and stripped.startswith('=>'):
                last_line = None  # Skip the previous line

        if last_line is not None:
            result.append(last_line)

        last_line = stripped

    if last_line is not None:
        result.append(last_line)

    return '\n'.join(result)

def convert_md_to_gemini():
    for file in os.listdir("output/markdown"):
        with open("output/markdown/" + str(file), "r") as f:
            gemini = md2gemini(f.read())
            f.close()

        f = open("output/gemini/" + str(file).replace(".md", ".gmi"), "w")
        gemini = merge_lines(gemini)
        f.write(gemini)
        f.close()
    for file in os.listdir("output/gemini"):
        processed = ""
        f = open("output/gemini/" + file, "r")
        for line in f.readlines():
            if line == "html\n":
                pass

            else:
                processed += line
        f.close()
        f = open("output/gemini/" + file, "w")
        f.write(processed)
        f.close()
    return "markdown was converted to gemini (2/3)"