1 files changed, 34 insertions, 33 deletions

@@ -20,19 +20,19 @@ int main(int argc, char** argv)
 
   std::random_device random_device;
   std::mt19937 random_engine{random_device()};
-  
+
   YAML::Node config = YAML::LoadFile(configfile);
-    
+
   twitter::auth auth;
   auth.setConsumerKey(config["consumer_key"].as<std::string>());
   auth.setConsumerSecret(config["consumer_secret"].as<std::string>());
   auth.setAccessKey(config["access_key"].as<std::string>());
   auth.setAccessSecret(config["access_secret"].as<std::string>());
-  
+
   twitter::client client(auth);
-  
-  verbly::data database {config["verbly_datafile"].as<std::string>()};
-  
+
+  verbly::database database {config["verbly_datafile"].as<std::string>()};
+
   std::vector<std::string> emojis;
   {
     std::ifstream emojifile(config["emoji_file"].as<std::string>());
@@ -49,33 +49,33 @@ int main(int argc, char** argv)
       {
         line.pop_back();
       }
-    
+
       emojis.push_back(line);
     }
   }
-  
+
   std::uniform_int_distribution<int> emojidist(0, emojis.size()-1);
   std::bernoulli_distribution continuedist(1.0/2.0);
-  
+
+  verbly::query<verbly::form> verbQuery = database.forms(
+    (verbly::notion::partOfSpeech == verbly::part_of_speech::verb)
+    && (verbly::pronunciation::rhymes));
+
   for (;;)
   {
     std::cout << "Generating tweet..." << std::endl;
-    
-    std::string exclamation;
-    while (exclamation.empty())
-    {
-      verbly::verb v = database.verbs().random().limit(1).has_pronunciation().run().front();
-      auto rhyms = database.nouns().rhymes_with(v).random().limit(1).is_not_proper().run();
-      if (!rhyms.empty())
-      {
-        auto n = rhyms.front();
-        if (n.base_form() != v.base_form())
-        {
-          exclamation = "god " + v.base_form() + " this " + n.base_form();
-        }
-      }
-    }
-    
+
+    verbly::form verb = verbQuery.first();
+
+    verbly::form noun = database.forms(
+      (verbly::notion::partOfSpeech == verbly::part_of_speech::noun)
+      && (verbly::pronunciation::rhymes %= verb)
+      && (verbly::form::proper == false)
+      && (verbly::form::id != verb.getId())
+      && !(verbly::word::usageDomains %= (verbly::notion::wnid == 106718862))).first();
+
+    std::string exclamation = "god " + verb.getText() + " this " + noun.getText();
+
     std::string emojibef;
     std::string emojiaft;
     int emn = 0;
@@ -86,7 +86,7 @@ int main(int argc, char** argv)
       emojibef += em;
       emojiaft = em + emojiaft;
     } while (continuedist(random_engine));
-    
+
     std::string result;
     if (emn > 3)
     {
@@ -94,22 +94,23 @@ int main(int argc, char** argv)
     } else {
       result = emojibef + " " + exclamation + " " + emojiaft;
     }
-    
+
     result.resize(140);
     std::cout << result << std::endl;
-    
+
     try
     {
       client.updateStatus(result);
-      
+
       std::cout << "Tweeted!" << std::endl;
+      std::cout << "Waiting..." << std::endl;
+
+      std::this_thread::sleep_for(std::chrono::hours(1));
+
+      std::cout << std::endl;
     } catch (const twitter::twitter_error& e)
     {
       std::cout << "Twitter error: " << e.what() << std::endl;
     }
-    
-    std::cout << "Waiting..." << std::endl;
-    
-    std::this_thread::sleep_for(std::chrono::hours(1));
   }
 }