/*
 * vdrmon.c: A plugin for the Video Disk Recorder
 *
 * See the README file for copyright information and how to reach the author.
 *
 * $Id$
 */

#include <vdr/plugin.h>
#include <vdr/thread.h>
#include <vdr/status.h>
#include <vdr/device.h>
#include <vdr/player.h>
#include "vdrmanagerthread.h"
#include "compressor.h"

#define VDRMANAGER_PORT		    6420
#define VDRMANAGER_SSL_PORT   6421
#define VDRMANAGER_CRT_FILE   "/etc/vdr/plugins/vdrmanager/vdrmanager.pem"
#define VDRMANAGER_KEY_FILE   "/etc/vdr/plugins/vdrmanager/vdrmanager.pem"

static const char *VERSION = "0.11";
static const char *DESCRIPTION = "VDR-Manager support plugin";

class cVdrManager: public cPlugin {
private:
	// Add any member variables or functions you may need here.
	cVdrManagerThread * Thread;
	int port;
	int sslport;
	const char * password;
	bool forceCheckSvdrp;
	int compressionMode;
	const char * certFile;
	const char * keyFile;
protected:
public:
	cVdrManager(void);
	virtual ~cVdrManager();
	virtual const char *Version(void) {
		return VERSION;
	}
	virtual const char *Description(void) {
		return DESCRIPTION;
	}
	virtual const char *CommandLineHelp(void);
	virtual bool Initialize(void);
	virtual bool Start(void);
	virtual void Stop(void);
	virtual void Housekeeping(void);
	virtual const char *MainMenuEntry(void) {
		return NULL;
	}
	virtual cOsdObject *MainMenuAction(void);
	virtual cMenuSetupPage *SetupMenu(void);
	virtual bool ProcessArgs(int argc, char *argv[]);
};

cVdrManager::cVdrManager(void) {
	// Initialize any member variables here.
	// DON'T DO ANYTHING ELSE THAT MAY HAVE SIDE EFFECTS, REQUIRE GLOBAL
	// VDR OBJECTS TO EXIST OR PRODUCE ANY OUTPUT!
	Thread = NULL;
	port = VDRMANAGER_PORT;
	sslport = VDRMANAGER_SSL_PORT;
	password = "";
	forceCheckSvdrp = false;
	compressionMode = COMPRESSION_NONE;
	certFile = VDRMANAGER_CRT_FILE;
	keyFile = VDRMANAGER_KEY_FILE;
}

cVdrManager::~cVdrManager() {
	// Clean up after yourself!
}

cOsdObject * cVdrManager::MainMenuAction(void) {
	return NULL;
}

cMenuSetupPage * cVdrManager::SetupMenu(void) {
	return NULL;
}

const char * cVdrManager::CommandLineHelp(void) {
	return
	    "  -p port[,sslport]     port number to listen to (sslport = port+1 if not given\n"
			"  -P password           password (none if not given). No password forces check against svdrphosts.conf.\n"
			"  -s                    force check against svdrphosts.conf, even if a password was given\n"
			"  -c compression        selects the compression mode to use (zlib or gzip). Default is zlib\n"
	    "  -k certfile[,keyfile] cert and key file for SSL (or one file for both)";
}

bool cVdrManager::ProcessArgs(int argc, char *argv[]) {
	int c;
	while ((c = getopt(argc, argv, "c::p:P:st:k:")) != -1)
		switch (c) {
		case 'p':
			port = atoi(optarg);
			{
			  const char * sep = strchr(optarg, ',');
			  if (sep)
			    sslport = atoi(sep+1);
			  else
			    sslport = port+1;
			}
			break;
		case 'P':
			password = optarg;
			break;
		case 's':
			forceCheckSvdrp = true;
			break;
		case 'c':
			if (!optarg) {
				compressionMode = COMPRESSION_ZLIB;
			} else if (optarg[0] == 'g') {
				compressionMode = COMPRESSION_GZIP;
			} else if (optarg[0] == 'z') {
				compressionMode = COMPRESSION_ZLIB;
			} else {
				return false;
			}
			break;
		case 'k':
		  {
		    const char * sep = strchr(optarg, ',');
		    if (sep == NULL) {
		      certFile = keyFile = optarg;
		    } else {
		      certFile = strndup(optarg, sep-optarg);
		      keyFile = sep;
		    }
		  }
		  break;
		case '?':
			return false;
		default:
			return false;
		}

// default port
	if (port <= 0)
		port = VDRMANAGER_PORT;
	if (sslport <= 0)
	  sslport = port+1;

	return true;
}

bool cVdrManager::Initialize(void) {
// Initialize any background activities the plugin shall perform.

// Start any background activities the plugin shall perform.
	Thread = new cVdrManagerThread(port, sslport, password, forceCheckSvdrp,
			                           compressionMode, certFile, keyFile);

	return Thread != NULL;
}

bool cVdrManager::Start(void) {
	Thread->Start();

	return true;
}

void cVdrManager::Stop(void) {
// Stop any background activities the plugin shall perform.
	Thread->Shutdown();
}

void cVdrManager::Housekeeping(void) {
// Perform any cleanup or other regular tasks.
}

VDRPLUGINCREATOR(cVdrManager);
// Don't touch this!