jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2020, the SerenityOS developers.
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7#include "ConnectionCache.h"
8#include <LibCore/EventLoop.h>
9#include <LibGemini/GeminiResponse.h>
10#include <LibGemini/Job.h>
11#include <RequestServer/GeminiRequest.h>
12
13namespace RequestServer {
14
15GeminiRequest::GeminiRequest(ConnectionFromClient& client, NonnullRefPtr<Gemini::Job> job, NonnullOwnPtr<Core::File>&& output_stream)
16 : Request(client, move(output_stream))
17 , m_job(move(job))
18{
19 m_job->on_finish = [this](bool success) {
20 Core::deferred_invoke([url = m_job->url(), socket = m_job->socket()] {
21 ConnectionCache::request_did_finish(url, socket);
22 });
23 if (auto* response = m_job->response()) {
24 set_downloaded_size(MUST(m_job->response_length()));
25 if (!response->meta().is_empty()) {
26 HashMap<DeprecatedString, DeprecatedString, CaseInsensitiveStringTraits> headers;
27 headers.set("meta", response->meta());
28 // Note: We're setting content-type to meta only on status==SUCCESS
29 // we should perhaps have a better mechanism for this, since we
30 // are already shoehorning the concept of "headers" here
31 if (response->status() >= 20 && response->status() < 30) {
32 headers.set("content-type", response->meta());
33 }
34 set_response_headers(headers);
35 }
36 }
37
38 // signal 100% request progress so any listeners can react
39 // appropriately
40 did_progress(downloaded_size(), downloaded_size());
41
42 did_finish(success);
43 };
44 m_job->on_progress = [this](Optional<u32> total, u32 current) {
45 did_progress(move(total), current);
46 };
47}
48
49void GeminiRequest::set_certificate(DeprecatedString, DeprecatedString)
50{
51}
52
53GeminiRequest::~GeminiRequest()
54{
55 m_job->on_finish = nullptr;
56 m_job->on_progress = nullptr;
57 m_job->cancel();
58}
59
60NonnullOwnPtr<GeminiRequest> GeminiRequest::create_with_job(Badge<GeminiProtocol>, ConnectionFromClient& client, NonnullRefPtr<Gemini::Job> job, NonnullOwnPtr<Core::File>&& output_stream)
61{
62 return adopt_own(*new GeminiRequest(client, move(job), move(output_stream)));
63}
64
65}