wartank/server/serv_web/serv_web.go

// package serv_web -- веб-сервер для графики
package serv_web

import (
	"fmt"
	"log"
	"os"
	"time"

	"github.com/gofiber/fiber/v2"
	// "github.com/gofiber/fiber/v2/middleware/compress"
	// "github.com/gofiber/template/html/v2"

	"wartank/pkg/types"
)

type СервВеб struct {
	серв   types.ИСервер
	роутер fiber.App
	порт   string
}

// НовСервВеб -- возвращает новый веб-сервер
func НовСервВеб(серв types.ИСервер) (*СервВеб, error) {
	if серв == nil {
		return nil, fmt.Errorf("НовСерВеб(): ИСервер==nil")
	}
	порт := os.Getenv("SERVER_HTTP_PORT")
	if порт == "" {
		return nil, fmt.Errorf("НовСерВеб(): env SERVER_HTTP_PORT not set")
	}
	// engine := html.New("./web/tmpl", ".tmpl.html")

	конф := fiber.Config{
		// Prefork:                      false,
		ServerHeader: "WarTank",
		//StrictRouting:                false,
		//CaseSensitive:                false,
		//Immutable:                    false,
		//UnescapePath:                 false,
		//ETag:                         false,
		BodyLimit:   10_000_000,
		Concurrency: 50,
		// Views:       engine,
		//ViewsLayout:                  "",
		//PassLocalsToViews:            false,
		ReadTimeout:     15 * time.Second,
		WriteTimeout:    15 * time.Second,
		IdleTimeout:     15 * time.Second,
		ReadBufferSize:  4096,
		WriteBufferSize: 4096,
		//CompressedFileSuffix:         "",
		//ProxyHeader:                  "",
		//GETOnly:                      false,
		//ErrorHandler:                 func(*fiber.Ctx, error) error { panic("not implemented") },
		//DisableKeepalive:             false,
		//DisableDefaultDate:           false,
		//DisableDefaultContentType:    false,
		//DisableHeaderNormalizing:     false,
		//DisableStartupMessage:        false,
		AppName: "WarTank.App",
		// StreamRequestBody:            false,
		// DisablePreParseMultipartForm: false,
		// ReduceMemoryUsage:            false,
		// JSONEncoder:                  func(interface{}) ([]byte, error) { panic("not implemented") },
		// JSONDecoder:                  func([]byte, interface{}) error { panic("not implemented") },
		// XMLEncoder:                   func(interface{}) ([]byte, error) { panic("not implemented") },
		Network: "tcp4",
		// EnableTrustedProxyCheck:      false,
		// TrustedProxies:               []string{},
		// EnableIPValidation:           false,
		// EnablePrintRoutes:            false,
		// ColorScheme: fiber.Colors{
		// 	Black:   "",
		// 	Red:     "",
		// 	Green:   "",
		// 	Yellow:  "",
		// 	Blue:    "",
		// 	Magenta: "",
		// 	Cyan:    "",
		// 	White:   "",
		// 	Reset:   "",
		// },
		// RequestMethods:           []string{},
		// EnableSplittingOnParsers: false,
	}
	сам := &СервВеб{
		серв:   серв,
		роутер: *fiber.New(конф),
		порт:   порт,
	}
	// compresConfig := compress.Config{
	// 	Level: compress.LevelBestCompression,
	// }
	// сам.роутер.Use(compress.New(compresConfig))
	// сам.роутер.Static("/static", "./web/static", fiber.Static{
	// 	Compress:      true,
	// 	ByteRange:     true,
	// 	Browse:        true,
	// 	Index:         "index.html",
	// 	CacheDuration: 30 * time.Second,
	// 	MaxAge:        3600,
	// })
	сам.роутер.Get("/list_bot/get", сам.гетСписБот)
	сам.роутер.Post("/list_bot/add", сам.постБотНов)
	return сам, nil
}

// Пуск -- запускае веб-сервер в работу
func (сам *СервВеб) Пуск() {
	фнПуск := func() {
		ош := сам.роутер.Listen(сам.порт)
		if ош != nil {
			log.Printf("СервВеб.Пуск(): при работе веб-сервера, ош=\n\t%v\n", ош)
			сам.серв.CancelApp()
		}
	}
	go фнПуск()
}

type ПостБотНов struct {
	Логин    string `form:"login"`
	Пароль   string `form:"pass"`
	ЕслиАвто string `form:"is_auto"`
}

// постБотНов -- добавляет бота на бото-ферму
func (сам *СервВеб) постБотНов(кнт *fiber.Ctx) error {
	бот := &ПостБотНов{}

	if err := кнт.BodyParser(бот); err != nil {
		return кнт.JSON(fiber.Map{
			"error": "СервВеб.постБотНов(): при парсинге тела запроса, ош=\n\t%" + err.Error(),
		})
	}
	if бот.Логин == "" {
		return кнт.JSON(fiber.Map{
			"error": "СервВеб.постБотНов(): пустой логин",
		})
	}
	if бот.Пароль == "" {
		return кнт.JSON(fiber.Map{
			"error": "СервВеб.постБотНов(): пустой пароль",
		})
	}
	if бот.ЕслиАвто == "" {
		return кнт.JSON(fiber.Map{
			"error": "СервВеб.постБотНов(): пустой еслиАвто",
		})
	}
	var еслиАвто bool
	if бот.ЕслиАвто == "true" {
		еслиАвто = true
	}
	ботоФерма := сам.серв.ServBots()
	ош := ботоФерма.НовБот(бот.Логин, бот.Пароль, еслиАвто)
	if ош != nil {
		return кнт.JSON(fiber.Map{
			"error": "СервВеб.постБотНов(): при создании бота %q, ош=\n\t%" + ош.Error(),
		})
	}
	return nil
}

// гетСписБот -- возвращает список ботов
func (сам *СервВеб) гетСписБот(кнт *fiber.Ctx) error {
	списБот := сам.серв.ServBots().ListBot()
	рез := []string{}
	for _, бот := range списБот {
		рез = append(рез, бот.Имя())
	}
	return кнт.JSON(рез)
}