Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
danielgross
GitHub Repository: danielgross/whatsapp-gpt
Path: blob/main/main.go
122 views
1
package main
2
3
import (
4
"bytes"
5
"context"
6
"fmt"
7
"net/http"
8
"net/url"
9
"os"
10
"os/signal"
11
"syscall"
12
13
_ "github.com/mattn/go-sqlite3"
14
"github.com/mdp/qrterminal/v3"
15
"go.mau.fi/whatsmeow"
16
waProto "go.mau.fi/whatsmeow/binary/proto"
17
"go.mau.fi/whatsmeow/store/sqlstore"
18
"go.mau.fi/whatsmeow/types"
19
"go.mau.fi/whatsmeow/types/events"
20
waLog "go.mau.fi/whatsmeow/util/log"
21
"google.golang.org/protobuf/proto"
22
)
23
24
type MyClient struct {
25
WAClient *whatsmeow.Client
26
eventHandlerID uint32
27
}
28
29
func (mycli *MyClient) register() {
30
mycli.eventHandlerID = mycli.WAClient.AddEventHandler(mycli.eventHandler)
31
}
32
33
func (mycli *MyClient) eventHandler(evt interface{}) {
34
switch v := evt.(type) {
35
case *events.Message:
36
newMessage := v.Message
37
msg := newMessage.GetConversation()
38
fmt.Println("Message from:", v.Info.Sender.User, "->", msg)
39
if msg == "" {
40
return
41
}
42
// Make a http request to localhost:5001/chat?q= with the message, and send the response
43
// URL encode the message
44
urlEncoded := url.QueryEscape(msg)
45
url := "http://localhost:5001/chat?q=" + urlEncoded
46
// Make the request
47
resp, err := http.Get(url)
48
if err != nil {
49
fmt.Println("Error making request:", err)
50
return
51
}
52
// Read the response
53
buf := new(bytes.Buffer)
54
buf.ReadFrom(resp.Body)
55
newMsg := buf.String()
56
// encode out as a string
57
response := &waProto.Message{Conversation: proto.String(string(newMsg))}
58
fmt.Println("Response:", response)
59
60
userJid := types.NewJID(v.Info.Sender.User, types.DefaultUserServer)
61
mycli.WAClient.SendMessage(context.Background(), userJid, "", response)
62
63
}
64
}
65
66
func main() {
67
dbLog := waLog.Stdout("Database", "DEBUG", true)
68
// Make sure you add appropriate DB connector imports, e.g. github.com/mattn/go-sqlite3 for SQLite
69
container, err := sqlstore.New("sqlite3", "file:examplestore.db?_foreign_keys=on", dbLog)
70
if err != nil {
71
panic(err)
72
}
73
// If you want multiple sessions, remember their JIDs and use .GetDevice(jid) or .GetAllDevices() instead.
74
deviceStore, err := container.GetFirstDevice()
75
if err != nil {
76
panic(err)
77
}
78
clientLog := waLog.Stdout("Client", "DEBUG", true)
79
client := whatsmeow.NewClient(deviceStore, clientLog)
80
// add the eventHandler
81
mycli := &MyClient{WAClient: client}
82
mycli.register()
83
84
if client.Store.ID == nil {
85
// No ID stored, new login
86
qrChan, _ := client.GetQRChannel(context.Background())
87
err = client.Connect()
88
if err != nil {
89
panic(err)
90
}
91
for evt := range qrChan {
92
if evt.Event == "code" {
93
// Render the QR code here
94
// e.g. qrterminal.GenerateHalfBlock(evt.Code, qrterminal.L, os.Stdout)
95
// or just manually `echo 2@... | qrencode -t ansiutf8` in a terminal
96
qrterminal.GenerateHalfBlock(evt.Code, qrterminal.L, os.Stdout)
97
// fmt.Println("QR code:", evt.Code)
98
} else {
99
fmt.Println("Login event:", evt.Event)
100
}
101
}
102
} else {
103
// Already logged in, just connect
104
err = client.Connect()
105
if err != nil {
106
panic(err)
107
}
108
}
109
110
// Listen to Ctrl+C (you can also do something else that prevents the program from exiting)
111
c := make(chan os.Signal)
112
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
113
<-c
114
115
client.Disconnect()
116
}
117
118