barandaBot/src/manageAdmin.go

package main

import (
	"bufio"
	"fmt"
	"log"
	"os"
	"regexp"
	"strconv"
	"strings"

	"github.com/dixonwille/wmenu"
	tb "gopkg.in/tucnak/telebot.v2"
)

func addBotAdmins(newAdminIDs []string) error {
	if redisClient == nil {
		return ErrNilPointer
	}
	errNum := 0
	if newAdminIDs == nil && cmdFlags.interactive {
		fmt.Println("Add the new admin IDs, comma-separated:")
		reader := bufio.NewReader(os.Stdin)
		line, err := reader.ReadString('\n')
		if err != nil {
			log.Printf("Error in reading new admin IDs: %v", err)
			return ErrStdRead
		}
		newAdminIDs = strings.Split(line, ",")
	}
	for _, newAdminID := range newAdminIDs {
		err := addBotAdmin(newAdminID)
		if err != nil {
			errNum++
			log.Printf("Error in adding new admin ID %s: %v", newAdminID, err)
		}
	}
	if errNum == len(newAdminIDs) {
		return ErrAddAdmin
	}
	return nil
}

func removeBotAdmins() error {
	if redisClient == nil {
		return ErrNilPointer
	}
	botAdmins, err := redisClient.SMembers(adminUsers).Result()
	if err != nil {
		log.Printf("Couldn't retrieve admins: %v", err)
		return ErrRedisRetrieveSet
	}
	menu := wmenu.NewMenu("Select the admin(s) you want to remove:")
	menu.AllowMultiple()
	menu.LoopOnInvalid()
	menu.Action(func(opts []wmenu.Opt) error {
		var returnErr error
		for _, opt := range opts {
			if opt.Value == nil {
				log.Println("Couldn't remove admin: nil token")
				returnErr = ErrNilPointer
			} else {
				err := removeBotAdmin(opt.Value.(int))
				if err != nil {
					log.Printf("Couldn't remove bot admin: %v", err)
				}
			}
		}
		return returnErr
	})
	for _, botAdmin := range botAdmins {
		adminID, err := strconv.Atoi(botAdmin)
		if err != nil {
			log.Printf("Error converting adminID from string to int: %v", err)
			return ErrAtoiConv
		}
		adminInfo, err := getUserInfo(adminID)
		menu.Option(adminInfo.Username+": "+adminInfo.FirstName+" "+adminInfo.LastName, adminID, false, nil)
	}
	err = menu.Run()
	if err != nil {
		log.Printf("Error in removeToken menu: %v", err)
		return ErrRemoveAdmin
	}
	return nil
}

func hasBotAdmins() (bool, error) {
	if redisClient == nil {
		return false, ErrNilPointer
	}

	adminNum, err := redisClient.SCard(adminUsers).Result()
	if err != nil {
		log.Printf("Error retrieving number of admins: %v", err)
		return false, ErrRedisRetrieveSet
	}

	if adminNum <= 0 {
		return false, nil
	}
	return true, nil
}

func isBotAdmin(userID int) (bool, error) {
	if redisClient == nil {
		return false, ErrNilPointer
	}
	admin, err := redisClient.SIsMember(adminUsers, strconv.Itoa(userID)).Result()
	if err != nil {
		log.Printf("Error checking if ID is bot admin: %v", err)
		return false, ErrRedisCheckSet
	}
	return admin, nil
}

func addBotAdmin(newAdminID string) error {
	if redisClient == nil {
		return ErrNilPointer
	}
	adminID := strings.TrimSpace(newAdminID)
	matched, err := regexp.MatchString("^\\d+$", adminID)
	if err != nil {
		log.Printf("Error in parsing admin ID: %v", err)
		return ErrIDParsing
	}
	if !matched {
		return ErrIDInvalid
	}

	ID, err := strconv.Atoi(adminID)
	if err != nil {
		log.Printf("Error converting user ID: %v", err)
		return ErrAtoiConv
	}
	chat, err := bot.ChatByID(adminID)
	if err != nil {
		log.Printf("Error retriving chat by id: %v", err)
		return ErrChatRetrieve
	}
	if chat.Type != tb.ChatPrivate {
		log.Printf("Admin must be a user!")
		return ErrAddAdmin
	}
	isUser, err := isUser(ID)
	if err != nil {
		log.Printf("Error checking if ID is bot user: %v", err)
		return ErrAddAdmin
	}
	if !isUser {
		err = addUser(&tb.User{ID: int(chat.ID), FirstName: chat.FirstName, LastName: chat.LastName, Username: chat.Username})
		if err != nil {
			log.Printf("Error adding user: %v", err)
			return ErrAddUser
		}
	}
	isAuth, err := isAuthrizedUser(ID)
	if err != nil {
		log.Printf("Error checking if ID is authorized: %v", err)
		return ErrAddAdmin
	}
	if !isAuth {
		err = authorizeUser(ID, true)
		if err != nil {
			log.Printf("Error authorizing user: %v", err)
			return ErrAddAuthUser
		}
	}
	err = redisClient.SAdd(adminUsers, adminID).Err()
	if err != nil {
		log.Printf("Error in adding new admin ID: %v", err)
		return ErrRedisAddSet
	}
	botStatus.hasAdmin = true

	user, err := getUserInfo(ID)
	if err != nil {
		log.Printf("Error getting user info: %v", err)
		return ErrGetUser
	}
	err = sendMsgWithMenu(user, newAdminMsg, false)
	if err != nil {
		log.Printf("Error sending message to new admin: %v", err)
		return ErrSendMsg
	}

	return nil
}

func removeBotAdmin(adminID int) error {
	if redisClient == nil {
		return ErrNilPointer
	}
	err := redisClient.SRem(adminUsers, strconv.Itoa(adminID)).Err()
	if err != nil {
		log.Printf("Error removing admin from set: %v", err)
		return ErrRedisRemSet
	}
	user, err := getUserInfo(adminID)
	if err != nil {
		log.Printf("Error getting user info: %v", err)
		return ErrGetUser
	}
	err = sendMsgWithMenu(user, delAdminMsg, false)
	if err != nil {
		log.Printf("Error sending message to removed admin: %v", err)
		return ErrSendMsg
	}

	hasAdmin, err := hasBotAdmins()
	if err != nil {
		log.Printf("Error checking if bot has admins: %v", err)
		return ErrRedisRetrieveSet
	}
	botStatus.hasAdmin = hasAdmin

	return nil
}