cmd/scrollback/main.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 main
  18
  19 import (
  20         "flag"
  21         "log"
  22         "os"
  23         "time"
  24
  25         "github.com/mattermost/mattermost-server/v6/model"
  26         "go.cypherpunks.ru/recfile"
  27         "go.stargrave.org/mmc"
  28 )
  29
  30 func main() {
  31         entrypoint := flag.String("entrypoint", mmc.GetEntrypoint(), "Entrypoint")
  32         pageNum := flag.Int("page", 0, "Page number")
  33         whole := flag.Bool("whole", false, "Whole history")
  34         direct := flag.Bool("direct", false, "Direct channel")
  35         flag.Parse()
  36         log.SetFlags(log.Lshortfile)
  37
  38         chId := flag.Arg(0)
  39         postId := flag.Arg(1)
  40         login, password := mmc.FindInNetrc(*entrypoint)
  41         if login == "" || password == "" {
  42                 log.Fatalln("no credentials found for:", *entrypoint)
  43         }
  44         c := model.NewAPIv4Client("https://" + *entrypoint)
  45         c.Login(login, password)
  46         me, _, err := c.GetMe("")
  47         if err != nil {
  48                 log.Fatalln(err)
  49         }
  50         users, err := mmc.GetUsers(c, nil)
  51         if err != nil {
  52                 log.Fatalln(err)
  53         }
  54         if *direct {
  55                 dc, _, err := c.CreateDirectChannel(me.Id, chId)
  56                 if err != nil {
  57                         log.Fatalln(err)
  58                 }
  59                 chId = dc.Id
  60         }
  61         var slice []*model.Post
  62         if *whole {
  63                 for n := 0; ; n++ {
  64                         log.Println("page:", n)
  65                         time.Sleep(mmc.SleepTime)
  66                         list, _, err := c.GetPostsBefore(chId, postId, n, mmc.PerPage, "", false)
  67                         if err != nil {
  68                                 log.Fatalln(err)
  69                         }
  70                         s := list.ToSlice()
  71                         slice = append(slice, s...)
  72                         if len(s) < mmc.PerPage {
  73                                 break
  74                         }
  75                 }
  76         } else {
  77                 list, _, err := c.GetPostsBefore(chId, postId, *pageNum, mmc.PerPage, "", false)
  78                 if err != nil {
  79                         log.Fatalln(err)
  80                 }
  81                 slice = list.ToSlice()
  82         }
  83         w := recfile.NewWriter(os.Stdout)
  84         for i := len(slice) - 1; i >= 0; i-- {
  85                 if err = mmc.PostToRec(
  86                         w, users,
  87                         mmc.Post{P: slice[i], E: model.WebsocketEventPosted},
  88                 ); err != nil {
  89                         log.Fatalln(err)
  90                 }
  91         }
  92         c.Logout()
  93 }