tools.h

#include <glob.h>
#include <list>
#include <unistd.h>
#include <string>
#include <sstream>
#include <vector>
#include <exception>
#include <iostream>
#include <fstream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <pthread.h>
#include <poll.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/inotify.h>
#include <cxxtools/regex.h>
#include <cxxtools/string.h>
#include <cxxtools/utf8codec.h>
#include <cxxtools/http/request.h>
#include <cxxtools/http/reply.h>
#include <cxxtools/query_params.h>
#include <cxxtools/serializationinfo.h> // only AdditionalMedia
#include <vdr/channels.h>
#include <vdr/timers.h>
#include <vdr/recording.h>
#include <vdr/plugin.h>
#include "utf8_checked.h"
#include "jsonparser.h"

#define LOWINT 2147483647;

//some defines for inotify memory allocation
#define EVENT_SIZE ( sizeof (struct inotify_event) )
#define BUF_LEN     ( 1024 * ( EVENT_SIZE + 16 ) )

#ifndef RESTFULAPI_EXTENSIONS
#define RESTFULAPI_EXTENSIONS

class Settings
{
  private:
    int port;
    std::string ip;
    std::string epgimage_dir;
    std::string channellogo_dir;
    bool activateHeaders;
  public:
    Settings() { initDefault(); }
    ~Settings() { };
    static Settings* get();
    void initDefault();
    int Port() { return port; }
    std::string Ip() { return ip; }
    std::string EpgImageDirectory() { return epgimage_dir; }
    std::string ChannelLogoDirectory() { return channellogo_dir; }
    bool Headers() { return activateHeaders; }
    bool SetPort(std::string v);
    bool SetIp(std::string v);
    bool SetEpgImageDirectory(std::string v);
    bool SetChannelLogoDirectory(std::string v);
    bool SetHeaders(std::string v);
};

class StreamExtension
{
  private:
    std::ostream *_out;
  public:
    StreamExtension(std::ostream *out);
    ~StreamExtension() { };
    std::ostream* getBasicStream();
    void write(const char* str);
    void write(cString& str) { write((const char*)str); }
    void writeHtmlHeader(std::string title);
    void writeXmlHeader();
    bool writeBinary(std::string path);
};

class HtmlHeader
{
  private:
    std::string _title;
    std::string _onload;
    std::vector< std::string > _stylesheets;
    std::vector< std::string > _scripts;
    std::vector< std::string > _metatags;
  public:
    HtmlHeader() { }
    ~HtmlHeader() { }
    void Title(std::string title) { _title = title; }
    std::string Title() { return _title; }

    void OnLoad(std::string onload) { _onload = onload; }
    std::string OnLoad() { return _onload; }

    void Stylesheet(std::string stylesheet) { _stylesheets.push_back(stylesheet); }
    std::vector< std::string >& Stylesheet() { return _stylesheets; }

    void Script(std::string script) { _scripts.push_back(script); }
    std::vector< std::string >& Scripts() { return _scripts; }

    void MetaTag(std::string metatag) { _metatags.push_back(metatag); }
    std::vector< std::string >& MataTags() { return _metatags; }

    void ToStream(StreamExtension* se);
};

class FileNotifier : public cThread
{
  private:
    int _filedescriptor;
    int _wd;
    int _mode;
    bool active;
    void Action(void);
  public:
    static const int EVENTS = 0x01;
    static const int CHANNELS = 0x02;
    FileNotifier() { active = false; };
    ~FileNotifier();
    void Initialize(int mode);
    void Stop();
    bool isActive() { return active; }
};

class FileCaches
{
  private:
    std::vector< std::string > eventImages;
    std::vector< std::string > channelLogos;
    FileNotifier notifierEvents;
    FileNotifier notifierChannels;
  public:
    FileCaches() {
         cacheEventImages();
         cacheChannelLogos();
         notifierEvents.Initialize(FileNotifier::EVENTS);
         notifierChannels.Initialize(FileNotifier::CHANNELS);
      };
    ~FileCaches() { };
    static FileCaches* get();
    void cacheEventImages();
    void cacheChannelLogos();
    void searchEventImages(int eventid, std::vector< std::string >& files);
    std::string searchChannelLogo(cChannel *channel);
    void addEventImage(std::string file);
    void addChannelLogo(std::string file);
    void removeEventImage(std::string file);
    void removeChannelLogo(std::string file);
    void stopNotifier() {
      notifierEvents.Stop();
      notifierChannels.Stop();
    };
};

class VdrExtension
{
  private:
    static bool MoveDirectory(std::string const & sourceDir, std::string const & targetDir, bool copy = false);
  public:
    static cChannel* getChannel(int number);
    static cChannel* getChannel(std::string id);
    static cTimer* getTimer(std::string id);
    static std::string getTimerID(cTimer* timer);
    static int scanForFiles(const std::string wildcardpath, std::vector< std::string >& files);
    static bool doesFileExistInFolder(std::string wildcardpath, std::string filename);
    static bool IsRadio(cChannel* channel);
    static bool IsRecording(cRecording* recording);
    static cTimer* TimerExists(cEvent* event);
    static std::vector< cTimer* > SortedTimers();
    static bool CompareTimers(cTimer* timer1, cTimer* timer2);
    static int RecordingLengthInSeconds(cRecording* recording);
    static cEvent* GetEventById(tEventID eventID, cChannel* channel = NULL);
    static std::string getRelativeVideoPath(cRecording* recording);
    static cEvent* getCurrentEventOnChannel(cChannel* channel);
    static std::string getVideoDiskSpace();
    static std::string FileSystemExchangeChars(std::string const & s, bool ToFileSystem);
    static std::string MoveRecording(cRecording const * recording, std::string const & name, bool copy = false);
};

class VdrMarks
{
  private:
    std::string cutComment(std::string str);
    bool validateMark(std::string mark);
    std::string getPath(cRecording* recording);
    bool parseLine(std::vector< std::string >& marks, std::string line);
  public:
    static VdrMarks* get();
    std::vector< std::string > readMarks(cRecording* recording);
    bool saveMarks(cRecording* recording, std::vector< std::string > marks);
    bool deleteMarks(cRecording* recording);
};

class StringExtension
{
  public:
    static std::string itostr(int i);
    static int strtoi(std::string str);
    static std::string replace(std::string const& text, std::string const& substring, std::string const& replacement);
    static std::string encodeToXml(const std::string &str);
    static cxxtools::String encodeToJson(const std::string &str);
    static cxxtools::String UTF8Decode(std::string str);
    static std::string toLowerCase(std::string str);
    static std::string trim(std::string str);
    static std::vector< std::string > split(std::string str, std::string s);
    static std::string timeToString(time_t time);
    static std::string dateToString(time_t time);
    static std::string addZeros(int value, int digits);
};

class QueryHandler
{
  private:
    std::string _url;
    std::string _service;
    std::vector< std::string > _params;
    cxxtools::QueryParams _options;
    cxxtools::QueryParams _body;
    JsonParser jsonParser;
    JsonObject* jsonObject;
    void parseRestParams(std::string params);
    std::string getJsonString(std::string name);
    int         getJsonInt(std::string name);
    bool        getJsonBool(std::string name);
    std::string _format;
  public:
    QueryHandler(std::string service, cxxtools::http::Request& request);
    ~QueryHandler();
    std::string getParamAsString(int level);              //Parameters are part of the url (the rest after you cut away the service path)
    std::string getOptionAsString(std::string name);      //Options are the normal url query parameters after the question mark
    std::string getBodyAsString(std::string name);        //Are variables in the body of the http-request -> for now only html/json are supported, xml is not implemented (!)
    int getParamAsInt(int level);
    int getOptionAsInt(std::string name);
    bool getOptionAsBool(std::string name);
    int getBodyAsInt(std::string name);
    bool getBodyAsBool(std::string name);
    JsonArray* getBodyAsArray(std::string name);
    bool isFormat(std::string format);
    std::string getFormat() { return _format; }
    static void addHeader(cxxtools::http::Reply& reply);
};

class BaseList
{
  protected:
    int iterator;
    int counter;
    int start;
    int limit;
  public:
    BaseList();
    virtual ~BaseList() { };
    virtual void activateLimit(int _start, int _limit);
    virtual bool filtered();
    virtual int Count() { return counter; }
};

class RestfulService
{
  protected:
    cxxtools::Regex* _regex;
    RestfulService* _parent;
    std::string _path;
    bool _internal;
    int _version;
  public:
    RestfulService(std::string path, bool internal = false, int version = 1, RestfulService* parent = NULL);
    ~RestfulService();
    RestfulService* Parent() { return _parent; }
    cxxtools::Regex* Regex() { return _regex; }
    std::string Path() { return _path; }
    bool Internal() { return _internal; }
    int Version() { return _version; }
};

class RestfulServices
{
  protected:
    std::vector< RestfulService* > services;
  public:
    RestfulServices() { };
    ~RestfulServices();
    static RestfulServices* get();
    void appendService(std::string path, bool internal = false, int version = 1, RestfulService* parent = NULL);
    void appendService(RestfulService* service);
    std::vector< RestfulService* > Services(bool internal = false, bool children = false);
};


// AdditionalMedia
cPlugin *GetScraperPlugin(void);

struct SerActor
{
  cxxtools::String Name;
  cxxtools::String Role;
  cxxtools::String Thumb;
};

struct SerImage
{
  cxxtools::String Path;
  int Width;
  int Height;
};

struct SerAdditionalMedia
{
  cxxtools::String Scraper;
  int SeriesId;
  cxxtools::String SeriesName;
  cxxtools::String SeriesOverview;
  cxxtools::String SeriesFirstAired;
  cxxtools::String SeriesNetwork;
  cxxtools::String SeriesGenre;
  float SeriesRating;
  cxxtools::String SeriesStatus;
  int EpisodeId;
  int EpisodeNumber;
  int EpisodeSeason;
  cxxtools::String EpisodeName;
  cxxtools::String EpisodeFirstAired;
  cxxtools::String EpisodeGuestStars;
  cxxtools::String EpisodeOverview;
  float EpisodeRating;
  cxxtools::String EpisodeImage;
  std::vector< struct SerImage > Posters;
  std::vector< struct SerImage > Banners;
  std::vector< struct SerImage > Fanarts;
  int MovieId;
  cxxtools::String MovieTitle;
  cxxtools::String MovieOriginalTitle;
  cxxtools::String MovieTagline;
  cxxtools::String MovieOverview;
  bool MovieAdult;
  cxxtools::String MovieCollectionName;
  int MovieBudget;
  int MovieRevenue;
  cxxtools::String MovieGenres;
  cxxtools::String MovieHomepage;
  cxxtools::String MovieReleaseDate;
  int MovieRuntime;
  float MoviePopularity;
  float MovieVoteAverage;
  cxxtools::String MoviePoster;
  cxxtools::String MovieFanart;
  cxxtools::String MovieCollectionPoster;
  cxxtools::String MovieCollectionFanart;
  std::vector< struct SerActor > Actors;
};

void operator<<= (cxxtools::SerializationInfo& si, const SerActor& a);
void operator<<= (cxxtools::SerializationInfo& si, const SerImage& i);
void operator<<= (cxxtools::SerializationInfo& si, const SerAdditionalMedia& am);

#endif

#ifndef __RESTFUL_BASICOSD_H
#define __RESTFUL_BASICOSD_H

class BasicOsd
{
  public:
    BasicOsd() { };
    virtual ~BasicOsd() { };
    virtual int Type() { return 0x00; };
};

#endif

#ifndef __RESTFUL_TASKS_H
#define __RESTFUL_TASKS_H

class BaseTask
{
  protected:
    int created; //time in s
  public:
    BaseTask() { created = (int)time(NULL); };
    virtual ~BaseTask() { };
    int Created() { return created; };
};

class TaskScheduler
{
  protected:
    std::list<BaseTask*> tasks;
    tChannelID _channel;
    cRecording* _recording;
    cMutex     _channelMutex;
    bool _bRewind;
  public:
    TaskScheduler() { _channel = tChannelID::InvalidID; _recording = NULL; };
    ~TaskScheduler();
    static TaskScheduler* get();
    void AddTask(BaseTask* task) { tasks.push_back(task); };
    void DoTasks();
    void SwitchableChannel(tChannelID channel);
    tChannelID SwitchableChannel();
    void SwitchableRecording(cRecording* recording) { _recording = recording; }
    cRecording* SwitchableRecording() { return _recording; }
    void SetRewind(bool bRewind) { _bRewind = bRewind; }
    bool IsRewind() { return _bRewind; }
};

#endif