// Copyright 2017 Citra Emulator Project
// Licensed under GPLv2 or any later version
// Refer to the license.txt file included.

#pragma once

#include <QPixmap>
#include <QStandardItem>
#include <QStandardItemModel>

#include "common/common_types.h"

namespace Column {
enum List {
    PASSWORD,
    ROOM_NAME,
    GAME_NAME,
    HOST,
    MEMBER,
    TOTAL,
};
}

class LobbyItem : public QStandardItem {
public:
    LobbyItem() : QStandardItem() {}
    LobbyItem(const QString& string) : QStandardItem(string) {}
    virtual ~LobbyItem() override {}
};

class LobbyItemPassword : public LobbyItem {
public:
    static const int PasswordRole = Qt::UserRole + 1;
    LobbyItemPassword() : LobbyItem() {}
    LobbyItemPassword(const bool has_password) : LobbyItem() {
        setData(has_password, PasswordRole);
    }

    QVariant data(int role) const override {
        if (role != Qt::DecorationRole) {
            return LobbyItem::data(role);
        }
        bool has_password = data(PasswordRole).toBool();
        return has_password ? QIcon(":/icons/lock.png") : QIcon();
    }

    bool operator<(const QStandardItem& other) const override {
        return data(PasswordRole).toBool() < other.data(PasswordRole).toBool();
    }
};

class LobbyItemName : public LobbyItem {
public:
    static const int NameRole = Qt::UserRole + 1;
    LobbyItemName() : LobbyItem() {}
    LobbyItemName(QString name) : LobbyItem() {
        setData(name, NameRole);
    }

    QVariant data(int role) const override {
        if (role != Qt::DisplayRole) {
            return LobbyItem::data(role);
        }
        return data(NameRole).toString();
    }
    bool operator<(const QStandardItem& other) const override {
        return data(NameRole).toString().localeAwareCompare(other.data(NameRole).toString()) < 0;
    }
};

class LobbyItemGame : public LobbyItem {
public:
    static const int TitleIDRole = Qt::UserRole + 1;
    static const int GameNameRole = Qt::UserRole + 2;
    static const int GameIconRole = Qt::UserRole + 3;

    LobbyItemGame() : LobbyItem() {}
    LobbyItemGame(u64 title_id, QString game_name, QPixmap smdh_icon) : LobbyItem() {
        setData(static_cast<unsigned long long>(title_id), TitleIDRole);
        setData(game_name, GameNameRole);
        if (!smdh_icon.isNull()) {
            setData(smdh_icon, GameIconRole);
        }
    }

    QVariant data(int role) const override {
        if (role == Qt::DecorationRole) {
            auto val = data(GameIconRole);
            if (val.isValid()) {
                val = val.value<QPixmap>().scaled(16, 16, Qt::KeepAspectRatio);
            }
            return val;
        } else if (role != Qt::DisplayRole) {
            return LobbyItem::data(role);
        }
        return data(GameNameRole).toString();
    }

    bool operator<(const QStandardItem& other) const override {
        return data(GameNameRole)
                   .toString()
                   .localeAwareCompare(other.data(GameNameRole).toString()) < 0;
    }
};

class LobbyItemHost : public LobbyItem {
public:
    static const int HostUsernameRole = Qt::UserRole + 1;
    static const int HostIPRole = Qt::UserRole + 2;
    static const int HostPortRole = Qt::UserRole + 3;

    LobbyItemHost() : LobbyItem() {}
    LobbyItemHost(QString username, QString ip, u16 port) : LobbyItem() {
        setData(username, HostUsernameRole);
        setData(ip, HostIPRole);
        setData(port, HostPortRole);
    }

    QVariant data(int role) const override {
        if (role != Qt::DisplayRole) {
            return LobbyItem::data(role);
        }
        return data(HostUsernameRole).toString();
    }

    bool operator<(const QStandardItem& other) const override {
        return data(HostUsernameRole)
                   .toString()
                   .localeAwareCompare(other.data(HostUsernameRole).toString()) < 0;
    }
};

class LobbyMember {
public:
    LobbyMember() {}
    LobbyMember(const LobbyMember& other) {
        username = other.username;
        title_id = other.title_id;
        game_name = other.game_name;
    }
    LobbyMember(const QString username, u64 title_id, const QString game_name)
        : username(username), title_id(title_id), game_name(game_name) {}
    ~LobbyMember() {}

    QString GetUsername() const {
        return username;
    }
    u64 GetTitleId() const {
        return title_id;
    }
    QString GetGameName() const {
        return game_name;
    }

private:
    QString username;
    u64 title_id;
    QString game_name;
};

Q_DECLARE_METATYPE(LobbyMember);

class LobbyItemMemberList : public LobbyItem {
public:
    static const int MemberListRole = Qt::UserRole + 1;
    static const int MaxPlayerRole = Qt::UserRole + 2;

    LobbyItemMemberList() : LobbyItem() {}
    LobbyItemMemberList(QList<QVariant> members, u32 max_players) : LobbyItem() {
        setData(members, MemberListRole);
        setData(max_players, MaxPlayerRole);
    }

    QVariant data(int role) const override {
        if (role != Qt::DisplayRole) {
            return LobbyItem::data(role);
        }
        auto members = data(MemberListRole).toList();
        return QString("%1 / %2").arg(QString::number(members.size()),
                                      data(MaxPlayerRole).toString());
    }

    bool operator<(const QStandardItem& other) const override {
        // sort by rooms that have the most players
        int left_members = data(MemberListRole).toList().size();
        int right_members = other.data(MemberListRole).toList().size();
        return left_members < right_members;
    }
};