mirror of https://github.com/bqv/weechat-xmpp
frankly i'm not sure
Tony Olagbaiye
7 years ago
parent
44915eb1b6
commit
22f22a0aee
@ -0,0 +1,244 @@
|
||||
#include <libwebsockets.h>
|
||||
#include <json.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "../weechat-plugin.h"
|
||||
#include "../slack.h"
|
||||
#include "../slack-workspace.h"
|
||||
#include "../slack-request.h"
|
||||
#include "../slack-user.h"
|
||||
#include "../request/slack-request-chat-postmessage.h"
|
||||
|
||||
static const char *const endpoint = "/api/chat.postMessage?"
|
||||
"token=%s&channel=%s&text=%s&"
|
||||
"as_user=true&link_names=true&mrkdwn=false&parse=full";
|
||||
|
||||
static inline int json_valid(json_object *object, struct t_slack_workspace *workspace)
|
||||
{
|
||||
if (!object)
|
||||
{
|
||||
weechat_printf(
|
||||
workspace->buffer,
|
||||
_("%s%s: error posting message: unexpected response from server"),
|
||||
weechat_prefix("error"), SLACK_PLUGIN_NAME);
|
||||
//__asm__("int3");
|
||||
return 0;
|
||||
}
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
static int callback_http(struct lws *wsi, enum lws_callback_reasons reason,
|
||||
void *user, void *in, size_t len)
|
||||
{
|
||||
struct t_slack_request *request = (struct t_slack_request *)user;
|
||||
|
||||
int status;
|
||||
|
||||
switch (reason)
|
||||
{
|
||||
/* because we are protocols[0] ... */
|
||||
case LWS_CALLBACK_CLIENT_CONNECTION_ERROR:
|
||||
weechat_printf(
|
||||
request->workspace->buffer,
|
||||
_("%s%s: (%d) error connecting to slack: %s"),
|
||||
weechat_prefix("error"), SLACK_PLUGIN_NAME, request->idx,
|
||||
in ? (char *)in : "(null)");
|
||||
request->client_wsi = NULL;
|
||||
break;
|
||||
|
||||
case LWS_CALLBACK_ESTABLISHED_CLIENT_HTTP:
|
||||
status = lws_http_client_http_response(wsi);
|
||||
weechat_printf(
|
||||
request->workspace->buffer,
|
||||
_("%s%s: (%d) posting message... (%d)"),
|
||||
weechat_prefix("network"), SLACK_PLUGIN_NAME, request->idx,
|
||||
status);
|
||||
break;
|
||||
|
||||
/* chunks of chunked content, with header removed */
|
||||
case LWS_CALLBACK_RECEIVE_CLIENT_HTTP_READ:
|
||||
{
|
||||
struct t_json_chunk *new_chunk, *last_chunk;
|
||||
|
||||
new_chunk = malloc(sizeof(*new_chunk));
|
||||
new_chunk->data = malloc((1024 * sizeof(char)) + 1);
|
||||
new_chunk->data[0] = '\0';
|
||||
new_chunk->next = NULL;
|
||||
|
||||
strncat(new_chunk->data, in, (int)len);
|
||||
|
||||
if (request->json_chunks)
|
||||
{
|
||||
for (last_chunk = request->json_chunks; last_chunk->next;
|
||||
last_chunk = last_chunk->next);
|
||||
last_chunk->next = new_chunk;
|
||||
}
|
||||
else
|
||||
{
|
||||
request->json_chunks = new_chunk;
|
||||
}
|
||||
}
|
||||
return 0; /* don't passthru */
|
||||
|
||||
/* uninterpreted http content */
|
||||
case LWS_CALLBACK_RECEIVE_CLIENT_HTTP:
|
||||
{
|
||||
char buffer[1024 + LWS_PRE];
|
||||
char *px = buffer + LWS_PRE;
|
||||
int lenx = sizeof(buffer) - LWS_PRE;
|
||||
|
||||
if (lws_http_client_read(wsi, &px, &lenx) < 0)
|
||||
return -1;
|
||||
}
|
||||
return 0; /* don't passthru */
|
||||
|
||||
case LWS_CALLBACK_COMPLETED_CLIENT_HTTP:
|
||||
{
|
||||
int chunk_count, i;
|
||||
char *json_string;
|
||||
json_object *response, *ok, *error;
|
||||
struct t_json_chunk *chunk_ptr;
|
||||
|
||||
chunk_count = 0;
|
||||
if (request->json_chunks)
|
||||
{
|
||||
chunk_count++;
|
||||
for (chunk_ptr = request->json_chunks; chunk_ptr->next;
|
||||
chunk_ptr = chunk_ptr->next)
|
||||
{
|
||||
chunk_count++;
|
||||
}
|
||||
}
|
||||
|
||||
json_string = malloc((1024 * sizeof(char) * chunk_count) + 1);
|
||||
json_string[0] = '\0';
|
||||
|
||||
chunk_ptr = request->json_chunks;
|
||||
for (i = 0; i < chunk_count; i++)
|
||||
{
|
||||
strncat(json_string, chunk_ptr->data, 1024);
|
||||
chunk_ptr = chunk_ptr->next;
|
||||
|
||||
free(request->json_chunks->data);
|
||||
free(request->json_chunks);
|
||||
request->json_chunks = chunk_ptr;
|
||||
}
|
||||
|
||||
weechat_printf(
|
||||
request->workspace->buffer,
|
||||
_("%s%s: (%d) got response: %s"),
|
||||
weechat_prefix("network"), SLACK_PLUGIN_NAME, request->idx,
|
||||
json_string);
|
||||
|
||||
response = json_tokener_parse(json_string);
|
||||
ok = json_object_object_get(response, "ok");
|
||||
if (!json_valid(ok, request->workspace))
|
||||
{
|
||||
json_object_put(response);
|
||||
free(json_string);
|
||||
return 0;
|
||||
}
|
||||
|
||||
if(json_object_get_boolean(ok))
|
||||
{
|
||||
/* wait for websocket to catch up */
|
||||
}
|
||||
else
|
||||
{
|
||||
error = json_object_object_get(response, "error");
|
||||
if (!json_valid(error, request->workspace))
|
||||
{
|
||||
json_object_put(response);
|
||||
free(json_string);
|
||||
return 0;
|
||||
}
|
||||
|
||||
weechat_printf(
|
||||
request->workspace->buffer,
|
||||
_("%s%s: (%d) failed to post message: %s"),
|
||||
weechat_prefix("error"), SLACK_PLUGIN_NAME, request->idx,
|
||||
json_object_get_string(error));
|
||||
}
|
||||
|
||||
json_object_put(response);
|
||||
free(json_string);
|
||||
}
|
||||
case LWS_CALLBACK_CLOSED_CLIENT_HTTP:
|
||||
request->client_wsi = NULL;
|
||||
/* Does not doing this cause a leak?
|
||||
lws_cancel_service(lws_get_context(wsi));*/ /* abort poll wait */
|
||||
break;
|
||||
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
return lws_callback_http_dummy(wsi, reason, user, in, len);
|
||||
}
|
||||
|
||||
static const struct lws_protocols protocols[] = {
|
||||
{
|
||||
"http",
|
||||
callback_http,
|
||||
0,
|
||||
0,
|
||||
},
|
||||
{ NULL, NULL, 0, 0 }
|
||||
};
|
||||
|
||||
struct t_slack_request *slack_request_chat_postmessage(
|
||||
struct t_slack_workspace *workspace,
|
||||
const char *token, const char *channel,
|
||||
const char *text)
|
||||
{
|
||||
struct t_slack_request *request;
|
||||
struct lws_context_creation_info ctxinfo;
|
||||
struct lws_client_connect_info ccinfo;
|
||||
|
||||
request = slack_request_alloc(workspace);
|
||||
|
||||
size_t urilen = snprintf(NULL, 0, endpoint, token, channel, text) + 1;
|
||||
request->uri = malloc(urilen);
|
||||
snprintf(request->uri, urilen, endpoint, token, channel, text);
|
||||
|
||||
memset(&ctxinfo, 0, sizeof(ctxinfo)); /* otherwise uninitialized garbage */
|
||||
ctxinfo.options = LWS_SERVER_OPTION_DO_SSL_GLOBAL_INIT;
|
||||
ctxinfo.port = CONTEXT_PORT_NO_LISTEN; /* we do not run any server */
|
||||
ctxinfo.protocols = protocols;
|
||||
|
||||
request->context = lws_create_context(&ctxinfo);
|
||||
if (!request->context)
|
||||
{
|
||||
weechat_printf(
|
||||
workspace->buffer,
|
||||
_("%s%s: (%d) error connecting to slack: lws init failed"),
|
||||
weechat_prefix("error"), SLACK_PLUGIN_NAME, request->idx);
|
||||
return NULL;
|
||||
}
|
||||
else
|
||||
{
|
||||
weechat_printf(
|
||||
workspace->buffer,
|
||||
_("%s%s: (%d) contacting slack.com:443"),
|
||||
weechat_prefix("network"), SLACK_PLUGIN_NAME, request->idx);
|
||||
}
|
||||
|
||||
memset(&ccinfo, 0, sizeof(ccinfo)); /* otherwise uninitialized garbage */
|
||||
ccinfo.context = request->context;
|
||||
ccinfo.ssl_connection = LCCSCF_USE_SSL;
|
||||
ccinfo.port = 443;
|
||||
ccinfo.address = "slack.com";
|
||||
ccinfo.path = request->uri;
|
||||
ccinfo.host = ccinfo.address;
|
||||
ccinfo.origin = ccinfo.address;
|
||||
ccinfo.method = "GET";
|
||||
ccinfo.protocol = protocols[0].name;
|
||||
ccinfo.pwsi = &request->client_wsi;
|
||||
ccinfo.userdata = request;
|
||||
|
||||
lws_client_connect_via_info(&ccinfo);
|
||||
|
||||
return request;
|
||||
}
|
||||
@ -0,0 +1,9 @@
|
||||
#ifndef _SLACK_REQUEST_CHAT_POSTMESSAGE_H_
|
||||
#define _SLACK_REQUEST_CHAT_POSTMESSAGE_H_
|
||||
|
||||
struct t_slack_request *slack_request_chat_postmessage(
|
||||
struct t_slack_workspace *workspace,
|
||||
const char *token, const char *channel,
|
||||
const char *text);
|
||||
|
||||
#endif /*SLACK_REQUEST_CHAT_POSTMESSAGE_H*/
|
||||
@ -0,0 +1,139 @@
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <regex.h>
|
||||
|
||||
#include "../weechat-plugin.h"
|
||||
#include "../slack.h"
|
||||
#include "../slack-workspace.h"
|
||||
#include "../slack-channel.h"
|
||||
#include "../slack-user.h"
|
||||
#include "../slack-message.h"
|
||||
|
||||
static const char format_regex[] = "<(.*?)>";
|
||||
static const size_t max_groups = 2;
|
||||
|
||||
char *slack_message_translate_code(struct t_slack_workspace *workspace,
|
||||
const char *code)
|
||||
{
|
||||
struct t_slack_channel *channel;
|
||||
struct t_slack_user *user;
|
||||
char *command;
|
||||
|
||||
switch (code[0])
|
||||
{
|
||||
case '#': /* channel */
|
||||
channel = slack_channel_search(workspace, code+1);
|
||||
if (channel)
|
||||
return strdup(channel->name);
|
||||
else
|
||||
return strdup(code);
|
||||
case '@': /* user */
|
||||
user = slack_user_search(workspace, code+1);
|
||||
if (user)
|
||||
return strdup(user->profile.display_name);
|
||||
else
|
||||
return strdup(code);
|
||||
case '!': /* special */
|
||||
command = strdup(code);
|
||||
command[0] = '@';
|
||||
return command;
|
||||
default: /* url */
|
||||
return strdup(code);
|
||||
}
|
||||
}
|
||||
|
||||
char *slack_message_decode(struct t_slack_workspace *workspace,
|
||||
const char *text)
|
||||
{
|
||||
int rc;
|
||||
regex_t reg;
|
||||
regmatch_t groups[max_groups];
|
||||
char msgbuf[100];
|
||||
char *decoded_text, *pos;
|
||||
const char *cursor;
|
||||
size_t offset;
|
||||
|
||||
if ((rc = regcomp(®, format_regex, REG_EXTENDED)))
|
||||
{
|
||||
regerror(rc, ®, msgbuf, sizeof(msgbuf));
|
||||
weechat_printf(
|
||||
workspace->buffer,
|
||||
_("%s%s: error compiling message formatting regex: %s"),
|
||||
weechat_prefix("error"), SLACK_PLUGIN_NAME,
|
||||
msgbuf);
|
||||
return strdup(text);
|
||||
}
|
||||
|
||||
decoded_text = malloc(SLACK_MESSAGE_MAX_LENGTH);
|
||||
if (!decoded_text)
|
||||
{
|
||||
regfree(®);
|
||||
weechat_printf(
|
||||
workspace->buffer,
|
||||
_("%s%s: error allocating space for message"),
|
||||
weechat_prefix("error"), SLACK_PLUGIN_NAME);
|
||||
return strdup(text);
|
||||
}
|
||||
pos = decoded_text;
|
||||
pos[0] = '\0';
|
||||
|
||||
for (cursor = text; regexec(®, cursor, max_groups, groups, 0) == 0; cursor += offset)
|
||||
{
|
||||
offset = groups[0].rm_eo;
|
||||
|
||||
char *copy = strdup(cursor);
|
||||
if (!copy)
|
||||
{
|
||||
regfree(®);
|
||||
weechat_printf(
|
||||
workspace->buffer,
|
||||
_("%s%s: error allocating space for message"),
|
||||
weechat_prefix("error"), SLACK_PLUGIN_NAME);
|
||||
return strdup(text);
|
||||
}
|
||||
copy[groups[1].rm_eo] = '\0';
|
||||
|
||||
char *match = strdup(copy + groups[1].rm_so);
|
||||
if (!match)
|
||||
{
|
||||
free(copy);
|
||||
regfree(®);
|
||||
weechat_printf(
|
||||
workspace->buffer,
|
||||
_("%s%s: error allocating space for message"),
|
||||
weechat_prefix("error"), SLACK_PLUGIN_NAME);
|
||||
return strdup(text);
|
||||
}
|
||||
copy[groups[0].rm_so] = '\0';
|
||||
|
||||
char *prematch = strdup(copy);
|
||||
if (!prematch)
|
||||
{
|
||||
free(match);
|
||||
free(copy);
|
||||
regfree(®);
|
||||
weechat_printf(
|
||||
workspace->buffer,
|
||||
_("%s%s: error allocating space for message"),
|
||||
weechat_prefix("error"), SLACK_PLUGIN_NAME);
|
||||
return strdup(text);
|
||||
}
|
||||
free(copy);
|
||||
|
||||
pos = strncat(decoded_text, prematch,
|
||||
SLACK_MESSAGE_MAX_LENGTH - strlen(decoded_text) - 1);
|
||||
free(prematch);
|
||||
|
||||
char *replacement = slack_message_translate_code(workspace, match);
|
||||
free(match);
|
||||
|
||||
pos = strncat(decoded_text, replacement,
|
||||
SLACK_MESSAGE_MAX_LENGTH - strlen(decoded_text) - 1);
|
||||
free(replacement);
|
||||
}
|
||||
pos = strncat(decoded_text, cursor,
|
||||
SLACK_MESSAGE_MAX_LENGTH - strlen(decoded_text) - 1);
|
||||
|
||||
regfree(®);
|
||||
return decoded_text;
|
||||
}
|
||||
@ -0,0 +1,9 @@
|
||||
#ifndef _SLACK_MESSAGE_H_
|
||||
#define _SLACK_MESSAGE_H_
|
||||
|
||||
#define SLACK_MESSAGE_MAX_LENGTH 40000
|
||||
|
||||
char *slack_message_decode(struct t_slack_workspace *workspace,
|
||||
const char *text);
|
||||
|
||||
#endif /*SLACK_MESSAGE_H*/
|
||||
Loading…
Reference in New Issue