/
channel.go
85 lines (76 loc) · 2.32 KB
/
channel.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package slacklog
import (
"sort"
"github.com/slack-go/slack"
)
// ChannelTable : チャンネルデータを保持する。
// channelsもchannelMapも保持するチャンネルデータは同じで、channelMapはチャンネ
// ルIDをキーとするmapとなっている。
// ユースケースに応じてchannelsとchannelMapは使い分ける。
type ChannelTable struct {
Channels []Channel
ChannelMap map[string]*Channel
}
// NewChannelTable : pathに指定したJSON形式のチャンネルデータを読み込み、
// ChannelTable を生成する。
// whitelistに指定したチャンネル名のみを読み込む。
func NewChannelTable(path string, whitelist []string) (*ChannelTable, error) {
var channels []Channel
if err := ReadFileAsJSON(path, true, &channels); err != nil {
return nil, err
}
channels = FilterChannel(channels, whitelist)
sort.Slice(channels, func(i, j int) bool {
return channels[i].Name < channels[j].Name
})
channelMap := make(map[string]*Channel, len(channels))
for i, ch := range channels {
channelMap[ch.ID] = &channels[i]
}
return &ChannelTable{
Channels: channels,
ChannelMap: channelMap,
}, nil
}
// FilterChannel : whitelistに指定したチャンネル名に該当するチャンネルのみを返
// す。
// whitelistに'*'が含まれる場合はchannelをそのまま返す。
func FilterChannel(channels []Channel, whitelist []string) []Channel {
if len(whitelist) == 0 {
return []Channel{}
}
allowed := map[string]struct{}{}
for _, s := range whitelist {
if s == "*" {
return channels
}
allowed[s] = struct{}{}
}
newChannels := make([]Channel, 0, len(whitelist))
for _, ch := range channels {
_, ok := allowed[ch.Name]
if ok {
newChannels = append(newChannels, ch)
}
}
return newChannels
}
// SortChannel sorts []Channel by name. It modify original slice.
func SortChannel(channels []Channel) {
sort.SliceStable(channels, func(i, j int) bool {
return channels[i].Name < channels[j].Name
})
}
// Channel represents channel object in Slack.
type Channel struct {
slack.Channel
Pins []ChannelPin `json:"pins"`
}
// ChannelPin represents a pinned message for a channel.
type ChannelPin struct {
ID string `json:"id"`
Typ string `json:"type"`
Created int64 `json:"created"`
User string `json:"user"`
Owner string `json:"owner"`
}