#pragma once

#include <AK/Lock.h>
#include <AK/Retainable.h>
#include <AK/RetainPtr.h>
#include <AK/HashTable.h>
#include <AK/Vector.h>
#include <Kernel/UnixTypes.h>

enum class SocketRole { None, Listener, Accepted, Connected };

class Socket : public Retainable<Socket> {
public:
    static RetainPtr<Socket> create(int domain, int type, int protocol, int& error);
    virtual ~Socket();

    int domain() const { return m_domain; }
    int type() const { return m_type; }
    int protocol() const { return m_protocol; }

    bool can_accept() const { return !m_pending.is_empty(); }
    RetainPtr<Socket> accept();
    bool is_connected() const { return m_connected; }
    bool listen(int backlog, int& error);

    virtual bool bind(const sockaddr*, socklen_t, int& error) = 0;
    virtual bool connect(const sockaddr*, socklen_t, int& error) = 0;
    virtual bool get_address(sockaddr*, socklen_t*) = 0;
    virtual bool is_local() const { return false; }
    virtual void attach_fd(SocketRole) = 0;
    virtual void detach_fd(SocketRole) = 0;
    virtual bool can_read(SocketRole) const = 0;
    virtual ssize_t read(SocketRole, byte*, size_t) = 0;
    virtual ssize_t write(SocketRole, const byte*, size_t) = 0;
    virtual bool can_write(SocketRole) const = 0;

    pid_t origin_pid() const { return m_origin_pid; }

protected:
    Socket(int domain, int type, int protocol);

    bool queue_connection_from(Socket&, int& error);

private:
    Lock m_lock;
    pid_t m_origin_pid { 0 };
    int m_domain { 0 };
    int m_type { 0 };
    int m_protocol { 0 };
    int m_backlog { 0 };
    bool m_connected { false };

    Vector<RetainPtr<Socket>> m_pending;
    Vector<RetainPtr<Socket>> m_clients;
};