Angela - Twitter Poster

package.json

@@ -58,6 +58,7 @@
     "babel-plugin-module-resolver": "^5.0.0",
     "bcryptjs": "^2.4.3",
     "body-parser": "^1.18.3",
+    "cheerio": "^1.0.0-rc.12",
     "cors": "^2.8.4",
     "cron": "^1.8.2",
     "dotenv": "^5.0.1",
@@ -77,6 +78,7 @@
     "redis": "^4.2.0",
     "sanitize-html": "^2.13.0",
     "supertest": "^6.3.4",
+    "twitter-api-v2": "^1.17.2",
     "uuid": "^3.4.0",
     "ws": "^8.17.1"
   },

src/controllers/socialMediaController.js

@@ -0,0 +1,141 @@
+// eslint-disable-next-line import/no-extraneous-dependencies
+const fetch = require('node-fetch');
+// eslint-disable-next-line import/no-extraneous-dependencies
+const cheerio = require('cheerio');
+// eslint-disable-next-line import/no-extraneous-dependencies
+const { TwitterApi } = require('twitter-api-v2');
+
+function extractTextAndImgUrl(htmlString) {
+  const $ = cheerio.load(htmlString);
+
+  const textContent = $('body').text().replace(/\+/g, '').trim();
+  const urlSrcs = [];
+  const base64Srcs = [];
+
+  $('img').each((i, img) => {
+    const src = $(img).attr('src');
+    if (src) {
+      if (src.startsWith('data:image')) {
+        base64Srcs.push(src);
+      } else {
+        urlSrcs.push(src);
+      }
+    }
+  });
+
+  return { textContent, urlSrcs, base64Srcs };
+}
+
+async function downloadImage(url) {
+  const response = await fetch(url);
+  const arrayBuffer = await response.arrayBuffer();
+  return {
+    buffer: Buffer.from(arrayBuffer),
+    mimeType: response.headers.get('content-type'),
+  };
+}
+
+async function getTwitterAccessToken(req, res) {
+  const twitterOAuth = new TwitterApi({
+    clientId: process.env.REACT_APP_TWITTER_CLIENT_ID,
+    clientSecret: process.env.REACT_APP_TWITTER_CLIENT_SECRET,
+  });
+
+  const { code, state, codeVerifier } = req.body;
+
+  if (!code || !state || !codeVerifier) {
+    return res.status(400).json({ error: 'Missing required parameters' });
+  }
+
+  try {
+    twitterOAuth
+      .loginWithOAuth2({ code, codeVerifier, redirectUri: 'http://localhost:3000/announcements' })
+      .then(async ({ client: loggedClient, accessToken, expiresIn, scope }) => {
+        try {
+          const { data } = await loggedClient.v2.me();
+          console.log('User data:', data);
+          console.log('scope:', scope);
+          res.json({
+            access_token: accessToken,
+            expires_in: expiresIn,
+          });
+        } catch (error) {
+          console.error('API Error:', error);
+        }
+      });
+  } catch (error) {
+    console.error('Error exchanging code for token:', error);
+    res.status(500).json({ error: 'Failed to obtain access token' });
+  }
+}
+
+async function createTweet(req, res) {
+  const TwitterClient = new TwitterApi({
+    appKey: process.env.REACT_APP_TWITTER_APP_KEY,
+    appSecret: process.env.REACT_APP_TWITTER_APP_SECRET,
+    accessToken: process.env.REACT_APP_TWITTER_ACCESS_TOKEN,
+    accessSecret: process.env.REACT_APP_TWITTER_ACCESS_SECRET,
+  });
+
+  const rwClient = TwitterClient.readWrite;
+  const { textContent, urlSrcs, base64Srcs } = extractTextAndImgUrl(req.body.EmailContent);
+
+  try {
+    let mediaIds = [];
+    // src type is url
+    if (urlSrcs && urlSrcs.length > 0) {
+      console.log('Uploading URL media...');
+      const urlMediaIds = await Promise.all(
+        urlSrcs.map(async (imageUrl) => {
+          try {
+            const { buffer, mimeType } = await downloadImage(imageUrl);
+            return await rwClient.v1.uploadMedia(buffer, { mimeType });
+          } catch (error) {
+            console.error(`Error uploading URL image: ${imageUrl}`, error);
+            return null;
+          }
+        }),
+      );
+      mediaIds = mediaIds.concat(urlMediaIds.filter((id) => id !== null));
+      console.log('URL media uploaded, IDs:', mediaIds);
+    }
+
+    // src type is base64
+    if (base64Srcs && base64Srcs.length > 0) {
+      console.log('Uploading base64 media...');
+      const base64MediaIds = await Promise.all(
+        base64Srcs.map(async (imgSrc) => {
+          try {
+            const base64Data = imgSrc.replace(/^data:image\/\w+;base64,/, '');
+            const buffer = Buffer.from(base64Data, 'base64');
+            const mimeType = imgSrc.split(';')[0].split(':')[1] || 'image/png';
+            return await rwClient.v1.uploadMedia(buffer, { mimeType });
+          } catch (error) {
+            console.error('Error uploading base64 image', error);
+            return null;
+          }
+        }),
+      );
+      mediaIds = mediaIds.concat(base64MediaIds.filter((id) => id !== null));
+      console.log('Base64 media uploaded, IDs:', mediaIds);
+    }
+
+    const tweetOptions = { text: textContent };
+
+    if (mediaIds && mediaIds.length > 0) {
+      tweetOptions.media = { media_ids: mediaIds };
+    }
+
+    const tweet = await rwClient.v2.tweet(tweetOptions);
+
+    res.status(200).json({ success: true, tweet });
+  } catch (error) {
+    console.error('[Backend] Network or other error: ', error);
+    res.status(500).json({ success: false, error: 'Internal server error' });
+  }
+}
+
+module.exports = {
+  getTwitterAccessToken,
+  createTweet,
+};

src/routes/socialMediaRouter.js

@@ -0,0 +1,13 @@
+const express = require('express');
+const { getTwitterAccessToken, createTweet } = require('../controllers/socialMediaController');
+
+const routes = function () {
+  const socialMediaRouter = express.Router();
+
+  socialMediaRouter.route('/getTwitterAccessToken').post(getTwitterAccessToken);
+  socialMediaRouter.route('/createTweet').post(createTweet);
+
+  return socialMediaRouter;
+};
+
+module.exports = routes;

src/startup/routes.js

@@ -125,6 +125,8 @@ const blueSquareEmailAssignmentRouter = require('../routes/BlueSquareEmailAssign
   userProfile,
 );
 
+const socialMediaRouter = require('../routes/socialMediaRouter')();
+
 module.exports = function (app) {
   app.use('/api', forgotPwdRouter);
   app.use('/api', loginRouter);
@@ -173,4 +175,5 @@ module.exports = function (app) {
   app.use('/api/bm', bmEquipmentRouter);
   app.use('/api/bm', bmConsumablesRouter);
   app.use('api', bmIssueRouter);
+  app.use('/api', socialMediaRouter);
 };