instream.hh

/*
 * Copyright (C) 2011 - 2022 Red Hat, Inc.
 *
 * This file is part of csdiff.
 *
 * csdiff 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, either version 3 of the License, or
 * any later version.
 *
 * csdiff 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 csdiff.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef H_GUARD_INSTREAM_H
#define H_GUARD_INSTREAM_H

#include <fstream>
#include <iostream>
#include <sstream>
#include <string>
#include <vector>

struct InFileException {
    std::string fileName;
    // TODO: details (errno?)

    InFileException(const std::string &fileName_):
        fileName(fileName_)
    {
    }
};

class InStream {
    public:
        InStream(const std::string &fileName, bool silent = false,
                 bool recordInputLocations = false);
        InStream(std::istringstream &str, bool silent = false);
        ~InStream() = default;

        const std::string& fileName()   const { return fileName_;       }
        std::istream& str()             const { return str_;            }
        bool silent()                   const { return silent_;         }
        bool recordInputLocations()     const { return recordInputLocations_; }
        bool anyError()                 const { return anyError_;       }

        void handleError(const std::string &msg = "", unsigned long line = 0UL);

    private:
        const std::string   fileName_;
        const bool          silent_;
        const bool          recordInputLocations_ = false;
        bool                anyError_ = false;
        std::ifstream       fileStr_;
        std::istream       &str_;
};

class InStreamLookAhead {
    public:
        InStreamLookAhead(
                InStream       &input,
                unsigned        size,
                bool            skipBOM,
                bool            skipWhiteSpaces);

        char operator[](const unsigned idx) const {
            return buf_.at(idx);
        }

    private:
        std::vector<char> buf_;
};

#endif /* H_GUARD_INSTREAM_H */