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

.gitignore

@@ -62,3 +62,6 @@ typings/
 
 # credentials
 credentials/*.json
+
+# webstorm ide files
+.idea

index.js

@@ -1,29 +1,42 @@
-const readline = require('readline-sync')
+const prompts = require('prompts')
 const robots = {
   text: require('./robots/text.js')
 }
+const trends = require('./searchHotTrends');
+
+async function askAndReturnAnswers() {
+  const questions = [
+    {
+      type: 'select',
+      name: 'searchTerm',
+      message: 'Choose one search term:',
+      choices: await trends.searchHotTrends(count=9),
+      validate: value => typeof value === 'string' ? value.trim() !== '' : false,
+    },
+    {
+      type: 'select',
+      name: 'prefix',
+      message: 'Choose one option:',
+      choices: ['Who is', 'What is', 'The history of'],
+      validate: value => typeof value === 'string' ? value.trim() !== '' : false,
+    }
+  ];
+
+  return new Promise(async (resolve, reject) => {
+    const promptOptions = {
+      onCancel: () => reject(new Error('The user has stopped answering'))
+    }
+    const response = await prompts(questions, promptOptions)
+    resolve(response)
+  });
+}
 
 async function start() {
-  const content = {}
-
-  content.searchTerm = askAndReturnSearchTerm()
-  content.prefix = askAndReturnPrefix()
+  const content = await askAndReturnAnswers()
 
   await robots.text(content)
 
-  function askAndReturnSearchTerm() {
-    return readline.question('Type a Wikipedia search 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
-  }
-
-  console.log(content)
+  console.log(content);
 }
 
-start()
+start();

package.json

@@ -18,7 +18,7 @@
   "homepage": "https://github.com/filipedeschamps/video-maker#readme",
   "dependencies": {
     "algorithmia": "^0.3.10",
-    "readline-sync": "^1.4.9",
+    "prompts": "^2.0.3",
     "sbd": "^1.0.14"
   }
 }

robots/text.js

@@ -10,8 +10,8 @@ async function robot(content) {
   async function fetchContentFromWikipedia(content) {
     const algorithmiaAuthenticated = algorithmia(algorithmiaApiKey)
     const wikipediaAlgorithm = algorithmiaAuthenticated.algo('web/WikipediaParser/0.1.2')
-    const wikipediaResponde = await wikipediaAlgorithm.pipe(content.searchTerm)
-    const wikipediaContent = wikipediaResponde.get()
+    const wikipediaResponse = await wikipediaAlgorithm.pipe(content.searchTerm)
+    const wikipediaContent = wikipediaResponse.get()
 
     content.sourceContentOriginal = wikipediaContent.content
   }

searchHotTrends.js

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