more testing

This commit is contained in:
Jannover5000r 2025-06-13 15:59:23 +02:00
parent a312f1d7ed
commit cf6f7ed843
3 changed files with 170 additions and 59 deletions

View file

@ -1,64 +1,20 @@
package Untis
import (
"fmt"
"log"
"os"
"os/signal"
"strings"
"github.com/bwmarrin/discordgo"
)
var BotToken string
func checkNilErr(e error) {
if e != nil {
log.Fatal("Error message")
}
}
func Run() {
// create a session
discord, err := discordgo.New("Bot " + BotToken)
checkNilErr(err)
// set intents to receive message events in guilds and DMs
discord.Identify.Intents = discordgo.IntentsGuildMessages | discordgo.IntentsDirectMessages
// add a event handler
discord.AddHandler(newMessage)
// open session
discord.Open()
defer discord.Close() // close session, after function termination
// keep bot running untill there is NO os interruption (ctrl + C)
fmt.Println("Bot running....")
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
<-c
}
func newMessage(discord *discordgo.Session, message *discordgo.MessageCreate) {
fmt.Printf(message.Content)
/* prevent bot responding to its own message
this is achived by looking into the message author id
if message.author.id is same as bot.author.id then just return
*/
if message.Author.ID == discord.State.User.ID {
log.Println("Run")
type res struct {
Name string
url string
key int
}
/* r := res{"Jannik", "google.com", 4}
r, err := json.Marshal(r)
if err != nil {
log.Printf("Error marshalling JSON: %v", err)
return
}
// respond to user message if it contains `!help` or `!bye`
switch {
case strings.Contains(message.Content, "!untis"):
discord.ChannelMessageSend(message.ChannelID, "Hello World😃")
case strings.Contains(message.Content, "!bye"):
discord.ChannelMessageSend(message.ChannelID, "Good Bye👋")
// add more cases if required
}
} */
}

163
main.go
View file

@ -1,14 +1,169 @@
package main
import (
"bytes"
"encoding/json"
"fmt"
"io/ioutil"
"log"
"net/http"
"os"
"os/signal"
"syscall"
"time"
Untis "untislogger/Bot"
"github.com/joho/godotenv"
)
func main() {
godotenv.Load()
Untis.BotToken = os.Getenv("DISCORD_TOKEN")
Untis.Run() // call the run function of Bot/Untis.go
// Global variable to store the current public IP
var currentPubIP string
// Discord webhook configuration
var discordWebhookURL string // Webhook URL from environment variable
func init() {
err := godotenv.Load()
if err != nil {
log.Printf("Error loading .env file: %v", err)
}
discordWebhookURL = os.Getenv("DISCORD_WEBHOOK_URL")
log.Println("Initializing application...")
log.Printf("DISCORD_WEBHOOK_URL: %q", discordWebhookURL)
// Check if Discord webhook is configured
if discordWebhookURL != "" {
log.Println("Discord webhook configured")
} else {
log.Println("No Discord webhook provided, Discord notifications will be disabled")
}
scheduleIPCheck() // Start the periodic IP check
}
func main() {
go Untis.Run() //starting API calls function
// Create a channel to receive OS signals
sigChan := make(chan os.Signal, 1)
signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
log.Println("Program is running. Press Ctrl+C to stop.")
// Block until we receive a signal
<-sigChan
log.Println("Shutting down...")
}
func testPubIP() {
resp, err := http.Get("https://api.ipify.org")
if err != nil {
log.Printf("Error getting public IP: %v", err)
return
}
defer resp.Body.Close()
ip, err := ioutil.ReadAll(resp.Body)
if err != nil {
log.Printf("Error reading response: %v", err)
return
}
log.Printf("Current public IP: %s", ip)
if string(ip) != currentPubIP {
log.Printf("IP changed sending notifications for new IP: %s", ip)
currentPubIP = string(ip)
notifyNewPubIP(currentPubIP)
}
}
func scheduleIPCheck() {
// Run once immediately
testPubIP()
// Then schedule to run every hour
ticker := time.NewTicker(1 * time.Hour)
go func() {
for range ticker.C {
testPubIP()
}
}()
}
func notifyNewPubIP(ip string) {
log.Printf("Notifying about new IP: %s", ip)
// Send Discord notification if webhook is configured
if discordWebhookURL != "" {
sendDiscordWebhook(ip)
} else {
log.Println("Discord webhook not configured, skipping Discord notification")
}
}
// DiscordWebhookPayload represents the structure for Discord webhook messages
type DiscordWebhookPayload struct {
Content string `json:"content"`
Embeds []Embed `json:"embeds,omitempty"`
}
type Embed struct {
Title string `json:"title"`
Description string `json:"description"`
Color int `json:"color"`
Timestamp string `json:"timestamp"`
Fields []Field `json:"fields,omitempty"`
}
type Field struct {
Name string `json:"name"`
Value string `json:"value"`
Inline bool `json:"inline"`
}
func sendDiscordWebhook(ip string) {
log.Println("Sending Discord webhook notification...")
// Create a rich embed message
embed := Embed{
Title: "🌐 Public IP Address Changed!",
Description: "Your public IP address has been updated.",
Color: 3066993, // Green color
Timestamp: time.Now().Format(time.RFC3339),
Fields: []Field{
{
Name: "New IP Address",
Value: fmt.Sprintf("`%s`", ip),
Inline: true,
},
{
Name: "Timestamp",
Value: time.Now().Format("2006-01-02 15:04:05"),
Inline: true,
},
},
}
payload := DiscordWebhookPayload{
Embeds: []Embed{embed},
}
jsonData, err := json.Marshal(payload)
if err != nil {
log.Printf("Error marshaling webhook payload: %v", err)
return
}
resp, err := http.Post(discordWebhookURL, "application/json", bytes.NewBuffer(jsonData))
if err != nil {
log.Printf("Error sending Discord webhook: %v", err)
return
}
defer resp.Body.Close()
if resp.StatusCode >= 200 && resp.StatusCode < 300 {
log.Println("Discord webhook notification sent successfully")
} else {
body, _ := ioutil.ReadAll(resp.Body)
log.Printf("Discord webhook failed with status %d: %s", resp.StatusCode, string(body))
}
}

Binary file not shown.