1 // mmc -- Mattermost client
2 // Copyright (C) 2023-2024 Sergey Matveev <stargrave@stargrave.org>
4 // This program is free software: you can redistribute it and/or modify
5 // it under the terms of the GNU Affero General Public License as
6 // published by the Free Software Foundation, either version 3 of the
9 // This program is distributed in the hope that it will be useful,
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 // GNU Affero General Public License for more details.
14 // You should have received a copy of the GNU Affero General Public License
15 // along with this program. If not, see <http://www.gnu.org/licenses/>.
39 "github.com/davecgh/go-spew/spew"
40 "github.com/mattermost/mattermost-server/v6/model"
41 "go.cypherpunks.ru/netrc"
42 "go.stargrave.org/mmc"
46 Newwin = flag.String("newwin", "cmd/newwin", "Path to newwin command")
51 func rewriteIfChanged(fn string, data string) {
52 if their, err := os.ReadFile(fn); err != nil ||
53 !bytes.Equal([]byte(data), their) {
54 if err = os.WriteFile(fn, []byte(data), 0o666); err != nil {
60 func mkFifo(pth string) {
61 if _, err := os.Stat(pth); err == nil {
64 if err := syscall.Mkfifo(pth, 0666); err != nil {
70 entrypoint := flag.String("entrypoint", mmc.GetEntrypoint(), "Entrypoint")
71 notifyCmd := flag.String("notify", "cmd/notify", "Path to notification handler")
72 heartbeatCh := flag.String("heartbeat-ch", "town-square", "Channel for heartbeating")
74 log.SetFlags(log.Lshortfile)
75 log.SetOutput(os.Stdout)
76 UmaskCur = syscall.Umask(0)
77 syscall.Umask(UmaskCur)
81 DebugFd, err = os.OpenFile(
82 "debug", os.O_WRONLY|os.O_APPEND, os.FileMode(0666),
89 login, password := netrc.Find(*entrypoint)
90 if login == "" || password == "" {
91 log.Fatalln("no credentials found for:", *entrypoint)
93 c := model.NewAPIv4Client("https://" + *entrypoint)
94 c.Login(login, password)
95 me, resp, err := c.GetMe("")
98 spew.Fdump(DebugFd, resp)
103 spew.Fdump(DebugFd, me)
105 log.Println("logged in")
107 time.Sleep(mmc.SleepTime)
108 teams, resp, err := c.GetTeamsForUser(me.Id, "")
111 spew.Fdump(DebugFd, resp)
116 spew.Fdump(DebugFd, teams)
120 var updateQueue []string
121 Chans := make(map[string]*model.Channel)
122 time.Sleep(mmc.SleepTime)
123 page, resp, err := c.GetChannelsForTeamForUser(Team.Id, me.Id, false, "")
126 spew.Fdump(DebugFd, resp)
131 spew.Fdump(DebugFd, page)
133 for _, ch := range page {
138 pth := path.Join("chans", strings.ReplaceAll(ch.Name, ".", "_"))
139 updateQueue = append(updateQueue, pth, ch.Id)
140 os.MkdirAll(pth, 0777)
141 rewriteIfChanged(path.Join(pth, "id"), ch.Id+"\n")
142 rewriteIfChanged(path.Join(pth, "info"), fmt.Sprintf(
148 if _, err := os.Stat(path.Join(pth, mmc.OutRec)); err != nil &&
149 errors.Is(err, fs.ErrNotExist) {
150 if _, err = os.OpenFile(
151 path.Join(pth, mmc.OutRec), os.O_WRONLY|os.O_CREATE, 0o666,
157 usersPth := path.Join(pth, "users")
159 go func(ch *model.Channel) {
161 time.Sleep(mmc.SleepTime)
162 fd, err := os.OpenFile(
163 usersPth, os.O_WRONLY|os.O_APPEND, os.FileMode(0666),
166 log.Println("OpenFile:", usersPth, err)
170 users, resp, err := c.GetUsersInChannel(ch.Id, n, mmc.PerPage, "")
173 spew.Fdump(DebugFd, resp)
175 log.Println("GetUsersInChannel:", err)
180 spew.Fdump(DebugFd, users)
182 for _, u := range users {
183 fmt.Fprintf(fd, "%s\n", u.Username)
185 if len(users) < mmc.PerPage {
193 pth = path.Join(pth, "in")
195 go func(ch *model.Channel) {
197 fd, err := os.OpenFile(pth, os.O_RDONLY, os.FileMode(0666))
201 data, err := io.ReadAll(fd)
206 if _, err = makePost(c, ch.Id, string(data)); err != nil {
207 log.Println("makePost:", err)
213 Users, err := mmc.GetUsers(c, DebugFd)
218 UsersDC := make(map[string]*model.Channel, len(Users))
219 for _, u := range Users {
220 pth := path.Join("users", strings.ReplaceAll(u.Username, ".", "_"))
221 os.MkdirAll(pth, 0777)
223 path.Join(pth, "name"),
224 fmt.Sprintf("%s %s\n", u.FirstName, u.LastName),
226 rewriteIfChanged(path.Join(pth, "email"), u.Email+"\n")
227 rewriteIfChanged(path.Join(pth, "id"), u.Id+"\n")
228 if _, err := os.Stat(path.Join(pth, mmc.OutRec)); err != nil &&
229 errors.Is(err, fs.ErrNotExist) {
230 if _, err = os.OpenFile(
231 path.Join(pth, mmc.OutRec), os.O_WRONLY|os.O_CREATE, 0o666,
237 if fi, err := os.Stat(path.Join(pth, mmc.OutRec)); err == nil && fi.Size() > 0 {
238 time.Sleep(mmc.SleepTime)
239 dc, resp, err := c.CreateDirectChannel(me.Id, u.Id)
242 spew.Fdump(DebugFd, resp)
244 log.Println("CreateDirectChannel:", err)
248 spew.Fdump(DebugFd, dc)
251 updateQueue = append(updateQueue, pth, dc.Id)
254 statusPth := path.Join(pth, "status")
256 go func(u *model.User) {
258 time.Sleep(mmc.SleepTime)
259 fd, err := os.OpenFile(
260 statusPth, os.O_WRONLY|os.O_APPEND, os.FileMode(0666),
263 log.Println("OpenFile:", statusPth, err)
266 status, resp, err := c.GetUserStatus(u.Id, "")
269 spew.Fdump(DebugFd, resp)
271 log.Println("GetUserStatus:", err)
276 spew.Fdump(DebugFd, status)
278 fmt.Fprintf(fd, "%s\n", status.Status)
283 pth = path.Join(pth, "in")
285 go func(u *model.User) {
286 var dc *model.Channel
288 fd, err := os.OpenFile(pth, os.O_RDONLY, os.FileMode(0666))
292 data, err := io.ReadAll(fd)
299 dc, resp, err = c.CreateDirectChannel(me.Id, u.Id)
302 spew.Fdump(DebugFd, resp)
304 log.Println("CreateDirectChannel:", err)
309 spew.Fdump(DebugFd, dc)
312 if _, err = makePost(c, dc.Id, string(data)); err != nil {
313 log.Println("makePost:", err)
319 UserStatus := make(map[string]string)
320 var UserStatusM sync.RWMutex
322 pth := path.Join("users", "status")
325 time.Sleep(mmc.SleepTime)
326 fd, err := os.OpenFile(pth, os.O_WRONLY|os.O_APPEND, os.FileMode(0666))
328 log.Println("OpenFile:", pth, err)
331 agg := make(map[string][]string)
333 for name, status := range UserStatus {
334 agg[status] = append(agg[status], name)
336 UserStatusM.RUnlock()
337 statuses := make([]string, 0, len(agg))
338 for status := range agg {
339 sort.Strings(agg[status])
340 statuses = append(statuses, status)
342 sort.Strings(statuses)
343 for _, status := range statuses {
344 fmt.Fprintln(fd, status+":", strings.Join(agg[status], " "))
350 log.Println("syncing", len(updateQueue)/2, "rooms")
351 for len(updateQueue) > 0 {
352 err := updatePosts(c, Users, updateQueue[0], updateQueue[1])
354 log.Println("updatePosts:", err)
356 updateQueue = updateQueue[2:]
358 log.Println("sync done")
359 if *notifyCmd != "" {
360 exec.Command(*notifyCmd, "sync done").Run()
364 os.MkdirAll("file", 0777)
365 pthGet := path.Join("file", "get")
367 pthOut := path.Join("file", "out")
370 time.Sleep(mmc.SleepTime)
371 fd, err := os.OpenFile(pthGet, os.O_RDONLY, os.FileMode(0666))
375 data, err := io.ReadAll(fd)
380 fileId := strings.TrimRight(string(data), " \n")
381 if len(fileId) == 0 {
384 fi, resp, err := c.GetFileInfo(fileId)
386 spew.Fdump(DebugFd, fi, resp)
396 data, resp, err = c.GetFile(fileId)
399 spew.Fdump(DebugFd, resp)
404 fd, err = os.OpenFile(
405 pthOut, os.O_WRONLY|os.O_APPEND, os.FileMode(0666),
411 w := tar.NewWriter(fd)
412 if err = w.WriteHeader(&tar.Header{
413 Format: tar.FormatPAX,
414 Typeflag: tar.TypeReg,
416 Size: int64(len(data)),
418 ModTime: time.Unix(fi.CreateAt/1000, 0),
419 PAXRecords: map[string]string{
421 "MM.MIMEType": fi.MimeType,
428 if _, err = io.Copy(w, bytes.NewReader(data)); err != nil {
441 needsShutdown := make(chan os.Signal)
442 wc, err := model.NewWebSocketClient4("wss://"+*entrypoint, c.AuthToken)
449 t := time.NewTicker(time.Minute)
453 if wc.ListenError != nil {
454 log.Println("ListenError:", wc.ListenError)
455 needsShutdown <- syscall.SIGTERM
459 if *heartbeatCh != "" {
460 if _, _, err = c.ViewChannel(
462 &model.ChannelView{ChannelId: Chans[*heartbeatCh].Id},
464 log.Println("ChannelView:", err)
467 case <-wc.PingTimeoutChannel:
468 log.Println("PING timeout")
469 needsShutdown <- syscall.SIGTERM
471 case e := <-wc.EventChannel:
472 if e == nil || !e.IsValid() {
476 spew.Fdump(DebugFd, e)
480 if userId, ok := data["user_id"]; ok && userId.(string) != "" {
481 user = Users[userId.(string)]
483 log.Println("unknown user for event:", e)
487 switch eventType := e.EventType(); eventType {
488 case model.WebsocketEventTyping:
489 if *notifyCmd != "" {
490 exec.Command(*notifyCmd, "typing: "+user.Username).Run()
492 case model.WebsocketEventPostEdited,
493 model.WebsocketEventPostDeleted,
494 model.WebsocketEventPosted:
495 chName, ok := data["channel_name"].(string)
500 if err = json.NewDecoder(
501 strings.NewReader(data["post"].(string)),
502 ).Decode(&post); err != nil {
506 switch model.ChannelType(data["channel_type"].(string)) {
507 case model.ChannelTypeDirect:
508 userId := strings.TrimPrefix(chName, me.Id+"__")
509 userId = strings.TrimSuffix(userId, "__"+me.Id)
510 user := Users[userId]
512 log.Println("unknown user:", userId)
515 recipient = path.Join("users", user.Username)
516 case model.ChannelTypeOpen:
518 case model.ChannelTypePrivate:
520 case model.ChannelTypeGroup:
521 recipient = path.Join("chans", chName)
524 recipient, Users, []mmc.Post{{P: &post, E: eventType}},
528 case model.WebsocketEventStatusChange:
529 status := data["status"].(string)
531 UserStatus[user.Username] = status
533 if *notifyCmd != "" {
534 exec.Command(*notifyCmd, fmt.Sprintf(
535 "status: %s -> %s", user.Username, status,
538 case model.WebsocketEventHello:
539 case model.WebsocketEventChannelViewed:
543 case resp := <-wc.ResponseChannel:
544 if resp == nil || !resp.IsValid() {
548 spew.Fdump(DebugFd, resp)
550 statuses := make(map[string]string)
551 for userId, status := range resp.Data {
552 status, ok := status.(string)
556 user := Users[userId]
560 statuses[user.Username] = status
562 if len(statuses) > 0 {
564 for u := range UserStatus {
565 delete(UserStatus, u)
567 for u, status := range statuses {
568 UserStatus[u] = status
576 signal.Notify(needsShutdown, syscall.SIGTERM, syscall.SIGINT)
579 log.Println("finished")