mirror of
https://github.com/SerenityOS/serenity.git
synced 2025-01-24 10:22:05 -05:00
6e19ab2bbc
We have a new, improved string type coming up in AK (OOM aware, no null state), and while it's going to use UTF-8, the name UTF8String is a mouthful - so let's free up the String name by renaming the existing class. Making the old one have an annoying name will hopefully also help with quick adoption :^)
65 lines
2.3 KiB
C++
65 lines
2.3 KiB
C++
/*
|
|
* Copyright (c) 2020, the SerenityOS developers.
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include "ConnectionCache.h"
|
|
#include <LibCore/EventLoop.h>
|
|
#include <LibGemini/GeminiResponse.h>
|
|
#include <LibGemini/Job.h>
|
|
#include <RequestServer/GeminiRequest.h>
|
|
|
|
namespace RequestServer {
|
|
|
|
GeminiRequest::GeminiRequest(ConnectionFromClient& client, NonnullRefPtr<Gemini::Job> job, NonnullOwnPtr<Core::Stream::File>&& output_stream)
|
|
: Request(client, move(output_stream))
|
|
, m_job(move(job))
|
|
{
|
|
m_job->on_finish = [this](bool success) {
|
|
Core::deferred_invoke([url = m_job->url(), socket = m_job->socket()] {
|
|
ConnectionCache::request_did_finish(url, socket);
|
|
});
|
|
if (auto* response = m_job->response()) {
|
|
set_downloaded_size(MUST(const_cast<Core::Stream::File&>(this->output_stream()).size()));
|
|
if (!response->meta().is_empty()) {
|
|
HashMap<DeprecatedString, DeprecatedString, CaseInsensitiveStringTraits> headers;
|
|
headers.set("meta", response->meta());
|
|
// Note: We're setting content-type to meta only on status==SUCCESS
|
|
// we should perhaps have a better mechanism for this, since we
|
|
// are already shoehorning the concept of "headers" here
|
|
if (response->status() >= 20 && response->status() < 30) {
|
|
headers.set("content-type", response->meta());
|
|
}
|
|
set_response_headers(headers);
|
|
}
|
|
}
|
|
|
|
// signal 100% request progress so any listeners can react
|
|
// appropriately
|
|
did_progress(downloaded_size(), downloaded_size());
|
|
|
|
did_finish(success);
|
|
};
|
|
m_job->on_progress = [this](Optional<u32> total, u32 current) {
|
|
did_progress(move(total), current);
|
|
};
|
|
}
|
|
|
|
void GeminiRequest::set_certificate(DeprecatedString, DeprecatedString)
|
|
{
|
|
}
|
|
|
|
GeminiRequest::~GeminiRequest()
|
|
{
|
|
m_job->on_finish = nullptr;
|
|
m_job->on_progress = nullptr;
|
|
m_job->cancel();
|
|
}
|
|
|
|
NonnullOwnPtr<GeminiRequest> GeminiRequest::create_with_job(Badge<GeminiProtocol>, ConnectionFromClient& client, NonnullRefPtr<Gemini::Job> job, NonnullOwnPtr<Core::Stream::File>&& output_stream)
|
|
{
|
|
return adopt_own(*new GeminiRequest(client, move(job), move(output_stream)));
|
|
}
|
|
|
|
}
|