Merged two bots into one
This commit is contained in:
Artyom Belousov
parent
68f1750beb
commit
480503d770
7 changed files with 177 additions and 181 deletions
136
cmd/bot/main.go
Normal file
136
cmd/bot/main.go
Normal file
|
|
@ -0,0 +1,136 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"math/rand"
|
||||
"os"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"gitlab.com/flygrounder/go-mtg-vk/internal/caching"
|
||||
"gitlab.com/flygrounder/go-mtg-vk/internal/cardsinfo"
|
||||
"gitlab.com/flygrounder/go-mtg-vk/internal/scenario"
|
||||
"gitlab.com/flygrounder/go-mtg-vk/internal/telegram"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
|
||||
environ "github.com/ydb-platform/ydb-go-sdk-auth-environ"
|
||||
"github.com/ydb-platform/ydb-go-sdk/v3"
|
||||
"gitlab.com/flygrounder/go-mtg-vk/internal/vk"
|
||||
)
|
||||
|
||||
type config struct {
|
||||
tgToken string
|
||||
vkGroupId int64
|
||||
vkSecretKey string
|
||||
vkToken string
|
||||
vkConfirmationString string
|
||||
ydbConnectionString string
|
||||
}
|
||||
|
||||
func getConfig() *config {
|
||||
var cfg config
|
||||
var exists bool
|
||||
var err error
|
||||
|
||||
cfg.tgToken, exists = os.LookupEnv("TG_TOKEN")
|
||||
if !exists {
|
||||
panic("TG_TOKEN environment variable not defined")
|
||||
}
|
||||
|
||||
vkGroupId, exists := os.LookupEnv("VK_GROUP_ID")
|
||||
if !exists {
|
||||
panic("VK_GROUP_ID environment variable not defined")
|
||||
}
|
||||
cfg.vkGroupId, err = strconv.ParseInt(vkGroupId, 10, 64)
|
||||
if err != nil {
|
||||
panic("VK_GROUP_ID is not a number")
|
||||
}
|
||||
|
||||
cfg.vkSecretKey, exists = os.LookupEnv("VK_SECRET_KEY")
|
||||
if !exists {
|
||||
panic("VK_SECRET_KEY environment variable not defined")
|
||||
}
|
||||
|
||||
cfg.vkToken, exists = os.LookupEnv("VK_TOKEN")
|
||||
if !exists {
|
||||
panic("VK_TOKEN environment variable not defined")
|
||||
}
|
||||
|
||||
cfg.vkConfirmationString, exists = os.LookupEnv("VK_CONFIRMATION_STRING")
|
||||
if !exists {
|
||||
panic("VK_CONFIRMATION_STRING environment variable not defined")
|
||||
}
|
||||
|
||||
cfg.ydbConnectionString, exists = os.LookupEnv("YDB_CONNECTION_STRING")
|
||||
if !exists {
|
||||
panic("YDB_CONNECTION_STRING environment variable not defined")
|
||||
}
|
||||
|
||||
return &cfg
|
||||
}
|
||||
|
||||
func main() {
|
||||
rand.Seed(time.Now().UTC().UnixNano())
|
||||
|
||||
cfg := getConfig()
|
||||
|
||||
bot, _ := tgbotapi.NewBotAPI(cfg.tgToken)
|
||||
r := gin.Default()
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||
defer cancel()
|
||||
|
||||
db, err := ydb.Open(ctx,
|
||||
cfg.ydbConnectionString,
|
||||
environ.WithEnvironCredentials(ctx),
|
||||
)
|
||||
if err != nil {
|
||||
panic(fmt.Errorf("connect error: %w", err))
|
||||
}
|
||||
defer func() { _ = db.Close(ctx) }()
|
||||
|
||||
cache := &caching.CacheClient{
|
||||
Storage: db.Table(),
|
||||
Expiration: 12 * time.Hour,
|
||||
Prefix: db.Name(),
|
||||
}
|
||||
err = cache.Init(context.Background())
|
||||
if err != nil {
|
||||
panic(fmt.Errorf("init error: %w", err))
|
||||
}
|
||||
|
||||
logger := log.New(os.Stdout, "", 0)
|
||||
fetcher := &cardsinfo.Fetcher{}
|
||||
handler := vk.Handler{
|
||||
Scenario: &scenario.Scenario{
|
||||
Sender: &vk.ApiSender{
|
||||
Token: cfg.vkToken,
|
||||
Logger: logger,
|
||||
},
|
||||
Logger: logger,
|
||||
InfoFetcher: fetcher,
|
||||
Cache: cache,
|
||||
},
|
||||
SecretKey: cfg.vkSecretKey,
|
||||
GroupId: cfg.vkGroupId,
|
||||
ConfirmationString: cfg.vkConfirmationString,
|
||||
}
|
||||
|
||||
tgHandler := telegram.Handler{
|
||||
Scenario: &scenario.Scenario{
|
||||
Sender: &telegram.Sender{
|
||||
API: bot,
|
||||
},
|
||||
Logger: logger,
|
||||
InfoFetcher: fetcher,
|
||||
Cache: cache,
|
||||
},
|
||||
}
|
||||
|
||||
r.POST("vk", handler.HandleMessage)
|
||||
r.POST("tg", tgHandler.HandleMessage)
|
||||
_ = r.Run(":8000")
|
||||
}
|
||||
|
|
@ -1,81 +0,0 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
|
||||
environ "github.com/ydb-platform/ydb-go-sdk-auth-environ"
|
||||
"github.com/ydb-platform/ydb-go-sdk/v3"
|
||||
"gitlab.com/flygrounder/go-mtg-vk/internal/caching"
|
||||
"gitlab.com/flygrounder/go-mtg-vk/internal/cardsinfo"
|
||||
"gitlab.com/flygrounder/go-mtg-vk/internal/scenario"
|
||||
"gitlab.com/flygrounder/go-mtg-vk/internal/telegram"
|
||||
)
|
||||
|
||||
const welcomeMessage = "Здравствуйте, вас приветствует бот для поиска цен на карты MTG, введите название карты, которая вас интересует."
|
||||
|
||||
func main() {
|
||||
token, exists := os.LookupEnv("TG_TOKEN")
|
||||
if !exists {
|
||||
panic("TG_TOKEN environment variable not defined")
|
||||
}
|
||||
bot, _ := tgbotapi.NewBotAPI(token)
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||
defer cancel()
|
||||
|
||||
dsn, exists := os.LookupEnv("YDB_CONNECTION_STRING")
|
||||
if !exists {
|
||||
panic("YDB_CONNECTION_STRING environment variable not defined")
|
||||
}
|
||||
|
||||
db, err := ydb.Open(ctx,
|
||||
dsn,
|
||||
environ.WithEnvironCredentials(ctx),
|
||||
)
|
||||
if err != nil {
|
||||
panic(fmt.Errorf("connect error: %w", err))
|
||||
}
|
||||
defer func() { _ = db.Close(ctx) }()
|
||||
|
||||
sender := &telegram.Sender{
|
||||
API: bot,
|
||||
}
|
||||
cache := &caching.CacheClient{
|
||||
Storage: db.Table(),
|
||||
Expiration: 12 * time.Hour,
|
||||
Prefix: db.Name(),
|
||||
}
|
||||
err = cache.Init(context.Background())
|
||||
if err != nil {
|
||||
panic(fmt.Errorf("init error: %w", err))
|
||||
}
|
||||
sc := &scenario.Scenario{
|
||||
Sender: sender,
|
||||
Logger: log.New(os.Stdout, "", 0),
|
||||
Cache: cache,
|
||||
InfoFetcher: &cardsinfo.Fetcher{},
|
||||
}
|
||||
|
||||
u := tgbotapi.NewUpdate(0)
|
||||
updates, _ := bot.GetUpdatesChan(u)
|
||||
for update := range updates {
|
||||
if update.Message == nil {
|
||||
continue
|
||||
}
|
||||
|
||||
if update.Message.Text == "/start" {
|
||||
sender.Send(update.Message.Chat.ID, welcomeMessage)
|
||||
continue
|
||||
}
|
||||
|
||||
go sc.HandleSearch(context.Background(), &scenario.UserMessage{
|
||||
Body: update.Message.Text,
|
||||
UserId: update.Message.Chat.ID,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
|
@ -1,73 +0,0 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"math/rand"
|
||||
"os"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"gitlab.com/flygrounder/go-mtg-vk/internal/caching"
|
||||
"gitlab.com/flygrounder/go-mtg-vk/internal/cardsinfo"
|
||||
"gitlab.com/flygrounder/go-mtg-vk/internal/scenario"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
environ "github.com/ydb-platform/ydb-go-sdk-auth-environ"
|
||||
"github.com/ydb-platform/ydb-go-sdk/v3"
|
||||
"gitlab.com/flygrounder/go-mtg-vk/internal/vk"
|
||||
)
|
||||
|
||||
func main() {
|
||||
rand.Seed(time.Now().UTC().UnixNano())
|
||||
|
||||
r := gin.Default()
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||
defer cancel()
|
||||
|
||||
dsn, exists := os.LookupEnv("YDB_CONNECTION_STRING")
|
||||
if !exists {
|
||||
panic("YDB_CONNECTION_STRING environment variable not defined")
|
||||
}
|
||||
|
||||
db, err := ydb.Open(ctx,
|
||||
dsn,
|
||||
environ.WithEnvironCredentials(ctx),
|
||||
)
|
||||
if err != nil {
|
||||
panic(fmt.Errorf("connect error: %w", err))
|
||||
}
|
||||
defer func() { _ = db.Close(ctx) }()
|
||||
|
||||
cache := &caching.CacheClient{
|
||||
Storage: db.Table(),
|
||||
Expiration: 12 * time.Hour,
|
||||
Prefix: db.Name(),
|
||||
}
|
||||
err = cache.Init(context.Background())
|
||||
if err != nil {
|
||||
panic(fmt.Errorf("init error: %w", err))
|
||||
}
|
||||
|
||||
groupId, _ := strconv.ParseInt(os.Getenv("VK_GROUP_ID"), 10, 64)
|
||||
logger := log.New(os.Stdout, "", 0)
|
||||
handler := vk.Handler{
|
||||
Scenario: &scenario.Scenario{
|
||||
Sender: &vk.ApiSender{
|
||||
Token: os.Getenv("VK_TOKEN"),
|
||||
Logger: logger,
|
||||
},
|
||||
Logger: logger,
|
||||
InfoFetcher: &cardsinfo.Fetcher{},
|
||||
Cache: cache,
|
||||
},
|
||||
SecretKey: os.Getenv("VK_SECRET_KEY"),
|
||||
GroupId: groupId,
|
||||
ConfirmationString: os.Getenv("VK_CONFIRMATION_STRING"),
|
||||
}
|
||||
|
||||
r.POST("callback/message", handler.HandleMessage)
|
||||
_ = r.Run(":8000")
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue