/*
 *   Copyright 2021 Oleg Zharkov
 *
 *   Licensed under the Apache License, Version 2.0 (the "License").
 *   You may not use this file except in compliance with the License.
 *   A copy of the License is located at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   or in the "license" file accompanying this file. This file 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.
 */ 

#ifndef COLLECTOR_H
#define	COLLECTOR_H

#include <string>
#include <fstream>
#include <iostream>
#include <sstream>
#include <unistd.h> // sleep
#include <istream>
#include <ostream>
#include <boost/asio.hpp>

#include "base64.h"
#include "source.h"

using boost::asio::ip::tcp;
using namespace std;
namespace bpt = boost::property_tree;

class Collector : public Source {
public: 
    
    ByteData bd;
    std::stringstream comp;
    
    Collector() {
        ResetStreams();
    }
        
    virtual int GetConfig();
    
    int Open();
    void Close();
    int Go();
    
    void UpdateProbeStatus();
    
    void ResetStreams() {
        comp.str("");
        comp.clear();
    }
};

#endif	/* COLLECTOR_H */

