cmd/sb/main.go

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