wartank/server/serv_bots/warbot/angar/base/bank/bank.go

package bank

import (
	"fmt"
	"log"
	"strconv"
	"strings"

	"wartank/pkg/components/section"
	"wartank/pkg/types"
	"wartank/server/serv_bots/warbot/angar/base/bank/bankmode"
	"wartank/server/serv_bots/warbot/angar/base/bank/banknet"
	"wartank/server/serv_bots/warbot/tank/tankstat/static_param"
)

/*
	Предоставляет объект банка на базе
*/

// Bank -- объект банка на базе
type Bank struct {
	*section.Section
	net       *banknet.BankNet
	silverBot types.IStatParam
	mode1     *bankmode.BankMode
	mode2     *bankmode.BankMode
}

// NewBank -- возвращает новый *Bank
func NewBank(base types.IBase) (*Bank, error) {
	section, err := section.NewSection(base.Bot(), `<span class="green2">Серебро</span><br/>`)
	if err != nil {
		return nil, fmt.Errorf("NewBank(): in create *Section, err=\n\t%w", err)
	}

	sf := &Bank{
		Section:   section,
		silverBot: static_param.NewStaticParam("silver_bot"),
		mode1:     bankmode.NewBankMode(),
		mode2:     bankmode.NewBankMode(),
	}
	sf.net, err = banknet.NewBankNet(sf)
	if err != nil {
		return nil, fmt.Errorf("NewBank(): in create NetBank, err=\n\t%w", err)
	}
	return sf, nil
}

// Run -- запускает банк в работу
func (sf *Bank) Run() error {
	log.Printf("Bank.Run()\n")
	go sf.run()
	return nil
}

// UpdateLst -- принудительно обновляет состояние банка
func (sf *Bank) UpdateLst() {
	if err := sf.net.UpdateLst("Банк"); err != nil {
		log.Printf("Bank.UpdateLst(): err=\n\t%v\n", err)
	}
}

// запускает банк в опрос
func (sf *Bank) run() {
	log.Printf("Bank.run()\n")
	sf.CountDown().Set(5)
	for range sf.CountDown().ChanSig() {
		// time.Sleep(time.Second * 30)
		count := sf.CountDown().Get()
		log.Printf("Bank countDown=%d\n", count)
		if err := sf.net.UpdateLst("банк"); err != nil {
			// log._rintf("ERRO Bank.Run():  при обновлении lstBank, err=\n\t%v\n", err)
			continue
		}
		if err := sf.getAllMode(); err != nil {
			// log._rintf("ERRO Bank.Run(): при получении списка режимов банка, err=\n\t%v\n", err)
			continue
		}
		if err := sf.makeProduct(); err != nil {
			log.Printf("ERRO Bank.Run(): при пуске производства банка, err=\n\t%v\n", err)
		}
	}
}

// Mode2 -- возвращает объект режима2
func (sf *Bank) Mode2() types.IBankMode {
	return sf.mode2
}

// Mode1 -- возвращает объект режима1
func (sf *Bank) Mode1() types.IBankMode {
	return sf.mode1
}

// SilverBot -- возвращает серебро от бота
func (sf *Bank) SilverBot() types.IStatParam {
	return sf.silverBot
}

// Запускает в производство серебро
func (sf *Bank) makeProduct() error {
	var (
		lstBank = sf.GetLst()
		ind     int
		strOut  string
		strLink string
		isFind  bool
	)
	time1 := sf.Mode1().Time()
	time2 := sf.Mode2().Time()
	if time1 > time2 {
		time1 = time2
	}
	for ind, strOut = range lstBank {
		if strings.Contains(strOut, time1) {
			ind += 7
			strLink = lstBank[ind]
			isFind = true
			break
		}
	}
	if isFind && strings.Contains(strLink, `>Начать производство</span>`) {
		lstLink := strings.Split(strLink, `<a class="simple-but border" href="`)
		strLink = lstLink[1]
		lstLink = strings.Split(strLink, `"><span><span>Начать производство</span></span></a>`)
		strLink = "http://wartank.ru/production/" + lstLink[0]
		lstBank, err := sf.net.Get(strLink)
		if err != nil {
			return fmt.Errorf("BankNet.makeProduct(): при выполнении GET-запроса начать производство, err=%w", err)
		}
		for _, strOut := range lstBank {
			if strings.Contains(strOut, `<title>База</title>`) { // Это база, а не банк
				return nil
			}
		}
		if err = sf.Update(lstBank); err != nil {
			return fmt.Errorf("BankNet.makeProduct(): при обновлении lstBank, err=%w", err)
		}
		if err := sf.CountDown().Parse(time1); err != nil {
			log.Printf("WARN Bank.makeProduct(): при установке времени производства банка(%v)\n\terr=%v\n", time1, err)
		}
	}
	return nil
}

// Получает все режимы банка
func (sf *Bank) getAllMode() error {
	var (
		lstBank  = sf.GetLst()
		ind      int
		strMode  string
		strMode1 string
		strMode2 string
	)
	{ // Получить первый режим
		for ind, strMode = range lstBank {
			if strings.Contains(strMode, `Кол-во: <span class="green2">`) {
				strMode1 = strMode
				break
			}
		}
		lstMode := strings.Split(strMode1, `Кол-во: <span class="green2">`)
		strMode1 = lstMode[1]
		lstMode = strings.Split(strMode1, `</span><br/>`)
		strMode1 = lstMode[0]
		iNum1, err := strconv.Atoi(strMode1)
		if err != nil {
			return fmt.Errorf("BankNet.getAllMode(): numSilver1(%v) not number, err=\n\t%w", strMode1, err)
		}
		sf.Mode1().Silver().SetVal(iNum1)
		// Установить время производства
		strTime1 := lstBank[ind+2]
		sf.Mode1().TimeSet(strTime1)
		if iNum1 <= 2 { // Если банк слишком слабый
			return nil
		}
	}
	{ // Получить второй режим
		for _ind := ind + 2; _ind < len(lstBank); _ind++ {
			strMode := lstBank[_ind]
			if strings.Contains(strMode, `Кол-во: <span class="green2">`) {
				strMode2 = strMode
				ind = _ind
				break
			}
		}
		lstMode := strings.Split(strMode2, `Кол-во: <span class="green2">`)
		strMode2 = lstMode[1]
		lstMode = strings.Split(strMode2, `</span><br/>`)
		strMode2 = lstMode[0]
		iNum2, err := strconv.Atoi(strMode2)
		if err != nil {
			return fmt.Errorf("BankNet.getAllMode(): numSilver2(%v) not number, err=\n\t%w", strMode2, err)
		}
		sf.Mode2().Silver().SetVal(iNum2)
		// Установить время производства
		strTime2 := lstBank[ind+2]
		sf.Mode2().TimeSet(strTime2)
	}
	return nil
}