Skip to content

Commit

Permalink
add capabilities to set presence and chat compose status
Browse files Browse the repository at this point in the history
  • Loading branch information
@dimaskiddo
dimaskiddo committed Apr 28, 2022
1 parent 69b27db commit 53b1c4f
Showing 1 changed file with 123 additions and 13 deletions.
136 changes: 123 additions & 13 deletions pkg/whatsapp/whatsapp.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -114,6 +114,9 @@ func WhatsAppLogin(jid string) (string, int, error) {
return "", 0, err
}

// Set WhatsApp Client Presence to Available
_ = WhatsAppClient[jid].SendPresence(types.PresenceAvailable)

// Get Generated QR Code and Timeout Information
qrImage, qrTimeout := WhatsAppGenerateQR(qrChanGenerate)

Expand Down Expand Up @@ -148,6 +151,9 @@ func WhatsAppReconnect(jid string) error {
return err
}

// Set WhatsApp Client Presence to Available
_ = WhatsAppClient[jid].SendPresence(types.PresenceAvailable)

return nil
}

Expand All @@ -159,8 +165,13 @@ func WhatsAppReconnect(jid string) error {

func WhatsAppLogout(jid string) error {
if WhatsAppClient[jid] != nil {
var err error

// Set WhatsApp Client Presence to Unavailable
_ = WhatsAppClient[jid].SendPresence(types.PresenceUnavailable)

// Logout WhatsApp Client and Disconnect from WebSocket
err := WhatsAppClient[jid].Logout()
err = WhatsAppClient[jid].Logout()
if err != nil {
// Force Disconnect
WhatsAppClient[jid].Disconnect()
Expand Down Expand Up @@ -232,25 +243,57 @@ func WhatsAppDecomposeJID(jid string) string {
return jid
}

func WhatsAppComposeStatus(jid string, rjid types.JID, isComposing bool, isAudio bool) {
// Set Compose Status
var typeCompose types.ChatPresence
if isComposing {
typeCompose = types.ChatPresenceComposing
} else {
typeCompose = types.ChatPresencePaused
}

// Set Compose Media Audio (Recording) or Text (Typing)
var typeComposeMedia types.ChatPresenceMedia
if isAudio {
typeComposeMedia = types.ChatPresenceMediaAudio
} else {
typeComposeMedia = types.ChatPresenceMediaText
}

// Send Chat Compose Status
_ = WhatsAppClient[jid].SendChatPresence(rjid, typeCompose, typeComposeMedia)
}

func WhatsAppSendText(jid string, rjid string, message string) error {
if WhatsAppClient[jid] != nil {
var err error

// Make Sure WhatsApp Client is OK
err := WhatsAppClientIsOK(jid)
err = WhatsAppClientIsOK(jid)
if err != nil {
return err
}

// Make Sure rJID is Proper JID Type
composeRJID := WhatsAppComposeJID(rjid)

// Set Chat Presence to Composing
WhatsAppComposeStatus(jid, composeRJID, true, false)

// Compose WhatsApp Proto
content := &waproto.Message{
Conversation: proto.String(message),
}

// Send WhatsApp Message Proto
_, err = WhatsAppClient[jid].SendMessage(WhatsAppComposeJID(rjid), "", content)
_, err = WhatsAppClient[jid].SendMessage(composeRJID, "", content)
if err != nil {
return err
}

// Set Chat Presence to Paused
WhatsAppComposeStatus(jid, composeRJID, false, false)

return nil
}

Expand All @@ -260,12 +303,20 @@ func WhatsAppSendText(jid string, rjid string, message string) error {

func WhatsAppSendLocation(jid string, rjid string, latitude float64, longitude float64) error {
if WhatsAppClient[jid] != nil {
var err error

// Make Sure WhatsApp Client is OK
err := WhatsAppClientIsOK(jid)
err = WhatsAppClientIsOK(jid)
if err != nil {
return err
}

// Make Sure rJID is Proper JID Type
composeRJID := WhatsAppComposeJID(rjid)

// Set Chat Presence to Composing
WhatsAppComposeStatus(jid, composeRJID, true, false)

// Compose WhatsApp Proto
content := &waproto.Message{
LocationMessage: &waproto.LocationMessage{
Expand All @@ -275,11 +326,14 @@ func WhatsAppSendLocation(jid string, rjid string, latitude float64, longitude f
}

// Send WhatsApp Message Proto
_, err = WhatsAppClient[jid].SendMessage(WhatsAppComposeJID(rjid), "", content)
_, err = WhatsAppClient[jid].SendMessage(composeRJID, "", content)
if err != nil {
return err
}

// Set Chat Presence to Paused
WhatsAppComposeStatus(jid, composeRJID, false, false)

return nil
}

Expand All @@ -289,14 +343,25 @@ func WhatsAppSendLocation(jid string, rjid string, latitude float64, longitude f

func WhatsAppSendDocument(jid string, rjid string, fileBytes []byte, fileType string, fileName string) error {
if WhatsAppClient[jid] != nil {
var err error

// Make Sure WhatsApp Client is OK
err := WhatsAppClientIsOK(jid)
err = WhatsAppClientIsOK(jid)
if err != nil {
return err
}

// Make Sure rJID is Proper JID Type
composeRJID := WhatsAppComposeJID(rjid)

// Set Chat Presence to Composing
WhatsAppComposeStatus(jid, composeRJID, true, false)

// Upload File to WhatsApp Storage Server
fileUploaded, err := WhatsAppClient[jid].Upload(context.Background(), fileBytes, whatsmeow.MediaDocument)
if err != nil {
return errors.New("Error While Uploading Media to WhatsApp Server")
}

// Compose WhatsApp Proto
content := &waproto.Message{
Expand All @@ -314,11 +379,14 @@ func WhatsAppSendDocument(jid string, rjid string, fileBytes []byte, fileType st
}

// Send WhatsApp Message Proto
_, err = WhatsAppClient[jid].SendMessage(WhatsAppComposeJID(rjid), "", content)
_, err = WhatsAppClient[jid].SendMessage(composeRJID, "", content)
if err != nil {
return err
}

// Set Chat Presence to Paused
WhatsAppComposeStatus(jid, composeRJID, false, false)

return nil
}

Expand All @@ -328,14 +396,25 @@ func WhatsAppSendDocument(jid string, rjid string, fileBytes []byte, fileType st

func WhatsAppSendImage(jid string, rjid string, imageBytes []byte, imageType string, imageCaption string) error {
if WhatsAppClient[jid] != nil {
var err error

// Make Sure WhatsApp Client is OK
err := WhatsAppClientIsOK(jid)
err = WhatsAppClientIsOK(jid)
if err != nil {
return err
}

// Make Sure rJID is Proper JID Type
composeRJID := WhatsAppComposeJID(rjid)

// Set Chat Presence to Composing
WhatsAppComposeStatus(jid, composeRJID, true, false)

// Upload Image to WhatsApp Storage Server
imageUploaded, err := WhatsAppClient[jid].Upload(context.Background(), imageBytes, whatsmeow.MediaImage)
if err != nil {
return errors.New("Error While Uploading Media to WhatsApp Server")
}

// Compose WhatsApp Proto
content := &waproto.Message{
Expand All @@ -352,11 +431,14 @@ func WhatsAppSendImage(jid string, rjid string, imageBytes []byte, imageType str
}

// Send WhatsApp Message Proto
_, err = WhatsAppClient[jid].SendMessage(WhatsAppComposeJID(rjid), "", content)
_, err = WhatsAppClient[jid].SendMessage(composeRJID, "", content)
if err != nil {
return err
}

// Set Chat Presence to Paused
WhatsAppComposeStatus(jid, composeRJID, false, false)

return nil
}

Expand All @@ -366,14 +448,25 @@ func WhatsAppSendImage(jid string, rjid string, imageBytes []byte, imageType str

func WhatsAppSendAudio(jid string, rjid string, audioBytes []byte, audioType string) error {
if WhatsAppClient[jid] != nil {
var err error

// Make Sure WhatsApp Client is OK
err := WhatsAppClientIsOK(jid)
err = WhatsAppClientIsOK(jid)
if err != nil {
return err
}

// Make Sure rJID is Proper JID Type
composeRJID := WhatsAppComposeJID(rjid)

// Set Chat Presence to Composing
WhatsAppComposeStatus(jid, composeRJID, true, true)

// Upload Audio to WhatsApp Storage Server
audioUploaded, err := WhatsAppClient[jid].Upload(context.Background(), audioBytes, whatsmeow.MediaAudio)
if err != nil {
return errors.New("Error While Uploading Media to WhatsApp Server")
}

// Compose WhatsApp Proto
content := &waproto.Message{
Expand All @@ -389,11 +482,14 @@ func WhatsAppSendAudio(jid string, rjid string, audioBytes []byte, audioType str
}

// Send WhatsApp Message Proto
_, err = WhatsAppClient[jid].SendMessage(WhatsAppComposeJID(rjid), "", content)
_, err = WhatsAppClient[jid].SendMessage(composeRJID, "", content)
if err != nil {
return err
}

// Set Chat Presence to Paused
WhatsAppComposeStatus(jid, composeRJID, false, true)

return nil
}

Expand All @@ -403,14 +499,25 @@ func WhatsAppSendAudio(jid string, rjid string, audioBytes []byte, audioType str

func WhatsAppSendVideo(jid string, rjid string, videoBytes []byte, videoType string, videoCaption string) error {
if WhatsAppClient[jid] != nil {
var err error

// Make Sure WhatsApp Client is OK
err := WhatsAppClientIsOK(jid)
err = WhatsAppClientIsOK(jid)
if err != nil {
return err
}

// Make Sure rJID is Proper JID Type
composeRJID := WhatsAppComposeJID(rjid)

// Set Chat Presence to Composing
WhatsAppComposeStatus(jid, composeRJID, true, false)

// Upload Video to WhatsApp Storage Server
videoUploaded, err := WhatsAppClient[jid].Upload(context.Background(), videoBytes, whatsmeow.MediaVideo)
if err != nil {
return errors.New("Error While Uploading Media to WhatsApp Server")
}

// Compose WhatsApp Proto
content := &waproto.Message{
Expand All @@ -427,11 +534,14 @@ func WhatsAppSendVideo(jid string, rjid string, videoBytes []byte, videoType str
}

// Send WhatsApp Message Proto
_, err = WhatsAppClient[jid].SendMessage(WhatsAppComposeJID(rjid), "", content)
_, err = WhatsAppClient[jid].SendMessage(composeRJID, "", content)
if err != nil {
return err
}

// Set Chat Presence to Paused
WhatsAppComposeStatus(jid, composeRJID, false, false)

return nil
}

Expand Down

0 comments on commit 53b1c4f

Please sign in to comment.