common.go

   1 // mmc -- Mattermost client
   2 // Copyright (C) 2023 Sergey Matveev <stargrave@stargrave.org>
   3 //
   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
   7 // License.
   8 //
   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.
  13 //
  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/>.
  16
  17 package mmc
  18
  19 import (
  20         "os"
  21         "strings"
  22         "time"
  23
  24         "github.com/davecgh/go-spew/spew"
  25         "github.com/mattermost/mattermost-server/v6/model"
  26         "go.cypherpunks.ru/recfile"
  27 )
  28
  29 const (
  30         PerPage    = 100
  31         OutRec     = "out.rec"
  32         OutRecLock = "out.rec.lock"
  33         Last       = "last"
  34 )
  35
  36 var SleepTime = 250 * time.Millisecond
  37
  38 type Post struct {
  39         P *model.Post
  40         E string
  41 }
  42
  43 func PostToRec(w *recfile.Writer, users map[string]*model.User, post Post) error {
  44         _, err := w.RecordStart()
  45         if err != nil {
  46                 return err
  47         }
  48         created := time.Unix(post.P.CreateAt/1000, 0)
  49         user := users[post.P.UserId]
  50         sender := "unknown"
  51         if user != nil {
  52                 sender = user.Username
  53         }
  54         fields := []recfile.Field{
  55                 {Name: "Id", Value: post.P.Id},
  56                 {Name: "Created", Value: created.Format("2006-01-02 15:04:05")},
  57                 {Name: "Sender", Value: sender},
  58         }
  59         if post.E != model.WebsocketEventPosted {
  60                 fields = append(fields, recfile.Field{Name: "Event", Value: post.E})
  61         }
  62         if post.P.RootId != "" {
  63                 fields = append(fields, recfile.Field{Name: "RootId", Value: post.P.RootId})
  64         }
  65         if post.P.Metadata != nil {
  66                 for _, fi := range post.P.Metadata.Files {
  67                         fields = append(
  68                                 fields,
  69                                 recfile.Field{Name: "File", Value: fi.Id},
  70                                 recfile.Field{Name: "FileName", Value: fi.Name},
  71                         )
  72                 }
  73         }
  74         if _, err = w.WriteFields(fields...); err != nil {
  75                 return err
  76         }
  77         _, err = w.WriteFieldMultiline("Text", strings.Split(post.P.Message, "\n"))
  78         return err
  79 }
  80
  81 func GetUsers(c *model.Client4, debugFd *os.File) (map[string]*model.User, error) {
  82         users := make(map[string]*model.User)
  83         for n := 0; ; n++ {
  84                 time.Sleep(SleepTime)
  85                 page, resp, err := c.GetUsers(n, PerPage, "")
  86                 if err != nil {
  87                         if debugFd != nil {
  88                                 spew.Fdump(debugFd, resp)
  89                         }
  90                         return nil, err
  91                 }
  92                 if debugFd != nil {
  93                         spew.Fdump(debugFd, page)
  94                 }
  95                 for _, u := range page {
  96                         users[u.Id] = u
  97                 }
  98                 if len(page) < PerPage {
  99                         break
 100                 }
 101         }
 102         return users, nil
 103 }
 104
 105 func GetEntrypoint() string {
 106         s := os.Getenv("MMC_ENTRYPOINT")
 107         if s == "" {
 108                 return "mm.invalid"
 109         }
 110         return s
 111 }