mirror of
https://github.com/jesperh1/csgo-hub-discord.git
synced 2025-05-16 01:08:10 +01:00
Init
This commit is contained in:
parent
0389f0cd28
commit
ec5e550fa8
14
go.mod
Normal file
14
go.mod
Normal file
@ -0,0 +1,14 @@
|
||||
module github.com/jesperbakhandskemager/csgo-hub-discord
|
||||
|
||||
go 1.19
|
||||
|
||||
require (
|
||||
github.com/bwmarrin/discordgo v0.26.1
|
||||
gopkg.in/yaml.v3 v3.0.1
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/gorilla/websocket v1.4.2 // indirect
|
||||
golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b // indirect
|
||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68 // indirect
|
||||
)
|
||||
15
go.sum
Normal file
15
go.sum
Normal file
@ -0,0 +1,15 @@
|
||||
github.com/bwmarrin/discordgo v0.26.1 h1:AIrM+g3cl+iYBr4yBxCBp9tD9jR3K7upEjl0d89FRkE=
|
||||
github.com/bwmarrin/discordgo v0.26.1/go.mod h1:NJZpH+1AfhIcyQsPeuBKsUtYrRnjkyu0kIVMCHkZtRY=
|
||||
github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc=
|
||||
github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
|
||||
golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b h1:7mWr3k41Qtv8XlltBkDkl8LoP3mpSgBW8BUoxtEdbXg=
|
||||
golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b/go.mod h1:T9bdIzuCu7OtxOm1hfPfRQxPLYneinmdGuTeoZ9dtd4=
|
||||
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68 h1:nxC68pudNYkKU6jWhgrqdreuFiOQWj1Fs7T3VrH4Pjw=
|
||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
259
main.go
Normal file
259
main.go
Normal file
@ -0,0 +1,259 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"flag"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"strings"
|
||||
|
||||
"github.com/bwmarrin/discordgo"
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
const API string = "https://steam.csgohub.xyz/"
|
||||
|
||||
var BotID string
|
||||
|
||||
var s *discordgo.Session
|
||||
|
||||
type YAMLFile struct {
|
||||
Config Config `yaml:"config"`
|
||||
}
|
||||
|
||||
type Config struct {
|
||||
STEAM_KEY string `yaml:"STEAM_KEY"`
|
||||
DISCORD_TOKEN string `yaml:"DISCORD_TOKEN"`
|
||||
}
|
||||
|
||||
func ReadConfig() (*Config, error) {
|
||||
config := &YAMLFile{}
|
||||
cfgFile, err := os.ReadFile("./config.yaml")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
err = yaml.Unmarshal(cfgFile, config)
|
||||
return &config.Config, err
|
||||
}
|
||||
|
||||
var (
|
||||
GuildID = flag.String("guild", "", "Test guild ID. If not passed - bot registers commands globally")
|
||||
RemoveCommands = flag.Bool("rmcmd", true, "Remove all commands after shutdowning or not")
|
||||
)
|
||||
|
||||
func init() {
|
||||
config, err := ReadConfig()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
s, err = discordgo.New("Bot " + config.DISCORD_TOKEN)
|
||||
if err != nil {
|
||||
log.Fatalf("Invalid bot parameters: %v", err)
|
||||
}
|
||||
u, err := s.User("@me")
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
return
|
||||
}
|
||||
|
||||
BotID = u.ID
|
||||
}
|
||||
|
||||
var (
|
||||
commands = []*discordgo.ApplicationCommand{
|
||||
{
|
||||
Name: "link-steam",
|
||||
Description: "Link your steam account",
|
||||
Type: discordgo.ChatApplicationCommand,
|
||||
},
|
||||
{
|
||||
Name: "show-team",
|
||||
Description: "Shows the CS:GO friend codes of the other users in VC",
|
||||
Type: discordgo.ChatApplicationCommand,
|
||||
},
|
||||
}
|
||||
|
||||
commandHandlers = map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate){
|
||||
"link-steam": func(s *discordgo.Session, i *discordgo.InteractionCreate) {
|
||||
if i.Member == nil {
|
||||
token := GetToken(i.User.ID)
|
||||
|
||||
s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
|
||||
Type: discordgo.InteractionResponseChannelMessageWithSource,
|
||||
Data: &discordgo.InteractionResponseData{
|
||||
Content: "Your verification link:",
|
||||
},
|
||||
})
|
||||
s.ChannelMessageSend(i.ChannelID, API+token)
|
||||
return
|
||||
}
|
||||
sender := i.Member.User.ID
|
||||
token := GetToken(sender)
|
||||
userDM, err := s.UserChannelCreate(sender)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
|
||||
Type: discordgo.InteractionResponseChannelMessageWithSource,
|
||||
Data: &discordgo.InteractionResponseData{
|
||||
Content: "Couldn't send you verification link",
|
||||
},
|
||||
})
|
||||
return
|
||||
}
|
||||
s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
|
||||
Type: discordgo.InteractionResponseChannelMessageWithSource,
|
||||
Data: &discordgo.InteractionResponseData{
|
||||
Content: "A verification link has been send to your DM's",
|
||||
},
|
||||
})
|
||||
s.ChannelMessageSend(userDM.ID, API+token)
|
||||
|
||||
},
|
||||
"show-team": func(s *discordgo.Session, i *discordgo.InteractionCreate) {
|
||||
// Find the guild for that channel.
|
||||
g, err := s.State.Guild(i.GuildID)
|
||||
if err != nil {
|
||||
// Could not find guild.
|
||||
log.Println("couldn't find guild")
|
||||
return
|
||||
}
|
||||
|
||||
var vcChannelId string
|
||||
var vcUsers []string
|
||||
|
||||
// Look for the message sender in that guild's current voice states.
|
||||
for _, vs := range g.VoiceStates {
|
||||
if vs.UserID == i.Member.User.ID {
|
||||
vcChannelId = vs.ChannelID
|
||||
}
|
||||
}
|
||||
|
||||
for _, vs := range g.VoiceStates {
|
||||
if vs.ChannelID == vcChannelId {
|
||||
vcUsers = append(vcUsers, vs.UserID)
|
||||
}
|
||||
}
|
||||
var returnString strings.Builder
|
||||
friends := GetFriendCodes(vcUsers)
|
||||
for _, friend := range friends {
|
||||
if friend.DiscordId != "" {
|
||||
returnString.WriteString("<@" + friend.DiscordId + ">: " + friend.FriendCode + "\n")
|
||||
}
|
||||
}
|
||||
if returnString.String() == "" {
|
||||
returnString.WriteString("No friend codes found")
|
||||
}
|
||||
|
||||
s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
|
||||
Type: discordgo.InteractionResponseChannelMessageWithSource,
|
||||
Data: &discordgo.InteractionResponseData{
|
||||
Content: returnString.String(),
|
||||
},
|
||||
})
|
||||
},
|
||||
}
|
||||
)
|
||||
|
||||
type userStruct struct {
|
||||
Id int `json:"id"`
|
||||
CreatedAt string `json:"created_at"`
|
||||
DiscordId string `json:"discord_id"`
|
||||
FriendCode string `json:"friend_code"`
|
||||
}
|
||||
|
||||
func GetFriendCode(users []string) []userStruct {
|
||||
var userArr []userStruct
|
||||
for _, u := range users {
|
||||
resp, err := http.Get("http://localhost:8383/api/v1/user/" + u)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
log.Println("quiting")
|
||||
return userArr
|
||||
}
|
||||
var userTemp userStruct
|
||||
bodyBytes, _ := io.ReadAll(resp.Body)
|
||||
resp.Body.Close()
|
||||
json.Unmarshal(bodyBytes, &userTemp)
|
||||
userArr = append(userArr, userTemp)
|
||||
}
|
||||
return userArr
|
||||
}
|
||||
|
||||
func GetFriendCodes(users []string) []userStruct {
|
||||
var userArr []userStruct
|
||||
|
||||
for _, u := range users {
|
||||
var us userStruct
|
||||
us.DiscordId = u
|
||||
userArr = append(userArr, us)
|
||||
}
|
||||
jsonReq, err := json.Marshal(userArr)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return userArr
|
||||
}
|
||||
userArr = nil
|
||||
|
||||
resp, err := http.Post("http://localhost:8383/api/v1/users", "application/json; charset=utf-8", bytes.NewBuffer(jsonReq))
|
||||
if err != nil {
|
||||
log.Fatalln(err)
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
bodyBytes, _ := io.ReadAll(resp.Body)
|
||||
json.Unmarshal(bodyBytes, &userArr)
|
||||
|
||||
return userArr
|
||||
}
|
||||
|
||||
func GetToken(discordId string) string {
|
||||
resp, err := http.Get("http://localhost:8383/api/v1/token/" + discordId)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
return ""
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
bodyBytes, _ := io.ReadAll(resp.Body)
|
||||
bodyString := strings.Replace(string(bodyBytes), "\"", "", -1)
|
||||
println(bodyString)
|
||||
return bodyString
|
||||
}
|
||||
|
||||
func main() {
|
||||
s.AddHandler(func(s *discordgo.Session, r *discordgo.Ready) { log.Println("Bot is up!") })
|
||||
s.AddHandler(func(s *discordgo.Session, i *discordgo.InteractionCreate) {
|
||||
if h, ok := commandHandlers[i.ApplicationCommandData().Name]; ok {
|
||||
h(s, i)
|
||||
}
|
||||
})
|
||||
s.Identify.Intents = discordgo.IntentsGuildVoiceStates
|
||||
err := s.Open()
|
||||
if err != nil {
|
||||
log.Fatalf("Cannot open the session: %v", err)
|
||||
}
|
||||
defer s.Close()
|
||||
|
||||
createdCommands, err := s.ApplicationCommandBulkOverwrite(s.State.User.ID, *GuildID, commands)
|
||||
|
||||
if err != nil {
|
||||
log.Fatalf("Cannot register commands: %v", err)
|
||||
}
|
||||
|
||||
stop := make(chan os.Signal, 1)
|
||||
signal.Notify(stop, os.Interrupt)
|
||||
<-stop
|
||||
log.Println("Gracefully shutting down")
|
||||
|
||||
if *RemoveCommands {
|
||||
for _, cmd := range createdCommands {
|
||||
err := s.ApplicationCommandDelete(s.State.User.ID, *GuildID, cmd.ID)
|
||||
if err != nil {
|
||||
log.Fatalf("Cannot delete %q command: %v", cmd.Name, err)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user