File.cpp

Go to the documentation of this file.

 
/*
 * Copyright (C) 2019-2021 Tobias Flaig.
 *
 * This file is part of nawa.
 *
 * nawa is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License,
 * version 3, as published by the Free Software Foundation.
 *
 * nawa 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with nawa.  If not, see <https://www.gnu.org/licenses/>.
 */
 
#include <cstring>
#include <fstream>
#include <nawa/Exception.h>
#include <nawa/request/File.h>
 
using namespace nawa;
using namespace std;
 
struct File::Data {
    string filename;
    string contentType;
    // TODO better solution can be used when shifting fcgi handler to use MimeMultipart
    shared_ptr<char[]> dataPtr;
    size_t size = 0;
 
    Data(shared_ptr<char[]> dataPtr, size_t size) : dataPtr(move(dataPtr)), size(size) {}
};
 
NAWA_DEFAULT_DESTRUCTOR_IMPL(File)
 
NAWA_COPY_CONSTRUCTOR_IMPL(File)
 
NAWA_COPY_ASSIGNMENT_OPERATOR_IMPL(File)
 
NAWA_MOVE_CONSTRUCTOR_IMPL(File)
 
NAWA_MOVE_ASSIGNMENT_OPERATOR_IMPL(File)
 
NAWA_COMPLEX_DATA_ACCESSORS_IMPL(File, filename, string)
 
NAWA_COMPLEX_DATA_ACCESSORS_IMPL(File, contentType, string)
 
File::File(std::shared_ptr<char[]> dataPtr, size_t size) {
    data = make_unique<Data>(move(dataPtr), size);
}
 
File::File(std::string const& data) {
    shared_ptr<char[]> dataPtr(new char[data.size()]);
    memcpy(dataPtr.get(), data.c_str(), data.size());
    this->data = make_unique<Data>(move(dataPtr), data.size());
}
 
size_t File::size() const noexcept {
    return data->size;
}
 
string File::toString() const {
    return string(data->dataPtr.get(), data->size);
}
 
void File::writeToDisk(string const& path) const {
    ofstream outfile;
    ios_base::iostate exceptionMask = outfile.exceptions() | ios::failbit;
    outfile.exceptions(exceptionMask);
    try {
        outfile.open(path, ofstream::out | ofstream::binary);
        outfile.write(data->dataPtr.get(), data->size);
        outfile.close();
    } catch (ios_base::failure const& e) {
        throw Exception(__PRETTY_FUNCTION__, 1, "Could not write file to disk.", e.what());
    }
}