diff options
Diffstat (limited to 'archived/projt-launcher/launcher/net/FileSink.cpp')
| -rw-r--r-- | archived/projt-launcher/launcher/net/FileSink.cpp | 181 |
1 files changed, 181 insertions, 0 deletions
diff --git a/archived/projt-launcher/launcher/net/FileSink.cpp b/archived/projt-launcher/launcher/net/FileSink.cpp new file mode 100644 index 0000000000..0d900feed4 --- /dev/null +++ b/archived/projt-launcher/launcher/net/FileSink.cpp @@ -0,0 +1,181 @@ +// SPDX-License-Identifier: GPL-3.0-only +// SPDX-FileCopyrightText: 2026 Project Tick +// SPDX-FileContributor: Project Tick Team +/* + * ProjT Launcher - Minecraft Launcher + * Copyright (C) 2026 Project Tick + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, version 3. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, write to the Free Software Foundation, + * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + * + * === Upstream License Block (Do Not Modify) ============================== + * + * + * + * Prism Launcher - Minecraft Launcher + * Copyright (c) 2022 flowln <flowlnlnln@gmail.com> + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, version 3. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, write to the Free Software Foundation, + * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + * + * This file incorporates work covered by the following copyright and + * permission notice: + * + * Copyright 2013-2021 MultiMC Contributors + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + * + * ======================================================================== */ + +#include "FileSink.h" + +#include "FileSystem.h" + +#include "net/Logging.h" + +namespace Net +{ + + Task::State FileSink::init(QNetworkRequest& request) + { + auto result = initCache(request); + if (result != Task::State::Running) + { + return result; + } + + // create a new save file and open it for writing + if (!FS::ensureFilePathExists(m_filename)) + { + qCCritical(taskNetLogC) << "Could not create folder for " + m_filename; + m_fail_reason = "Could not create folder"; + return Task::State::Failed; + } + + m_wroteAnyData = false; + m_output_file.reset(new PSaveFile(m_filename)); + if (!m_output_file->open(QIODevice::WriteOnly)) + { + qCCritical(taskNetLogC) << "Could not open " + m_filename + " for writing"; + m_fail_reason = "Could not open file"; + return Task::State::Failed; + } + + if (initAllValidators(request)) + return Task::State::Running; + m_fail_reason = "Failed to initialize validators"; + return Task::State::Failed; + } + + Task::State FileSink::write(QByteArray& data) + { + if (!writeAllValidators(data) || m_output_file->write(data) != data.size()) + { + qCCritical(taskNetLogC) << "Failed writing into " + m_filename; + m_output_file->cancelWriting(); + m_output_file.reset(); + m_wroteAnyData = false; + m_fail_reason = "Failed to write validators"; + return Task::State::Failed; + } + + m_wroteAnyData = true; + return Task::State::Running; + } + + Task::State FileSink::abort() + { + if (m_output_file) + { + m_output_file->cancelWriting(); + } + failAllValidators(); + return Task::State::Failed; + } + + Task::State FileSink::finalize(QNetworkReply& reply) + { + bool gotFile = false; + QVariant statusCodeV = reply.attribute(QNetworkRequest::HttpStatusCodeAttribute); + bool validStatus = false; + int statusCode = statusCodeV.toInt(&validStatus); + if (validStatus) + { + // this leaves out 304 Not Modified + gotFile = statusCode == 200 || statusCode == 203; + } + + // if we wrote any data to the save file, we try to commit the data to the real file. + // if it actually got a proper file, we write it even if it was empty + if (gotFile || m_wroteAnyData) + { + // ask validators for data consistency + // we only do this for actual downloads, not 'your data is still the same' cache hits + if (!finalizeAllValidators(reply)) + { + m_fail_reason = "Failed to finalize validators"; + return Task::State::Failed; + } + + // nothing went wrong... + if (!m_output_file->commit()) + { + qCCritical(taskNetLogC) << "Failed to commit changes to " << m_filename; + m_output_file->cancelWriting(); + m_fail_reason = "Failed to commit changes"; + return Task::State::Failed; + } + } + + // then get rid of the save file + m_output_file.reset(); + + return finalizeCache(reply); + } + + Task::State FileSink::initCache(QNetworkRequest&) + { + return Task::State::Running; + } + + Task::State FileSink::finalizeCache(QNetworkReply&) + { + return Task::State::Succeeded; + } + + bool FileSink::hasLocalData() + { + QFileInfo info(m_filename); + return info.exists() && info.size() != 0; + } +} // namespace Net |