272 lines
6.9 KiB
JavaScript
272 lines
6.9 KiB
JavaScript
module.exports = function(request, fs) {
|
||
const config = require('../config')
|
||
this.downloadFile = (url) => {
|
||
return new Promise(resolve => {
|
||
request(url, { encoding: 'binary' }, (error, response, body) => {
|
||
if(!error && response.statusCode === 200) {
|
||
resolve({ success: true, data: body })
|
||
} else {
|
||
resolve({ success: false, data: null })
|
||
}
|
||
})
|
||
}).catch((err) => {
|
||
console.log(`Downloading media file failed for unknown reason. Details:`, err)
|
||
})
|
||
}
|
||
|
||
this.writeToDisk = (data, filename) => {
|
||
return new Promise(resolve => {
|
||
fs.writeFile(filename, data, 'binary', (error, result) => {
|
||
if(!error) {
|
||
resolve({ success: true })
|
||
} else {
|
||
resolve({ success: false })
|
||
}
|
||
})
|
||
}).catch((err) => {
|
||
console.log(`Writing media file to disk failed for unknown reason. Details:`, err)
|
||
})
|
||
}
|
||
|
||
this.logTimestamp = (date) => {
|
||
return date.toGMTString()
|
||
}
|
||
|
||
this.cleanUrl = (url) => {
|
||
return url.replace(/&/g, '&')
|
||
}
|
||
|
||
this.teddifyUrl = (url) => {
|
||
try {
|
||
let u = new URL(url)
|
||
if(u.host === 'www.reddit.com' || u.host === 'reddit.com') {
|
||
url = url.replace(u.host, config.domain)
|
||
if(u.pathname.startsWith('/gallery/'))
|
||
url = url.replace('/gallery/', '/comments/')
|
||
}
|
||
if(u.host === 'i.redd.it' || u.host === 'v.redd.it') {
|
||
let image_exts = ['png', 'jpg', 'jpeg']
|
||
let video_exts = ['mp4', 'gif', 'gifv']
|
||
let file_ext = getFileExtension(url)
|
||
if(image_exts.includes(file_ext))
|
||
url = url.replace(`${u.host}/`, `${config.domain}/pics/w:null_`)
|
||
if(video_exts.includes(file_ext) || !image_exts.includes(file_ext))
|
||
url = url.replace(u.host, `${config.domain}/vids`) + '.mp4'
|
||
}
|
||
|
||
} catch(e) { }
|
||
return url
|
||
}
|
||
|
||
this.kFormatter = (num) => {
|
||
return Math.abs(num) > 999 ? Math.sign(num)*((Math.abs(num)/1000).toFixed(1)) + 'k' : Math.sign(num)*Math.abs(num)
|
||
}
|
||
|
||
this.timeDifference = (time) => {
|
||
time = parseInt(time) * 1000
|
||
let ms_per_minute = 60 * 1000
|
||
let ms_per_hour = ms_per_minute * 60
|
||
let ms_per_day = ms_per_hour * 24
|
||
let ms_per_month = ms_per_day * 30
|
||
let ms_per_year = ms_per_day * 365
|
||
let current = + new Date()
|
||
|
||
let elapsed = Math.abs(time - current)
|
||
let r = ''
|
||
let e
|
||
|
||
if(elapsed < ms_per_minute) {
|
||
e = Math.round(elapsed/1000)
|
||
r = `${e} seconds ago`
|
||
if(e === 1)
|
||
r = 'just now'
|
||
return r
|
||
}
|
||
|
||
else if(elapsed < ms_per_hour) {
|
||
e = Math.round(elapsed/ms_per_minute)
|
||
r = `${e} minutes ago`
|
||
if(r === 1)
|
||
r = 'a minute ago'
|
||
return r
|
||
}
|
||
|
||
else if(elapsed < ms_per_day ) {
|
||
e = Math.round(elapsed/ms_per_hour)
|
||
r = `${e} hours ago`
|
||
if(e === 1)
|
||
r = 'an hour ago'
|
||
return r
|
||
}
|
||
|
||
else if(elapsed < ms_per_month) {
|
||
e = Math.round(elapsed/ms_per_day)
|
||
r = `${e} days ago`
|
||
if(e === 1)
|
||
r = '1 day ago'
|
||
return r
|
||
}
|
||
|
||
else if(elapsed < ms_per_year) {
|
||
e = Math.round(elapsed/ms_per_month)
|
||
r = `${e} months ago`
|
||
if(e === 1)
|
||
r = '1 month ago'
|
||
return r
|
||
}
|
||
|
||
else {
|
||
e = Math.round(elapsed/ms_per_year)
|
||
r = `${e} years ago`
|
||
if(e === 1)
|
||
r = '1 year ago'
|
||
return r
|
||
}
|
||
}
|
||
|
||
this.toUTCString = (time) => {
|
||
let d = new Date()
|
||
d.setTime(time*1000)
|
||
return d.toUTCString()
|
||
}
|
||
|
||
this.unescape = (s) => {
|
||
/* It would make much more sense to rename this function to something
|
||
* like "formatter".
|
||
*/
|
||
if(s) {
|
||
var re = /&(?:amp|#38|lt|#60|gt|#62|apos|#39|quot|#34);/g;
|
||
var unescaped = {
|
||
'&': '&',
|
||
'&': '&',
|
||
'<': '<',
|
||
'<': '<',
|
||
'>': '>',
|
||
'>': '>',
|
||
''': "'",
|
||
''': "'",
|
||
'"': '"',
|
||
'"': '"'
|
||
}
|
||
if(config.convert_urls.reddit) {
|
||
let str = s.replace(re, (m) => {
|
||
return unescaped[m]
|
||
})
|
||
let r = /([A-z.]+\.)?(reddit(\.com)|redd(\.it))/gm;
|
||
let result = str.replace(r, config.domain)
|
||
return result
|
||
} else {
|
||
return s.replace(re, (m) => {
|
||
return unescaped[m]
|
||
})
|
||
}
|
||
} else {
|
||
return ''
|
||
}
|
||
}
|
||
|
||
this.deleteFiles = (files, callback) => {
|
||
var i = files.length
|
||
files.forEach((filepath) => {
|
||
fs.unlink(filepath, (err) => {
|
||
i--
|
||
if(err) {
|
||
callback(err)
|
||
return
|
||
} else if(i <= 0) {
|
||
callback(null)
|
||
}
|
||
})
|
||
})
|
||
}
|
||
|
||
this.isGif = (url) => {
|
||
try {
|
||
url = new URL(url)
|
||
let pathname = url.pathname
|
||
let file_ext = pathname.substring(pathname.lastIndexOf('.') + 1)
|
||
if(file_ext === 'gif' || file_ext === 'gifv') {
|
||
return true
|
||
} else {
|
||
return false
|
||
}
|
||
} catch (error) {
|
||
console.error(`Invalid url supplied to isGif(). URL: ${url}`, error)
|
||
return false
|
||
}
|
||
}
|
||
|
||
this.getFileExtension = (url) => {
|
||
try {
|
||
url = new URL(url)
|
||
let pathname = url.pathname
|
||
let file_ext = pathname.substring(pathname.lastIndexOf('.') + 1)
|
||
if(file_ext) {
|
||
return file_ext
|
||
} else {
|
||
return ''
|
||
}
|
||
} catch (error) {
|
||
console.error(`Invalid url supplied to getFileExtension(). URL: ${url}`, error)
|
||
return ''
|
||
}
|
||
}
|
||
|
||
this.formatLinkFlair = async (post) => {
|
||
if (!config.flairs_enabled) {
|
||
return ''
|
||
}
|
||
|
||
const wrap = (inner) => `<span class="flair">${inner}</span>`
|
||
|
||
if (post.link_flair_text === null)
|
||
return ''
|
||
|
||
if (post.link_flair_type === 'text')
|
||
return wrap(post.link_flair_text)
|
||
|
||
if (post.link_flair_type === 'richtext') {
|
||
let flair = ''
|
||
for (let fragment of post.link_flair_richtext) {
|
||
if (fragment.e === 'text')
|
||
flair += fragment.t
|
||
else if (fragment.e === 'emoji')
|
||
flair += `<span class="emoji" style="background-image: url(${await downloadAndSave(fragment.u, 'flair_')})"></span>`
|
||
}
|
||
return wrap(flair)
|
||
}
|
||
|
||
return ''
|
||
}
|
||
|
||
this.formatUserFlair = async (post) => {
|
||
if (!config.flairs_enabled) {
|
||
return ''
|
||
}
|
||
|
||
// Generate the entire HTML here for consistency in both pug and HTML
|
||
const wrap = (inner) => `<span class="flair">${inner}</span>`
|
||
|
||
if (post.author_flair_text === null)
|
||
return ''
|
||
|
||
if (post.author_flair_type === 'text')
|
||
return wrap(post.author_flair_text)
|
||
|
||
if (post.author_flair_type === 'richtext') {
|
||
let flair = ''
|
||
for (let fragment of post.author_flair_richtext) {
|
||
// `e` seems to mean `type`
|
||
if (fragment.e === 'text')
|
||
flair += fragment.t // `t` is the text
|
||
else if (fragment.e === 'emoji')
|
||
flair += `<span class="emoji" style="background-image: url(${await downloadAndSave(fragment.u, 'flair_')})"></span>` // `u` is the emoji URL
|
||
}
|
||
return wrap(flair)
|
||
}
|
||
|
||
return ''
|
||
}
|
||
|
||
}
|