Spaces:
Sleeping
Sleeping
| // (c) Jisin0 | |
| package plugins | |
| import ( | |
| "context" | |
| "fmt" | |
| "strings" | |
| "github.com/Jisin0/Go-Filter-Bot/utils" | |
| "github.com/PaulSonOfLars/gotgbot/v2" | |
| "github.com/PaulSonOfLars/gotgbot/v2/ext" | |
| "go.mongodb.org/mongo-driver/bson" | |
| ) | |
| func Broadcast(bot *gotgbot.Bot, ctx *ext.Context) error { | |
| // Function to handle /broadcast command | |
| if !utils.IsAdmin(ctx.EffectiveUser.Id) { | |
| ctx.Message.Reply(bot, "<b>Only bot admins can use this command !</b>", &gotgbot.SendMessageOpts{ParseMode: gotgbot.ParseModeHTML, ReplyParameters: &gotgbot.ReplyParameters{MessageId: ctx.Message.MessageId}}) | |
| return nil | |
| } | |
| var ( | |
| update = ctx.Message | |
| filter = bson.D{} | |
| ) | |
| if strings.HasPrefix(update.Text, "/concast") { | |
| filter = bson.D{{Key: "connected", Value: bson.D{{Key: "$exists", Value: true}}}} | |
| } | |
| cursor, err := DB.Ucol.Find(context.TODO(), filter) | |
| if err != nil { | |
| fmt.Printf("broadcast.find: %v\n", err) | |
| return nil | |
| } | |
| if update.ReplyToMessage == nil { | |
| update.Reply(bot, "Please reply this command to the message you would like to broadcast !", &gotgbot.SendMessageOpts{}) | |
| return nil | |
| } | |
| msg := update.ReplyToMessage | |
| var ( | |
| isText = false | |
| isMedia = false | |
| caption string | |
| markup = &gotgbot.InlineKeyboardMarkup{} | |
| total int | |
| sent int | |
| failed int | |
| id int64 | |
| ) | |
| if msg.Text != "" { | |
| isText = true | |
| } else { | |
| isMedia = true | |
| caption = msg.OriginalCaptionHTML() | |
| } | |
| if msg.ReplyMarkup != nil { | |
| markup = &gotgbot.InlineKeyboardMarkup{InlineKeyboard: msg.ReplyMarkup.InlineKeyboard} | |
| } | |
| stat, err := update.Reply(bot, "<code>Starting broadcast ...</code>", &gotgbot.SendMessageOpts{ParseMode: gotgbot.ParseModeHTML}) | |
| if err != nil { | |
| fmt.Printf("broadcast.startmsg: %v\n", err) | |
| return nil | |
| } | |
| // TODO: create User type and remove this bson.M trash. | |
| for cursor.Next(context.TODO()) { | |
| var doc bson.M | |
| cursor.Decode(&doc) | |
| // Just in case, to prevent unwanted crashes | |
| rawID, ok := doc["_id"] | |
| if !ok { | |
| continue | |
| } | |
| if _, ok := rawID.(int32); ok { | |
| id = int64(rawID.(int32)) | |
| } else { | |
| id = rawID.(int64) | |
| } | |
| if isText { | |
| _, err := msg.Copy( | |
| bot, | |
| id, | |
| &gotgbot.CopyMessageOpts{ReplyMarkup: markup}, | |
| ) | |
| if err != nil { | |
| fmt.Printf("broadcast.sendmsg: %v\n", err) | |
| failed += 1 | |
| } else { | |
| sent += 1 | |
| } | |
| } else if isMedia { | |
| _, err := msg.Copy( | |
| bot, | |
| id, | |
| &gotgbot.CopyMessageOpts{Caption: &caption, ParseMode: gotgbot.ParseModeHTML, ReplyMarkup: markup}, | |
| ) | |
| if err != nil { | |
| failed += 1 | |
| } else { | |
| sent += 1 | |
| } | |
| } | |
| total += 1 | |
| // Update stat message every 20 requests to prevent floodwaits | |
| if total%20 == 0 { | |
| _, _, er := stat.EditText( | |
| bot, | |
| fmt.Sprintf(` | |
| <u>Live Broadcast Stats :</u> | |
| Success : %v | |
| Failed : %v | |
| Total : %v | |
| `, sent, failed, total), | |
| &gotgbot.EditMessageTextOpts{ParseMode: gotgbot.ParseModeHTML}, | |
| ) | |
| if er != nil { | |
| fmt.Printf("broadcast.progress.edit: %v\n", err) | |
| } | |
| } | |
| } | |
| _, _, err = stat.EditText( | |
| bot, | |
| fmt.Sprintf(` | |
| <b><u>Broadcast Completed :</u></b> | |
| Success : %v | |
| Failed : %v | |
| Total : %v | |
| `, sent, failed, total), | |
| &gotgbot.EditMessageTextOpts{ParseMode: gotgbot.ParseModeHTML}, | |
| ) | |
| if err != nil { | |
| fmt.Printf("broadcast.progress.complete: %v\n", err) | |
| } | |
| return nil | |
| } | |