You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
156 lines
3.0 KiB
Go
156 lines
3.0 KiB
Go
package core
|
|
|
|
import (
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/Rhymen/go-whatsapp"
|
|
"github.com/hugot/go-deltachat/deltabot"
|
|
"github.com/hugot/go-deltachat/deltachat"
|
|
)
|
|
|
|
type BridgeContext struct {
|
|
Config *Config
|
|
WhappConn *whatsapp.Conn
|
|
DCContext *deltachat.Context
|
|
DCClient *deltachat.Client
|
|
DB *Database
|
|
MessageTracker *MessageTracker
|
|
DCUserID uint32
|
|
DCUserChatID uint32
|
|
}
|
|
|
|
func NewBridgeContext(config *Config, msgTrackerFlushInterval time.Duration) *BridgeContext {
|
|
db := NewDatabase(config.App.DataFolder + "/app.db")
|
|
messageTracker := NewMessageTracker(db, msgTrackerFlushInterval)
|
|
|
|
return &BridgeContext{
|
|
Config: config,
|
|
DB: db,
|
|
MessageTracker: messageTracker,
|
|
}
|
|
}
|
|
|
|
// Do all the initialization stuff like intializing databases & services, configuring
|
|
// clients, connecting to remote servers, logging in etc.
|
|
func (b *BridgeContext) Init(
|
|
whappHandler whatsapp.Handler,
|
|
botCommands []deltabot.Command,
|
|
) error {
|
|
err := b.DB.Init()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
dcClient, err := BootstrapDcClientFromConfig(*b.Config, b)
|
|
|
|
b.SendLog("Whapp-Deltachat started.")
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for i := 0; i < 10; i++ {
|
|
err = CreateAndLoginWhappConnection(b.Config.App.DataFolder, b)
|
|
|
|
if err == nil {
|
|
break
|
|
}
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
b.WhappConn.AddHandler(whappHandler)
|
|
|
|
bot := &deltabot.Bot{}
|
|
|
|
for _, command := range botCommands {
|
|
bot.AddCommand(command)
|
|
}
|
|
|
|
dcClient.On(deltachat.DC_EVENT_INCOMING_MSG, bot.HandleMessage)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (b *BridgeContext) Close() error {
|
|
_, err := b.WhappConn.Disconnect()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = b.MessageTracker.Flush()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
b.DCClient.Close()
|
|
|
|
err = b.DB.Close()
|
|
|
|
return err
|
|
}
|
|
|
|
// Find or create a deltachat verified group chat for a whatsapp JID and return it's ID.
|
|
func (b *BridgeContext) GetOrCreateDCIDForJID(JID string) (uint32, error) {
|
|
if DCID, _ := b.DB.GetDCIDForWhappJID(JID); DCID != nil {
|
|
return *DCID, nil
|
|
}
|
|
|
|
chatName := JID
|
|
chat, ok := b.WhappConn.Store.Chats[JID]
|
|
|
|
if ok {
|
|
chatName = chat.Name
|
|
} else if sender, ok := b.WhappConn.Store.Contacts[JID]; ok {
|
|
chatName = sender.Name
|
|
}
|
|
|
|
DCID := b.DCContext.CreateGroupChat(true, chatName)
|
|
|
|
err := b.DB.StoreDCIDForJID(JID, DCID)
|
|
|
|
if err != nil {
|
|
return DCID, err
|
|
}
|
|
|
|
b.DCContext.AddContactToChat(DCID, b.DCUserID)
|
|
|
|
return DCID, err
|
|
}
|
|
|
|
func (b *BridgeContext) SendLog(logString string) {
|
|
b.DCContext.SendTextMessage(b.DCUserChatID, logString)
|
|
}
|
|
|
|
func (b *BridgeContext) MessageWasSent(ID string) bool {
|
|
sent, err := b.MessageTracker.WasSent(ID)
|
|
|
|
if err != nil {
|
|
log.Println(err)
|
|
b.SendLog(err.Error())
|
|
}
|
|
|
|
return sent
|
|
}
|
|
|
|
func (b *BridgeContext) ShouldMessageBeSent(info whatsapp.MessageInfo) bool {
|
|
// Skip if the message has already been sent
|
|
if b.MessageWasSent(info.Id) {
|
|
return false
|
|
}
|
|
|
|
// send if not from user
|
|
if !info.FromMe {
|
|
return true
|
|
}
|
|
|
|
// If from user, only send when it is enabled in the config
|
|
return b.Config.App.ShowFromMe
|
|
}
|