Compare commits
2 commits
ca09e35027
...
76316016b5
Author | SHA1 | Date | |
---|---|---|---|
76316016b5 | |||
239923e298 |
6 changed files with 261 additions and 7 deletions
|
@ -11,6 +11,8 @@ import (
|
||||||
"github.com/Cynosphere/comcord/lib"
|
"github.com/Cynosphere/comcord/lib"
|
||||||
"github.com/Cynosphere/comcord/state"
|
"github.com/Cynosphere/comcord/state"
|
||||||
"github.com/bwmarrin/discordgo"
|
"github.com/bwmarrin/discordgo"
|
||||||
|
"github.com/mgutz/ansi"
|
||||||
|
tsize "github.com/kopoli/go-terminal-size"
|
||||||
)
|
)
|
||||||
|
|
||||||
var REGEX_EMOTE = regexp.MustCompile(`<(?:\x{200b}|&)?a?:(\w+):(\d+)>`)
|
var REGEX_EMOTE = regexp.MustCompile(`<(?:\x{200b}|&)?a?:(\w+):(\d+)>`)
|
||||||
|
@ -58,7 +60,7 @@ func ListGuildsCommand(session *discordgo.Session) {
|
||||||
fmt.Print("\n\r")
|
fmt.Print("\n\r")
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetSortedChannels(session *discordgo.Session, guildId string, withCategories bool) []*discordgo.Channel {
|
func GetSortedChannels(session *discordgo.Session, guildId string, withCategories bool, withPrivate bool) []*discordgo.Channel {
|
||||||
channels := make([]*discordgo.Channel, 0)
|
channels := make([]*discordgo.Channel, 0)
|
||||||
guild, err := session.State.Guild(guildId)
|
guild, err := session.State.Guild(guildId)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -69,6 +71,20 @@ func GetSortedChannels(session *discordgo.Session, guildId string, withCategorie
|
||||||
categories := make(map[string][]*discordgo.Channel)
|
categories := make(map[string][]*discordgo.Channel)
|
||||||
|
|
||||||
for _, channel := range guild.Channels {
|
for _, channel := range guild.Channels {
|
||||||
|
if channel.Type != discordgo.ChannelTypeGuildText && channel.Type != discordgo.ChannelTypeGuildNews {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
perms, err := session.State.UserChannelPermissions(session.State.User.ID, channel.ID)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
private := perms & discordgo.PermissionViewChannel == 0
|
||||||
|
if private && !withPrivate {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
categoryID := "0"
|
categoryID := "0"
|
||||||
if channel.ParentID != "" {
|
if channel.ParentID != "" {
|
||||||
categoryID = channel.ParentID
|
categoryID = channel.ParentID
|
||||||
|
@ -79,9 +95,6 @@ func GetSortedChannels(session *discordgo.Session, guildId string, withCategorie
|
||||||
categories[categoryID] = make([]*discordgo.Channel, 0)
|
categories[categoryID] = make([]*discordgo.Channel, 0)
|
||||||
}
|
}
|
||||||
|
|
||||||
if channel.Type != discordgo.ChannelTypeGuildText && channel.Type != discordgo.ChannelTypeGuildNews {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
categories[categoryID] = append(categories[categoryID], channel)
|
categories[categoryID] = append(categories[categoryID], channel)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -134,6 +147,17 @@ func GetSortedChannels(session *discordgo.Session, guildId string, withCategorie
|
||||||
if channel.Type != discordgo.ChannelTypeGuildText && channel.Type != discordgo.ChannelTypeGuildNews {
|
if channel.Type != discordgo.ChannelTypeGuildText && channel.Type != discordgo.ChannelTypeGuildNews {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
perms, err := session.State.UserChannelPermissions(session.State.User.ID, channel.ID)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
private := perms & discordgo.PermissionViewChannel == 0
|
||||||
|
if private && !withPrivate {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
channels = append(channels, channel)
|
channels = append(channels, channel)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -153,7 +177,7 @@ func ListChannelsCommand(session *discordgo.Session) {
|
||||||
}
|
}
|
||||||
|
|
||||||
longest := 0
|
longest := 0
|
||||||
channels := GetSortedChannels(session, currentGuild, true)
|
channels := GetSortedChannels(session, currentGuild, true, false)
|
||||||
|
|
||||||
for _, channel := range channels {
|
for _, channel := range channels {
|
||||||
perms, err := session.State.UserChannelPermissions(session.State.User.ID, channel.ID)
|
perms, err := session.State.UserChannelPermissions(session.State.User.ID, channel.ID)
|
||||||
|
@ -224,8 +248,166 @@ func ListChannelsCommand(session *discordgo.Session) {
|
||||||
fmt.Print("\n\r")
|
fmt.Print("\n\r")
|
||||||
}
|
}
|
||||||
|
|
||||||
func ListUsersCommand(session *discordgo.Session) {
|
type ListedMember struct {
|
||||||
|
Name string
|
||||||
|
Bot bool
|
||||||
|
Status discordgo.Status
|
||||||
|
Position int
|
||||||
|
}
|
||||||
|
|
||||||
|
func ListUsersCommand(session *discordgo.Session) {
|
||||||
|
currentGuild := state.GetCurrentGuild()
|
||||||
|
currentChannel := state.GetCurrentChannel()
|
||||||
|
|
||||||
|
if currentGuild == "" {
|
||||||
|
fmt.Print("<not in a guild>\n\r")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if currentChannel == "" {
|
||||||
|
fmt.Print("<not in a channel>\n\r")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
guild, err := session.State.Guild(state.GetCurrentGuild())
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
channel, err := session.State.Channel(currentChannel)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Print("\n\r")
|
||||||
|
fmt.Printf("[you are in '%s' in '#%s' among %d]\n\r", guild.Name, channel.Name, guild.MemberCount)
|
||||||
|
fmt.Print("\n\r")
|
||||||
|
|
||||||
|
longest := 0
|
||||||
|
|
||||||
|
sortedMembers := make([]ListedMember, 0)
|
||||||
|
|
||||||
|
for _, presence := range guild.Presences {
|
||||||
|
if presence.Status == discordgo.StatusOffline {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
perms, err := session.State.UserChannelPermissions(presence.User.ID, currentChannel)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if perms & discordgo.PermissionViewChannel == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
member, err := session.State.Member(currentGuild, presence.User.ID)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
length := utf8.RuneCountInString(member.User.Username) + 3
|
||||||
|
if length > longest {
|
||||||
|
longest = length
|
||||||
|
}
|
||||||
|
|
||||||
|
position := 0
|
||||||
|
for _, id := range member.Roles {
|
||||||
|
role, err := session.State.Role(currentGuild, id)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if role.Hoist && role.Position > position {
|
||||||
|
position = role.Position
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
sortedMembers = append(sortedMembers, ListedMember{
|
||||||
|
Name: member.User.Username,
|
||||||
|
Bot: member.User.Bot,
|
||||||
|
Status: presence.Status,
|
||||||
|
Position: position,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
membersByPosition := make(map[int][]ListedMember)
|
||||||
|
for _, member := range sortedMembers {
|
||||||
|
_, has := membersByPosition[member.Position]
|
||||||
|
if !has {
|
||||||
|
membersByPosition[member.Position] = make([]ListedMember, 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
membersByPosition[member.Position] = append(membersByPosition[member.Position], member)
|
||||||
|
}
|
||||||
|
for _, members := range membersByPosition {
|
||||||
|
sort.Slice(members, func(i, j int) bool {
|
||||||
|
return members[i].Name < members[j].Name
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
positions := make([]int, 0, len(membersByPosition))
|
||||||
|
for k := range membersByPosition {
|
||||||
|
positions = append(positions, k)
|
||||||
|
}
|
||||||
|
sort.Slice(positions, func(i, j int) bool {
|
||||||
|
return positions[i] > positions[j]
|
||||||
|
})
|
||||||
|
|
||||||
|
size, err := tsize.GetSize()
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
columns := int(math.Floor(float64(size.Width) / float64(longest)))
|
||||||
|
|
||||||
|
index := 0
|
||||||
|
for _, position := range positions {
|
||||||
|
members := membersByPosition[position]
|
||||||
|
for _, member := range members {
|
||||||
|
|
||||||
|
statusColor := "reset"
|
||||||
|
if member.Status == discordgo.StatusOnline {
|
||||||
|
statusColor = "green+b"
|
||||||
|
} else if member.Status == discordgo.StatusIdle {
|
||||||
|
statusColor = "yellow+b"
|
||||||
|
} else if member.Status == discordgo.StatusDoNotDisturb {
|
||||||
|
statusColor = "red+b"
|
||||||
|
}
|
||||||
|
|
||||||
|
nameColor := "reset"
|
||||||
|
if member.Bot {
|
||||||
|
nameColor = "yellow"
|
||||||
|
}
|
||||||
|
|
||||||
|
nameAndStatus := ansi.Color(" \u2022 ", statusColor) + ansi.Color(member.Name, nameColor)
|
||||||
|
nameLength := utf8.RuneCountInString(member.Name) + 3
|
||||||
|
|
||||||
|
index++
|
||||||
|
|
||||||
|
pad := 0
|
||||||
|
if index % columns != 0 {
|
||||||
|
pad = longest - nameLength
|
||||||
|
}
|
||||||
|
if pad < 0 {
|
||||||
|
pad = 0
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf(nameAndStatus + strings.Repeat(" ", pad))
|
||||||
|
|
||||||
|
if index % columns == 0 {
|
||||||
|
fmt.Print("\n\r")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if index % columns != 0 {
|
||||||
|
fmt.Print("\n\r")
|
||||||
|
}
|
||||||
|
fmt.Print("\n\r")
|
||||||
|
|
||||||
|
if channel.Topic != "" {
|
||||||
|
fmt.Print("--Topic" + strings.Repeat("-", 73) + "\n\r")
|
||||||
|
for _, line := range strings.Split(channel.Topic, "\n") {
|
||||||
|
fmt.Print(line + "\n\r")
|
||||||
|
}
|
||||||
|
fmt.Print(strings.Repeat("-", 80) + "\n\r")
|
||||||
|
fmt.Print("\n\r")
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func SwitchGuild(session *discordgo.Session, input string) {
|
func SwitchGuild(session *discordgo.Session, input string) {
|
||||||
|
@ -248,7 +430,7 @@ func SwitchGuild(session *discordgo.Session, input string) {
|
||||||
state.SetCurrentGuild(target)
|
state.SetCurrentGuild(target)
|
||||||
last := state.GetLastChannel(target)
|
last := state.GetLastChannel(target)
|
||||||
if last == "" {
|
if last == "" {
|
||||||
channels := GetSortedChannels(session, target, false)
|
channels := GetSortedChannels(session, target, false, false)
|
||||||
topChannel := channels[0]
|
topChannel := channels[0]
|
||||||
|
|
||||||
state.SetCurrentChannel(topChannel.ID)
|
state.SetCurrentChannel(topChannel.ID)
|
||||||
|
@ -271,3 +453,41 @@ func SwitchGuildsCommand(session *discordgo.Session) {
|
||||||
SwitchGuild(session, input)
|
SwitchGuild(session, input)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func SwitchChannelsCommand(session *discordgo.Session) {
|
||||||
|
currentGuild := state.GetCurrentGuild()
|
||||||
|
|
||||||
|
if currentGuild == "" {
|
||||||
|
fmt.Print("<not in a guild>\n\r")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
lib.MakePrompt(session, ":channel> ", false, func(session *discordgo.Session, input string, interrupt bool) {
|
||||||
|
fmt.Print("\r")
|
||||||
|
if input == "" {
|
||||||
|
ListUsersCommand(session)
|
||||||
|
} else {
|
||||||
|
target := ""
|
||||||
|
|
||||||
|
channels := GetSortedChannels(session, currentGuild, false, false)
|
||||||
|
|
||||||
|
for _, channel := range channels {
|
||||||
|
if strings.Index(strings.ToLower(channel.Name), strings.ToLower(input)) > -1 {
|
||||||
|
target = channel.ID
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if target == "" {
|
||||||
|
fmt.Print("<channel not found>\n\r")
|
||||||
|
} else {
|
||||||
|
state.SetCurrentChannel(target)
|
||||||
|
state.SetLastChannel(currentGuild, target)
|
||||||
|
|
||||||
|
ListUsersCommand(session)
|
||||||
|
|
||||||
|
// TODO: update presence
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
|
@ -46,6 +46,16 @@ func Setup() {
|
||||||
Run: SwitchGuildsCommand,
|
Run: SwitchGuildsCommand,
|
||||||
Description: "goto guild",
|
Description: "goto guild",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
commandMap["g"] = Command{
|
||||||
|
Run: SwitchChannelsCommand,
|
||||||
|
Description: "goto channel",
|
||||||
|
}
|
||||||
|
|
||||||
|
commandMap["w"] = Command{
|
||||||
|
Run: ListUsersCommand,
|
||||||
|
Description: "who is in channel",
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetCommand(key string) (Command, bool) {
|
func GetCommand(key string) (Command, bool) {
|
||||||
|
|
|
@ -18,6 +18,23 @@ func SendMode(session *discordgo.Session) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
channel, err := session.State.Channel(channelId)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Print("<error getting channel: ", err.Error(), ">\n\r")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
perms, err := session.State.UserChannelPermissions(session.State.User.ID, channel.ID)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Print("<failed to check permissions: ", err.Error(), ">\n\r")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if perms & discordgo.PermissionSendMessages == 0 {
|
||||||
|
fmt.Print("<you do not have permission to send messages here>\n\r")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
length := utf8.RuneCountInString(session.State.User.Username) + 2
|
length := utf8.RuneCountInString(session.State.User.Username) + 2
|
||||||
curLength := state.GetNameLength()
|
curLength := state.GetNameLength()
|
||||||
|
|
||||||
|
|
|
@ -38,6 +38,10 @@ func MessageUpdate(session *discordgo.Session, msg *discordgo.MessageUpdate) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if msg.Content == msg.BeforeUpdate.Content {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
channel, err := session.State.Channel(msg.ChannelID)
|
channel, err := session.State.Channel(msg.ChannelID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
|
|
1
go.mod
1
go.mod
|
@ -8,6 +8,7 @@ require (
|
||||||
github.com/containerd/console v1.0.3 // indirect
|
github.com/containerd/console v1.0.3 // indirect
|
||||||
github.com/ergochat/readline v0.0.5 // indirect
|
github.com/ergochat/readline v0.0.5 // indirect
|
||||||
github.com/gorilla/websocket v1.4.2 // indirect
|
github.com/gorilla/websocket v1.4.2 // indirect
|
||||||
|
github.com/kopoli/go-terminal-size v0.0.0-20170219200355-5c97524c8b54 // indirect
|
||||||
github.com/mattn/go-colorable v0.1.13 // indirect
|
github.com/mattn/go-colorable v0.1.13 // indirect
|
||||||
github.com/mattn/go-isatty v0.0.17 // indirect
|
github.com/mattn/go-isatty v0.0.17 // indirect
|
||||||
github.com/mattn/go-runewidth v0.0.13 // indirect
|
github.com/mattn/go-runewidth v0.0.13 // indirect
|
||||||
|
|
2
go.sum
2
go.sum
|
@ -29,6 +29,8 @@ github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/ad
|
||||||
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
|
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
|
||||||
github.com/klauspost/cpuid/v2 v2.0.10/go.mod h1:g2LTdtYhdyuGPqyWyv7qRAmj1WBqxuObKfj5c0PQa7c=
|
github.com/klauspost/cpuid/v2 v2.0.10/go.mod h1:g2LTdtYhdyuGPqyWyv7qRAmj1WBqxuObKfj5c0PQa7c=
|
||||||
github.com/klauspost/cpuid/v2 v2.0.12/go.mod h1:g2LTdtYhdyuGPqyWyv7qRAmj1WBqxuObKfj5c0PQa7c=
|
github.com/klauspost/cpuid/v2 v2.0.12/go.mod h1:g2LTdtYhdyuGPqyWyv7qRAmj1WBqxuObKfj5c0PQa7c=
|
||||||
|
github.com/kopoli/go-terminal-size v0.0.0-20170219200355-5c97524c8b54 h1:0SMHxjkLKNawqUjjnMlCtEdj6uWZjv0+qDZ3F6GOADI=
|
||||||
|
github.com/kopoli/go-terminal-size v0.0.0-20170219200355-5c97524c8b54/go.mod h1:bm7MVZZvHQBfqHG5X59jrRE/3ak6HvK+/Zb6aZhLR2s=
|
||||||
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||||
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||||
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||||
|
|
Loading…
Reference in a new issue