DirMonInvoker.h

Go to the documentation of this file.

#ifndef DIRMONINVOKER_H_W09J90F0
#define DIRMONINVOKER_H_W09J90F0

#include "uscxml/config.h"
#include "uscxml/plugins/InvokerImpl.h"
#include "uscxml/interpreter/Logging.h"

#include <map>
#include <set>
#include <thread>
#include <mutex>
#include <sys/stat.h>

#ifdef BUILD_AS_PLUGINS
#include "uscxml/plugins/Plugins.h"
#endif

namespace uscxml {

class DirectoryWatchMonitor;

class DirectoryWatch {
public:
    enum Action {
        ADDED = 1,
        MODIFIED = 2,
        DELETED = 4,
        EXISTING = 8
    };

    DirectoryWatch(const std::string& dir, bool recurse = false) : _dir(dir), _recurse(recurse), _lastChecked(0) {}
    ~DirectoryWatch();

    void addMonitor(DirectoryWatchMonitor* monitor) {
        _monitors.insert(monitor);
    }
    void removeMonitor(DirectoryWatchMonitor* monitor) {
        _monitors.erase(monitor);
    }
    void updateEntries(bool reportAsExisting = false);
    void reportAsDeleted();

    std::map<std::string, struct stat> getAllEntries() {
        std::map<std::string, struct stat> entries;
        entries.insert(_knownEntries.begin(), _knownEntries.end());

        std::map<std::string, DirectoryWatch*>::iterator dirIter = _knownDirs.begin();
        while(dirIter != _knownDirs.end()) {
            std::map<std::string, struct stat> dirEntries = dirIter->second->getAllEntries();
            std::map<std::string, struct stat>::iterator dirEntryIter = dirEntries.begin();
            while(dirEntryIter != dirEntries.end()) {
                entries[dirIter->first + '/' + dirEntryIter->first] = dirEntryIter->second;
                dirEntryIter++;
            }
            dirIter++;
        }

        return entries;
    }

    void setLogger(Logger logger) {
        _logger = logger;
    }

protected:
    DirectoryWatch(const std::string& dir, const std::string& relDir) : _dir(dir), _relDir(relDir), _recurse(true), _lastChecked(0) {}

    std::string _dir;
    std::string _relDir;
    Logger _logger;

    bool _recurse;
    std::map<std::string, struct stat> _knownEntries;
    std::map<std::string, DirectoryWatch*> _knownDirs;
    std::set<DirectoryWatchMonitor*> _monitors;
    typedef std::set<DirectoryWatchMonitor*> _monitors_t;
    time_t _lastChecked;
};

class DirectoryWatchMonitor {
public:
    virtual void handleChanges(DirectoryWatch::Action action, const std::string dir, const std::string file, struct stat fileStat) = 0;
};

class DirMonInvoker : public InvokerImpl, public DirectoryWatchMonitor {
public:
    DirMonInvoker();
    virtual ~DirMonInvoker();
    virtual std::shared_ptr<InvokerImpl> create(InvokerCallbacks* callbacks);

    virtual std::list<std::string> getNames() {
        std::list<std::string> names;
        names.push_back("dirmon");
        names.push_back("DirectoryMonitor");
        names.push_back("http://uscxml.tk.informatik.tu-darmstadt.de/#dirmon");
        return names;
    }

    virtual Data getDataModelVariables();
    virtual void eventFromSCXML(const Event& event);
    virtual void invoke(const std::string& source, const Event& invokeEvent);
    virtual void uninvoke();

    virtual void handleChanges(DirectoryWatch::Action action, const std::string dir, const std::string file, struct stat fileStat);

    static void run(void* instance);

protected:
    bool _reportExisting;
    bool _reportHidden;
    bool _recurse;

    std::string _dir;
    std::set<std::string> _suffixes;

    bool _isRunning;
    std::thread* _thread;
    std::recursive_mutex _mutex;

    DirectoryWatch* _watcher;
};

#ifdef BUILD_AS_PLUGINS
PLUMA_INHERIT_PROVIDER(DirMonInvoker, InvokerImpl)
#endif

}


#endif /* end of include guard: DIRMONINVOKER_H_W09J90F0 */