bot.go 8.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202
  1. // package bot -- бот для игры в wartank
  2. package bot
  3. import (
  4. "context"
  5. "fmt"
  6. "log"
  7. "strings"
  8. "time"
  9. . "wartank/app/lev0/alias"
  10. . "wartank/app/lev0/types"
  11. "wartank/app/lev2/arena/arena_angar"
  12. "wartank/app/lev3/bot/bot_config"
  13. "wartank/app/lev3/bot/bot_net"
  14. "wartank/app/lev3/bot/bot_stat/tank_stat"
  15. "wartank/kernel"
  16. . "wartank/kernel/kernel_types"
  17. )
  18. // ВарБот -- бот для игры в вартанк
  19. type Бот struct {
  20. прилож ИПриложение
  21. хран ИЯдроХранилище
  22. стата ИТанкСтат
  23. errFinal error // Финальная ошибка работы, если была
  24. ангар ИАренаАнгар
  25. сеть ИБотСеть
  26. еслиРаботает ИБезопБул
  27. еслиАвтозапуск ИБезопБулНаблюд
  28. конфиг *bot_config.БотКонфиг // Конфиг бота для хранения в базе
  29. кнт context.Context // Контекст бота
  30. фтОтмена func() // Функция отменя контекста бота
  31. лог ИЯдроЛог
  32. конт ИБотКонтекст
  33. }
  34. // ЗагрузитьВарБот -- загружает бота из базы
  35. func ЗагрузитьВарБот(конт ИЯдроКонтекст, номер АБотНомер) *Бот {
  36. лог := kernel.НовЛог("ВарБот")
  37. лог.Инфо("ЗагрузитьВарБот()\n")
  38. лог.Паника(номер == 0, "ЗагрузитьВарБот(): номер пустой")
  39. стрНомер := fmt.Sprint(номер)
  40. лог.Инфо("ЗагрузитьВарБот(): номер=%q\n", стрНомер)
  41. хран := конт.Получ("kernStore").(ИЯдроХранилище)
  42. binData, err := хран.Get("/bots/" + стрНомер)
  43. if err != nil {
  44. if !strings.Contains(err.Error(), "not found") {
  45. лог.Паника(true, "ЗагрузитьВарБот(): in load bot '%v' from store, err=\n\t%v\n", номер, err)
  46. }
  47. лог.Паника(err != nil, "ЗагрузитьВарБот(): in load bot '%v' from store, err=\n\t%v\n", номер, err)
  48. }
  49. конфиг := &bot_config.БотКонфиг{}
  50. конфиг.Unmarshal(binData)
  51. сам := создатьЯдроВарБот(конт, конфиг)
  52. go сам.рестарт()
  53. _ = ИБот(сам)
  54. return сам
  55. }
  56. // Каждые два часа перезапускает себя
  57. func (сам *Бот) рестарт() {
  58. time.Sleep(time.Hour * 2)
  59. сам.прилож.Отменить()
  60. сам.лог.Инфо("рестарт(): бот %q перезагружен\n", сам.конфиг.Логин_)
  61. }
  62. // НовВарБот -- возвращает новый WarBot
  63. func НовВарБот(конт ИЯдроКонтекст, номер АБотНомер, логин, пароль string, еслиАвто bool) *Бот {
  64. лог := kernel.НовЛог("ВарБот")
  65. лог.Инфо("НовВарБот()\n")
  66. лог.Паника(логин == "", "НовВарБот(): логин пустой")
  67. лог.Паника(пароль == "", "НовВарБот(): пароль пустой")
  68. лог.Инфо("НовВарБот(): name=%q\n", логин)
  69. config := &bot_config.БотКонфиг{
  70. ЕслиАвтозапуск_: еслиАвто,
  71. Логин_: логин,
  72. Пароль_: пароль,
  73. Номер_: номер,
  74. }
  75. сам := создатьЯдроВарБот(конт, config)
  76. сам.сохрКонфиг()
  77. _ = ИБот(сам)
  78. return сам
  79. }
  80. // Создаёт ядро бота
  81. func создатьЯдроВарБот(конт ИЯдроКонтекст, конфиг *bot_config.БотКонфиг) *Бот {
  82. лог := kernel.НовЛог("ВарБот")
  83. лог.Инфо("создатьЯдроВарБот()\n")
  84. лог.Паника(конфиг == nil, "создатьЯдроВарБот(): ВарБотКонфиг==nil")
  85. ctx, fnCancel := context.WithCancel(конт.Конт())
  86. приложение := конт.Получ("приложение").(ИПриложение)
  87. контБот := kernel.НовКонтекст()
  88. сам := &Бот{
  89. конт: контБот,
  90. прилож: приложение,
  91. хран: конт.Получ("kernStore").(ИЯдроХранилище),
  92. стата: tank_stat.НовТанкСтат(конт),
  93. еслиРаботает: kernel.НовБезопБул(),
  94. конфиг: конфиг,
  95. кнт: ctx,
  96. фтОтмена: fnCancel,
  97. лог: лог,
  98. }
  99. сам.еслиАвтозапуск = kernel.НовБезопБулНаблюд(сам.Имя(), сам.автозапускИзм)
  100. сам.конт.Уст("бот", сам)
  101. сам.конт.Уст("приложение", приложение)
  102. сам.сеть = bot_net.НовБотСеть(сам.конт)
  103. сам.ангар = arena_angar.НовАнгар(сам.конт)
  104. if сам.конфиг.ЕслиАвтозапуск_ {
  105. сам.еслиАвтозапуск.Уст()
  106. сам.Пуск()
  107. }
  108. return сам
  109. }
  110. // ЕслиРабота -- возвращает признак, что бот подключен
  111. func (сам *Бот) ЕслиРабота() bool {
  112. return сам.еслиРаботает.Получ()
  113. }
  114. // Номер -- возвращает номер бота
  115. func (сам *Бот) Номер() АБотНомер {
  116. return сам.конфиг.Номер()
  117. }
  118. // Имя -- возвращает имя бота
  119. func (сам *Бот) Имя() string {
  120. return сам.конфиг.Логин()
  121. }
  122. // Пароль -- возвращает пароль бота
  123. func (сам *Бот) Пароль() string {
  124. return сам.конфиг.Пароль_
  125. }
  126. // Пуск -- запускает бот в работу
  127. func (сам *Бот) Пуск() {
  128. if сам.еслиРаботает.Получ() {
  129. return
  130. }
  131. сам.ангар.Пуск()
  132. сам.еслиРаботает.Уст()
  133. }
  134. // Error -- возвращает финальную ошибку работы, если была
  135. func (сам *Бот) Error() error {
  136. return сам.errFinal
  137. }
  138. // Ангар -- возвращает ангар игры
  139. func (сам *Бот) Ангар() ИАренаАнгар {
  140. return сам.ангар
  141. }
  142. // Стата -- возвращает статистику танка
  143. func (сам *Бот) Стата() ИТанкСтат {
  144. return сам.стата
  145. }
  146. // Сеть -- возвращает ссылку на свой сетевой клиент
  147. func (сам *Бот) Сеть() ИБотСеть {
  148. return сам.сеть
  149. }
  150. // Автозапуск -- возвращает признак автоматического запуска бота
  151. func (сам *Бот) Автозапуск() ИБезопБулНаблюд {
  152. return сам.еслиАвтозапуск
  153. }
  154. // Обратный вызов автоматического запуска бота
  155. func (сам *Бот) автозапускИзм(знач bool) {
  156. сам.лог.Отладка("Бот.автозапускИзм()")
  157. // сам.еслиАвтозапуск.Уст()
  158. сам.конфиг.ЕслиАвтозапуск_ = знач
  159. сам.сохрКонфиг()
  160. }
  161. // Сохраняет конфиг бота
  162. func (сам *Бот) сохрКонфиг() {
  163. log.Printf("ВарБот.сохрКонфиг()")
  164. strConf := сам.конфиг.Marshall()
  165. стрНомер := fmt.Sprint(сам.Номер())
  166. err := сам.хран.Set("/bots/"+стрНомер, strConf)
  167. сам.лог.Паника(err != nil, "ВарБот.сохрКонфиг(): err=\n\t%v\n", err)
  168. }
  169. // Контекст -- возвращает контекст бота
  170. func (сам *Бот) Контекст() context.Context {
  171. return сам.кнт
  172. }
  173. // Закончить -- отменяет контекст бота
  174. func (сам *Бот) Закончить() {
  175. сам.фтОтмена()
  176. }
  177. // КонтБот -- возвращает контекст бота
  178. func (сам *Бот) КонтБот() ИБотКонтекст {
  179. return сам.конт
  180. }