Short link (#179)

SaTT-Wallet · May 18, 2023 · 1efe8a4 · 1efe8a4
2 parents d0c1d60 + 8981d64
commit 1efe8a4
Show file tree

Hide file tree

Showing 5 changed files with 720 additions and 17,237 deletions.
diff --git a/controllers/campaign.controller.js b/controllers/campaign.controller.js
@@ -9,6 +9,8 @@ const GridFsStorage = require('multer-gridfs-storage')
 const { create } = require('ipfs-http-client')
 var mongoose = require('mongoose')
 var fs = require('fs')
+const axios = require('axios');
+
 const cron = require('node-cron')
 //const ipfs = IPFS('ipfs.infura.io', '5001', {protocol: 'https'})
 const {
@@ -163,6 +165,8 @@ const { Constants, TronConstant, wrapConstants } = require('../conf/const')
 const { BigNumber } = require('ethers')
 const { token } = require('morgan')
 const { request } = require('http')
+const { URL } = require('url');
+const { http, https } = require('follow-redirects');
 const { Console } = require('console')
 
 //const conn = mongoose.createConnection(mongoConnection().mongoURI)
@@ -2830,34 +2834,32 @@ module.exports.initStat = () => {
         rejected: 0,
     }
 }
-module.exports.expandUrl = (req, res) => {
+module.exports.expandUrl = async (req, res) => {
+    const shortUrl = req.query.shortUrl;
+    const options = {
+      method: 'HEAD',
+      url: shortUrl,
+      maxRedirects: 5,
+      headers: {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/92.0.4515.107 Safari/537.36'
+      },
+      agent: shortUrl.startsWith('https') ? https : http
+    };
+
     try {
-        var child_process = require('child_process')
-        let { shortUrl } = req.query
-
-        function runCmd(cmd) {
-            var resp = child_process.execSync(cmd)
-            var result = resp.toString('UTF8')
-            return result
-        }
-        var cmd = `curl -sLI ${shortUrl} | grep -i Location`
-        var result = runCmd(cmd)
+      const response = await axios(options);
+      const expandedUrl = response.request.res.responseUrl || shortUrl;
+      const parsedUrl = new URL(expandedUrl);
 
-        return responseHandler.makeResponseData(
-            res,
-            200,
-            'shorted successfully',
-            result.split('Location: ')[1] || result.split('location: ')[1]
-        )
+      return responseHandler.makeResponseData(res, 200, 'success', parsedUrl)
     } catch (err) {
         return responseHandler.makeResponseError(
             res,
             500,
             err.message ? err.message : err.error
         )
     }
-}
-
+  };
 module.exports.statLinkCampaign = async (req, res) => {
     try {
         let id_campaign = req.params.hash

diff --git a/controllers/profile.controller.js b/controllers/profile.controller.js
@@ -1087,23 +1087,23 @@ module.exports.ProfilPrivacy = async (req, res) => {
         let tiktokProfile = await TikTokProfile.findOne({ userId })
         let getUrl = `https://open-api.tiktok.com/oauth/refresh_token?client_key=${process.env.TIKTOK_KEY}&grant_type=refresh_token&refresh_token=${tiktokProfile.refreshToken}`
         let resMedia = await rp.get(getUrl)
-        const linkedinData = {
+        const linkedinData = await{
             url: 'https://open.tiktokapis.com/v2/video/list/?fields=cover_image_url,id,title',
             method: 'POST',
             body: {
                 max_count: 20,
             },
             headers: {
-                Authorization: 'Bearer ' + resMedia?.data.access_token,
+                Authorization: 'Bearer ' + resMedia?.data.data.access_token,
             },
             json: true,
         }
         let postData = await rp.post('https://open.tiktokapis.com/v2/video/list/?fields=cover_image_url,id,title',{
             max_count: 20,
         },{headers :{
-            Authorization: 'Bearer ' + resMedia?.data.access_token,
+            Authorization: 'Bearer ' + resMedia?.data.data.access_token,
         }})
-        if (postData.data.videos.length === 0) {
+        if (postData.data.data.videos.length === 0) {
             privacy = 'private'
         } else {
             privacy = 'public'