/*
 * Copyright (c) 2025, ayeteadoe <ayeteadoe@gmail.com>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <LibCore/System.h>
#include <RequestServer/RequestPipe.h>

#if defined(AK_OS_WINDOWS)
#    include <AK/Windows.h>
#endif

namespace RequestServer {

RequestPipe::RequestPipe(int const reader_fd, int const writer_fd)
    : m_reader_fd(reader_fd)
    , m_writer_fd(writer_fd)
{
    VERIFY(m_reader_fd >= 0);
    VERIFY(m_writer_fd >= 0);
}

RequestPipe::RequestPipe(RequestPipe&& other)
    : m_reader_fd(exchange(other.m_reader_fd, -1))
    , m_writer_fd(exchange(other.m_writer_fd, -1))
{
}

RequestPipe& RequestPipe::operator=(RequestPipe&& other)
{
    m_reader_fd = exchange(other.m_reader_fd, -1);
    m_writer_fd = exchange(other.m_writer_fd, -1);
    return *this;
}

RequestPipe::~RequestPipe()
{
    if (m_writer_fd != -1)
        MUST(Core::System::close(m_writer_fd));
}

ErrorOr<RequestPipe> RequestPipe::create()
{
#if defined(AK_OS_WINDOWS)
    int socket_fds[2] {};
    TRY(Core::System::socketpair(AF_LOCAL, SOCK_STREAM, 0, socket_fds));
    int option = 1;
    TRY(Core::System::ioctl(socket_fds[0], FIONBIO, option));
    TRY(Core::System::ioctl(socket_fds[1], FIONBIO, option));
    return RequestPipe(socket_fds[0], socket_fds[1]);
#else
    auto fds = TRY(Core::System::pipe2(O_NONBLOCK));
    return RequestPipe(fds[0], fds[1]);
#endif
}

ErrorOr<ssize_t> RequestPipe::write(ReadonlyBytes bytes)
{
#if defined(AK_OS_WINDOWS)
    auto sent = ::send(m_writer_fd, reinterpret_cast<char const*>(bytes.data()), bytes.size(), 0);
    if (sent == SOCKET_ERROR)
        return Error::from_windows_error();
    return sent;
#else
    return Core::System::write(m_writer_fd, bytes);
#endif
}

}