Split rawr-ebooks and rawr-gen

Also wrote README

1 files changed, 169 insertions, 0 deletions

@@ -0,0 +1,169 @@
+#include <cstdio>
+#include <list>
+#include <map>
+#include "kgramstats.h"
+#include <ctime>
+#include <vector>
+#include <cstdlib>
+#include <fstream>
+#include <iostream>
+#include <twitcurl.h>
+#include <unistd.h>
+#include <yaml-cpp/yaml.h>
+
+using namespace::std;
+
+int main(int argc, char** args)
+{
+        srand(time(NULL));
+        
+        YAML::Node config = YAML::LoadFile("config.yml");       
+        ifstream infile(config["corpus"].as<std::string>().c_str());
+        string corpus;
+        string line;
+        while (getline(infile, line))
+        {
+                corpus += " " + line;
+        }
+
+    cout << "Preprocessing corpus..." << endl;
+        kgramstats* stats = new kgramstats(corpus, 5);
+
+    cout << "Generating..." << endl;
+        for (;;)
+        {
+                vector<string> doc = stats->randomSentence(rand() % 25 + 5);
+                string hi;
+                for (vector<string>::iterator it = doc.begin(); it != doc.end(); ++it)
+                {
+                        hi += *it + " ";
+                }
+        
+                hi = hi.substr(0,140);
+
+            twitCurl twitterObj;
+            std::string tmpStr, tmpStr2;
+            std::string replyMsg;
+            char tmpBuf[1024];
+                std::string username(config["username"].as<std::string>());
+                std::string password(config["password"].as<std::string>());
+
+            /* Set twitter username and password */
+            twitterObj.setTwitterUsername(username);
+            twitterObj.setTwitterPassword(password);
+        
+            /* OAuth flow begins */
+            /* Step 0: Set OAuth related params. These are got by registering your app at twitter.com */
+            twitterObj.getOAuth().setConsumerKey( config["consumer_key"].as<std::string>() );
+            twitterObj.getOAuth().setConsumerSecret( config["consumer_secret"].as<std::string>() );
+
+            /* Step 1: Check if we alredy have OAuth access token from a previous run */
+            std::string myOAuthAccessTokenKey("");
+            std::string myOAuthAccessTokenSecret("");
+            std::ifstream oAuthTokenKeyIn;
+            std::ifstream oAuthTokenSecretIn;
+
+            oAuthTokenKeyIn.open( "twitterClient_token_key.txt" );
+            oAuthTokenSecretIn.open( "twitterClient_token_secret.txt" );
+
+            memset( tmpBuf, 0, 1024 );
+            oAuthTokenKeyIn >> tmpBuf;
+            myOAuthAccessTokenKey = tmpBuf;
+
+            memset( tmpBuf, 0, 1024 );
+            oAuthTokenSecretIn >> tmpBuf;
+            myOAuthAccessTokenSecret = tmpBuf;
+
+            oAuthTokenKeyIn.close();
+            oAuthTokenSecretIn.close();
+
+            if( myOAuthAccessTokenKey.size() && myOAuthAccessTokenSecret.size() )
+            {
+                /* If we already have these keys, then no need to go through auth again */
+                printf( "\nUsing:\nKey: %s\nSecret: %s\n\n", myOAuthAccessTokenKey.c_str(), myOAuthAccessTokenSecret.c_str() );
+
+                twitterObj.getOAuth().setOAuthTokenKey( myOAuthAccessTokenKey );
+                twitterObj.getOAuth().setOAuthTokenSecret( myOAuthAccessTokenSecret );
+            }
+            else
+            {
+                /* Step 2: Get request token key and secret */
+                std::string authUrl;
+                twitterObj.oAuthRequestToken( authUrl );
+
+                /* Step 3: Get PIN  */
+                memset( tmpBuf, 0, 1024 );
+                printf( "\nDo you want to visit twitter.com for PIN (0 for no; 1 for yes): " );
+                gets( tmpBuf );
+                tmpStr = tmpBuf;
+                if( std::string::npos != tmpStr.find( "1" ) )
+                {
+                    /* Ask user to visit twitter.com auth page and get PIN */
+                    memset( tmpBuf, 0, 1024 );
+                    printf( "\nPlease visit this link in web browser and authorize this application:\n%s", authUrl.c_str() );
+                    printf( "\nEnter the PIN provided by twitter: " );
+                    gets( tmpBuf );
+                    tmpStr = tmpBuf;
+                    twitterObj.getOAuth().setOAuthPin( tmpStr );
+                }
+                else
+                {
+                    /* Else, pass auth url to twitCurl and get it via twitCurl PIN handling */
+                    twitterObj.oAuthHandlePIN( authUrl );
+                }
+
+                /* Step 4: Exchange request token with access token */
+                twitterObj.oAuthAccessToken();
+
+                /* Step 5: Now, save this access token key and secret for future use without PIN */
+                twitterObj.getOAuth().getOAuthTokenKey( myOAuthAccessTokenKey );
+                twitterObj.getOAuth().getOAuthTokenSecret( myOAuthAccessTokenSecret );
+
+                /* Step 6: Save these keys in a file or wherever */
+                std::ofstream oAuthTokenKeyOut;
+                std::ofstream oAuthTokenSecretOut;
+
+                oAuthTokenKeyOut.open( "twitterClient_token_key.txt" );
+                oAuthTokenSecretOut.open( "twitterClient_token_secret.txt" );
+
+                oAuthTokenKeyOut.clear();
+                oAuthTokenSecretOut.clear();
+
+                oAuthTokenKeyOut << myOAuthAccessTokenKey.c_str();
+                oAuthTokenSecretOut << myOAuthAccessTokenSecret.c_str();
+
+                oAuthTokenKeyOut.close();
+                oAuthTokenSecretOut.close();
+            }
+            /* OAuth flow ends */
+
+            /* Account credentials verification */
+            if( twitterObj.accountVerifyCredGet() )
+            {
+                twitterObj.getLastWebResponse( replyMsg );
+                printf( "\ntwitterClient:: twitCurl::accountVerifyCredGet web response:\n%s\n", replyMsg.c_str() );
+            }
+            else
+            {
+                twitterObj.getLastCurlError( replyMsg );
+                printf( "\ntwitterClient:: twitCurl::accountVerifyCredGet error:\n%s\n", replyMsg.c_str() );
+            }
+        
+            /* Post a new status message */
+            replyMsg = "";
+            if( twitterObj.statusUpdate( hi ) )
+            {
+                twitterObj.getLastWebResponse( replyMsg );
+                printf( "\ntwitterClient:: twitCurl::statusUpdate web response:\n%s\n", replyMsg.c_str() );
+            }
+            else
+            {
+                twitterObj.getLastCurlError( replyMsg );
+                printf( "\ntwitterClient:: twitCurl::statusUpdate error:\n%s\n", replyMsg.c_str() );
+            }
+
+                sleep(900);
+        }
+        
+        return 0;
+}
\ No newline at end of file