bot.go

package main

import (
	"github.com/jbrodriguez/mlog"
	"errors"
	"fmt"
	"strconv"
)

var mbot map[int]bot


type bot struct {
	ID int
	Name string
	Login string
	Password string
	Connected bool
	UseProxy bool
	AddressProxy string
}

func init() {
	// при необходимости иницилизировать map
	if mbot == nil {
		mbot = make(map[int]bot)
	}
}

// newBot Создать нового бота и запомнить его в map
func newBot() (int, error) {
	var err error
	mlog.Trace("Функция: newBot")

	// найти новый незанятый id
	id := 0
	mlog.Trace("Функция: newBot. Поиск нового незанятого id")
	for i := 1; i < 10000; i++ {
		_, ok := mbot[i]
		if !ok {
			id = i
			break
		}
	}
	if id == 0 {	// превышен индекс
		err = errors.New("Превышено количество индексов ботов. id > 999")
		mlog.Error(err)
	}else{
		b := bot{}	// инициализировать бота
		b.ID = id
		b.Name = fmt.Sprintf("Бот %s", strconv.Itoa(id))
		mlog.Trace("Функция: newBot. Новый бот.", b)
		mbot[id] = b
	}

	return id, err
}

// delBot Найти бота по идентификатору и удалить
func delBot(id int) error {
	mlog.Trace("Функция: delBot")
	_, ok := mbot[id]
	if ok {
		delete(mbot, id)
		mlog.Trace("Функция: delBot. Бот с id = %d удален.", id)
	}
	return nil
}

// lstBot Функция подготавливает и возвращает массив с информацией по текущим ботам
func lstBot() ([]map[string]string, error) {
	mlog.Trace("Функция: lstBot")
	list := make([]map[string]string, len(mbot))
	i := 0
	for id, bot := range mbot {
		inf := make(map[string]string)
		inf["id"] = strconv.Itoa(id)
		inf["name"] = bot.Name
		inf["login"] = bot.Login
		inf["password"] = bot.Password
		list[i] = inf
		i++
	}
	return list, nil
}

// updBot Функция обновляет информацию бота по указанному идентификатору
func updBot(id int, inf map[string]interface{}) error {
	mlog.Trace("Функция: updBot")
	var err error
	bot, ok := mbot[id]
	if !ok {
		err = errors.New("В mbot нет бота с указанным индексом.")
		mlog.Error(err)
		return err
	}
	for key, value := range inf {
		switch key {
		case "name":
			valueStr, ok := value.(string)
			if ok {
				bot.Name = valueStr

			}else {
				mlog.Trace("Ошибка при приведении значения к строке")
			}
		case "login":
			valueStr, ok := value.(string)
			if ok {
				bot.Login = valueStr

			}else {
				mlog.Trace("Ошибка при приведении значения к строке")
			}
		case "password":
			valueStr, ok := value.(string)
			if ok {
				bot.Password = valueStr

			}else {
				mlog.Trace("Ошибка при приведении значения к строке")
			}
		}
	}
	mbot[id] = bot
	mlog.Trace("Функция: updBot. Обновлена информация бота с id = %d.", id)
	return nil
}

// connectBotToServer Функция получает uid и token с сайта mail.ru и подключается к игровому серверу
func connectBotToServer(id int) error {
	mlog.Trace("Функция: connectBotToServer")
	var err error
	bot, ok := mbot[id]
	if !ok {
		err = errors.New(fmt.Sprintf("В mbot нет бота с указанным индексом. id = %d", id))
		mlog.Error(err)
		return err
	}
	if bot.Connected {
		mlog.Warning("Бот уже подключен к серверу.")
		return nil
	}
	uid, uid2, token, err := getUidAndToken(bot.Login, bot.Password)
	if err != nil {
		mlog.Error(err)
		return err
	}
	mlog.Trace("Функция: connectBotToServer. uid = %v, uid2 = %v, token = %v", uid, uid2, token)
	bot.Connected = true
	mbot[id] = bot
	mlog.Trace("Функция: connectBotToServer. Бот подключен к серверу.")
	return nil
}

// disconnectBotFromServer Функция отключает бота от игрового сервера
func disconnectBotFromServer(id int) error {
	mlog.Trace("Функция: disconnectBotFromServer")
	var err error
	bot, ok := mbot[id]
	if !ok {
		err = errors.New(fmt.Sprintf("В mbot нет бота с указанным индексом. id = %d", id))
		mlog.Error(err)
		return err
	}
	if !bot.Connected {
		mlog.Warning("Бот уже отключен от сервера.")
		return nil
	}
	bot.Connected = false
	mbot[id] = bot
	mlog.Trace("Функция: disconnectBotFromServer. Бот отключен от сервера.")
	return nil
}