warbot.go 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151
  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. // Name -- возвращает имя бота
  74. func (sf *WarBot) Name() string {
  75. return sf.name
  76. }
  77. // Pass -- возвращает пароль бота
  78. func (sf *WarBot) Pass() string {
  79. return sf.pass
  80. }
  81. // Run -- запускает бот в работу
  82. func (sf *WarBot) Run() (err error) {
  83. if sf.isRun.Get() {
  84. return nil
  85. }
  86. err = sf.botNet.Login()
  87. if err != nil {
  88. return fmt.Errorf("WarBot.Run(): bot(%q) in net login, err=\n\t%w", sf.name, err)
  89. }
  90. if err := sf.angar.Run(); err != nil {
  91. return fmt.Errorf("WarBot.Run(): bot(%q) in run angar, err=\n\t%w", sf.name, err)
  92. }
  93. sf.isRun.Set()
  94. return nil
  95. }
  96. // Error -- возвращает финальную ошибку работы, если была
  97. func (sf *WarBot) Error() error {
  98. return sf.errFinal
  99. }
  100. // Angar -- возвращает ангар игры
  101. func (sf *WarBot) Angar() types.IAngar {
  102. return sf.angar
  103. }
  104. // Tank -- возвращает объект танка
  105. func (sf *WarBot) Tank() types.ITank {
  106. return sf.tank
  107. }
  108. // BotNet -- возвращает ссылку на свой сетевой клиент
  109. func (sf *WarBot) BotNet() types.IBotNet {
  110. return sf.botNet
  111. }
  112. // IsAutoGame -- возвращает признак автоматичского запуска бота
  113. func (sf *WarBot) IsAutoGame() bool {
  114. return sf.isAutoRun.Get()
  115. }
  116. // SetAutoGame -- устанавливает признак автоматического запуска бота
  117. func (sf *WarBot) SetAutoGame() {
  118. sf.isAutoRun.Set()
  119. sf.config.IsAutoRun_ = true
  120. sf.saveConfig()
  121. }
  122. // ResetAutoGame -- сбрасывает признак автоматического запуска бота
  123. func (sf *WarBot) ResetAutoGame() {
  124. sf.isAutoRun.Set()
  125. sf.config.IsAutoRun_ = false
  126. sf.saveConfig()
  127. }
  128. // Сохраняет конфиг бота
  129. func (sf *WarBot) saveConfig() {
  130. err := sf.store.Put("/bots/"+sf.name, sf.config.Marshall())
  131. if err != nil {
  132. log.Printf("WarBot.saveConfig(): err=\n\t%v\n", err)
  133. }
  134. sf.server.CancelApp()
  135. }