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

def merge_lines(input_text):
    lines = input_text.strip().splitlines()
    result = []
    previous_line_was_arrow = False

    for line in lines:
        stripped = line.strip()
        if stripped == '' or stripped == '*':
            if previous_line_was_arrow:
                continue

        result.append(line)
        previous_line_was_arrow = stripped.startswith('=>')

    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)"