warbot.go 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170
  1. // package warbot -- бот для игры в wartank
  2. package warbot
  3. import (
  4. "fmt"
  5. "log"
  6. "wartank/pkg/components/safebool"
  7. "wartank/pkg/types"
  8. "wartank/server/serv_bots/warbot/angar"
  9. "wartank/server/serv_bots/warbot/tank"
  10. "wartank/server/serv_bots/warbot/warbot_config"
  11. "wartank/server/serv_bots/warbot/warbot_net"
  12. )
  13. // WarBot -- бот для игры в вартанк
  14. type WarBot struct {
  15. server types.IServer
  16. store types.IStore
  17. tank *tank.Tank
  18. errFinal error // Финальная ошибка работы, если была
  19. angar types.IAngar
  20. botNet types.IBotNet
  21. name string // Имя бота
  22. pass string // Пароль бота
  23. isRun *safebool.SafeBool
  24. isAutoRun *safebool.SafeBool
  25. config *warbot_config.WarBotConfig // Конфиг бота для хранения в базе
  26. }
  27. // NewWarBot -- возвращает новый WarBot
  28. func NewWarBot(server types.IServer, name string) (*WarBot, error) {
  29. { // Предусловия
  30. if server == nil {
  31. return nil, fmt.Errorf("NewWarBot(): IApp is nil")
  32. }
  33. if name == "" {
  34. return nil, fmt.Errorf("NewWarBot(): name is empty")
  35. }
  36. }
  37. tank, err := tank.NewTank(server)
  38. if err != nil {
  39. return nil, fmt.Errorf("NewWarBot(): in create Tank, err=\n\t%w", err)
  40. }
  41. sf := &WarBot{
  42. server: server,
  43. store: server.Store(),
  44. tank: tank,
  45. name: name,
  46. isRun: safebool.NewSafeBool(),
  47. isAutoRun: safebool.NewSafeBool(),
  48. config: &warbot_config.WarBotConfig{},
  49. }
  50. strData, err := sf.store.Get("/bots/" + sf.name)
  51. if err != nil {
  52. return nil, fmt.Errorf("NewWarBot(): in get binData for WarBotConfig from store, err=\n\t%w", err)
  53. }
  54. if err = sf.config.Unmarshall(strData); err != nil {
  55. return nil, fmt.Errorf("NewWarBot(): in unmarshall WarBotConfig from store, err=\n\t%w", err)
  56. }
  57. sf.pass = sf.config.Password_
  58. { // WarBotNet
  59. sf.botNet, err = warbot_net.NewWarBotNet(server, sf)
  60. if err != nil {
  61. return nil, fmt.Errorf("NewWarBot(): in make IBotNet, err=\n\t%w", err)
  62. }
  63. }
  64. if sf.config.IsAutoRun_ {
  65. sf.isAutoRun.Set()
  66. }
  67. sf.angar, err = angar.NewAngar(sf.server, sf)
  68. if err != nil {
  69. return nil, fmt.Errorf("NewWarBot(): bot(%q) in make IAngar, err=\n\t%w", sf.name, err)
  70. }
  71. return sf, nil
  72. }
  73. // IsRun -- возвращает признак, что бот подключен
  74. func (sf *WarBot) IsRun() bool {
  75. return sf.isRun.Get()
  76. }
  77. // Name -- возвращает имя бота
  78. func (sf *WarBot) Name() string {
  79. return sf.name
  80. }
  81. // Pass -- возвращает пароль бота
  82. func (sf *WarBot) Pass() string {
  83. return sf.pass
  84. }
  85. // Make -- создание компонентов бота
  86. func (sf *WarBot) Make() error {
  87. if sf.isRun.Get() {
  88. return nil
  89. }
  90. err := sf.botNet.Login()
  91. if err != nil {
  92. return fmt.Errorf("WarBot.Make(): bot(%q) in net login, err=\n\t%w", sf.name, err)
  93. }
  94. if err := sf.angar.Make(); err != nil {
  95. return fmt.Errorf("WarBot.Make(): при создании компонентов ангара, err=\n\t%w", err)
  96. }
  97. return nil
  98. }
  99. // Run -- запускает бот в работу
  100. func (sf *WarBot) Run() (err error) {
  101. if sf.isRun.Get() {
  102. return nil
  103. }
  104. if err := sf.angar.Run(); err != nil {
  105. return fmt.Errorf("WarBot.Run(): bot(%q) in run angar, err=\n\t%w", sf.name, err)
  106. }
  107. sf.isRun.Set()
  108. return nil
  109. }
  110. // Error -- возвращает финальную ошибку работы, если была
  111. func (sf *WarBot) Error() error {
  112. return sf.errFinal
  113. }
  114. // Angar -- возвращает ангар игры
  115. func (sf *WarBot) Angar() types.IAngar {
  116. return sf.angar
  117. }
  118. // Tank -- возвращает объект танка
  119. func (sf *WarBot) Tank() types.ITank {
  120. return sf.tank
  121. }
  122. // BotNet -- возвращает ссылку на свой сетевой клиент
  123. func (sf *WarBot) BotNet() types.IBotNet {
  124. return sf.botNet
  125. }
  126. // IsAutoGame -- возвращает признак автоматичского запуска бота
  127. func (sf *WarBot) IsAutoGame() bool {
  128. return sf.isAutoRun.Get()
  129. }
  130. // SetAutoGame -- устанавливает признак автоматического запуска бота
  131. func (sf *WarBot) SetAutoGame() {
  132. log.Printf("WarBot.SetAutoGame()")
  133. sf.isAutoRun.Set()
  134. sf.config.IsAutoRun_ = true
  135. sf.saveConfig()
  136. }
  137. // ResetAutoGame -- сбрасывает признак автоматического запуска бота
  138. func (sf *WarBot) ResetAutoGame() {
  139. log.Printf("WarBot.ResetAutoGame()")
  140. sf.isAutoRun.Set()
  141. sf.config.IsAutoRun_ = false
  142. sf.saveConfig()
  143. }
  144. // Сохраняет конфиг бота
  145. func (sf *WarBot) saveConfig() {
  146. log.Printf("WarBot.saveConfig()")
  147. strConf := sf.config.Marshall()
  148. err := sf.store.Put("/bots/"+sf.name, strConf)
  149. if err != nil {
  150. log.Printf("WarBot.saveConfig(): err=\n\t%v\n", err)
  151. }
  152. }