warbot.go 8.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236
  1. // package warbot -- бот для игры в wartank
  2. package warbot
  3. import (
  4. "context"
  5. "fmt"
  6. "log"
  7. "strings"
  8. "wartank/pkg/components/safebool"
  9. "wartank/pkg/types"
  10. "wartank/server/serv_bots/warbot/angar"
  11. "wartank/server/serv_bots/warbot/tank"
  12. "wartank/server/serv_bots/warbot/warbot_config"
  13. "wartank/server/serv_bots/warbot/warbot_net"
  14. )
  15. // ВарБот -- бот для игры в вартанк
  16. type ВарБот struct {
  17. сервер types.ИСервер
  18. store types.IStore
  19. танк *tank.Танк
  20. errFinal error // Финальная ошибка работы, если была
  21. ангар types.ИАнгар
  22. сеть types.ИБотСеть
  23. еслиРаботает *safebool.SafeBool
  24. еслиАвтозапуск *safebool.SafeBool
  25. конфиг *warbot_config.ВарБотКонфиг // Конфиг бота для хранения в базе
  26. кнт context.Context // Контекст бота
  27. фтОтмена func() // Функция отменя контекста бота
  28. }
  29. // ЗагрузитьВарБот -- загружает бота из базы
  30. func ЗагрузитьВарБот(сервер types.ИСервер, логин string) (*ВарБот, error) {
  31. { // Предусловия
  32. if сервер == nil {
  33. return nil, fmt.Errorf("ЗагрузитьВарБот(): IApp is nil")
  34. }
  35. if логин == "" {
  36. return nil, fmt.Errorf("ЗагрузитьВарБот(): name is empty")
  37. }
  38. }
  39. log.Printf("ЗагрузитьВарБот(): name=%q\n", логин)
  40. store := сервер.Store()
  41. strData, err := store.Get("/bots/" + логин)
  42. if err != nil {
  43. if !strings.Contains(err.Error(), "not found") {
  44. return nil, fmt.Errorf("ЗагрузитьВарБот(): in load bot %q from store, err=\n\t%w", логин, err)
  45. }
  46. return nil, fmt.Errorf("ЗагрузитьВарБот(): bot %q not found in store", логин)
  47. }
  48. конфиг := &warbot_config.ВарБотКонфиг{}
  49. if err = конфиг.Unmarshall(strData); err != nil {
  50. return nil, fmt.Errorf("ЗагрузитьВарБот(): in unmarshall WarBotConfig(%q) from store, err=\n\t%w", логин, err)
  51. }
  52. sf, err := создатьЯдроВарБот(сервер, конфиг)
  53. if err != nil {
  54. return nil, fmt.Errorf("ЗагрузитьВарБот(): in make core for bot %q, err=\n\t%w", логин, err)
  55. }
  56. return sf, nil
  57. }
  58. // НовВарБот -- возвращает новый WarBot
  59. func НовВарБот(сервер types.ИСервер, логин, пароль string) (*ВарБот, error) {
  60. { // Предусловия
  61. if сервер == nil {
  62. return nil, fmt.Errorf("НовВарБот(): IApp is nil")
  63. }
  64. if логин == "" {
  65. return nil, fmt.Errorf("НовВарБот(): login is empty")
  66. }
  67. if пароль == "" {
  68. return nil, fmt.Errorf("НовВарБот(): pass is empty")
  69. }
  70. }
  71. log.Printf("НовВарБот(): name=%q\n", логин)
  72. config := &warbot_config.ВарБотКонфиг{
  73. ЕслиАвтозапуск_: false,
  74. Логин_: логин,
  75. Пароль_: пароль,
  76. }
  77. sf, err := создатьЯдроВарБот(сервер, config)
  78. if err != nil {
  79. return nil, fmt.Errorf("НовВарБот(): in make core bot %q, err=\n\t%w", логин, err)
  80. }
  81. if err := sf.save(); err != nil {
  82. return nil, fmt.Errorf("НовВарБот(): in self save to store, err=\n\t%w", err)
  83. }
  84. return sf, nil
  85. }
  86. // Создаёт ядро бота
  87. func создатьЯдроВарБот(server types.ИСервер, config *warbot_config.ВарБотКонфиг) (*ВарБот, error) {
  88. { // Предусловия
  89. if server == nil {
  90. return nil, fmt.Errorf("NewWarBot(): IServer==nil")
  91. }
  92. if config == nil {
  93. return nil, fmt.Errorf("NewWarBot(): WarBotConfig==nil")
  94. }
  95. }
  96. танк, ош := tank.NewTank()
  97. if ош != nil {
  98. return nil, fmt.Errorf("NewWarBot(): при создании параметров танка, ош=\n\t%w", ош)
  99. }
  100. ctx, fnCancel := context.WithCancel(server.CtxApp())
  101. sf := &ВарБот{
  102. сервер: server,
  103. store: server.Store(),
  104. танк: танк,
  105. еслиРаботает: safebool.NewSafeBool(),
  106. еслиАвтозапуск: safebool.NewSafeBool(),
  107. конфиг: config,
  108. кнт: ctx,
  109. фтОтмена: fnCancel,
  110. }
  111. var err error
  112. { // WarBotNet
  113. sf.сеть, err = warbot_net.NewWarBotNet(sf)
  114. if err != nil {
  115. return nil, fmt.Errorf("NewWarBot(): in make IBotNet, err=\n\t%w", err)
  116. }
  117. }
  118. sf.ангар, err = angar.НовАнгар(sf)
  119. if err != nil {
  120. return nil, fmt.Errorf("NewWarBot(): bot(%q) in make IAngar, err=\n\t%w", sf.Имя(), err)
  121. }
  122. if sf.конфиг.ЕслиАвтозапуск_ {
  123. sf.еслиАвтозапуск.Уст()
  124. sf.Пуск()
  125. }
  126. return sf, nil
  127. }
  128. // Сервер -- возвращает ссылку на объект сервера
  129. func (sf *ВарБот) Сервер() types.ИСервер {
  130. return sf.сервер
  131. }
  132. // ЕслиПуск -- возвращает признак, что бот подключен
  133. func (sf *ВарБот) ЕслиПуск() bool {
  134. return sf.еслиРаботает.Получ()
  135. }
  136. // Имя -- возвращает имя бота
  137. func (sf *ВарБот) Имя() string {
  138. return sf.конфиг.Логин_
  139. }
  140. // Пароль -- возвращает пароль бота
  141. func (sf *ВарБот) Пароль() string {
  142. return sf.конфиг.Пароль_
  143. }
  144. // Пуск -- запускает бот в работу
  145. func (sf *ВарБот) Пуск() (err error) {
  146. if sf.еслиРаботает.Получ() {
  147. return nil
  148. }
  149. if err := sf.ангар.Пуск(); err != nil {
  150. return fmt.Errorf("WarBot.Run(): bot(%q) in run angar, err=\n\t%w", sf.Имя(), err)
  151. }
  152. sf.еслиРаботает.Уст()
  153. return nil
  154. }
  155. // Error -- возвращает финальную ошибку работы, если была
  156. func (sf *ВарБот) Error() error {
  157. return sf.errFinal
  158. }
  159. // Ангар -- возвращает ангар игры
  160. func (sf *ВарБот) Ангар() types.ИАнгар {
  161. return sf.ангар
  162. }
  163. // Танк -- возвращает объект танка
  164. func (sf *ВарБот) Танк() types.ИТанк {
  165. return sf.танк
  166. }
  167. // Сеть -- возвращает ссылку на свой сетевой клиент
  168. func (sf *ВарБот) Сеть() types.ИБотСеть {
  169. return sf.сеть
  170. }
  171. // АвтоИграЕсли -- возвращает признак автоматичского запуска бота
  172. func (sf *ВарБот) АвтоИграЕсли() bool {
  173. return sf.еслиАвтозапуск.Получ()
  174. }
  175. // АвтоИграУст -- устанавливает признак автоматического запуска бота
  176. func (sf *ВарБот) АвтоИграУст() {
  177. log.Printf("WarBot.SetAutoGame()")
  178. sf.еслиАвтозапуск.Уст()
  179. sf.конфиг.ЕслиАвтозапуск_ = true
  180. sf.saveConfig()
  181. }
  182. // АвтоИграСброс -- сбрасывает признак автоматического запуска бота
  183. func (sf *ВарБот) АвтоИграСброс() {
  184. log.Printf("WarBot.ResetAutoGame()")
  185. sf.еслиАвтозапуск.Уст()
  186. sf.конфиг.ЕслиАвтозапуск_ = false
  187. sf.saveConfig()
  188. }
  189. // Сохраняет конфиг бота
  190. func (sf *ВарБот) saveConfig() {
  191. log.Printf("WarBot.saveConfig()")
  192. strConf := sf.конфиг.Marshall()
  193. err := sf.store.Put("/bots/"+sf.Имя(), strConf)
  194. if err != nil {
  195. log.Printf("WarBot.saveConfig(): err=\n\t%v\n", err)
  196. }
  197. }
  198. // Кнт -- возвращает контекст бота
  199. func (sf *ВарБот) Кнт() context.Context {
  200. return sf.кнт
  201. }
  202. // Закончить -- отменяет контекст бота
  203. func (sf *ВарБот) Закончить() {
  204. sf.фтОтмена()
  205. }
  206. // Сохраняет себя в базу
  207. func (sf *ВарБот) save() error {
  208. err := sf.store.Put("/bots/"+sf.Имя(), sf.конфиг.Marshall())
  209. if err != nil {
  210. return fmt.Errorf("WarBot.save(): in self save to store bot(%q), err=\n\t%w", sf.Имя(), err)
  211. }
  212. return nil
  213. }