#pragma once

#include <AK/AKString.h>
#include <AK/HashMap.h>
#include <AK/Vector.h>
#include <LibGUI/GTableModel.h>
#include <unistd.h>

class ProcessTableModel final : public GTableModel {
public:
    ProcessTableModel();
    virtual ~ProcessTableModel() override;

    virtual int row_count() const override;
    virtual int column_count() const override;
    virtual String column_name(int column) const override;
    virtual ColumnMetadata column_metadata(int column) const override;
    virtual GModelIndex selected_index() const override;
    virtual void set_selected_index(GModelIndex) override;
    virtual String data(int row, int column) const override;
    virtual void update() override;

    pid_t selected_pid() const;

private:
    struct ProcessState {
        pid_t pid;
        unsigned nsched;
        String name;
        String state;
        String user;
        String priority;
        size_t linear;
        size_t physical;
        float cpu_percent;
    };

    struct Process {
        ProcessState current_state;
        ProcessState previous_state;
    };

    HashMap<pid_t, OwnPtr<Process>> m_processes;
    Vector<pid_t> m_pids;
    int m_selected_row { -1 };
};