mirror of https://github.com/matrix-org/gomatrix
127 lines
2.8 KiB
Go
127 lines
2.8 KiB
Go
package main
|
|
|
|
//import "github.com/kr/pretty"
|
|
import (
|
|
"os"
|
|
"time"
|
|
"log"
|
|
"github.com/matrix-org/gomatrix"
|
|
"fmt"
|
|
)
|
|
|
|
// Login to a local homeserver and set the user ID and access token on success.
|
|
func Login(server, username, password string) *gomatrix.Client {
|
|
cli, _ := gomatrix.NewClient(server, "", "")
|
|
resp, err := cli.Login(&gomatrix.ReqLogin{
|
|
Type: "m.login.password",
|
|
User: username,
|
|
Password: password,
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
cli.SetCredentials(resp.UserID, resp.AccessToken)
|
|
return cli
|
|
}
|
|
|
|
func ProcessResponse(resSync *gomatrix.RespSync) {
|
|
for _,v := range resSync.Rooms.Join {
|
|
//fmt.Printf("%v: %# v\n\n\n", k, pretty.Formatter(v.Timeline.Events))
|
|
for _, e := range v.Timeline.Events {
|
|
if e.Type == "m.room.message" {
|
|
fmt.Printf("%v: %v\n", e.Sender, e.Content["body"])
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
|
|
func ExtractRooms(resSync *gomatrix.RespSync) map[string]string{
|
|
rooms := map[string]string{}
|
|
for roomID,v := range resSync.Rooms.Join {
|
|
//fmt.Printf("%v: %# v\n\n\n", k, pretty.Formatter(v.Timeline.Events))
|
|
for _, e := range v.Timeline.Events {
|
|
if e.Type == "m.room.name" {
|
|
rooms[fmt.Sprint(e.Content["name"])] = roomID
|
|
}
|
|
}
|
|
|
|
}
|
|
return rooms
|
|
}
|
|
|
|
func main () {
|
|
log.Print("Logging in...")
|
|
username := os.Args[2]
|
|
inviteUser := os.Args[4]
|
|
roomname := "TestRoomFor" + username
|
|
roomalias := roomname
|
|
cli := Login(os.Args[1], username, os.Args[3])
|
|
log.Println("Done!")
|
|
cli.SetDisplayName(os.Args[2])
|
|
log.Print("Creating room...")
|
|
resp, err := cli.CreateRoom(&gomatrix.ReqCreateRoom{
|
|
RoomAliasName: roomalias,
|
|
Name : roomname,
|
|
Preset: "public_chat",
|
|
})
|
|
id := "!JHMwXpSMhcSnyAGDWB:matrix.org"
|
|
if err != nil {
|
|
log.Println(err)
|
|
log.Println("Failed!")
|
|
id = "#" + roomalias + ":matrix.org"
|
|
} else {
|
|
log.Println("Done!")
|
|
fmt.Println("Room:", resp.RoomID)
|
|
id = resp.RoomID
|
|
log.Print("Joining room...")
|
|
if _, err := cli.JoinRoom(id, "", nil); err != nil {
|
|
panic(err)
|
|
}
|
|
log.Println("Done!")
|
|
fmt.Println("Room:", id)
|
|
}
|
|
|
|
log.Print("Syncing...")
|
|
filter := `{"room":{"timeline":{"limit":50}}}`
|
|
resSync, err := cli.SyncRequest(5, "", filter, false, "")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
ProcessResponse(resSync)
|
|
|
|
nextbatch := resSync.NextBatch
|
|
|
|
//get room id
|
|
rooms := ExtractRooms(resSync)
|
|
log.Println(rooms)
|
|
id = rooms[roomname]
|
|
|
|
|
|
log.Print("Sending message...")
|
|
if _, err := cli.SendText(id, "Hello") ; err != nil {
|
|
panic(err)
|
|
}
|
|
log.Println("Done!")
|
|
|
|
log.Print("Inviting user...")
|
|
if _, err := cli.InviteUser(id, &gomatrix.ReqInviteUser{UserID: inviteUser}) ; err != nil {
|
|
log.Println(err)
|
|
}
|
|
log.Println("Done!")
|
|
|
|
|
|
|
|
for {
|
|
resSync, err := cli.SyncRequest(5, nextbatch, filter, false, "")
|
|
ProcessResponse(resSync)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
nextbatch = resSync.NextBatch
|
|
time.Sleep(2 * time.Second)
|
|
}
|
|
log.Println("Done!")
|
|
}
|