X-Git-Url: https://git.toastfreeware.priv.at/gregoa/bti.git/blobdiff_plain/e67f72404840f2e46d20edfe043a41b33e6c8562..fd562ddb1a223ca6623c05761305a103e6a85dfb:/bti.c diff --git a/bti.c b/bti.c index bd311cd..ae32c37 100644 --- a/bti.c +++ b/bti.c @@ -1,5 +1,7 @@ /* * Copyright (C) 2008 Greg Kroah-Hartman + * Copyright (C) 2009 Bart Trojanowski + * Copyright (C) 2009 Amir Mohammad Saied * * This program is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License as published by the @@ -15,6 +17,8 @@ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. */ +#define _GNU_SOURCE + #include #include #include @@ -26,12 +30,15 @@ #include #include #include +#include +#include #include -#include #include #include #include -#include "bti_version.h" +#include +#include +#include #define zalloc(size) calloc(size, 1) @@ -39,23 +46,28 @@ #define dbg(format, arg...) \ do { \ if (debug) \ - printf("%s: " format , __func__ , ## arg); \ + fprintf(stdout, "bti: %s: " format , __func__ , \ + ## arg); \ } while (0) static int debug; +static int verbose; enum host { - HOST_TWITTER = 0, + HOST_TWITTER = 0, HOST_IDENTICA = 1, + HOST_CUSTOM = 2 }; enum action { - ACTION_UPDATE = 0, + ACTION_UPDATE = 0, ACTION_FRIENDS = 1, - ACTION_USER = 2, + ACTION_USER = 2, ACTION_REPLIES = 4, - ACTION_PUBLIC = 8 + ACTION_PUBLIC = 8, + ACTION_GROUP = 16, + ACTION_UNKNOWN = 32 }; struct session { @@ -67,9 +79,20 @@ struct session { char *homedir; char *logfile; char *user; + char *group; + char *hosturl; + char *hostname; + char *configfile; + char *replyto; int bash; + int interactive; + int shrink_urls; + int dry_run; + int page; enum host host; enum action action; + void *readline_handle; + char *(*readline)(const char *); }; struct bti_curl_buffer { @@ -80,27 +103,132 @@ struct bti_curl_buffer { static void display_help(void) { - fprintf(stdout, "bti - send tweet to twitter\n"); - fprintf(stdout, "Version: " BTI_VERSION "\n"); + fprintf(stdout, "bti - send tweet to twitter or identi.ca\n"); + fprintf(stdout, "Version: " VERSION "\n"); fprintf(stdout, "Usage:\n"); fprintf(stdout, " bti [options]\n"); fprintf(stdout, "options are:\n"); fprintf(stdout, " --account accountname\n"); fprintf(stdout, " --password password\n"); + fprintf(stdout, " --action action\n"); + fprintf(stdout, " ('update', 'friends', 'public', 'replies', " + "'group' or 'user')\n"); + fprintf(stdout, " --user screenname\n"); + fprintf(stdout, " --group groupname\n"); fprintf(stdout, " --proxy PROXY:PORT\n"); fprintf(stdout, " --host HOST\n"); fprintf(stdout, " --logfile logfile\n"); + fprintf(stdout, " --config configfile\n"); + fprintf(stdout, " --replyto ID\n"); + fprintf(stdout, " --shrink-urls\n"); + fprintf(stdout, " --page PAGENUMBER\n"); fprintf(stdout, " --bash\n"); - fprintf(stdout, " --action action\n"); - fprintf(stdout, " --user screenname\n"); fprintf(stdout, " --debug\n"); + fprintf(stdout, " --verbose\n"); + fprintf(stdout, " --dry-run\n"); fprintf(stdout, " --version\n"); fprintf(stdout, " --help\n"); } static void display_version(void) { - fprintf(stdout, "bti - version %s\n", BTI_VERSION); + fprintf(stdout, "bti - version %s\n", VERSION); +} + +static char *get_string(const char *name) +{ + char *temp; + char *string; + + string = zalloc(1000); + if (!string) + exit(1); + if (name != NULL) + fprintf(stdout, "%s", name); + if (!fgets(string, 999, stdin)) + return NULL; + temp = strchr(string, '\n'); + if (temp) + *temp = '\0'; + return string; +} + +/* + * Try to get a handle to a readline function from a variety of different + * libraries. If nothing is present on the system, then fall back to an + * internal one. + * + * Logic originally based off of code in the e2fsutils package in the + * lib/ss/get_readline.c file, which is licensed under the MIT license. + * + * This keeps us from having to relicense the bti codebase if readline + * ever changes its license, as there is no link-time dependancy. + * It is a run-time thing only, and we handle any readline-like library + * in the same manner, making bti not be a derivative work of any + * other program. + */ +static void session_readline_init(struct session *session) +{ + /* Libraries we will try to use for readline/editline functionality */ + const char *libpath = "libreadline.so.6:libreadline.so.5:" + "libreadline.so.4:libreadline.so:libedit.so.2:" + "libedit.so:libeditline.so.0:libeditline.so"; + void *handle = NULL; + char *tmp, *cp, *next; + int (*bind_key)(int, void *); + void (*insert)(void); + + /* default to internal function if we can't or won't find anything */ + session->readline = get_string; + if (!isatty(0)) + return; + session->interactive = 1; + + tmp = malloc(strlen(libpath)+1); + if (!tmp) + return; + strcpy(tmp, libpath); + for (cp = tmp; cp; cp = next) { + next = strchr(cp, ':'); + if (next) + *next++ = 0; + if (*cp == 0) + continue; + handle = dlopen(cp, RTLD_NOW); + if (handle) { + dbg("Using %s for readline library\n", cp); + break; + } + } + free(tmp); + if (!handle) { + dbg("No readline library found.\n"); + return; + } + + session->readline_handle = handle; + session->readline = (char *(*)(const char *))dlsym(handle, "readline"); + if (session->readline == NULL) { + /* something odd happened, default back to internal stuff */ + session->readline_handle = NULL; + session->readline = get_string; + return; + } + + /* + * If we found a library, turn off filename expansion + * as that makes no sense from within bti. + */ + bind_key = (int (*)(int, void *))dlsym(handle, "rl_bind_key"); + insert = (void (*)(void))dlsym(handle, "rl_insert"); + if (bind_key && insert) + bind_key('\t', insert); +} + +static void session_readline_cleanup(struct session *session) +{ + if (session->readline_handle) + dlclose(session->readline_handle); } static struct session *session_alloc(void) @@ -117,6 +245,7 @@ static void session_free(struct session *session) { if (!session) return; + free(session->replyto); free(session->password); free(session->account); free(session->tweet); @@ -124,6 +253,10 @@ static void session_free(struct session *session) free(session->time); free(session->homedir); free(session->user); + free(session->group); + free(session->hosturl); + free(session->hostname); + free(session->configfile); free(session); } @@ -155,17 +288,17 @@ static void bti_curl_buffer_free(struct bti_curl_buffer *buffer) free(buffer); } -static const char *twitter_user_url = "http://twitter.com/statuses/user_timeline/"; -static const char *twitter_update_url = "https://twitter.com/statuses/update.xml"; -static const char *twitter_public_url = "http://twitter.com/statuses/public_timeline.xml"; -static const char *twitter_friends_url = "https://twitter.com/statuses/friends_timeline.xml"; -static const char *twitter_replies_url = "http://twitter.com/statuses/replies.xml"; +static const char *twitter_host = "https://twitter.com/statuses"; +static const char *identica_host = "https://identi.ca/api/statuses"; +static const char *twitter_name = "twitter"; +static const char *identica_name = "identi.ca"; -static const char *identica_user_url = "http://identi.ca/api/statuses/user_timeline/"; -static const char *identica_update_url = "http://identi.ca/api/statuses/update.xml"; -static const char *identica_public_url = "http://identi.ca/api/statuses/public_timeline.xml"; -static const char *identica_friends_url = "http://identi.ca/api/statuses/friends_timeline.xml"; -static const char *identica_replies_url = "http://identi.ca/api/statuses/replies.xml"; +static const char *user_uri = "/user_timeline/"; +static const char *update_uri = "/update.xml"; +static const char *public_uri = "/public_timeline.xml"; +static const char *friends_uri = "/friends_timeline.xml"; +static const char *replies_uri = "/replies.xml"; +static const char *group_uri = "/../laconica/groups/timeline/"; static CURL *curl_init(void) { @@ -182,17 +315,23 @@ static CURL *curl_init(void) return curl; } -void parse_statuses(xmlDocPtr doc, xmlNodePtr current) +static void parse_statuses(xmlDocPtr doc, xmlNodePtr current) { xmlChar *text = NULL; xmlChar *user = NULL; + xmlChar *created = NULL; + xmlChar *id = NULL; xmlNodePtr userinfo; current = current->xmlChildrenNode; while (current != NULL) { if (current->type == XML_ELEMENT_NODE) { + if (!xmlStrcmp(current->name, (const xmlChar *)"created_at")) + created = xmlNodeListGetString(doc, current->xmlChildrenNode, 1); if (!xmlStrcmp(current->name, (const xmlChar *)"text")) text = xmlNodeListGetString(doc, current->xmlChildrenNode, 1); + if (!xmlStrcmp(current->name, (const xmlChar *)"id")) + id = xmlNodeListGetString(doc, current->xmlChildrenNode, 1); if (!xmlStrcmp(current->name, (const xmlChar *)"user")) { userinfo = current->xmlChildrenNode; while (userinfo != NULL) { @@ -204,12 +343,22 @@ void parse_statuses(xmlDocPtr doc, xmlNodePtr current) userinfo = userinfo->next; } } - if (user && text) { - printf("[%s] %s\n", user, text); + + if (user && text && created && id) { + if (verbose) + printf("[%s] {%s} (%.16s) %s\n", + user, id, created, text); + else + printf("[%s] %s\n", + user, text); xmlFree(user); xmlFree(text); + xmlFree(created); + xmlFree(id); user = NULL; text = NULL; + created = NULL; + id = NULL; } } current = current->next; @@ -222,8 +371,9 @@ static void parse_timeline(char *document) { xmlDocPtr doc; xmlNodePtr current; - doc = xmlReadMemory(document, strlen(document), "timeline.xml", NULL, XML_PARSE_NOERROR); + doc = xmlReadMemory(document, strlen(document), "timeline.xml", + NULL, XML_PARSE_NOERROR); if (doc == NULL) return; @@ -251,7 +401,8 @@ static void parse_timeline(char *document) return; } -size_t curl_callback(void *buffer, size_t size, size_t nmemb, void *userp) +static size_t curl_callback(void *buffer, size_t size, size_t nmemb, + void *userp) { struct bti_curl_buffer *curl_buf = userp; size_t buffer_size = size * nmemb; @@ -280,10 +431,9 @@ size_t curl_callback(void *buffer, size_t size, size_t nmemb, void *userp) static int send_request(struct session *session) { + char endpoint[100]; char user_password[500]; char data[500]; - /* is there usernames longer than 22 chars? */ - char user_url[70]; struct bti_curl_buffer *curl_buf; CURL *curl = NULL; CURLcode res; @@ -302,6 +452,9 @@ static int send_request(struct session *session) if (!curl) return -EINVAL; + if (!session->hosturl) + session->hosturl = strdup(twitter_host); + switch (session->action) { case ACTION_UPDATE: snprintf(user_password, sizeof(user_password), "%s:%s", @@ -317,71 +470,58 @@ static int send_request(struct session *session) CURLFORM_COPYCONTENTS, "bti", CURLFORM_END); + if (session->replyto) + curl_formadd(&formpost, &lastptr, + CURLFORM_COPYNAME, "in_reply_to_status_id", + CURLFORM_COPYCONTENTS, session->replyto, + CURLFORM_END); + curl_easy_setopt(curl, CURLOPT_HTTPPOST, formpost); slist = curl_slist_append(slist, "Expect:"); curl_easy_setopt(curl, CURLOPT_HTTPHEADER, slist); - switch (session->host) { - case HOST_TWITTER: - curl_easy_setopt(curl, CURLOPT_URL, twitter_update_url); - break; - case HOST_IDENTICA: - curl_easy_setopt(curl, CURLOPT_URL, identica_update_url); - break; - } + + sprintf(endpoint, "%s%s", session->hosturl, update_uri); + curl_easy_setopt(curl, CURLOPT_URL, endpoint); curl_easy_setopt(curl, CURLOPT_USERPWD, user_password); break; case ACTION_FRIENDS: snprintf(user_password, sizeof(user_password), "%s:%s", session->account, session->password); - switch (session->host) { - case HOST_TWITTER: - curl_easy_setopt(curl, CURLOPT_URL, twitter_friends_url); - break; - case HOST_IDENTICA: - curl_easy_setopt(curl, CURLOPT_URL, identica_friends_url); - break; - } + sprintf(endpoint, "%s%s?page=%d", session->hosturl, + friends_uri, session->page); + curl_easy_setopt(curl, CURLOPT_URL, endpoint); curl_easy_setopt(curl, CURLOPT_USERPWD, user_password); break; case ACTION_USER: - switch (session->host) { - case HOST_TWITTER: - sprintf(user_url, "%s%s.xml", twitter_user_url, session->user); - curl_easy_setopt(curl, CURLOPT_URL, user_url); - break; - case HOST_IDENTICA: - sprintf(user_url, "%s%s.xml", identica_user_url, session->user); - curl_easy_setopt(curl, CURLOPT_URL, user_url); - break; - } + sprintf(endpoint, "%s%s%s.xml?page=%d", session->hosturl, + user_uri, session->user, session->page); + curl_easy_setopt(curl, CURLOPT_URL, endpoint); break; case ACTION_REPLIES: snprintf(user_password, sizeof(user_password), "%s:%s", session->account, session->password); - switch (session->host) { - case HOST_TWITTER: - curl_easy_setopt(curl, CURLOPT_URL, twitter_replies_url); - break; - case HOST_IDENTICA: - curl_easy_setopt(curl, CURLOPT_URL, identica_replies_url); - break; - } + sprintf(endpoint, "%s%s?page=%d", session->hosturl, replies_uri, + session->page); + curl_easy_setopt(curl, CURLOPT_URL, endpoint); curl_easy_setopt(curl, CURLOPT_USERPWD, user_password); break; case ACTION_PUBLIC: - switch (session->host) { - case HOST_TWITTER: - curl_easy_setopt(curl, CURLOPT_URL, twitter_public_url); - break; - case HOST_IDENTICA: - curl_easy_setopt(curl, CURLOPT_URL, identica_public_url); - break; - } + sprintf(endpoint, "%s%s?page=%d", session->hosturl, public_uri, + session->page); + curl_easy_setopt(curl, CURLOPT_URL, endpoint); + + break; + case ACTION_GROUP: + sprintf(endpoint, "%s%s%s.xml?page=%d", session->hosturl, + group_uri, session->group, session->page); + curl_easy_setopt(curl, CURLOPT_URL, endpoint); + break; + default: break; } @@ -397,10 +537,13 @@ static int send_request(struct session *session) curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, curl_callback); curl_easy_setopt(curl, CURLOPT_WRITEDATA, curl_buf); - res = curl_easy_perform(curl); - if (res && !session->bash) { - fprintf(stderr, "error(%d) trying to perform operation\n", res); - return -EINVAL; + if (!session->dry_run) { + res = curl_easy_perform(curl); + if (res && !session->bash) { + fprintf(stderr, "error(%d) trying to perform " + "operation\n", res); + return -EINVAL; + } } curl_easy_cleanup(curl); @@ -423,13 +566,9 @@ static void parse_configfile(struct session *session) char *action = NULL; char *user = NULL; char *file; + int shrink_urls = 0; - /* config file is ~/.bti */ - file = alloca(strlen(session->homedir) + 7); - - sprintf(file, "%s/.bti", session->homedir); - - config_file = fopen(file, "r"); + config_file = fopen(session->configfile, "r"); /* No error if file does not exist or is unreadable. */ if (config_file == NULL) @@ -488,6 +627,18 @@ static void parse_configfile(struct session *session) c += 5; if (c[0] != '\0') user = strdup(c); + } else if (!strncasecmp(c, "shrink-urls", 11) && + (c[11] == '=')) { + c += 12; + if (!strncasecmp(c, "true", 4) || + !strncasecmp(c, "yes", 3)) + shrink_urls = 1; + } else if (!strncasecmp(c, "verbose", 7) && + (c[7] == '=')) { + c += 8; + if (!strncasecmp(c, "true", 4) || + !strncasecmp(c, "yes", 3)) + verbose = 1; } } while (!feof(config_file)); @@ -496,10 +647,19 @@ static void parse_configfile(struct session *session) if (account) session->account = account; if (host) { - if (strcasecmp(host, "twitter") == 0) + if (strcasecmp(host, "twitter") == 0) { session->host = HOST_TWITTER; - if (strcasecmp(host, "identica") == 0) + session->hosturl = strdup(twitter_host); + session->hostname = strdup(twitter_name); + } else if (strcasecmp(host, "identica") == 0) { session->host = HOST_IDENTICA; + session->hosturl = strdup(identica_host); + session->hostname = strdup(identica_name); + } else { + session->host = HOST_CUSTOM; + session->hosturl = strdup(host); + session->hostname = strdup(host); + } free(host); } if (proxy) { @@ -512,19 +672,23 @@ static void parse_configfile(struct session *session) if (action) { if (strcasecmp(action, "update") == 0) session->action = ACTION_UPDATE; - if (strcasecmp(action, "friends") == 0) + else if (strcasecmp(action, "friends") == 0) session->action = ACTION_FRIENDS; - if (strcasecmp(action, "user") == 0) + else if (strcasecmp(action, "user") == 0) session->action = ACTION_USER; - if (strcasecmp(action, "replies") == 0) + else if (strcasecmp(action, "replies") == 0) session->action = ACTION_REPLIES; - if (strcasecmp(action, "public") == 0) + else if (strcasecmp(action, "public") == 0) session->action = ACTION_PUBLIC; + else if (strcasecmp(action, "group") == 0) + session->action = ACTION_GROUP; + else + session->action = ACTION_UNKNOWN; free(action); } - if (user) { + if (user) session->user = user; - } + session->shrink_urls = shrink_urls; /* Free buffer and close file. */ free(line); @@ -535,7 +699,6 @@ static void log_session(struct session *session, int retval) { FILE *log_file; char *filename; - char *host; /* Only log something if we have a log file set */ if (!session->logfile) @@ -549,62 +712,384 @@ static void log_session(struct session *session, int retval) log_file = fopen(filename, "a+"); if (log_file == NULL) return; - switch (session->host) { - case HOST_TWITTER: - host = "twitter"; - break; - case HOST_IDENTICA: - host = "identi.ca"; - break; - default: - host = "unknown"; - break; - } - if (session->action == ACTION_UPDATE) { + switch (session->action) { + case ACTION_UPDATE: if (retval) fprintf(log_file, "%s: host=%s tweet failed\n", - session->time, host); + session->time, session->hostname); else fprintf(log_file, "%s: host=%s tweet=%s\n", - session->time, host, session->tweet); - } else if (session->action == ACTION_FRIENDS) { + session->time, session->hostname, + session->tweet); + break; + case ACTION_FRIENDS: fprintf(log_file, "%s: host=%s retrieving friends timeline\n", - session->time, host); - } else if (session->action == ACTION_PUBLIC) { + session->time, session->hostname); + break; + case ACTION_USER: + fprintf(log_file, "%s: host=%s retrieving %s's timeline\n", + session->time, session->hostname, session->user); + break; + case ACTION_REPLIES: + fprintf(log_file, "%s: host=%s retrieving replies\n", + session->time, session->hostname); + break; + case ACTION_PUBLIC: fprintf(log_file, "%s: host=%s retrieving public timeline\n", - session->time, host); + session->time, session->hostname); + break; + case ACTION_GROUP: + fprintf(log_file, "%s: host=%s retrieving group timeline\n", + session->time, session->hostname); + break; + default: + break; } fclose(log_file); } +static char *get_string_from_stdin(void) +{ + char *temp; + char *string; + + string = zalloc(1000); + if (!string) + return NULL; + + if (!fgets(string, 999, stdin)) + return NULL; + temp = strchr(string, '\n'); + if (temp) + *temp = '\0'; + return string; +} + +static void read_password(char *buf, size_t len, char *host) +{ + char pwd[80]; + int retval; + struct termios old; + struct termios tp; + + tcgetattr(0, &tp); + old = tp; + + tp.c_lflag &= (~ECHO); + tcsetattr(0, TCSANOW, &tp); + + fprintf(stdout, "Enter password for %s: ", host); + fflush(stdout); + tcflow(0, TCOOFF); + retval = scanf("%79s", pwd); + tcflow(0, TCOON); + fprintf(stdout, "\n"); + + tcsetattr(0, TCSANOW, &old); + + strncpy(buf, pwd, len); + buf[len-1] = '\0'; +} + +static int find_urls(const char *tweet, int **pranges) +{ + /* + * magic obtained from + * http://www.geekpedia.com/KB65_How-to-validate-an-URL-using-RegEx-in-Csharp.html + */ + static const char *re_magic = + "(([a-zA-Z][0-9a-zA-Z+\\-\\.]*:)/{1,3}" + "[0-9a-zA-Z;/~?:@&=+$\\.\\-_'()%]+)" + "(#[0-9a-zA-Z;/?:@&=+$\\.\\-_!~*'()%]+)?"; + pcre *re; + const char *errptr; + int erroffset; + int ovector[10] = {0,}; + const size_t ovsize = sizeof(ovector)/sizeof(*ovector); + int startoffset, tweetlen; + int i, rc; + int rbound = 10; + int rcount = 0; + int *ranges = malloc(sizeof(int) * rbound); + + re = pcre_compile(re_magic, + PCRE_NO_AUTO_CAPTURE, + &errptr, &erroffset, NULL); + if (!re) { + fprintf(stderr, "pcre_compile @%u: %s\n", erroffset, errptr); + exit(1); + } + + tweetlen = strlen(tweet); + for (startoffset = 0; startoffset < tweetlen; ) { + + rc = pcre_exec(re, NULL, tweet, strlen(tweet), startoffset, 0, + ovector, ovsize); + if (rc == PCRE_ERROR_NOMATCH) + break; + + if (rc < 0) { + fprintf(stderr, "pcre_exec @%u: %s\n", + erroffset, errptr); + exit(1); + } + + for (i = 0; i < rc; i += 2) { + if ((rcount+2) == rbound) { + rbound *= 2; + ranges = realloc(ranges, sizeof(int) * rbound); + } + + ranges[rcount++] = ovector[i]; + ranges[rcount++] = ovector[i+1]; + } + + startoffset = ovector[1]; + } + + pcre_free(re); + + *pranges = ranges; + return rcount; +} + +/** + * bidirectional popen() call + * + * @param rwepipe - int array of size three + * @param exe - program to run + * @param argv - argument list + * @return pid or -1 on error + * + * The caller passes in an array of three integers (rwepipe), on successful + * execution it can then write to element 0 (stdin of exe), and read from + * element 1 (stdout) and 2 (stderr). + */ +static int popenRWE(int *rwepipe, const char *exe, const char *const argv[]) +{ + int in[2]; + int out[2]; + int err[2]; + int pid; + int rc; + + rc = pipe(in); + if (rc < 0) + goto error_in; + + rc = pipe(out); + if (rc < 0) + goto error_out; + + rc = pipe(err); + if (rc < 0) + goto error_err; + + pid = fork(); + if (pid > 0) { + /* parent */ + close(in[0]); + close(out[1]); + close(err[1]); + rwepipe[0] = in[1]; + rwepipe[1] = out[0]; + rwepipe[2] = err[0]; + return pid; + } else if (pid == 0) { + /* child */ + close(in[1]); + close(out[0]); + close(err[0]); + close(0); + rc = dup(in[0]); + close(1); + rc = dup(out[1]); + close(2); + rc = dup(err[1]); + + execvp(exe, (char **)argv); + exit(1); + } else + goto error_fork; + + return pid; + +error_fork: + close(err[0]); + close(err[1]); +error_err: + close(out[0]); + close(out[1]); +error_out: + close(in[0]); + close(in[1]); +error_in: + return -1; +} + +static int pcloseRWE(int pid, int *rwepipe) +{ + int rc, status; + close(rwepipe[0]); + close(rwepipe[1]); + close(rwepipe[2]); + rc = waitpid(pid, &status, 0); + return status; +} + +static char *shrink_one_url(int *rwepipe, char *big) +{ + int biglen = strlen(big); + char *small; + int smalllen; + int rc; + + rc = dprintf(rwepipe[0], "%s\n", big); + if (rc < 0) + return big; + + smalllen = biglen + 128; + small = malloc(smalllen); + if (!small) + return big; + + rc = read(rwepipe[1], small, smalllen); + if (rc < 0 || rc > biglen) + goto error_free_small; + + if (strncmp(small, "http://", 7)) + goto error_free_small; + + smalllen = rc; + while (smalllen && isspace(small[smalllen-1])) + small[--smalllen] = 0; + + free(big); + return small; + +error_free_small: + free(small); + return big; +} + +static char *shrink_urls(char *text) +{ + int *ranges; + int rcount; + int i; + int inofs = 0; + int outofs = 0; + const char *const shrink_args[] = { + "bti-shrink-urls", + NULL + }; + int shrink_pid; + int shrink_pipe[3]; + int inlen = strlen(text); + + dbg("before len=%u\n", inlen); + + shrink_pid = popenRWE(shrink_pipe, shrink_args[0], shrink_args); + if (shrink_pid < 0) + return text; + + rcount = find_urls(text, &ranges); + if (!rcount) + return text; + + for (i = 0; i < rcount; i += 2) { + int url_start = ranges[i]; + int url_end = ranges[i+1]; + int long_url_len = url_end - url_start; + char *url = strndup(text + url_start, long_url_len); + int short_url_len; + int not_url_len = url_start - inofs; + + dbg("long url[%u]: %s\n", long_url_len, url); + url = shrink_one_url(shrink_pipe, url); + short_url_len = url ? strlen(url) : 0; + dbg("short url[%u]: %s\n", short_url_len, url); + + if (!url || short_url_len >= long_url_len) { + /* The short url ended up being too long + * or unavailable */ + if (inofs) { + strncpy(text + outofs, text + inofs, + not_url_len + long_url_len); + } + inofs += not_url_len + long_url_len; + outofs += not_url_len + long_url_len; + + } else { + /* copy the unmodified block */ + strncpy(text + outofs, text + inofs, not_url_len); + inofs += not_url_len; + outofs += not_url_len; + + /* copy the new url */ + strncpy(text + outofs, url, short_url_len); + inofs += long_url_len; + outofs += short_url_len; + } + + free(url); + } + + /* copy the last block after the last match */ + if (inofs) { + int tail = inlen - inofs; + if (tail) { + strncpy(text + outofs, text + inofs, tail); + outofs += tail; + } + } + + free(ranges); + + (void)pcloseRWE(shrink_pid, shrink_pipe); + + text[outofs] = 0; + dbg("after len=%u\n", outofs); + return text; +} + int main(int argc, char *argv[], char *envp[]) { static const struct option options[] = { { "debug", 0, NULL, 'd' }, + { "verbose", 0, NULL, 'V' }, { "account", 1, NULL, 'a' }, { "password", 1, NULL, 'p' }, { "host", 1, NULL, 'H' }, { "proxy", 1, NULL, 'P' }, { "action", 1, NULL, 'A' }, { "user", 1, NULL, 'u' }, + { "group", 1, NULL, 'G' }, { "logfile", 1, NULL, 'L' }, + { "shrink-urls", 0, NULL, 's' }, { "help", 0, NULL, 'h' }, { "bash", 0, NULL, 'b' }, + { "dry-run", 0, NULL, 'n' }, + { "page", 1, NULL, 'g' }, { "version", 0, NULL, 'v' }, + { "config", 1, NULL, 'c' }, + { "replyto", 1, NULL, 'r' }, { } }; struct session *session; pid_t child; char *tweet; + static char password[80]; int retval = 0; int option; char *http_proxy; time_t t; + int page_nr; debug = 0; - rl_bind_key('\t', rl_insert); + verbose = 0; session = session_alloc(); if (!session) { @@ -617,8 +1102,13 @@ int main(int argc, char *argv[], char *envp[]) session->time = strdup(ctime(&t)); session->time[strlen(session->time)-1] = 0x00; + /* Get the home directory so we can try to find a config file */ session->homedir = strdup(getenv("HOME")); + /* set up a default config file location (traditionally ~/.bti) */ + session->configfile = zalloc(strlen(session->homedir) + 7); + sprintf(session->configfile, "%s/.bti", session->homedir); + curl_global_init(CURL_GLOBAL_ALL); /* Set environment variables first, before reading command line options @@ -634,7 +1124,7 @@ int main(int argc, char *argv[], char *envp[]) parse_configfile(session); while (1) { - option = getopt_long_only(argc, argv, "dqe:p:P:H:a:A:u:h", + option = getopt_long_only(argc, argv, "dp:P:H:a:A:u:c:hg:G:sr:nVv", options, NULL); if (option == -1) break; @@ -642,12 +1132,24 @@ int main(int argc, char *argv[], char *envp[]) case 'd': debug = 1; break; + case 'V': + verbose = 1; + break; case 'a': if (session->account) free(session->account); session->account = strdup(optarg); dbg("account = %s\n", session->account); break; + case 'g': + page_nr = atoi(optarg); + dbg("page = %d\n", page_nr); + session->page = page_nr; + break; + case 'r': + session->replyto = strdup(optarg); + dbg("in_reply_to_status_id = %s\n", session->replyto); + break; case 'p': if (session->password) free(session->password); @@ -663,14 +1165,18 @@ int main(int argc, char *argv[], char *envp[]) case 'A': if (strcasecmp(optarg, "update") == 0) session->action = ACTION_UPDATE; - if (strcasecmp(optarg, "friends") == 0) + else if (strcasecmp(optarg, "friends") == 0) session->action = ACTION_FRIENDS; - if (strcasecmp(optarg, "user") == 0) + else if (strcasecmp(optarg, "user") == 0) session->action = ACTION_USER; - if (strcasecmp(optarg, "replies") == 0) + else if (strcasecmp(optarg, "replies") == 0) session->action = ACTION_REPLIES; - if (strcasecmp(optarg, "public") == 0) + else if (strcasecmp(optarg, "public") == 0) session->action = ACTION_PUBLIC; + else if (strcasecmp(optarg, "group") == 0) + session->action = ACTION_GROUP; + else + session->action = ACTION_UNKNOWN; dbg("action = %d\n", session->action); break; case 'u': @@ -679,25 +1185,63 @@ int main(int argc, char *argv[], char *envp[]) session->user = strdup(optarg); dbg("user = %s\n", session->user); break; + + case 'G': + if (session->group) + free(session->group); + session->group = strdup(optarg); + dbg("group = %s\n", session->group); + break; case 'L': if (session->logfile) free(session->logfile); session->logfile = strdup(optarg); dbg("logfile = %s\n", session->logfile); break; + case 's': + session->shrink_urls = 1; + break; case 'H': - if (strcasecmp(optarg, "twitter") == 0) + if (session->hosturl) + free(session->hosturl); + if (session->hostname) + free(session->hostname); + if (strcasecmp(optarg, "twitter") == 0) { session->host = HOST_TWITTER; - if (strcasecmp(optarg, "identica") == 0) + session->hosturl = strdup(twitter_host); + session->hostname = strdup(twitter_name); + } else if (strcasecmp(optarg, "identica") == 0) { session->host = HOST_IDENTICA; + session->hosturl = strdup(identica_host); + session->hostname = strdup(identica_name); + } else { + session->host = HOST_CUSTOM; + session->hosturl = strdup(optarg); + session->hostname = strdup(optarg); + } dbg("host = %d\n", session->host); break; case 'b': session->bash = 1; break; + case 'c': + if (session->configfile) + free(session->configfile); + session->configfile = strdup(optarg); + dbg("configfile = %s\n", session->configfile); + + /* + * read the config file now. Yes, this could override previously + * set options from the command line, but the user asked for it... + */ + parse_configfile(session); + break; case 'h': display_help(); goto exit; + case 'n': + session->dry_run = 1; + break; case 'v': display_version(); goto exit; @@ -707,29 +1251,58 @@ int main(int argc, char *argv[], char *envp[]) } } + session_readline_init(session); + /* + * Show the version to make it easier to determine what + * is going on here + */ + if (debug) + display_version(); + + if (session->action == ACTION_UNKNOWN) { + fprintf(stderr, "Unknown action, valid actions are:\n"); + fprintf(stderr, "'update', 'friends', 'public', " + "'replies', 'group' or 'user'.\n"); + goto exit; + } + + if (session->host == HOST_TWITTER && session->action == ACTION_GROUP) { + fprintf(stderr, "Groups only work in Identi.ca.\n"); + goto exit; + } + + if (session->action == ACTION_GROUP && !session->group) { + fprintf(stdout, "Enter group name: "); + session->group = session->readline(NULL); + } + if (!session->account) { - fprintf(stdout, "Enter twitter account: "); - session->account = readline(NULL); + fprintf(stdout, "Enter account for %s: ", session->hostname); + session->account = session->readline(NULL); } if (!session->password) { - fprintf(stdout, "Enter twitter password: "); - session->password = readline(NULL); + read_password(password, sizeof(password), session->hostname); + session->password = strdup(password); } if (session->action == ACTION_UPDATE) { - if (session->bash) - tweet = readline(NULL); + if (session->bash || !session->interactive) + tweet = get_string_from_stdin(); else - tweet = readline("tweet: "); + tweet = session->readline("tweet: "); if (!tweet || strlen(tweet) == 0) { dbg("no tweet?\n"); return -1; } + if (session->shrink_urls) + tweet = shrink_urls(tweet); + session->tweet = zalloc(strlen(tweet) + 10); if (session->bash) - sprintf(session->tweet, "$ %s", tweet); + sprintf(session->tweet, "%c %s", + getuid() ? '$' : '#', tweet); else sprintf(session->tweet, "%s", tweet); @@ -738,8 +1311,11 @@ int main(int argc, char *argv[], char *envp[]) } if (!session->user) - session->user = session->account; + session->user = strdup(session->account); + if (session->page == 0) + session->page = 1; + dbg("config file = %s\n", session->configfile); dbg("account = %s\n", session->account); dbg("password = %s\n", session->password); dbg("host = %d\n", session->host); @@ -762,6 +1338,7 @@ int main(int argc, char *argv[], char *envp[]) log_session(session, retval); exit: + session_readline_cleanup(session); session_free(session); return retval;; }