Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- var fs = require('fs');
- var readline = require('readline');
- var google = require('googleapis');
- var googleAuth = require('google-auth-library');
- var atob = require('atob');
- var pg = require('pg');
- const Language = require('@google-cloud/language');
- const language = Language({
- projectId: 'fine-balm-165720',
- keyFilename: 'MyProject.json'
- });
- global.fetch = require('node-fetch');
- var cc = require('cryptocompare');
- // If modifying these scopes, delete your previously saved credentials
- // at ~/.credentials/gmail-nodejs-quickstart.json
- var SCOPES = ['https://mail.google.com/'];
- var TOKEN_DIR = (process.env.HOME || process.env.HOMEPATH ||
- process.env.USERPROFILE) + '/.credentials/';
- var TOKEN_PATH = TOKEN_DIR + 'gmail-nodejs-quickstart.json';
- // Load client secrets from a local file.
- fs.readFile('client_secret.json', function processClientSecrets(err, content) {
- if (err) {
- console.log('Error loading client secret file: ' + err);
- return;
- }
- // Authorize a client with the loaded credentials, then call the
- // Gmail API.
- authorize(JSON.parse(content), listMessages);
- });
- /**
- * Create an OAuth2 client with the given credentials, and then execute the
- * given callback function.
- *
- * @param {Object} credentials The authorization client credentials.
- * @param {function} callback The callback to call with the authorized client.
- */
- function authorize(credentials, callback) {
- var clientSecret = credentials.installed.client_secret;
- var clientId = credentials.installed.client_id;
- var redirectUrl = credentials.installed.redirect_uris[0];
- var auth = new googleAuth();
- var oauth2Client = new auth.OAuth2(clientId, clientSecret, redirectUrl);
- // Check if we have previously stored a token.
- fs.readFile(TOKEN_PATH, function (err, token) {
- if (err) {
- getNewToken(oauth2Client, callback);
- } else {
- oauth2Client.credentials = JSON.parse(token);
- callback(oauth2Client);
- }
- });
- }
- /**
- * Get and store new token after prompting for user authorization, and then
- * execute the given callback with the authorized OAuth2 client.
- *
- * @param {google.auth.OAuth2} oauth2Client The OAuth2 client to get token for.
- * @param {getEventsCallback} callback The callback to call with the authorized
- * client.
- */
- function getNewToken(oauth2Client, callback) {
- var authUrl = oauth2Client.generateAuthUrl({
- access_type: 'offline',
- scope: SCOPES
- });
- console.log('Authorize this app by visiting this url: ', authUrl);
- var rl = readline.createInterface({
- input: process.stdin,
- output: process.stdout
- });
- rl.question('Enter the code from that page here: ', function (code) {
- rl.close();
- oauth2Client.getToken(code, function (err, token) {
- if (err) {
- console.log('Error while trying to retrieve access token', err);
- return;
- }
- oauth2Client.credentials = token;
- storeToken(token);
- callback(oauth2Client);
- });
- });
- }
- /**
- * Store token to disk be used in later program executions.
- *
- * @param {Object} token The token to store to disk.
- */
- function storeToken(token) {
- try {
- fs.mkdirSync(TOKEN_DIR);
- } catch (err) {
- if (err.code != 'EEXIST') {
- throw err;
- }
- }
- fs.writeFile(TOKEN_PATH, JSON.stringify(token));
- console.log('Token stored to ' + TOKEN_PATH);
- }
- /**
- * Lists the messages content in the user's account.
- *
- * @param {google.auth.OAuth2} auth An authorized OAuth2 client.
- * @param {labelIds} messages only from this label are displayed
- */
- var gmail = google.gmail('v1');
- function listMessages(auth) {
- gmail.users.messages.list({
- auth: auth,
- userId: 'me',
- maxResults: '500',
- q: 'newsletter@coindesk.com',
- labelIds: 'INBOX'
- }, function (err, response) {
- if (err) {
- console.log('The API returned an error: ' + err);
- return;
- }
- var messages = response.messages;
- if (undefined !== messages && messages.length == 0) {
- console.log('Messages not found.');
- } else {
- getMessageBody(messages, auth)
- }
- });
- }
- function getMessageBody(messages, auth) {
- if (undefined !== messages) {
- for (var i = 0; i < messages.length; i++) {
- var message = messages[i];
- gmail.users.messages.get({
- auth: auth,
- 'userId': 'me',
- 'id': message.id
- }, function (err, response) {
- if (err) {
- console.log('The API returned an error: ' + err);
- return;
- }
- var message = response.id;
- //modifyMessage(message, auth); // Label change
- var body = response.payload.parts[0].body.data;
- cc.price('BTC', 'USD')
- .then(prices => {
- language.detectSentiment(atob(body).split('(http://www.facebook.com')[0])
- .then((results) => {
- const sentiment = results[0];
- console.log(`Text: ${atob(body)}`); // VIEW TEXT MESSAGE
- console.log(atob(body).split('(http://www.facebook.com')[0]);
- console.log('ID:' + message + ` Sentiment score: ${sentiment.score}`);
- console.log(`Sentiment magnitude: ${sentiment.magnitude}`);
- console.log('');
- var sentimentScore = sentiment.score;
- sql(message, sentimentScore, prices)
- })
- .catch((err) => {
- console.error('ERROR:', err);
- });
- });
- }
- );
- }
- }
- }
- function modifyMessage(message, auth) {
- gmail.users.messages.modify({
- 'auth': auth,
- 'userId': 'me',
- 'id': message,
- 'resource': {
- "addLabelIds": [
- "Label_2"
- ],
- "removeLabelIds": [
- "INBOX"
- ]
- }
- }, function (err, response) {
- if (err) {
- console.log('The API returned an error: ' + err);
- return;
- }
- console.log('Success, label changed for message: ' + message)
- }
- );
- }
- // TODO ADD EVERYTHING TO THIS FUCKING DATABASE
- function sql(message, sentiment, price) {
- var username = 'kfpyhrzizvnthq';
- var password = '23fb755df500316578d72b2a62b443e2d638e1b17e60407f1838804eba1d7897';
- var host = 'ec2-184-73-199-72.compute-1.amazonaws.com';
- var conStringPri = 'postgres://' + username + ':' + password + '@' + host +
- ':5432/d4et7h51ifgegu';
- pg.defaults.ssl = true;
- var today = new Date().toISOString().replace(/T/, ' ').// replace T with a space
- replace(/\..+/, ''); // delete the dot and everything after
- pg.connect(conStringPri, function (err, client, done) {
- client.query('INSERT INTO antek(ID, DATE, SCORE, PRICE) values($1, $2, $3, $4)',
- [message, today, sentiment, price]);
- }
- );
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement