mirror of
https://github.com/FranP-code/Baileys.git
synced 2025-10-13 00:32:22 +00:00
Initial Conversion to typescript
This commit is contained in:
Adhiraj Singh
97
Example/ConversationExtract.js
Normal file
97
Example/ConversationExtract.js
Normal file
@@ -0,0 +1,97 @@
|
||||
const WhatsAppWeb = require("../WhatsAppWeb")
|
||||
const fs = require("fs")
|
||||
|
||||
/**
|
||||
* Extract all your WhatsApp conversations & save them to a file
|
||||
* produceAnonData => should the Id of the chat be recorded
|
||||
* */
|
||||
function extractChats (authCreds, outputFile, produceAnonData=false, offset=null) {
|
||||
let client = new WhatsAppWeb() // instantiate an instance
|
||||
// internal extract function
|
||||
const extract = function () {
|
||||
let rows = 0
|
||||
let chats = Object.keys(client.chats)
|
||||
let encounteredOffset
|
||||
if (offset) {
|
||||
encounteredOffset = false
|
||||
} else {
|
||||
encounteredOffset = true
|
||||
fs.writeFileSync(outputFile, "chat,input,output\n") // write header to file
|
||||
}
|
||||
|
||||
const extractChat = function (index) {
|
||||
const id = chats[index]
|
||||
if (id.includes("g.us") || !encounteredOffset) { // skip groups
|
||||
if (id === offset) {
|
||||
encounteredOffset = true
|
||||
}
|
||||
if (index+1 < chats.length) {
|
||||
return extractChat(index+1)
|
||||
}
|
||||
return
|
||||
}
|
||||
console.log("extracting for " + id + "...")
|
||||
|
||||
var curInput = ""
|
||||
var curOutput = ""
|
||||
var lastMessage
|
||||
return client.loadEntireConversation (id, m => {
|
||||
var text
|
||||
if (!m.message) { // if message not present, return
|
||||
return
|
||||
} else if (m.message.conversation) { // if its a plain text message
|
||||
text = m.message.conversation
|
||||
} else if (m.message.extendedTextMessage && m.message.extendedTextMessage.contextInfo) { // if its a reply to a previous message
|
||||
const mText = m.message.extendedTextMessage.text
|
||||
const quotedMessage = m.message.extendedTextMessage.contextInfo.quotedMessage
|
||||
// if it's like a '.' and the quoted message has no text, then just forget it
|
||||
if (mText.length <= 2 && !quotedMessage.conversation) {
|
||||
return
|
||||
}
|
||||
// if somebody sent like a '.', then the text should be the quoted message
|
||||
if (mText.length <= 2) {
|
||||
text = quotedMessage.conversation
|
||||
} else { // otherwise just use this text
|
||||
text = mText
|
||||
}
|
||||
} else {
|
||||
return
|
||||
}
|
||||
// if the person who sent the message has switched, flush the row
|
||||
if (lastMessage && !m.key.fromMe && lastMessage.key.fromMe) {
|
||||
|
||||
let row = "" + (produceAnonData ? "" : id) + ",\"" + curInput + "\",\"" + curOutput + "\"\n"
|
||||
fs.appendFileSync (outputFile, row)
|
||||
rows += 1
|
||||
curInput = ""
|
||||
curOutput = ""
|
||||
}
|
||||
|
||||
if (m.key.fromMe) {
|
||||
curOutput += curOutput === "" ? text : ("\n"+text)
|
||||
} else {
|
||||
curInput += curInput === "" ? text : ("\n"+text)
|
||||
}
|
||||
|
||||
lastMessage = m
|
||||
}, 50, false) // load from the start, in chunks of 50
|
||||
.then (() => console.log("finished extraction for " + id))
|
||||
.then (() => {
|
||||
if (index+1 < chats.length) {
|
||||
return extractChat(index+1)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
extractChat(0)
|
||||
.then (() => {
|
||||
console.log("extracted all; total " + rows + " rows")
|
||||
client.logout ()
|
||||
})
|
||||
}
|
||||
client.connect (authCreds)
|
||||
.then (() => extract())
|
||||
.catch (err => console.log("got error: " + error))
|
||||
}
|
||||
let creds = null//JSON.parse(fs.readFileSync("auth_info.json"))
|
||||
extractChats(creds, "output.csv")
|
||||
144
Example/example.ts
Normal file
144
Example/example.ts
Normal file
@@ -0,0 +1,144 @@
|
||||
import {
|
||||
WAClient,
|
||||
AuthenticationCredentialsBase64,
|
||||
getNotificationType,
|
||||
MessageType,
|
||||
decodeMediaMessage,
|
||||
Presence,
|
||||
MessageOptions,
|
||||
Mimetype,
|
||||
} from '../WAClient/WAClient'
|
||||
import fs from 'fs'
|
||||
|
||||
async function example() {
|
||||
const client = new WAClient() // instantiate
|
||||
client.autoReconnect = true // auto reconnect on disconnect
|
||||
client.logUnhandledMessages = false // set to true to see what kind of stuff you can implement
|
||||
|
||||
let authInfo: AuthenticationCredentialsBase64 = null
|
||||
try {
|
||||
const file = fs.readFileSync('auth_info.json') // load a closed session back if it exists
|
||||
authInfo = JSON.parse(file)
|
||||
} catch {}
|
||||
|
||||
// connect or timeout in 20 seconds
|
||||
const [user, chats, contacts, unread] = await client.connect(authInfo, 20 * 1000)
|
||||
|
||||
console.log('oh hello ' + user.name + ' (' + user.id + ')')
|
||||
console.log('you have ' + unread.length + ' unread messages')
|
||||
console.log('you have ' + chats.length + ' chats & ' + contacts.length + ' contacts')
|
||||
|
||||
authInfo = client.base64EncodedAuthInfo() // get all the auth info we need to restore this session
|
||||
fs.writeFileSync('./auth_info.json', JSON.stringify(authInfo, null, '\t')) // save this info to a file
|
||||
/* Note: one can take this auth_info.json file and login again from any computer without having to scan the QR code,
|
||||
and get full access to one's WhatsApp. Despite the convenience, be careful with this file */
|
||||
|
||||
client.setOnPresenceUpdate((json) => console.log(json.id + ' presence is ' + json.type))
|
||||
client.setOnMessageStatusChange((json) => {
|
||||
const participant = json.participant ? ' (' + json.participant + ')' : '' // participant exists when the message is from a group
|
||||
console.log(
|
||||
json.to +
|
||||
participant +
|
||||
' acknowledged message(s) ' +
|
||||
json.ids +
|
||||
' as ' +
|
||||
json.type +
|
||||
' at ' +
|
||||
json.timestamp,
|
||||
)
|
||||
})
|
||||
client.setOnUnreadMessage(async (m) => {
|
||||
const [notificationType, messageType] = getNotificationType(m) // get what type of notification it is -- message, group add notification etc.
|
||||
console.log('got notification of type: ' + notificationType)
|
||||
|
||||
if (notificationType !== 'message') {
|
||||
return
|
||||
}
|
||||
if (m.key.fromMe) {
|
||||
console.log('relayed my own message')
|
||||
return
|
||||
}
|
||||
|
||||
let sender = m.key.remoteJid
|
||||
if (m.key.participant) {
|
||||
// participant exists if the message is in a group
|
||||
sender += ' (' + m.key.participant + ')'
|
||||
}
|
||||
if (messageType === MessageType.text) {
|
||||
const text = m.message.conversation
|
||||
console.log(sender + ' sent: ' + text)
|
||||
} else if (messageType === MessageType.extendedText) {
|
||||
const text = m.message.extendedTextMessage.text
|
||||
console.log(sender + ' sent: ' + text + ' and quoted message: ' + JSON.stringify(m.message))
|
||||
} else if (messageType === MessageType.contact) {
|
||||
const contact = m.message.contactMessage
|
||||
console.log(sender + ' sent contact (' + contact.displayName + '): ' + contact.vcard)
|
||||
} else if (messageType === MessageType.location || messageType === MessageType.liveLocation) {
|
||||
const locMessage = m.message[messageType]
|
||||
console.log(
|
||||
sender +
|
||||
' sent location (lat: ' +
|
||||
locMessage.degreesLatitude +
|
||||
', long: ' +
|
||||
locMessage.degreesLongitude +
|
||||
'), saving thumbnail...',
|
||||
)
|
||||
decodeMediaMessage(m.message, 'loc_thumb_in_' + m.key.id)
|
||||
|
||||
if (messageType === MessageType.liveLocation) {
|
||||
console.log(
|
||||
sender +
|
||||
' sent live location for duration: ' +
|
||||
m.duration / 60 +
|
||||
' minutes, seq number: ' +
|
||||
locMessage.sequenceNumber,
|
||||
)
|
||||
}
|
||||
} else {
|
||||
// if it is a media (audio, image, video) message
|
||||
// decode, decrypt & save the media.
|
||||
// The extension to the is applied automatically based on the media type
|
||||
try {
|
||||
const savedFile = await decodeMediaMessage(m.message, 'media_in_' + m.key.id)
|
||||
console.log(sender + ' sent media, saved at: ' + savedFile)
|
||||
} catch (err) {
|
||||
console.log('error in decoding message: ' + err)
|
||||
}
|
||||
}
|
||||
// send a reply after 3 seconds
|
||||
setTimeout(async () => {
|
||||
await client.sendReadReceipt(m.key.remoteJid, m.key.id) // send read receipt
|
||||
await client.updatePresence(m.key.remoteJid, Presence.available) // tell them we're available
|
||||
await client.updatePresence(m.key.remoteJid, Presence.composing) // tell them we're composing
|
||||
|
||||
const options: MessageOptions = { quoted: m }
|
||||
let content
|
||||
let type: MessageType
|
||||
const rand = Math.random()
|
||||
if (rand > 0.66) {
|
||||
// choose at random
|
||||
content = 'hello!' // send a "hello!" & quote the message recieved
|
||||
type = MessageType.text
|
||||
} else if (rand > 0.33) {
|
||||
// choose at random
|
||||
content = { degreesLatitude: 32.123123, degreesLongitude: 12.12123123 }
|
||||
type = MessageType.location
|
||||
} else {
|
||||
content = fs.readFileSync('example/ma_gif.mp4') // load the gif
|
||||
options.mimetype = Mimetype.gif
|
||||
}
|
||||
const response = await client.sendMessage(m.key.remoteJid, content, type, options)
|
||||
console.log("sent message with ID '" + response.messageID + "' successfully: " + (response.status === 200))
|
||||
}, 3 * 1000)
|
||||
}, true) // set to false to not relay your own sent messages
|
||||
|
||||
/* example of custom functionality for tracking battery */
|
||||
client.registerCallback(['action', null, 'battery'], (json) => {
|
||||
const batteryLevelStr = json[2][0][1].value
|
||||
const batterylevel = parseInt(batteryLevelStr)
|
||||
console.log('battery level: ' + batterylevel)
|
||||
})
|
||||
client.setOnUnexpectedDisconnect((err) => console.log('disconnected unexpectedly: ' + err))
|
||||
}
|
||||
|
||||
example().catch((err) => console.log(`encountered error: ${err}`))
|
||||
Reference in New Issue
Block a user