/*
 * Copyright (c) 2020-2022, the SerenityOS developers.
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <AK/ByteString.h>
#include <AK/Forward.h>
#include <AK/NonnullOwnPtr.h>
#include <LibURL/URL.h>

namespace Gemini {

class Line {
public:
    Line(ByteString string)
        : m_text(move(string))
    {
    }

    virtual ~Line() = default;

    virtual ByteString render_to_html() const = 0;

protected:
    ByteString m_text;
};

class Document : public RefCounted<Document> {
public:
    ByteString render_to_html() const;

    static NonnullRefPtr<Document> parse(StringView source, const URL::URL&);

    const URL::URL& url() const { return m_url; }

private:
    explicit Document(const URL::URL& url)
        : m_url(url)
    {
    }

    void read_lines(StringView);

    Vector<NonnullOwnPtr<Line>> m_lines;
    URL::URL m_url;
    bool m_inside_preformatted_block { false };
    bool m_inside_unordered_list { false };
};

class Text : public Line {
public:
    Text(ByteString line)
        : Line(move(line))
    {
    }
    virtual ~Text() override = default;
    virtual ByteString render_to_html() const override;
};

class Link : public Line {
public:
    Link(ByteString line, Document const&);
    virtual ~Link() override = default;
    virtual ByteString render_to_html() const override;

private:
    URL::URL m_url;
    ByteString m_name;
};

class Preformatted : public Line {
public:
    Preformatted(ByteString line)
        : Line(move(line))
    {
    }
    virtual ~Preformatted() override = default;
    virtual ByteString render_to_html() const override;
};

class UnorderedList : public Line {
public:
    UnorderedList(ByteString line)
        : Line(move(line))
    {
    }
    virtual ~UnorderedList() override = default;
    virtual ByteString render_to_html() const override;
};

class Control : public Line {
public:
    enum Kind {
        UnorderedListStart,
        UnorderedListEnd,
        PreformattedStart,
        PreformattedEnd,
    };
    Control(Kind kind)
        : Line("")
        , m_kind(kind)
    {
    }
    virtual ~Control() override = default;
    virtual ByteString render_to_html() const override;

private:
    Kind m_kind;
};

class Heading : public Line {
public:
    Heading(ByteString line, int level)
        : Line(move(line))
        , m_level(level)
    {
    }
    virtual ~Heading() override = default;
    virtual ByteString render_to_html() const override;

private:
    int m_level { 1 };
};

}