133 lines
2.5 KiB
Go
133 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"os/user"
|
|
"strconv"
|
|
//"strings"
|
|
"time"
|
|
//"errors"
|
|
|
|
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
|
|
)
|
|
|
|
const (
|
|
telegramTokenID = "TELEGRAM_BOT_TOKEN"
|
|
telegramChatID = "TELEGRAM_CHAT_ID"
|
|
)
|
|
|
|
type HostInfo struct {
|
|
hostname string
|
|
date string
|
|
}
|
|
|
|
func sendInfo() HostInfo{
|
|
hostname, err := os.Hostname()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
Info := HostInfo {
|
|
hostname: hostname,
|
|
date: time.Now().Format("2006-05-05 15:04:05"),
|
|
}
|
|
return Info
|
|
}
|
|
|
|
func checkId() bool {
|
|
currentUser, err := user.Current()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if currentUser.Uid == "0" {
|
|
return true
|
|
} else {
|
|
return false
|
|
}
|
|
}
|
|
|
|
func telegramBot(botToken string, chatID int64) {
|
|
theHostInfo := sendInfo()
|
|
bot, err := tgbotapi.NewBotAPI(botToken)
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
updateConfig := tgbotapi.NewUpdate(0)
|
|
updateConfig.Timeout = 30
|
|
updates := bot.GetUpdatesChan(updateConfig)
|
|
|
|
|
|
for update := range updates {
|
|
if update.Message.Chat.ID != chatID {
|
|
continue
|
|
}
|
|
if update.Message == nil {
|
|
continue
|
|
}
|
|
if !update.Message.IsCommand() {
|
|
continue
|
|
}
|
|
|
|
var out string
|
|
switch update.Message.Command() {
|
|
case "umount":
|
|
out = umount(theHostInfo)
|
|
case "default":
|
|
out = fmt.Sprintf("Comando %s no reconocido", update.Message.Command())
|
|
}
|
|
data := fmt.Sprintf("Mensaje de host %s: %s\n", theHostInfo.hostname, out)
|
|
msg := tgbotapi.NewMessage(chatID, data)
|
|
_, err := bot.Send(msg)
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func umount(h HostInfo) string {
|
|
switch h.hostname {
|
|
case "alfa":
|
|
cmd := exec.Command("umount", "/home/kar/pelis/")
|
|
output, err := cmd.CombinedOutput()
|
|
if err != nil {
|
|
outErr := err.Error()
|
|
return string(outErr)
|
|
}
|
|
if string(output) == "" {
|
|
output = []byte("Se logro desmontar el directorio")
|
|
}
|
|
return string(output)
|
|
|
|
case "jellyfin ":
|
|
cmd := exec.Command("umount", "/mnt/pelis")
|
|
output, err := cmd.CombinedOutput()
|
|
if err != nil {
|
|
outErr := err.Error()
|
|
return string(outErr)
|
|
}
|
|
return string(output)
|
|
}
|
|
return "error"
|
|
}
|
|
|
|
|
|
func main() {
|
|
botToken := os.Getenv(telegramTokenID)
|
|
tgChatId := os.Getenv(telegramChatID)
|
|
|
|
chatID, err := strconv.ParseInt(tgChatId, 10, 64)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
|
|
if checkId() {
|
|
telegramBot(botToken, chatID)
|
|
} else {
|
|
fmt.Println("Programa no tiene privilegios suficientes")
|
|
os.Exit(1)
|
|
}
|
|
}
|