Add files via upload

maicosilver · web-flow · commit 08e426d90e05 · 2021-01-27T16:14:00.000-03:00
diff --git a/lib/database.js b/lib/database.js
@@ -0,0 +1,33 @@
+const path = require('path')
+const fs = require('fs')
+
+class Database {
+    constructor(filepath, ...args) {
+        this.file = path.resolve(filepath)
+        this.load()
+        this._jsonargs = args
+    }
+
+    get data() {
+        return this._data
+    }
+
+    set data(value) {
+        this._data = value
+        this.save()
+    }
+
+    load() {
+        return this._data = fs.existsSync(this.file) ? JSON.parse(fs.readFileSync(this.file)) : {}
+    }
+
+    save() {
+        let dirname = path.dirname(this.file)
+        !fs.existsSync(dirname) && fs.mkdirSync(dirname, { recursive: true })
+        fs.writeFileSync(this.file, JSON.stringify(this._data, ...this._jsonargs))
+        return this.file
+    }
+}
+
+module.exports = Database
+
diff --git a/lib/print.js b/lib/print.js
@@ -0,0 +1,30 @@
+let { WA_MESSAGE_STUB_TYPES, URL_REGEX } = require('@adiwajshing/baileys')
+let chalk = require('chalk')
+
+module.exports = function (m, conn = {user: {}}) {
+  let sender = typeof m.text == 'string' || m.mtype ? [m.sender] : m.messageStubParameters
+  sender = sender.map(v => v.split`@`[0] + (conn.getName(v) ? ' ~' + conn.getName(v) : '')).join` & `
+  let chat = conn.getName(m.chat)
+  let ansi = '\x1b['
+  console.log(
+    `${chalk.red('%s')} ${chalk.bgYellow(chalk.black('%s'))} ${chalk.bgGreen(chalk.black('%s'))}\n${chalk.green('%s')} ${chalk.yellow('%s')} ${chalk.blue('to')} ${chalk.green('%s')} ${chalk.bgYellow(chalk.black('%s'))}`,
+    (conn.user.jid + ' ~' + conn.user.name).replace('@s.whatsapp.net', ''),
+    (m.messageTimestamp ? new Date(1000 * (m.messageTimestamp.low || m.messageTimestamp)) : new Date()).toTimeString(),
+    m.messageStubType ? WA_MESSAGE_STUB_TYPES[m.messageStubType] : '',
+    sender,
+    (m ? m.exp : '?') + (global.DATABASE.data.users[m.sender] ? '|' + global.DATABASE.data.users[m.sender].exp : ''),
+    m.chat + (chat ? ' ~' + chat : ''),
+    m.mtype ? m.mtype.replace(/message$/i, '') : ''
+  )
+  if (typeof m.text == 'string') console.log(
+    '%s\n',
+    m.text
+      .replace(RegExp('(https?:\\/\\/)?' + URL_REGEX.toString().slice(1, -4), URL_REGEX.flags), chalk.blue('$&'))
+      .replace(/(^|\s|\n)_(.+)?_($|\s|\n)/g, `$1${ansi}3m$2${ansi}23m$3`)
+      .replace(/(^|\s|\n)\*(.+)?\*($|\s|\n)/g, `$1${ansi}1m$2${ansi}22m$3`)
+      .replace(/(^|\s|\n)~(.+)?~($|\s|\n)/g, `$1${ansi}9m$2${ansi}29m$3`)
+      .replace(/(^|\s|\n)\x60{3}(.+)?\x60{3}($|\s|\n)/g, `$1$2$3`)
+  )
+}
+
+delete require.cache[require.resolve(__filename)]
diff --git a/lib/simple.js b/lib/simple.js
@@ -0,0 +1,203 @@
+const fs = require('fs')
+const util = require('util')
+const path = require('path')
+const FileType = require('file-type')
+const fetch = require('node-fetch')
+const { spawn } = require('child_process')
+const { MessageType } = require('@adiwajshing/baileys')
+
+exports.WAConnection = (_WAConnection) => {
+	class WAConnection extends _WAConnection {
+		constructor(...args) {
+		  super(...args)
+	  }
+
+    async waitEvent(eventName) {
+      return await (new Promise(resolve => this.once(eventName, resolve)))
+    }
+
+  	async sendFile(jid, path, filename = '', caption = '', quoted, ptt = false, options = {}) {
+	  	let file = Buffer.isBuffer(path) ? path : /^data:.*?\/.*?;base64,/i.test(path) ? Buffer.from(path.split`,`[1], 'base64') : /^https?:\/\//.test(path) ? await (await fetch(path)).buffer() : fs.existsSync(path) ? fs.readFileSync(path) : Buffer.alloc(0)
+  		const type = await FileType.fromBuffer(file) || {
+        mime: 'application/octet-stream',
+        ext: '.bin'
+      }
+      if (!type) {
+        options.asDocument = true
+      }
+  		let mtype = ''
+  		let opt = { filename, caption }
+      if (!options.asDocument) {
+        if (/audio/.test(type.mime)) file = await (ptt ? toPTT : toAudio)(file, type.ext)
+   		  else if (/video/.test(type.mime)) file = await toVideo(file, type.ext)
+    		if (/image/.test(type.mime)) mtype = MessageType.image
+    		else if (/video/.test(type.mime)) mtype = MessageType.video
+     		else opt.caption = filename
+     		if (/audio/.test(type.mime)) {
+    			mtype = MessageType.audio
+    			opt.ptt = ptt
+    		} else if (/pdf/.test(type.ext)) mtype = MessageType.pdf
+    		else if (!mtype) {
+          mtype = MessageType.document
+          opt.mimetype = type.mime
+        }
+        delete options.asDocument
+      } else {
+        mtype = MessageType.document
+        opt.mimetype = type.mime
+      }
+      if (quoted) opt.quoted = quoted
+  		return await this.sendMessage(jid, file, mtype, {...opt, ...options})
+  	}
+
+  	reply(jid, text, quoted, options) {
+  		return this.sendMessage(jid, text, MessageType.extendedText, { quoted, ...options })
+  	}
+	
+	  fakeReply(jid, text = '', fakeJid = this.user.jid, fakeText = '', fakeGroupJid) {
+  		return this.reply(jid, text, { key: { fromMe: fakeJid == this.user.jid, participant: fakeJid, ...(fakeGroupJid ? { remoteJid: fakeGroupJid } : {}) }, message: { conversation: fakeText }})
+  	}
+
+  	parseMention(text) {
+  		return [...text.matchAll(/@(\d{5,16})/g)].map(v => v[1] + '@s.whatsapp.net')
+  	}
+
+	  getName(jid)  {
+  		let v = jid === this.user.jid ? this.user : this.contacts[jid] || { notify: jid.replace(/@.+/, '') }
+  		return v.name || v.vname || v.notify
+  	}
+
+	  async downloadM(m) {
+      if (!m) return Buffer.alloc(0)
+    	if (!m.message) return Buffer.alloc(0)
+	  	if (!m.message[Object.keys(m.message)[0]].url) await this.updateMediaMessage(m)
+  		return await this.downloadMediaMessage(m)
+  	}
+	}
+
+  return WAConnection
+}
+
+exports.smsg = (conn, m, hasParent) => {
+  if (!m) return m
+	if (m.key) {
+		m.id = m.key.id
+    m.isBaileys = m.id.startsWith('3EB0') && m.id.length === 12
+		m.chat = m.key.remoteJid
+		m.fromMe = m.key.fromMe
+		m.isGroup = m.chat.endsWith('@g.us')
+		m.sender = m.fromMe ? conn.user.jid : m.isGroup ? m.participant : m.chat
+	}
+	if (m.message) {
+		m.mtype = Object.keys(m.message)[0]
+		m.msg = m.message[m.mtype]
+		if (m.mtype === 'ephemeralMessage') {
+			exports.smsg(conn, m.msg)
+			m.mtype = m.msg.mtype
+			m.msg = m.msg.msg
+		}
+		m.quoted = m.msg.contextInfo ? m.msg.contextInfo.quotedMessage : null
+		m.mentionedJid = m.msg.contextInfo ? m.msg.contextInfo.mentionedJid : []
+		if (m.quoted) {
+		  let type = Object.keys(m.quoted)[0]
+			m.quoted = m.quoted[type]
+      if (typeof m.quoted == 'string') m.quoted = { text: m.quoted }
+			m.quoted.mtype = type
+			m.quoted.id = m.msg.contextInfo.stanzaId
+      m.quoted.isBaileys = m.quoted.id.startsWith('3EB0') && m.quoted.id.length === 12
+			m.quoted.sender = m.msg.contextInfo.participant
+		  m.quoted.fromMe = m.quoted.sender == conn.user.jid
+			m.quoted.text = m.quoted.text || m.quoted.caption || ''
+			m.getQuotedObj = async () => {
+        let q
+        await conn.findMessage(m.chat, 25, s => {
+          q = s
+          return s.key ? s.key.id == m.quoted.id : false
+        })
+        return q && m.quoted.id == q.key.id ? exports.smsg(conn, q) : false
+      }
+		}
+		m.text = m.msg.text || m.msg.caption || m.msg || ''
+    m.reply = (text, chatId, options) => conn.reply(chatId ? chatId : m.chat, text, m,  options)
+	}
+}
+
+exports.logic = (check, inp, out) => {
+	if (inp.length !== out.length) throw new Error('Input and Output must have same length')
+	for (let i in inp) if (util.isDeepStrictEqual(check, inp[i])) return out[i]
+	return null
+}
+
+function toAudio(buffer, ext) {
+  return new Promise((resolve, reject) => {
+    let tmp = path.join(__dirname, '../tmp', (new Date * 1) + '.' + ext)
+    let out = tmp.replace(new RegExp(ext + '$'), 'mp3')
+    fs.writeFileSync(tmp, buffer)
+    spawn('ffmpeg', [
+      '-y',
+      '-i',tmp,
+      '-vn',
+      '-c:a','aac',
+      '-b:a','128k',
+      '-ar','44100',
+      '-f', 'mp3',
+      out
+    ])
+    .on('error', reject)
+    .on('error', () => fs.unlinkSync(tmp))
+    .on('close', () => {
+      resolve(fs.readFileSync(out))
+      fs.unlinkSync(tmp)
+      if (fs.existsSync(out)) fs.unlinkSync(out)
+    })
+  })
+}
+
+function toPTT(buffer, ext) {
+  return new Promise((resolve, reject) => {
+    let tmp = path.join(__dirname, '../tmp', (new Date * 1) + '.' + ext)
+    let out = tmp.replace(new RegExp(ext + '$'), 'opus')
+    fs.writeFileSync(tmp, buffer)
+    spawn('ffmpeg', [
+      '-y',
+      '-i',tmp,
+      '-vn',
+      '-c:a','libopus',
+      '-b:a','128k',
+      '-vbr','on',
+      '-compression_level','10',
+      out,
+    ])
+    .on('error', reject)
+    .on('error', () => fs.unlinkSync(tmp))
+    .on('close', () => {
+      resolve(fs.readFileSync(out))
+      fs.unlinkSync(tmp)
+      if (fs.existsSync(out)) fs.unlinkSync(out)
+    })
+  })
+}
+
+function toVideo(buffer, ext) {
+  return new Promise((resolve, reject) => {
+    let tmp = path.join(__dirname, '../tmp', (new Date * 1) + '.' + ext)
+    let out = tmp.replace(new RegExp(ext + '$'), 'mp4')
+    fs.writeFileSync(tmp, buffer)
+    spawn('ffmpeg', [
+      '-y',
+      '-i',tmp,
+      '-c:v','libx264',
+      '-c:a','aac',
+      '-ab','192k',
+      '-ar','44100',
+      out
+    ])
+    .on('error', reject)
+    .on('error', () => fs.unlinkSync(tmp))
+    .on('close', () => {
+      resolve(fs.readFileSync(out))
+      fs.unlinkSync(tmp)
+      if (fs.existsSync(out)) fs.unlinkSync(out)
+    })
+  })
+}
diff --git a/lib/webp.js b/lib/webp.js
