#pragma once

#include "libparse.h"
#include <ast_novel.h>

namespace example_novel {
    class LIBPARSE_EXPORT FragmentExistsCheck : public lib_parse::CheckProvider {
    private:
        void exists_check(std::shared_ptr<const ast_gen::GlobalElement> root, std::shared_ptr<const ast_gen::ElementAccess> target) const;

    public:
        // 通过 CheckProvider 继承
        virtual void validCheck(std::shared_ptr<const ast_gen::ElementAccess> root) const override;
        virtual QString name() const override;
    };


    class FragmentGraphHelper;
    class LIBPARSE_EXPORT FragmentGraphCheck : public std::enable_shared_from_this<FragmentGraphCheck>, public lib_parse::CheckProvider {
    private:
        QHash<QString, std::shared_ptr<FragmentGraphHelper>> elements_store;
        QList<std::shared_ptr<const FragmentGraphHelper>> fragments_sort_list;

        QList<std::shared_ptr<FragmentGraphHelper>> refers_cycle_check(
                std::shared_ptr<FragmentGraphHelper> item, QList<std::shared_ptr<FragmentGraphHelper>> prevs = QList<std::shared_ptr<FragmentGraphHelper>>()) const;

    public:
        void setElement(std::shared_ptr<FragmentGraphHelper> inst);
        std::shared_ptr<FragmentGraphHelper> getElement(const QString &signature) const;

        QList<std::shared_ptr<FragmentGraphHelper>> getHangoutNodes();
        bool nodeDismantle(std::shared_ptr<FragmentGraphHelper> inst);

        QList<std::shared_ptr<const FragmentGraphHelper>> fragmentsSequence() const;

        // CheckProvider interface
    public:
        virtual void validCheck(std::shared_ptr<const ast_gen::ElementAccess> root) const override;
        virtual QString name() const override;
    };

    class FragmentGraphHelper : public std::enable_shared_from_this<FragmentGraphHelper> {
    private:
        std::shared_ptr<const example_novel::PointDefines> node_peer;
        QList<std::shared_ptr<FragmentGraphHelper>> next_nodes;
        uint indegree = 0;

    public:
        FragmentGraphHelper(std::shared_ptr<const example_novel::PointDefines> node);
        
        std::shared_ptr<const example_novel::PointDefines> nodePeer() const;

        void appendNext(std::shared_ptr<FragmentGraphHelper> node);
        QList<std::shared_ptr<FragmentGraphHelper>> nextList() const;

        uint& inDegree();
    };


    class LIBPARSE_EXPORT StoryOrderCheck : public lib_parse::CheckProvider {
    private:
        uint sort_index = 1;
        /**
         * 获取合适的文档节点.
         * 
         * \param pnode
         * \return 
         */
        QList<std::shared_ptr<const ast_gen::ElementAccess>> valid_docs_peak(std::shared_ptr<const ast_gen::ElementAccess> pnode) const;

    public:
        // 通过 CheckProvider 继承
        QString name() const override;
        void validCheck(std::shared_ptr<const ast_gen::ElementAccess> root) const override;
    };
} // namespace example_novel