/** @module root/router */ 'use strict'; // vim: set ts=2 sw=2 et tw=80: const express = require('express'); const router = express.Router(); const mongoose = require('mongoose'); const Favorite = mongoose.model('Favorite'); const fetch = require('node-fetch'); const querystring = require('querystring'); const promiseAny = require('promise-any'); const eventBus = require('../../pubsub'); const fetch64 = require('fetch-base64'); router.get('/', (req, res) => { if (req.query.access_token) { req.app.locals.imgur = req.query; res.redirect('/'); } else { res.status(200, {'Content-Type': 'text/html'}); res.end(`
`); } }); async function fetchImgur(req, res, method, url, body) { const endpoint = 'https://api.imgur.com/3'; const response = await fetch(endpoint + url, { method: method, headers: { 'Content-Type': 'application/x-www-form-urlencoded', 'Authorization' : 'Bearer ' + req.app.locals.imgur.access_token }, body: body ? querystring.stringify(body) : undefined }); const json = await response.json(); if (json.status == 409 && url.match(/follow\/tag/)) { return {}; } if (Math.floor(json.status / 100) != 2) { res.status(json.status).json(json); return false; } return json; } router.post('/ordeal', async (req, res) => { try { if ((!req.body.replace && (!req.body.name || !req.body.album)) || (req.body.replace && (!req.body.name || !req.body.oldName))) { res.status(400).json({ error: "Malformed ordeal request" }); return; } const ordealResponse = { ordealSuccess: true }; const uploadData = { image: req.body.dataURL.substring('data:image/png;base64,'.length), type: 'base64', title: req.body.name, name: req.body.name + '.png', }; const uploadJson = await fetchImgur(req, res, 'POST', '/upload', uploadData); if (!uploadJson) { return; } const imageId = uploadJson.data.id; const albumData = { 'ids[]': imageId, }; if (!req.body.replace) { if (req.body.album) { albumData.title = req.body.album; } const albumJson = await fetchImgur(req, res, 'POST', '/album', albumData); if (!albumJson) { return; } if (req.body.favorites) { if (!await fetchImgur(req, res, 'POST', '/album/' + albumJson.data.id + '/favorite', {})) { return; } } } else { req.body.album = undefined; const submissions = await fetchImgur(req, res, 'GET', '/account/me/submissions'); if (!submissions) { return; } const submission = submissions.data .filter(e => !e.images && e.in_gallery && e.title == req.body.oldName)[0]; if (!submission) { res.status(404).json({ error: "No image in submissions found" }); return; } console.log('submission', submission.id, submission.title); const result = {}; result.image = submission; ordealResponse.views = submission.views; ordealResponse.votes = {}; ordealResponse.votes.ups = submission.ups; ordealResponse.votes.downs = submission.downs; const albumIds = await fetchImgur(req, res, 'GET', '/account/me/albums/ids'); if (!albumIds) { return; } console.log('albumIds', albumIds.data.join(',')); let reject = false; const promises = []; for (const albumId of albumIds.data) { promises.push(new Promise(async (res, rej) => { if (reject) { return reject; } const images = await fetchImgur(req, res, 'GET', '/album/' + albumId + '/images'); if (!images || reject) { rej(); return; } console.log('album', albumId, 'images', images.data.map(e => e.id)); const image = images.data.filter(e => e.id == result.image.id)[0]; if (!image || reject) { rej(); return; } reject = true; console.log('the album', albumId, 'image', image.id); res(albumId); })); } try { result.albumId = await promiseAny(promises); } catch(_) { res.status(400).json({ error: "Image not found in any album" }); return; } if (!await fetchImgur(req, res, 'DELETE', '/image/' + result.image.id)) { return; } if (!await fetchImgur(req, res, 'PUT', '/album/' + result.albumId + '/add', { 'ids[]': uploadJson.data.id })) { return; } } if (!await fetchImgur(req, res, 'POST', '/gallery/image/' + uploadJson.data.id, { title: req.body.name, tags: req.body.tags })) { return; } if (req.body.tags) { const tagsData = { 'tags': req.body.tags }; if (!await fetchImgur(req, res, 'POST', '/gallery/tags/' + uploadJson.data.id, tagsData)) { return; } for (const tag of req.body.tags.split(',')) { if (!await fetchImgur(req, res, 'POST', '/account/me/follow/tag/' + tag, {})) { return; } } } res.json(ordealResponse); } catch(e) { console.error(e); res.status(500).json({ error: e.toString() }); } }); router.post('/search', async (req, res) => { try { if (!req.body.name) { res.status(400).json({ error: 'Bad parameters for /imgur/search' }); return; } const results = await fetchImgur(req, res, 'GET', '/gallery/search/time/top/1?q=' + encodeURIComponent(req.body.name) + '&q_type=png'); if (!results) { return; } if (results.data.length == 0) { res.status(404).json({ error: 'No image found' }); return; } let image = results.data[0]; if (image.is_album) { image = image.images[0]; } if (!await fetchImgur(req, res, 'POST', '/gallery/' + results.data[0].id + '/vote/up', {})) { return; } if (!await fetchImgur(req, res, 'POST', '/image/' + image.id + '/favorite', {})) { return; } const fetch64res = await fetch64.remote(image.link); const fav = { _id: mongoose.Types.ObjectId(), name: req.body.name, dataURL: fetch64res[1], bookmarked: false, }; await new Favorite(fav).save(); eventBus.emit('favorite.created', fav); res.json({ success: true }); } catch(e) { console.error(e); res.status(500).json({ error: e.toString() }); } }); /** router for /root */ module.exports = router;