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