znc-push/push.cpp

1323 lines
30 KiB
C++
Raw Normal View History

/**
2011-09-29 13:30:06 +00:00
* ZNC Push Module
*
2011-09-29 13:30:06 +00:00
* Allows the user to enter a Push user and API token, and sends
* channel highlights and personal messages to Push.
*
* Copyright (c) 2011 John Reese
* Licensed under the MIT license
*/
#define REQUIRESSL
#include <znc/znc.h>
#include <znc/Chan.h>
#include <znc/User.h>
#include <znc/IRCNetwork.h>
#include <znc/Modules.h>
#include <znc/FileUtils.h>
#include "time.h"
// Forward declaration
class CPushMod;
/**
* Socket class for generating HTTP requests.
*/
class CPushSocket : public CSocket
{
public:
CPushSocket(CModule *p) : CSocket(p)
{
EnableReadLine();
parent = (CPushMod*) p;
first = true;
crlf = "\r\n";
user_agent = "ZNC Push";
}
// Implemented after CPushMod
void Request(bool post, const CString& host, const CString& url, MCString& parameters, const CString& auth="");
virtual void ReadLine(const CString& data);
virtual void Disconnected();
private:
CPushMod *parent;
bool first;
// Too lazy to add CString("\r\n\") everywhere
CString crlf;
// User agent to use
CString user_agent;
/**
* Shorthand for encoding a string for a URL.
*
* @param str String to be encoded
* @return Encoded string
*/
CString urlencode(const CString& str)
{
return str.Escape_n(CString::EASCII, CString::EURL);
}
};
/**
* Push notification module.
*/
2011-09-29 13:30:06 +00:00
class CPushMod : public CModule
{
protected:
// Application name
CString app;
// Time last notification was sent for a given context
map <CString, unsigned int> last_notification_time;
// Time of last message by user to a given context
map <CString, unsigned int> last_reply_time;
// Time of last activity by user for a given context
map <CString, unsigned int> last_active_time;
// Time of last activity by user in any context
unsigned int idle_time;
// User object
CUser *user;
// Configuration options
MCString options;
MCString defaults;
public:
2011-09-29 13:30:06 +00:00
MODCONSTRUCTOR(CPushMod) {
app = "ZNC";
idle_time = time(NULL);
// Current user
user = GetUser();
// Push service information
defaults["service"] = "";
defaults["username"] = "";
defaults["secret"] = "";
// Condition strings
defaults["channel_conditions"] = "all";
defaults["query_conditions"] = "all";
// Notification conditions
2011-01-14 14:45:47 +00:00
defaults["away_only"] = "no";
defaults["client_count_less_than"] = "0";
defaults["highlight"] = "";
defaults["idle"] = "0";
defaults["last_active"] = "180";
defaults["last_notification"] = "300";
defaults["nick_blacklist"] = "";
defaults["replied"] = "yes";
2011-01-14 03:27:26 +00:00
// Notification settings
defaults["message_length"] = "100";
defaults["message_uri"] = "";
defaults["debug"] = "off";
}
2011-09-29 13:30:06 +00:00
virtual ~CPushMod() {}
public:
/**
* Debugging messages. Prints to *push when the debug option is enabled.
*
* @param data Debug message
*/
void PutDebug(const CString& data)
{
if (options["debug"] == "on")
{
PutModule(data);
}
}
protected:
/**
* Performs string expansion on a set of keywords.
* Given an initial string and a dictionary of string replacments,
* iterate over the dictionary, expanding keywords one-by-one.
*
* @param content String contents
* @param replace Dictionary of string replacements
* @return Result of string replacements
*/
CString expand(const CString& content, MCString& replace)
{
CString result = content.c_str();
for(MCString::iterator i = replace.begin(); i != replace.end(); i++)
{
result.Replace(i->first, i->second);
}
return result;
}
/**
* Send a message to the currently-configured Notifo account.
* Requires (and assumes) that the user has already configured their
* username and API secret using the 'set' command.
*
* @param message Message to be sent to the user
* @param title Message title to use
* @param context Channel or nick context
*/
2011-09-29 13:30:06 +00:00
void send_message(const CString& message, const CString& title="New Message", const CString& context="*push", const CNick& nick=CString("*push"))
{
// Set the last notification time
last_notification_time[context] = time(NULL);
2011-01-14 03:46:02 +00:00
// Shorten message if needed
unsigned int message_length = options["message_length"].ToUInt();
CString short_message = message;
if (message_length > 0)
{
short_message = message.Ellipsize(message_length);
}
// Generate an ISO8601 date string
time_t rawtime;
struct tm * timeinfo;
time(&rawtime);
timeinfo = localtime(&rawtime);
char iso8601 [20];
strftime(iso8601, 20, "%Y-%m-%d %H:%M:%S", timeinfo);
// URI string replacements
MCString replace;
replace["{context}"] = context;
replace["{nick}"] = nick.GetNick();
replace["{datetime}"] = CString(iso8601);
replace["{unixtime}"] = CString(time(NULL));
CString uri = expand(options["message_uri"], replace);
// Set up the connection profile
CString service = options["service"];
bool use_post = true;
int use_port = 443;
bool use_ssl = true;
CString service_host;
CString service_url;
CString service_auth;
MCString params;
// Service-specific profiles
if (service == "notifo")
{
if (options["username"] == "" || options["secret"] == "")
{
PutModule("Error: username or secret not set");
return;
}
service_host = "api.notifo.com";
service_url = "/v1/send_notification";
// BASIC auth, base64-encoded username:password
service_auth = options["username"] + CString(":") + options["secret"];
service_auth.Base64Encode();
params["to"] = options["username"];
params["msg"] = short_message;
params["label"] = app;
params["title"] = title;
params["uri"] = uri;
}
else if (service == "boxcar")
{
if (options["username"] == "")
{
PutModule("Error: username not set");
return;
}
CString boxcar_api_key = "puSd2qp2gCDZO7nWkvb9";
CString boxcar_api_secret = "wLQQKSyGybIOkggbiKipefeYGLni9B3FPZabopHp";
service_host = "boxcar.io";
service_url = "/devices/providers/" + boxcar_api_key + "/notifications";
params["email"] = options["username"];
params["notification[from_screen_name]"] = context;
params["notification[message]"] = short_message;
params["notification[source_url]"] = uri;
}
else if (service == "nma")
{
if (options["secret"] == "")
{
PutModule("Error: secret not set");
return;
}
service_host = "www.notifymyandroid.com";
service_url = "/publicapi/notify";
params["apikey"] = options["secret"];
params["application"] = app;
params["event"] = title;
params["description"] = short_message;
params["url"] = uri;
}
else if (service == "prowl")
{
if (options["secret"] == "")
{
PutModule("Error: secret not set");
return;
}
service_host = "api.prowlapp.com";
service_url = "/publicapi/add";
params["apikey"] = options["secret"];
params["application"] = app;
params["event"] = title;
params["description"] = short_message;
params["url"] = uri;
}
else
{
PutModule("Error: service type not selected");
return;
}
// Create the socket connection, write to it, and add it to the queue
CPushSocket *sock = new CPushSocket(this);
sock->Connect(service_host, use_port, use_ssl);
sock->Request(use_post, service_host, service_url, params, service_auth);
AddSocket(sock);
}
/**
* Evaluate a boolean expression using condition values.
* All tokens must be separated by spaces, using "and" and "or" for
* boolean operators, "(" and ")" to enclose sub-expressions, and
* condition option names to evaluate each condition.
*
* @param expression Boolean expression string
* @param context Notification context
* @param nick Sender nick
* @param message Message contents
* @return Result of boolean evaluation
*/
bool eval(const CString& expression, const CString& context=CString(""), const CNick& nick=CNick(""), const CString& message=" ")
{
CString padded = expression.Replace_n("(", " ( ");
padded.Replace(")", " ) ");
VCString tokens;
padded.Split(" ", tokens, false);
PutDebug("Evaluating message: <" + nick.GetNick() + "> " + message);
bool result = eval_tokens(tokens.begin(), tokens.end(), context, nick, message);
return result;
}
#define expr(x, y) else if (token == x) { \
bool result = y; \
dbg += CString(x) + "/" + CString(result ? "true" : "false") + " "; \
value = oper ? value && result : value || result; \
}
/**
* Evaluate a tokenized boolean expression, or sub-expression.
*
* @param pos Token vector iterator current position
* @param end Token vector iterator end position
* @param context Notification context
* @param nick Sender nick
* @param message Message contents
* @return Result of boolean expression
*/
bool eval_tokens(VCString::iterator pos, VCString::iterator end, const CString& context, const CNick& nick, const CString& message)
{
bool oper = true;
bool value = true;
CString dbg = "";
for(; pos != end; pos++)
{
CString token = pos->AsLower();
if (token == "(")
{
// recursively evaluate sub-expressions
bool inner = eval_tokens(++pos, end, context, nick, message);
dbg += "( inner/" + CString(inner ? "true" : "false") + " ) ";
value = oper ? value && inner : value || inner;
// search ahead to the matching parenthesis token
unsigned int parens = 1;
while(pos != end)
{
if (*pos == "(")
{
parens++;
}
else if (*pos == ")")
{
parens--;
}
if (parens == 0)
{
break;
}
pos++;
}
}
else if (token == ")")
{
pos++;
PutDebug(dbg);
return value;
}
else if (token == "and")
{
dbg += "and ";
oper = true;
}
else if (token == "or")
{
dbg += "or ";
oper = false;
}
expr("true", true)
expr("false", false)
expr("away_only", away_only())
expr("client_count_less_than", client_count_less_than())
expr("highlight", highlight(message))
expr("idle", idle())
expr("last_active", last_active(context))
expr("last_notification", last_notification(context))
expr("nick_blacklist", nick_blacklist(nick))
expr("replied", replied(context))
else
{
PutModule("Error: Unexpected token \"" + token + "\"");
}
}
PutDebug(dbg);
return value;
}
#undef expr
2011-01-14 14:45:47 +00:00
protected:
/**
* Check if the away status condition is met.
*
* @return True if away_only is not "yes" or away status is set
*/
bool away_only()
{
CString value = options["away_only"].AsLower();
return value != "yes" || GetNetwork()->IsIRCAway();
2011-01-14 14:45:47 +00:00
}
/**
* Check how many clients are connected to ZNC.
*
* @return Number of connected clients
*/
unsigned int client_count()
{
return user->GetUserClients().size();
}
/**
* Check if the client_count condition is met.
*
* @return True if client_count is less than client_count_less_than or if client_count_less_than is zero
*/
bool client_count_less_than()
{
unsigned int value = options["client_count_less_than"].ToUInt();
return value == 0 || client_count() < value;
}
/**
* Determine if the given message matches any highlight rules.
*
* @param message Message contents
* @return True if message matches a highlight
*/
bool highlight(const CString& message)
{
CString msg = " " + message.AsLower() + " ";
VCString values;
options["highlight"].Split(" ", values, false);
for (VCString::iterator i = values.begin(); i != values.end(); i++)
{
CString value = i->AsLower();
char prefix = value[0];
2011-09-29 13:30:06 +00:00
bool push = true;
if (prefix == '-')
{
2011-09-29 13:30:06 +00:00
push = false;
value.LeftChomp(1);
}
else if (prefix == '_')
{
value = " " + value.LeftChomp_n(1) + " ";
}
value = "*" + value + "*";
if (msg.WildCmp(value))
{
2011-09-29 13:30:06 +00:00
return push;
}
}
CNick nick = user->GetNick();
if (message.find(nick.GetNick()) != string::npos)
{
return true;
}
return false;
}
/**
* Check if the idle condition is met.
*
* @return True if idle is zero or elapsed time is greater than idle
*/
bool idle()
{
unsigned int value = options["idle"].ToUInt();
unsigned int now = time(NULL);
return value == 0
|| idle_time + value < now;
}
/**
* Check if the last_active condition is met.
*
* @param context Channel or nick context
* @return True if last_active is zero or elapsed time is greater than last_active
*/
bool last_active(const CString& context)
{
unsigned int value = options["last_active"].ToUInt();
unsigned int now = time(NULL);
return value == 0
|| last_active_time.count(context) < 1
|| last_active_time[context] + value < now;
}
/**
* Check if the last_notification condition is met.
*
* @param context Channel or nick context
* @return True if last_notification is zero or elapsed time is greater than last_nofication
*/
bool last_notification(const CString& context)
{
unsigned int value = options["last_notification"].ToUInt();
unsigned int now = time(NULL);
return value == 0
|| last_notification_time.count(context) < 1
|| last_notification_time[context] + value < now;
}
/**
* Check if the nick_blacklist condition is met.
*
* @param nick Nick that sent the message
* @return True if nick is not in the blacklist
*/
bool nick_blacklist(const CNick& nick)
{
VCString blacklist;
options["nick_blacklist"].Split(" ", blacklist, false);
CString name = nick.GetNick().AsLower();
for (VCString::iterator i = blacklist.begin(); i != blacklist.end(); i++)
{
if (name.WildCmp(i->AsLower()))
{
return false;
}
}
return true;
}
/**
* Check if the replied condition is met.
*
* @param context Channel or nick context
* @return True if last_reply_time > last_notification_time or if replied is not "yes"
*/
bool replied(const CString& context)
{
CString value = options["replied"].AsLower();
return value != "yes"
|| last_notification_time[context] == 0
|| last_notification_time[context] < last_reply_time[context];
}
/**
* Determine when to notify the user of a channel message.
*
* @param nick Nick that sent the message
* @param channel Channel the message was sent to
* @param message Message contents
* @return Notification should be sent
*/
bool notify_channel(const CNick& nick, const CChan& channel, const CString& message)
{
CString context = channel.GetName();
CString expression = options["channel_conditions"].AsLower();
if (expression != "all")
{
return eval(expression, context, nick, message);
}
2011-01-14 14:48:56 +00:00
return away_only()
&& client_count_less_than()
&& highlight(message)
&& idle()
&& last_active(context)
&& last_notification(context)
&& nick_blacklist(nick)
&& replied(context)
&& true;
}
2011-01-13 22:40:41 +00:00
/**
* Determine when to notify the user of a private message.
2011-01-13 22:40:41 +00:00
*
* @param nick Nick that sent the message
* @return Notification should be sent
2011-01-13 22:40:41 +00:00
*/
bool notify_pm(const CNick& nick, const CString& message)
2011-01-13 22:40:41 +00:00
{
CString context = nick.GetNick();
CString expression = options["query_conditions"].AsLower();
if (expression != "all")
{
return eval(expression, context, nick, message);
}
return away_only()
&& client_count_less_than()
&& idle()
&& last_active(context)
&& last_notification(context)
&& nick_blacklist(nick)
&& replied(context)
&& true;
2011-01-13 22:40:41 +00:00
}
protected:
/**
* Handle the plugin being loaded. Retrieve plugin config values.
*
* @param args Plugin arguments
* @param message Message to show the user after loading
*/
bool OnLoad(const CString& args, CString& message)
{
for (MCString::iterator i = defaults.begin(); i != defaults.end(); i++)
{
CString value = GetNV(i->first);
if (value != "")
{
options[i->first] = value;
}
else
{
options[i->first] = defaults[i->first];
}
}
return true;
}
/**
* Handle channel messages.
*
* @param nick Nick that sent the message
* @param channel Channel the message was sent to
* @param message Message contents
*/
EModRet OnChanMsg(CNick& nick, CChan& channel, CString& message)
{
if (notify_channel(nick, channel, message))
{
CString title = "Highlight";
CString msg = channel.GetName();
msg += ": <" + nick.GetNick();
msg += "> " + message;
send_message(msg, title, channel.GetName());
}
return CONTINUE;
}
/**
* Handle channel actions.
*
* @param nick Nick that sent the action
* @param channel Channel the message was sent to
* @param message Message contents
*/
EModRet OnChanAction(CNick& nick, CChan& channel, CString& message)
{
if (notify_channel(nick, channel, message))
{
CString title = "Highlight";
CString msg = channel.GetName();
msg += ": " + nick.GetNick();
msg += " " + message;
send_message(msg, title, channel.GetName());
}
return CONTINUE;
}
2011-01-13 22:40:41 +00:00
/**
* Handle a private message.
2011-01-13 22:40:41 +00:00
*
* @param nick Nick that sent the message
* @param message Message contents
*/
EModRet OnPrivMsg(CNick& nick, CString& message)
{
if (notify_pm(nick, message))
{
CString title = "Private Message";
CString msg = "From " + nick.GetNick();
msg += ": " + message;
send_message(msg, title, nick.GetNick());
}
return CONTINUE;
}
/**
* Handle a private action.
*
* @param nick Nick that sent the action
* @param message Message contents
*/
EModRet OnPrivAction(CNick& nick, CString& message)
{
if (notify_pm(nick, message))
{
CString title = "Private Message";
CString msg = "* " + nick.GetNick();
msg += " " + message;
send_message(msg, title, nick.GetNick());
}
2011-01-13 22:40:41 +00:00
return CONTINUE;
}
/**
* Handle a message sent by the user.
*
* @param target Target channel or nick
* @param message Message contents
*/
EModRet OnUserMsg(CString& target, CString& message)
{
last_reply_time[target] = last_active_time[target] = idle_time = time(NULL);
return CONTINUE;
}
/**
* Handle an action sent by the user.
*
* @param target Target channel or nick
* @param message Message contents
*/
EModRet OnUserAction(CString& target, CString& message)
{
last_reply_time[target] = last_active_time[target] = idle_time = time(NULL);
return CONTINUE;
}
/**
* Handle the user joining a channel.
*
* @param channel Channel name
* @param key Channel key
*/
EModRet OnUserJoin(CString& channel, CString& key)
{
idle_time = time(NULL);
return CONTINUE;
}
/**
* Handle the user parting a channel.
*
* @param channel Channel name
* @param message Part message
*/
EModRet OnUserPart(CString& channel, CString& message)
{
idle_time = time(NULL);
return CONTINUE;
}
/**
* Handle the user setting the channel topic.
*
* @param channel Channel name
* @param topic Topic message
*/
EModRet OnUserTopic(CString& channel, CString& topic)
{
idle_time = time(NULL);
return CONTINUE;
}
/**
* Handle the user requesting the channel topic.
*
* @param channel Channel name
*/
EModRet OnUserTopicRequest(CString& channel)
{
idle_time = time(NULL);
return CONTINUE;
}
/**
2011-09-29 13:30:06 +00:00
* Handle direct commands to the *push virtual user.
*
* @param command Command string
*/
void OnModCommand(const CString& command)
{
VCString tokens;
int token_count = command.Split(" ", tokens, false);
if (token_count < 1)
{
return;
}
CString action = tokens[0].AsLower();
// SET command
if (action == "set")
{
if (token_count < 3)
{
PutModule("Usage: set <option> <value>");
return;
}
CString option = tokens[1].AsLower();
CString value = command.Token(2, true, " ");
MCString::iterator pos = options.find(option);
if (pos == options.end())
{
PutModule("Error: invalid option name");
}
else
{
value.Trim();
if (option == "channel_conditions" || option == "query_conditions")
{
if (value != "all")
{
eval(value);
}
}
else if (option == "service")
{
value.MakeLower();
if (value == "notifo")
{
PutModule("Note: Notifo requires setting both 'username' and 'secret' options");
}
else if (value == "boxcar")
{
PutModule("Note: Boxcar requires setting the 'username' option");
}
else if (value == "nma")
{
PutModule("Note: NMA requires setting the 'secret' option");
}
else if (value == "prowl")
{
PutModule("Note: Prowl requires setting the 'secret' option");
}
else
{
PutModule("Error: unknown service name");
return;
}
}
options[option] = value;
SetNV(option, options[option]);
}
}
// APPEND command
else if (action == "append")
{
if (token_count < 3)
{
PutModule("Usage: append <option> <value>");
return;
}
CString option = tokens[1].AsLower();
CString value = command.Token(2, true, " ");
MCString::iterator pos = options.find(option);
if (pos == options.end())
{
PutModule("Error: invalid option name");
}
else if (option == "service")
{
PutModule("Error: cannot append to this option");
}
else
{
options[option] += " " + value;
options[option].Trim();
SetNV(option, options[option]);
}
}
// PREPEND command
else if (action == "prepend")
{
if (token_count < 3)
{
PutModule("Usage: prepend <option> <value>");
return;
}
CString option = tokens[1].AsLower();
CString value = command.Token(2, true, " ");
MCString::iterator pos = options.find(option);
if (pos == options.end())
{
PutModule("Error: invalid option name");
}
else if (option == "service")
{
PutModule("Error: cannot prepend to this option");
}
else
{
options[option] = value + " " + options[option];
options[option].Trim();
SetNV(option, options[option]);
}
}
// UNSET command
else if (action == "unset")
{
if (token_count != 2)
{
PutModule("Usage: unset <option>");
return;
}
CString option = tokens[1].AsLower();
MCString::iterator pos = options.find(option);
if (pos == options.end())
{
PutModule("Error: invalid option name");
}
else
{
options[option] = defaults[option];
DelNV(option);
}
}
// GET command
else if (action == "get")
{
if (token_count > 2)
{
PutModule("Usage: get [<option>]");
return;
}
if (token_count < 2)
{
CTable table;
table.AddColumn("Option");
table.AddColumn("Value");
for (MCString::iterator i = options.begin(); i != options.end(); i++)
{
table.AddRow();
table.SetCell("Option", i->first);
table.SetCell("Value", i->second);
}
PutModule(table);
return;
}
CString option = tokens[1].AsLower();
MCString::iterator pos = options.find(option);
if (pos == options.end())
{
PutModule("Error: invalid option name");
}
else
{
PutModule(option + CString(": \"") + options[option] + CString("\""));
}
}
// SAVE command
else if (action == "save")
{
if (token_count < 2)
{
PutModule("Usage: save <filepath>");
}
CString file_path = command.Token(1, true, " ");
int status = options.WriteToDisk(file_path);
if (status == MCString::MCS_SUCCESS)
{
PutModule("Options saved to " + file_path);
}
else
{
switch (status)
{
case MCString::MCS_EOPEN:
case MCString::MCS_EWRITE:
case MCString::MCS_EWRITEFIL:
PutModule("Failed to save options to " + file_path);
break;
default:
PutModule("Failure");
break;
}
}
}
// LOAD command
else if (action == "load")
{
if (token_count < 2)
{
PutModule("Usage: load <filename>");
}
CString file_path = command.Token(1, true, " ");
2011-09-29 13:30:06 +00:00
if (!CFile::Exists(file_path))
{
PutModule("File does not exist: " + file_path);
return;
}
int status = options.ReadFromDisk(file_path);
if (status == MCString::MCS_SUCCESS)
{
PutModule("Options loaded from " + file_path);
// Restore any defaults that aren't in the loaded dictionary,
// and save loaded options to ZNC's data store
for (MCString::iterator i = defaults.begin(); i != defaults.end(); i++)
{
CString option = i->first;
MCString::iterator pos = options.find(option);
if (pos == options.end())
{
options[option] = defaults[option];
DelNV(option);
}
else
{
SetNV(option, options[option]);
}
}
}
else
{
switch (status)
{
case MCString::MCS_EOPEN:
case MCString::MCS_EREADFIL:
PutModule("Failed to read options from " + file_path);
break;
default:
PutModule("Failure");
break;
}
}
}
// STATUS command
else if (action == "status")
{
CTable table;
table.AddColumn("Condition");
table.AddColumn("Status");
2011-01-14 14:45:47 +00:00
table.AddRow();
table.SetCell("Condition", "away");
table.SetCell("Status", GetNetwork()->IsIRCAway() ? "yes" : "no");
2011-01-14 14:45:47 +00:00
table.AddRow();
table.SetCell("Condition", "client_count");
table.SetCell("Status", CString(client_count()));
unsigned int now = time(NULL);
unsigned int ago = now - idle_time;
table.AddRow();
table.SetCell("Condition", "idle");
table.SetCell("Status", CString(ago) + " seconds");
if (token_count > 1)
{
CString context = tokens[1];
table.AddRow();
table.SetCell("Condition", "last_active");
if (last_active_time.count(context) < 1)
{
table.SetCell("Status", "n/a");
}
else
{
ago = now - last_active_time[context];
table.SetCell("Status", CString(ago) + " seconds");
}
table.AddRow();
table.SetCell("Condition", "last_notification");
if (last_notification_time.count(context) < 1)
{
table.SetCell("Status", "n/a");
}
else
{
ago = now - last_notification_time[context];
table.SetCell("Status", CString(ago) + " seconds");
}
table.AddRow();
table.SetCell("Condition", "replied");
table.SetCell("Status", replied(context) ? "yes" : "no");
}
PutModule(table);
}
// SUBSCRIBE command
else if (action == "subscribe")
{
// Set up the connection profile
CString service = options["service"];
bool use_post = true;
int use_port = 443;
bool use_ssl = true;
CString service_host;
CString service_url;
CString service_auth;
MCString params;
if (service == "boxcar")
{
if (options["username"] == "")
{
PutModule("Error: username not set");
return;
}
CString boxcar_api_key = "puSd2qp2gCDZO7nWkvb9";
service_host = "boxcar.io";
service_url = "/devices/providers/" + boxcar_api_key + "/notifications/subscribe";
params["email"] = options["username"];
}
else
{
PutModule("Error: service does not support subscribe command");
return;
}
// Create the socket connection, write to it, and add it to the queue
CPushSocket *sock = new CPushSocket(this);
sock->Connect(service_host, use_port, use_ssl);
sock->Request(use_post, service_host, service_url, params, service_auth);
AddSocket(sock);
}
2011-01-14 03:51:53 +00:00
// SEND command
else if (action == "send")
{
CString message = command.Token(1, true, " ", true);
send_message(message);
}
2011-01-14 03:58:49 +00:00
// HELP command
else if (action == "help")
{
2011-09-29 13:30:06 +00:00
PutModule("View the detailed documentation at https://github.com/jreese/znc-push/blob/master/README.md");
2011-01-14 03:58:49 +00:00
}
// EVAL command
else if (action == "eval")
{
CString value = command.Token(1, true, " ");
PutModule(eval(value) ? "true" : "false");
}
else
{
2011-01-14 03:58:49 +00:00
PutModule("Error: invalid command, try `help`");
}
}
};
/**
* Send an HTTP request.
*
* @param post POST command
* @param host Host domain
* @param url Resource path
* @param parameters Query parameters
* @param auth Basic authentication string
*/
void CPushSocket::Request(bool post, const CString& host, const CString& url, MCString& parameters, const CString& auth)
{
parent->PutDebug("Building notification to " + host + url + "...");
// query string for the request
bool more = false;
CString query;
CString key;
CString value;
for (MCString::iterator param = parameters.begin(); param != parameters.end(); param++)
{
key = urlencode(param->first);
value = urlencode(param->second);
if (more)
{
query += "&" + key + "=" + value;
}
else
{
query += key + "=" + value;
more = true;
}
}
parent->PutDebug("Query string: " + query);
// Request headers and POST body
CString request;
if (post)
{
request += "POST " + url + " HTTP/1.1" + crlf;
request += "Content-Type: application/x-www-form-urlencoded" + crlf;
request += "Content-Length: " + CString(query.length()) + crlf;
}
else
{
request += "GET " + url + "?" + query + " HTTP/1.1" + crlf;
}
request += "Host: " + host + crlf;
request += "Connection: close" + crlf;
request += "User-Agent: " + user_agent + crlf;
if (auth != "")
{
request += "Authorization: Basic " + auth + crlf;
}
request += crlf;
if (post)
{
request += query;
}
Write(request);
parent->PutDebug("Request sending");
}
/**
* Read each line of data returned from the HTTP request.
*/
void CPushSocket::ReadLine(const CString& data)
{
if (first)
{
CString status = data.Token(1);
CString message = data.Token(2, true);
parent->PutDebug(status);
parent->PutDebug(message);
first = false;
}
else
{
parent->PutDebug(data);
}
}
void CPushSocket::Disconnected()
{
parent->PutDebug("Disconnected.");
Close(CSocket::CLT_AFTERWRITE);
}
2011-09-29 13:30:06 +00:00
MODULEDEFS(CPushMod, "Send highlights and personal messages to a push notification service")