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