changed message and added toggle for default message
This commit is contained in:
parent
f3636c15c9
commit
7646684136
2 changed files with 25 additions and 9 deletions
|
|
@ -16,9 +16,10 @@ import (
|
||||||
"syscall"
|
"syscall"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/joho/godotenv"
|
|
||||||
"untislogger/Untis"
|
"untislogger/Untis"
|
||||||
|
|
||||||
|
"github.com/joho/godotenv"
|
||||||
|
|
||||||
"github.com/bwmarrin/discordgo"
|
"github.com/bwmarrin/discordgo"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -43,6 +44,8 @@ var (
|
||||||
|
|
||||||
var encryptionKey []byte
|
var encryptionKey []byte
|
||||||
|
|
||||||
|
var WebHook = true
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
godotenv.Load(".env")
|
godotenv.Load(".env")
|
||||||
keyStr := os.Getenv("ENC_KEY")
|
keyStr := os.Getenv("ENC_KEY")
|
||||||
|
|
@ -135,7 +138,7 @@ func saveAccount(userID, username, password string) error {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
if err := os.WriteFile(accountsFile, newData, 0644); err != nil {
|
if err := os.WriteFile(accountsFile, newData, 0o644); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -143,10 +146,10 @@ func saveAccount(userID, username, password string) error {
|
||||||
timetableFile := getTimetableFile(userID)
|
timetableFile := getTimetableFile(userID)
|
||||||
timetableFilledFile := getTimetableFilledFile(userID)
|
timetableFilledFile := getTimetableFilledFile(userID)
|
||||||
if _, err := os.Stat(timetableFile); os.IsNotExist(err) {
|
if _, err := os.Stat(timetableFile); os.IsNotExist(err) {
|
||||||
os.WriteFile(timetableFile, []byte("[]"), 0644)
|
os.WriteFile(timetableFile, []byte("[]"), 0o644)
|
||||||
}
|
}
|
||||||
if _, err := os.Stat(timetableFilledFile); os.IsNotExist(err) {
|
if _, err := os.Stat(timetableFilledFile); os.IsNotExist(err) {
|
||||||
os.WriteFile(timetableFilledFile, []byte("[]"), 0644)
|
os.WriteFile(timetableFilledFile, []byte("[]"), 0o644)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
|
|
@ -355,7 +358,7 @@ func deleteAccount(userID string) error {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
if err := os.WriteFile(accountsFile, newData, 0644); err != nil {
|
if err := os.WriteFile(accountsFile, newData, 0o644); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -371,7 +374,15 @@ func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
|
||||||
if m.Author.ID == s.State.User.ID {
|
if m.Author.ID == s.State.User.ID {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
if m.GuildID != "" && m.Content == "!WebHook" {
|
||||||
|
_ = s.ChannelMessageDelete(m.ChannelID, m.ID)
|
||||||
|
WebHook = !WebHook
|
||||||
|
if WebHook {
|
||||||
|
s.ChannelMessageSend(m.ChannelID, "Sending Webhook set to True")
|
||||||
|
} else {
|
||||||
|
s.ChannelMessageSend(m.ChannelID, "Sending Webhook set to False")
|
||||||
|
}
|
||||||
|
}
|
||||||
// Handle "!addaccount" only in guilds (not in DMs)
|
// Handle "!addaccount" only in guilds (not in DMs)
|
||||||
if m.GuildID != "" && m.Content == "!addaccount" {
|
if m.GuildID != "" && m.Content == "!addaccount" {
|
||||||
// Delete the command for privacy
|
// Delete the command for privacy
|
||||||
|
|
|
||||||
11
main.go
11
main.go
|
|
@ -33,6 +33,8 @@ type NamedTimetableEntry struct {
|
||||||
ActivityType string `json:"activityType"`
|
ActivityType string `json:"activityType"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var WebHook = true
|
||||||
|
|
||||||
// init and main//
|
// init and main//
|
||||||
func init() {
|
func init() {
|
||||||
godotenv.Load(".env")
|
godotenv.Load(".env")
|
||||||
|
|
@ -202,15 +204,18 @@ func Run(userID string) {
|
||||||
status, _ := NextCodeForTime(codeByStartTime, now)
|
status, _ := NextCodeForTime(codeByStartTime, now)
|
||||||
|
|
||||||
var message string
|
var message string
|
||||||
if status != "" {
|
if status == "cancelled" {
|
||||||
|
message = fmt.Sprintf("Next lesson: **%s** at **%s**. is **%s**", subject, nextTime, status)
|
||||||
|
} else if status != "" && status != "cancelled" {
|
||||||
message = fmt.Sprintf("Next lesson: **%s** in room **%s** at **%s**. Status: **%s**", subject, room, nextTime, status)
|
message = fmt.Sprintf("Next lesson: **%s** in room **%s** at **%s**. Status: **%s**", subject, room, nextTime, status)
|
||||||
} else {
|
} else {
|
||||||
message = fmt.Sprintf("Next lesson: **%s** in room **%s** at **%s**.", subject, room, nextTime)
|
message = fmt.Sprintf("Next lesson: **%s** in room **%s** at **%s**.", subject, room, nextTime)
|
||||||
}
|
}
|
||||||
|
|
||||||
if userID == "default" {
|
if userID == "default" {
|
||||||
// The default user still sends to the webhook
|
if WebHook {
|
||||||
sendDiscordWebhook(subject, room, nextTime, status)
|
sendDiscordWebhook(subject, room, nextTime, status)
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
BotStart.SendDM(userID, message)
|
BotStart.SendDM(userID, message)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue