Adicionando busca automática de termos no Google Trends e generalizando a função de user input

.gitignore

@@ -60,5 +60,5 @@ typings/
 # next.js build output
 .next
 
-# credentials
-credentials/*.json
+# webstorm ide files
+.idea

index.js

@@ -1,29 +1,38 @@
-const readline = require('readline-sync')
-const robots = {
-  text: require('./robots/text.js')
-}
-
-async function start() {
-  const content = {}
+const readline = require('readline-sync');
+const trends = require('./searchHotTrends');
 
-  content.searchTerm = askAndReturnSearchTerm()
-  content.prefix = askAndReturnPrefix()
+function start() {
+  const searchFields = {};
 
-  await robots.text(content)
+  askAndReturnSearchTerm()
+      .then(term => {
+        searchFields.searchTerm = term;
+        searchFields.prefix = askAndReturnPrefix();
+        console.log(searchFields);
+      });
 
   function askAndReturnSearchTerm() {
-    return readline.question('Type a Wikipedia search term: ')
+    let typedTerm = readline.question('Type a search term or press <Enter> to get a list of hot terms: ');
+    if (typedTerm)
+      return new Promise(resolve => resolve(typedTerm));
+    else {
+      console.log('Auto searching for hot terms online...');
+      return trends.searchHotTrends()
+          .then(function (hotTerms) {
+            return getUserOptionInput(hotTerms, 'Choose on term: ');
+          });
+    }
   }
 
-  function askAndReturnPrefix() {
-    const prefixes = ['Who is', 'What is', 'The history of']
-    const selectedPrefixIndex = readline.keyInSelect(prefixes, 'Choose one option: ')
-    const selectedPrefixText = prefixes[selectedPrefixIndex]
-
-    return selectedPrefixText
+  function getUserOptionInput(options, message) {
+    const selectedOption = readline.keyInSelect(options, message);
+    return options[selectedOption];
   }
 
-  console.log(content)
+  function askAndReturnPrefix() {
+    const prefixes = ['Who is', 'What is', 'The history of'];
+    return getUserOptionInput(prefixes, 'Choose one option: ');
+  }
 }
 
-start()
+start();

package.json

@@ -17,8 +17,7 @@
   },
   "homepage": "https://github.com/filipedeschamps/video-maker#readme",
   "dependencies": {
-    "algorithmia": "^0.3.10",
-    "readline-sync": "^1.4.9",
-    "sbd": "^1.0.14"
+    "google-trends-api": "^4.9.0",
+    "readline-sync": "^1.4.9"
   }
 }

searchHotTrends.js

@@ -0,0 +1,23 @@
+const googleTrends = require('google-trends-api');
+
+module.exports = {
+  searchHotTrends: function () {
+    let trendsSettings = {
+      trendDate: new Date(),
+      geo: 'BR',
+      hl: "pt-BR"
+    };
+
+    return googleTrends.realTimeTrends(trendsSettings)
+      .then(results => {
+        let hotTrendStories = JSON.parse(results)['storySummaries']['trendingStories'];
+        let hotTrends = [];
+        hotTrendStories.forEach((story) => hotTrends = hotTrends.concat(story['entityNames']));
+        return hotTrends.slice(0, 9);
+      })
+      .catch(error => {
+        console.error('An error occurs: ', error);
+        return [];
+      });
+  }
+};