web_api.go 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446
  1. // package web_api -- веб-сервер для API
  2. package web_api
  3. import (
  4. "fmt"
  5. "log"
  6. "time"
  7. "github.com/gofiber/fiber/v2"
  8. "wartank/pkg/alias"
  9. "wartank/pkg/types"
  10. )
  11. // ВебАпи -- веб-сервер для API
  12. type ВебАпи struct {
  13. серв types.ИСервер
  14. файбер fiber.App
  15. }
  16. // НовВебАпи -- возвращает новый веб-сервер для API
  17. func НовВебАпи(вебСервер types.ИВебСервер) (*ВебАпи, error) {
  18. if вебСервер == nil {
  19. return nil, fmt.Errorf("НовВебАпи(): ИВебСервер==nil")
  20. }
  21. сам := &ВебАпи{
  22. серв: вебСервер.Сервер(),
  23. файбер: *вебСервер.Файбер(),
  24. }
  25. сам.файбер.Post("/api/login", сам.логин)
  26. сам.файбер.Post("/api/add_bot", сам.добавитьБота)
  27. сам.файбер.Get("/api/uptime", сам.аптаймСервер)
  28. сам.файбер.Get("/api/count_start", сам.стартНомер)
  29. сам.файбер.Get("/api/bot/:number/mine/level", сам.шахтаУровень)
  30. сам.файбер.Get("/api/bot/:number/mine/mode", сам.шахтаРежим)
  31. сам.файбер.Get("/api/bot/:number/mine/count_product", сам.шахтаРаботаКоличество)
  32. сам.файбер.Get("/api/bot/:number/mine/name_product", сам.шахтаРаботаИмя)
  33. сам.файбер.Get("/api/bot/:number/mine/back_time", сам.шахтаВремяОсталось)
  34. сам.файбер.Get("/api/bot/:number/tank/fuel", сам.танкТопливо)
  35. сам.файбер.Get("/api/bot/:number/angar/silver", сам.ангарСеребро)
  36. сам.файбер.Get("/api/bot/:number/polygon/level", сам.полигонУровень)
  37. сам.файбер.Get("/api/bot/:number/polygon/mode", сам.полигонРежим)
  38. сам.файбер.Get("/api/bot/:number/polygon/count_product", сам.полигонРаботаКоличество)
  39. сам.файбер.Get("/api/bot/:number/polygon/name_product", сам.полигонРаботаИмя)
  40. сам.файбер.Get("/api/bot/:number/polygon/back_time", сам.полигонВремяОсталось)
  41. сам.файбер.Get("/api/bot/:number/arsenal/level", сам.арсеналУровень)
  42. сам.файбер.Get("/api/bot/:number/arsenal/mode", сам.арсеналРежим)
  43. сам.файбер.Get("/api/bot/:number/arsenal/count_product", сам.полигонРаботаКоличество)
  44. сам.файбер.Get("/api/bot/:number/arsenal/name_product", сам.арсеналРаботаИмя)
  45. сам.файбер.Get("/api/bot/:number/arsenal/back_time", сам.ареналВремяОсталось)
  46. return сам, nil
  47. }
  48. // Возвращает имя производства снаряда на оружейке
  49. func (сам *ВебАпи) арсеналРаботаИмя(кнт *fiber.Ctx) error {
  50. номер, ош := кнт.ParamsInt("number")
  51. if ош != nil {
  52. сообщ := fmt.Sprintf("[Тип: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  53. return кнт.SendString(сообщ)
  54. }
  55. ботНомер := alias.БотНомер(номер)
  56. бот := сам.серв.ServBots().Get(ботНомер)
  57. if бот == nil {
  58. return кнт.SendString("[Тип: нет такого бота]")
  59. }
  60. имя := бот.Ангар().База().Арсенал().ПродуктИмяСейчас()
  61. if имя == "" {
  62. return кнт.SendString("[Тип: пустое имя]")
  63. }
  64. return кнт.SendString("[Тип: " + имя + "]")
  65. }
  66. // Возвращает режим оружейки
  67. func (сам *ВебАпи) арсеналРежим(кнт *fiber.Ctx) error {
  68. номер, ош := кнт.ParamsInt("number")
  69. if ош != nil {
  70. сообщ := fmt.Sprintf("[Режим: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  71. return кнт.SendString(сообщ)
  72. }
  73. ботНомер := alias.БотНомер(номер)
  74. бот := сам.серв.ServBots().Get(ботНомер)
  75. if бот == nil {
  76. return кнт.SendString("[Режим: нет такого бота]")
  77. }
  78. сценаРежим := бот.Ангар().База().Арсенал().СценаРежим()
  79. стрРежим := fmt.Sprint(сценаРежим.Режим())
  80. if стрРежим == "" {
  81. return кнт.SendString("[Режим: пустой режим]")
  82. }
  83. return кнт.SendString("[Режим: " + стрРежим + "]")
  84. }
  85. // Возвращает уровень арсенала
  86. func (сам *ВебАпи) арсеналУровень(кнт *fiber.Ctx) error {
  87. номер, ош := кнт.ParamsInt("number")
  88. if ош != nil {
  89. сообщ := fmt.Sprintf("[Уровень: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  90. return кнт.SendString(сообщ)
  91. }
  92. ботНомер := alias.БотНомер(номер)
  93. бот := сам.серв.ServBots().Get(ботНомер)
  94. if бот == nil {
  95. return кнт.SendString("[Уровень: нет такого бота]")
  96. }
  97. уровень := бот.Ангар().База().Арсенал().Уровень()
  98. стрУровень := fmt.Sprint(уровень.Получ())
  99. if стрУровень == "" {
  100. return кнт.SendString("[Уровень: пустой уровень]")
  101. }
  102. return кнт.SendString("[Уровень: " + стрУровень + "]")
  103. }
  104. // Возвращает время, которое осталось на арсенале
  105. func (сам *ВебАпи) ареналВремяОсталось(кнт *fiber.Ctx) error {
  106. номер, ош := кнт.ParamsInt("number")
  107. if ош != nil {
  108. сообщ := fmt.Sprintf("[Время: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  109. return кнт.SendString(сообщ)
  110. }
  111. ботНомер := alias.БотНомер(номер)
  112. бот := сам.серв.ServBots().Get(ботНомер)
  113. if бот == nil {
  114. return кнт.SendString("[Время: нет такого бота]")
  115. }
  116. время := бот.Ангар().База().Арсенал().ВремяОстат()
  117. стрВремя := время.String()
  118. if стрВремя == "" {
  119. return кнт.SendString("[Время: пустой остаток времени]")
  120. }
  121. return кнт.SendString("[Время: " + стрВремя + "]")
  122. }
  123. // Возвращает время, которое осталось на полигоне
  124. func (сам *ВебАпи) полигонВремяОсталось(кнт *fiber.Ctx) error {
  125. номер, ош := кнт.ParamsInt("number")
  126. if ош != nil {
  127. сообщ := fmt.Sprintf("[Время: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  128. return кнт.SendString(сообщ)
  129. }
  130. ботНомер := alias.БотНомер(номер)
  131. бот := сам.серв.ServBots().Get(ботНомер)
  132. if бот == nil {
  133. return кнт.SendString("[Время: нет такого бота]")
  134. }
  135. время := бот.Ангар().База().Полигон().ВремяОстат()
  136. стрВремя := время.String()
  137. if стрВремя == "" {
  138. return кнт.SendString("[Время: пустой остаток времени]")
  139. }
  140. return кнт.SendString("[Время: " + стрВремя + "]")
  141. }
  142. // Возвращает имя добычи ресурса на полигоне
  143. func (сам *ВебАпи) полигонРаботаИмя(кнт *fiber.Ctx) error {
  144. номер, ош := кнт.ParamsInt("number")
  145. if ош != nil {
  146. сообщ := fmt.Sprintf("[Тип: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  147. return кнт.SendString(сообщ)
  148. }
  149. ботНомер := alias.БотНомер(номер)
  150. бот := сам.серв.ServBots().Get(ботНомер)
  151. if бот == nil {
  152. return кнт.SendString("[Тип: нет такого бота]")
  153. }
  154. имя := бот.Ангар().База().Полигон().ПродуктИмяСейчас()
  155. if имя == "" {
  156. return кнт.SendString("[Тип: пустое имя]")
  157. }
  158. return кнт.SendString("[Тип: " + имя + "]")
  159. }
  160. // Возвращает количесто добычи на полигоне
  161. func (сам *ВебАпи) полигонРаботаКоличество(кнт *fiber.Ctx) error {
  162. номер, ош := кнт.ParamsInt("number")
  163. if ош != nil {
  164. сообщ := fmt.Sprintf("[Кол: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  165. return кнт.SendString(сообщ)
  166. }
  167. ботНомер := alias.БотНомер(номер)
  168. бот := сам.серв.ServBots().Get(ботНомер)
  169. if бот == nil {
  170. return кнт.SendString("[Кол: нет такого бота]")
  171. }
  172. колич := бот.Ангар().База().Полигон().ПродуктКолСейчас()
  173. стрКолич := fmt.Sprint(колич)
  174. if стрКолич == "" {
  175. return кнт.SendString("[Кол: пустое кол]")
  176. }
  177. return кнт.SendString("[Кол: +" + стрКолич + "]")
  178. }
  179. // Возвращает режим полигона
  180. func (сам *ВебАпи) полигонРежим(кнт *fiber.Ctx) error {
  181. номер, ош := кнт.ParamsInt("number")
  182. if ош != nil {
  183. сообщ := fmt.Sprintf("[Режим: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  184. return кнт.SendString(сообщ)
  185. }
  186. ботНомер := alias.БотНомер(номер)
  187. бот := сам.серв.ServBots().Get(ботНомер)
  188. if бот == nil {
  189. return кнт.SendString("[Режим: нет такого бота]")
  190. }
  191. сценаРежим := бот.Ангар().База().Полигон().СценаРежим()
  192. стрРежим := fmt.Sprint(сценаРежим.Режим())
  193. if стрРежим == "" {
  194. return кнт.SendString("[Режим: пустой режим]")
  195. }
  196. return кнт.SendString("[Режим: " + стрРежим + "]")
  197. }
  198. // Возвращает уровень полигона
  199. func (сам *ВебАпи) полигонУровень(кнт *fiber.Ctx) error {
  200. номер, ош := кнт.ParamsInt("number")
  201. if ош != nil {
  202. сообщ := fmt.Sprintf("[Уровень: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  203. return кнт.SendString(сообщ)
  204. }
  205. ботНомер := alias.БотНомер(номер)
  206. бот := сам.серв.ServBots().Get(ботНомер)
  207. if бот == nil {
  208. return кнт.SendString("[Уровень: нет такого бота]")
  209. }
  210. уровень := бот.Ангар().База().Полигон().Уровень()
  211. стрУровень := fmt.Sprint(уровень.Получ())
  212. if стрУровень == "" {
  213. return кнт.SendString("[Уровень: пустой уровень]")
  214. }
  215. return кнт.SendString("[Уровень: " + стрУровень + "]")
  216. }
  217. // Возвращает количество серебра в ангаре
  218. func (сам *ВебАпи) ангарСеребро(кнт *fiber.Ctx) error {
  219. номер, ош := кнт.ParamsInt("number")
  220. if ош != nil {
  221. сообщ := fmt.Sprintf("[Серебро: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  222. return кнт.SendString(сообщ)
  223. }
  224. ботНомер := alias.БотНомер(номер)
  225. бот := сам.серв.ServBots().Get(ботНомер)
  226. if бот == nil {
  227. return кнт.SendString("[Серебро: нет такого бота]")
  228. }
  229. серебро := бот.Ангар().СереброВсего().Получ()
  230. if серебро == 0 {
  231. return кнт.SendString("[Серебро: пустое кол]")
  232. }
  233. стрСеребро := fmt.Sprint(серебро)
  234. return кнт.SendString("[Серебро: " + стрСеребро + "]")
  235. }
  236. // Возвращает количество топлива в танке
  237. func (сам *ВебАпи) танкТопливо(кнт *fiber.Ctx) error {
  238. номер, ош := кнт.ParamsInt("number")
  239. if ош != nil {
  240. сообщ := fmt.Sprintf("[Топливо: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  241. return кнт.SendString(сообщ)
  242. }
  243. ботНомер := alias.БотНомер(номер)
  244. бот := сам.серв.ServBots().Get(ботНомер)
  245. if бот == nil {
  246. return кнт.SendString("[Топливо: нет такого бота]")
  247. }
  248. топливо := бот.Ангар().Топливо().Получ()
  249. if топливо == 0 {
  250. return кнт.SendString("[Топливо: пустое кол]")
  251. }
  252. стрТопливо := fmt.Sprint(топливо)
  253. return кнт.SendString("[Топливо: " + стрТопливо + "]")
  254. }
  255. // Возвращает имя добычи ресурса в шахте
  256. func (сам *ВебАпи) шахтаРаботаИмя(кнт *fiber.Ctx) error {
  257. номер, ош := кнт.ParamsInt("number")
  258. if ош != nil {
  259. сообщ := fmt.Sprintf("[Тип: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  260. return кнт.SendString(сообщ)
  261. }
  262. ботНомер := alias.БотНомер(номер)
  263. бот := сам.серв.ServBots().Get(ботНомер)
  264. if бот == nil {
  265. return кнт.SendString("[Тип: нет такого бота]")
  266. }
  267. имя := бот.Ангар().База().Шахта().ПродуктИмяСейчас()
  268. if имя == "" {
  269. return кнт.SendString("[Тип: пустое имя]")
  270. }
  271. return кнт.SendString("[Тип: " + имя + "]")
  272. }
  273. // Возвращает количесто добычи в шахте
  274. func (сам *ВебАпи) шахтаРаботаКоличество(кнт *fiber.Ctx) error {
  275. номер, ош := кнт.ParamsInt("number")
  276. if ош != nil {
  277. сообщ := fmt.Sprintf("[Кол: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  278. return кнт.SendString(сообщ)
  279. }
  280. ботНомер := alias.БотНомер(номер)
  281. бот := сам.серв.ServBots().Get(ботНомер)
  282. if бот == nil {
  283. return кнт.SendString("[Кол: нет такого бота]")
  284. }
  285. колич := бот.Ангар().База().Шахта().ПродуктКолСейчас()
  286. стрКолич := fmt.Sprint(колич)
  287. if стрКолич == "" {
  288. return кнт.SendString("[Кол: пустое кол]")
  289. }
  290. return кнт.SendString("[Кол: " + стрКолич + "]")
  291. }
  292. // Возвращает режим шахты
  293. func (сам *ВебАпи) шахтаРежим(кнт *fiber.Ctx) error {
  294. номер, ош := кнт.ParamsInt("number")
  295. if ош != nil {
  296. сообщ := fmt.Sprintf("[Режим: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  297. return кнт.SendString(сообщ)
  298. }
  299. ботНомер := alias.БотНомер(номер)
  300. бот := сам.серв.ServBots().Get(ботНомер)
  301. if бот == nil {
  302. return кнт.SendString("[Режим: нет такого бота]")
  303. }
  304. сценаРежим := бот.Ангар().База().Шахта().СценаРежим()
  305. стрРежим := fmt.Sprint(сценаРежим.Режим())
  306. if стрРежим == "" {
  307. return кнт.SendString("[Режим: пустой режим]")
  308. }
  309. return кнт.SendString("[Режим: " + стрРежим + "]")
  310. }
  311. // Возвращает уровень шахты
  312. func (сам *ВебАпи) шахтаУровень(кнт *fiber.Ctx) error {
  313. номер, ош := кнт.ParamsInt("number")
  314. if ош != nil {
  315. сообщ := fmt.Sprintf("[Уровень: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  316. return кнт.SendString(сообщ)
  317. }
  318. ботНомер := alias.БотНомер(номер)
  319. бот := сам.серв.ServBots().Get(ботНомер)
  320. if бот == nil {
  321. return кнт.SendString("[Уровень: нет такого бота]")
  322. }
  323. уровень := бот.Ангар().База().Шахта().Уровень()
  324. стрУровень := fmt.Sprint(уровень.Получ())
  325. if стрУровень == "" {
  326. return кнт.SendString("[Уровень: пустой уровень]")
  327. }
  328. return кнт.SendString("[Уровень: " + стрУровень + "]")
  329. }
  330. // Возвращает время, которое осталось на шахте
  331. func (сам *ВебАпи) шахтаВремяОсталось(кнт *fiber.Ctx) error {
  332. номер, ош := кнт.ParamsInt("number")
  333. if ош != nil {
  334. сообщ := fmt.Sprintf("[Время: неправильный номер бота(%q), err=%v]", номер, ош.Error())
  335. return кнт.SendString(сообщ)
  336. }
  337. ботНомер := alias.БотНомер(номер)
  338. бот := сам.серв.ServBots().Get(ботНомер)
  339. if бот == nil {
  340. return кнт.SendString("[Время: нет такого бота]")
  341. }
  342. время := бот.Ангар().База().Шахта().ВремяОстат()
  343. стрВремя := время.String()
  344. if стрВремя == "" {
  345. return кнт.SendString("[Время: пустой остаток времени]")
  346. }
  347. return кнт.SendString("[Время: " + стрВремя + "]")
  348. }
  349. // Возвращает счётчик запусков сервера
  350. func (сам *ВебАпи) стартНомер(кнт *fiber.Ctx) error {
  351. return кнт.SendString("[Старт: " + fmt.Sprint(сам.серв.Стат().СчётСтарт()) + "]")
  352. }
  353. // Возвращает аптайм сервера
  354. func (сам *ВебАпи) аптаймСервер(кнт *fiber.Ctx) error {
  355. return кнт.SendString("[Аптайм: " + сам.серв.Стат().ВремяВсего() + "]")
  356. }
  357. type AddBotRequest struct {
  358. Логин_ string `json:"login" form:"login_bot"`
  359. Пароль string `json:"password" form:"password_bot"`
  360. }
  361. // Запрос добавления нового бота на ботоферму
  362. func (сам *ВебАпи) добавитьБота(кнт *fiber.Ctx) error {
  363. запрос := &AddBotRequest{}
  364. if err := кнт.BodyParser(запрос); err != nil {
  365. return кнт.Render("add_bot", fiber.Map{
  366. "err": "ошибка в разборе формы",
  367. })
  368. }
  369. log.Printf("ВебАпи.добавитьБота(): : %#+v\n", запрос)
  370. if запрос.Логин_ == "" {
  371. return кнт.Render("add_bot", fiber.Map{
  372. "err": "пустой логин бота",
  373. })
  374. }
  375. if запрос.Пароль == "" {
  376. return кнт.Render("add_bot", fiber.Map{
  377. "err": "пустой пароль бота",
  378. })
  379. }
  380. сам.серв.ServBots().НовБот(запрос.Логин_, запрос.Пароль, true)
  381. return кнт.Redirect("/gui/list_bot")
  382. }
  383. type LoginRequest struct {
  384. Логин_ string `json:"login" form:"login"`
  385. Пароль_ string `json:"password" form:"password"`
  386. КонтрольноеСлово_ string `json:"control_word" form:"control_word"`
  387. }
  388. // Вызывается при попытке войти
  389. func (сам *ВебАпи) логин(кнт *fiber.Ctx) error {
  390. запрос := new(LoginRequest)
  391. if err := кнт.BodyParser(запрос); err != nil {
  392. return кнт.JSON(fiber.Map{
  393. "status": "ошибка в разборе формы",
  394. })
  395. }
  396. log.Printf("ВебАпи.логин(): : %#+v\n", *запрос)
  397. if запрос.Логин_ == "" || запрос.Пароль_ == "" {
  398. return кнт.JSON(fiber.Map{
  399. "status": "пустой логин/пароль",
  400. })
  401. }
  402. if запрос.Логин_ != "svi" || запрос.Пароль_ != "Lera_07091978" {
  403. return кнт.JSON(fiber.Map{
  404. "status": "ошибка логина/пароля",
  405. })
  406. }
  407. кнт.Cookie(&fiber.Cookie{
  408. Name: "login",
  409. Value: "svi",
  410. Expires: time.Now().Add(24 * time.Hour),
  411. HTTPOnly: true,
  412. })
  413. return кнт.Redirect("/gui/list_bot")
  414. }