warbot.go 4.8 KB

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