]> ToastFreeware Gitweb - gregoa/bti.git/blobdiff - bti.c
Merge branch 'master' of gregkh@master.kernel.org:/pub/scm/linux/kernel/git/gregkh/bti
[gregoa/bti.git] / bti.c
diff --git a/bti.c b/bti.c
index 5f4a6cb20da98f2f13300ad3d5eafa0fc75460eb..28d205dc960b069d07eb926c8f2bbe8eb15cbff6 100644 (file)
--- a/bti.c
+++ b/bti.c
@@ -16,6 +16,8 @@
  * 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  */
 
+#define _GNU_SOURCE
+
 #include <stdio.h>
 #include <stdlib.h>
 #include <stddef.h>
@@ -35,7 +37,6 @@
 #include <libxml/parser.h>
 #include <libxml/tree.h>
 #include <pcre.h>
-#include "bti_version.h"
 
 
 #define zalloc(size)   calloc(size, 1)
@@ -43,7 +44,8 @@
 #define dbg(format, arg...)                                            \
        do {                                                            \
                if (debug)                                              \
-                       printf("%s: " format , __func__ , ## arg);      \
+                       fprintf(stdout, "bti: %s: " format , __func__ , \
+                               ## arg);                                \
        } while (0)
 
 
@@ -75,6 +77,7 @@ struct session {
        int bash;
        int shrink_urls;
        int dry_run;
+       int page;
        enum host host;
        enum action action;
 };
@@ -88,28 +91,31 @@ struct bti_curl_buffer {
 static void display_help(void)
 {
        fprintf(stdout, "bti - send tweet to twitter or identi.ca\n");
-       fprintf(stdout, "Version: " BTI_VERSION "\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' or 'user')\n");
+       fprintf(stdout, "    ('update', 'friends', 'public', 'replies' "
+               "or 'user')\n");
        fprintf(stdout, "  --user screenname\n");
        fprintf(stdout, "  --proxy PROXY:PORT\n");
        fprintf(stdout, "  --host HOST\n");
        fprintf(stdout, "  --logfile logfile\n");
        fprintf(stdout, "  --shrink-urls\n");
+       fprintf(stdout, "  --page PAGENUMBER\n");
        fprintf(stdout, "  --bash\n");
        fprintf(stdout, "  --debug\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 struct session *session_alloc(void)
@@ -191,15 +197,18 @@ 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;
        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 *)"user")) {
@@ -213,12 +222,16 @@ void parse_statuses(xmlDocPtr doc, xmlNodePtr current)
                                        userinfo = userinfo->next;
                                }
                        }
-                       if (user && text) {
-                               printf("[%s] %s\n", user, text);
+
+                       if (user && text && created) {
+                               printf("[%s] (%.16s) %s\n",
+                                       user, created, text);
                                xmlFree(user);
                                xmlFree(text);
+                               xmlFree(created);
                                user = NULL;
                                text = NULL;
+                               created = NULL;
                        }
                }
                current = current->next;
@@ -231,8 +244,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;
 
@@ -260,7 +274,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;
@@ -331,10 +346,12 @@ static int send_request(struct session *session)
                curl_easy_setopt(curl, CURLOPT_HTTPHEADER, slist);
                switch (session->host) {
                case HOST_TWITTER:
-                       curl_easy_setopt(curl, CURLOPT_URL, twitter_update_url);
+                       curl_easy_setopt(curl, CURLOPT_URL,
+                                        twitter_update_url);
                        break;
                case HOST_IDENTICA:
-                       curl_easy_setopt(curl, CURLOPT_URL, identica_update_url);
+                       curl_easy_setopt(curl, CURLOPT_URL,
+                                        identica_update_url);
                        break;
                }
                curl_easy_setopt(curl, CURLOPT_USERPWD, user_password);
@@ -345,10 +362,12 @@ static int send_request(struct session *session)
                         session->account, session->password);
                switch (session->host) {
                case HOST_TWITTER:
-                       curl_easy_setopt(curl, CURLOPT_URL, twitter_friends_url);
+                       sprintf(user_url, "%s?page=%d", twitter_friends_url, session->page);
+                       curl_easy_setopt(curl, CURLOPT_URL, user_url);
                        break;
                case HOST_IDENTICA:
-                       curl_easy_setopt(curl, CURLOPT_URL, identica_friends_url);
+                       sprintf(user_url, "%s?page=%d", identica_friends_url, session->page);
+                       curl_easy_setopt(curl, CURLOPT_URL, user_url);
                        break;
                }
                curl_easy_setopt(curl, CURLOPT_USERPWD, user_password);
@@ -357,11 +376,11 @@ static int send_request(struct session *session)
        case ACTION_USER:
                switch (session->host) {
                case HOST_TWITTER:
-                       sprintf(user_url, "%s%s.xml", twitter_user_url, session->user);
+                       sprintf(user_url, "%s%s.xml?page=%d", twitter_user_url, session->user, session->page);
                        curl_easy_setopt(curl, CURLOPT_URL, user_url);
                        break;
                case HOST_IDENTICA:
-                       sprintf(user_url, "%s%s.xml", identica_user_url, session->user);
+                       sprintf(user_url, "%s%s.xml?page=%d", identica_user_url, session->user, session->page);
                        curl_easy_setopt(curl, CURLOPT_URL, user_url);
                        break;
                }
@@ -372,10 +391,12 @@ static int send_request(struct session *session)
                         session->account, session->password);
                switch (session->host) {
                case HOST_TWITTER:
-                       curl_easy_setopt(curl, CURLOPT_URL, twitter_replies_url);
+                       sprintf(user_url, "%s?page=%d", twitter_replies_url, session->page);
+                       curl_easy_setopt(curl, CURLOPT_URL, user_url);
                        break;
                case HOST_IDENTICA:
-                       curl_easy_setopt(curl, CURLOPT_URL, identica_replies_url);
+                       sprintf(user_url, "%s?page=%d", identica_replies_url, session->page);
+                       curl_easy_setopt(curl, CURLOPT_URL, user_url);
                        break;
                }
                curl_easy_setopt(curl, CURLOPT_USERPWD, user_password);
@@ -384,10 +405,12 @@ static int send_request(struct session *session)
        case ACTION_PUBLIC:
                switch (session->host) {
                case HOST_TWITTER:
-                       curl_easy_setopt(curl, CURLOPT_URL, twitter_public_url);
+                       sprintf(user_url, "%s?page=%d", twitter_public_url, session->page);
+                       curl_easy_setopt(curl, CURLOPT_URL, user_url);
                        break;
                case HOST_IDENTICA:
-                       curl_easy_setopt(curl, CURLOPT_URL, identica_public_url);
+                       sprintf(user_url, "%s?page=%d", identica_public_url, session->page);
+                       curl_easy_setopt(curl, CURLOPT_URL, user_url);
                        break;
                }
 
@@ -411,7 +434,8 @@ static int send_request(struct session *session)
        if (!session->dry_run) {
                res = curl_easy_perform(curl);
                if (res && !session->bash) {
-                       fprintf(stderr, "error(%d) trying to perform operation\n", res);
+                       fprintf(stderr, "error(%d) trying to perform "
+                               "operation\n", res);
                        return -EINVAL;
                }
        }
@@ -544,9 +568,8 @@ static void parse_configfile(struct session *session)
                        session->action = ACTION_UNKNOWN;
                free(action);
        }
-       if (user) {
+       if (user)
                session->user = user;
-       }
        session->shrink_urls = shrink_urls;
 
        /* Free buffer and close file.  */
@@ -634,7 +657,10 @@ static char *get_string_from_stdin(void)
 
 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
+       /*
+        * 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;/~?:@&=+$\\.\\-_'()%]+)"
@@ -655,23 +681,24 @@ static int find_urls(const char *tweet, int **pranges)
                        &errptr, &erroffset, NULL);
        if (!re) {
                fprintf(stderr, "pcre_compile @%u: %s\n", erroffset, errptr);
-               exit (1);
+               exit(1);
        }
 
        tweetlen = strlen(tweet);
-       for (startoffset=0; startoffset<tweetlen; ) {
+       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);
+               if (rc < 0) {
+                       fprintf(stderr, "pcre_exec @%u: %s\n",
+                               erroffset, errptr);
+                       exit(1);
                }
 
-               for (i=0; i<rc; i+=2) {
+               for (i = 0; i < rc; i += 2) {
                        if ((rcount+2) == rbound) {
                                rbound *= 2;
                                ranges = realloc(ranges, sizeof(int) * rbound);
@@ -711,19 +738,20 @@ static int popenRWE(int *rwepipe, const char *exe, const char *const argv[])
        int rc;
 
        rc = pipe(in);
-       if (rc<0)
+       if (rc < 0)
                goto error_in;
 
        rc = pipe(out);
-       if (rc<0)
+       if (rc < 0)
                goto error_out;
 
        rc = pipe(err);
-       if (rc<0)
+       if (rc < 0)
                goto error_err;
 
        pid = fork();
-       if (pid > 0) { // parent
+       if (pid > 0) {
+               /* parent */
                close(in[0]);
                close(out[1]);
                close(err[1]);
@@ -731,7 +759,8 @@ static int popenRWE(int *rwepipe, const char *exe, const char *const argv[])
                rwepipe[1] = out[0];
                rwepipe[2] = err[0];
                return pid;
-       } else if (pid == 0) { // child
+       } else if (pid == 0) {
+               /* child */
                close(in[1]);
                close(out[0]);
                close(err[0]);
@@ -742,7 +771,7 @@ static int popenRWE(int *rwepipe, const char *exe, const char *const argv[])
                close(2);
                rc = dup(err[1]);
 
-               execvp(exe, (char**)argv);
+               execvp(exe, (char **)argv);
                exit(1);
        } else
                goto error_fork;
@@ -799,7 +828,7 @@ static char *shrink_one_url(int *rwepipe, char *big)
        while (smalllen && isspace(small[smalllen-1]))
                        small[--smalllen] = 0;
 
-       free (big);
+       free(big);
        return small;
 
 error_free_small:
@@ -829,8 +858,10 @@ static char *shrink_urls(char *text)
                return text;
 
        rcount = find_urls(text, &ranges);
+       if (!rcount)
+               return text;
 
-       for (i=0; i<rcount; i+=2) {
+       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;
@@ -844,7 +875,8 @@ static char *shrink_urls(char *text)
                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
+                       /* The short url ended up being too long
+                        * or unavailable */
                        if (inofs) {
                                strncpy(text + outofs, text + inofs,
                                                not_url_len + long_url_len);
@@ -853,21 +885,21 @@ static char *shrink_urls(char *text)
                        outofs += not_url_len + long_url_len;
 
                } else {
-                       // copy the unmodified block
+                       /* copy the unmodified block */
                        strncpy(text + outofs, text + inofs, not_url_len);
                        inofs += not_url_len;
                        outofs += not_url_len;
 
-                       // copy the new url
+                       /* copy the new url */
                        strncpy(text + outofs, url, short_url_len);
                        inofs += long_url_len;
                        outofs += short_url_len;
                }
 
-               free (url);
+               free(url);
        }
 
-       // copy the last block after the last match
+       /* copy the last block after the last match */
        if (inofs) {
                int tail = inlen - inofs;
                if (tail) {
@@ -900,6 +932,7 @@ int main(int argc, char *argv[], char *envp[])
                { "help", 0, NULL, 'h' },
                { "bash", 0, NULL, 'b' },
                { "dry-run", 0, NULL, 'n' },
+               { "page", 1, NULL, 'g' },
                { "version", 0, NULL, 'v' },
                { }
        };
@@ -910,6 +943,7 @@ int main(int argc, char *argv[], char *envp[])
        int option;
        char *http_proxy;
        time_t t;
+       int page_nr;
 
        debug = 0;
        rl_bind_key('\t', rl_insert);
@@ -942,7 +976,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, "dqe:p:P:H:a:A:u:hg:sn",
                                          options, NULL);
                if (option == -1)
                        break;
@@ -956,6 +990,11 @@ int main(int argc, char *argv[], char *envp[])
                        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 'p':
                        if (session->password)
                                free(session->password);
@@ -1023,9 +1062,17 @@ int main(int argc, char *argv[], char *envp[])
                }
        }
 
+       /*
+        * 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' or 'user'.\n");
+               fprintf(stderr, "'update', 'friends', 'public', "
+                       "'replies' or 'user'.\n");
                goto exit;
        }
 
@@ -1054,7 +1101,7 @@ int main(int argc, char *argv[], char *envp[])
 
                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);
 
@@ -1065,6 +1112,8 @@ int main(int argc, char *argv[], char *envp[])
        if (!session->user)
                session->user = strdup(session->account);
 
+       if (session->page == 0)
+               session->page = 1;
        dbg("account = %s\n", session->account);
        dbg("password = %s\n", session->password);
        dbg("host = %d\n", session->host);