warbot.go 9.0 KB

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