csv_file.h

Go to the documentation of this file.

// Copyright (C) 2014-2020 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#ifndef CSV_FILE_H
#define CSV_FILE_H

#include <exceptions/exceptions.h>
#include <boost/lexical_cast.hpp>
#include <boost/shared_ptr.hpp>
#include <fstream>
#include <ostream>
#include <string>
#include <vector>

namespace isc {
namespace util {

class CSVFileError : public Exception {
public:
    CSVFileError(const char* file, size_t line, const char* what) :
        isc::Exception(file, line, what) { };
};

class CSVRow {
public:

    CSVRow(const size_t cols = 0, const char separator = ',');

    CSVRow(const std::string& text, const char separator = ',');

    size_t getValuesCount() const {
        return (values_.size());
    }

    void parse(const std::string& line);

    std::string readAt(const size_t at) const;

    std::string readAtEscaped(const size_t at) const;

    void trim(const size_t count);

    template<typename T>
    T readAndConvertAt(const size_t at) const {
        T cast_value;
        try {
            cast_value = boost::lexical_cast<T>(readAt(at).c_str());

        } catch (const boost::bad_lexical_cast& ex) {
            isc_throw(CSVFileError, ex.what());
        }
        return (cast_value);
    }

    std::string render() const;

    void writeAt(const size_t at, const char* value);

    void writeAt(const size_t at, const std::string& value) {
        writeAt(at, value.c_str());
    }

    void writeAtEscaped(const size_t at, const std::string& value);

    template<typename T>
    void append(const T value) {
        try {
            values_.push_back(boost::lexical_cast<std::string>(value));
        } catch (const boost::bad_lexical_cast& ex) {
            isc_throw(CSVFileError, "unable to stringify the value to be "
                      "appended to the CSV file row.");
        }
    }

    template<typename T>
    void writeAt(const size_t at, const T value) {
        checkIndex(at);
        try {
            values_[at] = boost::lexical_cast<std::string>(value);
        } catch (const boost::bad_lexical_cast& ex) {
            isc_throw(CSVFileError, "unable to stringify the value to be"
                      " written in the CSV file row at position '"
                      << at << "'");
        }
    }

    bool operator==(const CSVRow& other) const {
        return (render() == other.render());
    }

    bool operator!=(const CSVRow& other) const {
        return (render() != other.render());
    }

    static std::string escapeCharacters(const std::string& orig_str,
                                        const std::string& characters);

    static std::string unescapeCharacters(const std::string& escaped_str);

private:

    void checkIndex(const size_t at) const;

    std::string separator_;

    std::vector<std::string> values_;

    static const std::string escape_tag;
};

std::ostream& operator<<(std::ostream& os, const CSVRow& row);

class CSVFile {
public:

    CSVFile(const std::string& filename);

    virtual ~CSVFile();

    void addColumn(const std::string& col_name);

    void append(const CSVRow& row) const;

    void close();

    bool exists() const;

    void flush() const;

    size_t getColumnCount() const {
        return (cols_.size());
    }

    std::string getFilename() const {
        return (filename_);
    }

    std::string getReadMsg() const {
        return (read_msg_);
    }

    size_t getColumnIndex(const std::string& col_name) const;

    std::string getColumnName(const size_t col_index) const;

    bool next(CSVRow& row, const bool skip_validation = false);


    virtual void open(const bool seek_to_end = false);

    virtual void recreate();

    void setReadMsg(const std::string& read_msg) {
        read_msg_ = read_msg;
    }

    static CSVRow EMPTY_ROW() {
        static CSVRow row(0);
        return (row);
    }

protected:

    void addColumnInternal(const std::string& col_name);

    virtual bool validate(const CSVRow& row);

protected:

    virtual bool validateHeader(const CSVRow& header);

private:
    void checkStreamStatusAndReset(const std::string& operation) const;

    std::streampos size() const;

    std::string filename_;

    boost::shared_ptr<std::fstream> fs_;

    std::vector<std::string> cols_;

    std::string read_msg_;
};

} // namespace isc::util
} // namespace isc

#endif // CSV_FILE_H