]> Sergey Matveev's repositories - mmc.git/blob - cmd/scrollback/main.go
cmd/scrollback -whole
[mmc.git] / 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 }