119 lines
2.6 KiB
Go
119 lines
2.6 KiB
Go
package bot
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"git.dev.m-and-m.ovh/mderasse/tesla/api"
|
|
)
|
|
|
|
var specialChars = []string{
|
|
"\\",
|
|
"_",
|
|
"*",
|
|
"[",
|
|
"]",
|
|
"(",
|
|
")",
|
|
"~",
|
|
"`",
|
|
">",
|
|
"<",
|
|
"&",
|
|
"#",
|
|
"+",
|
|
"-",
|
|
"=",
|
|
"|",
|
|
"{",
|
|
"}",
|
|
".",
|
|
"!",
|
|
"(",
|
|
")",
|
|
}
|
|
|
|
var carFilter api.AvailabilityQueryParams = api.AvailabilityQueryParams{
|
|
Model: "my",
|
|
Condition: "new",
|
|
Options: api.OptionsParams{
|
|
Trim: []string{"SRRWD", "LRAWD"},
|
|
},
|
|
Arrangeby: "Price",
|
|
Order: "asc",
|
|
Market: "FR",
|
|
Language: "fr",
|
|
SuperRegion: "north america",
|
|
Lng: 3.0511,
|
|
Lat: 50.624,
|
|
Zip: "59100",
|
|
Range: 0,
|
|
Region: "FR",
|
|
}
|
|
|
|
// botConfig contain configuration for the telegram bot.
|
|
type botConfig struct {
|
|
AlertChatIds []int64
|
|
BotName string
|
|
WhiteListChatIds []int64
|
|
Token string
|
|
}
|
|
|
|
func initBotConfig() (*botConfig, error) {
|
|
alertChatIdsArrStr := strings.Split(os.Getenv("TELEGRAM_ALERT_CHAT_IDS"), ",")
|
|
whitelistChatIdsArrStr := strings.Split(os.Getenv("TELEGRAM_WHITELIST_CHAT_IDS"), ",")
|
|
botName := os.Getenv("TELEGRAM_BOT_NAME")
|
|
token := os.Getenv("TELEGRAM_TOKEN")
|
|
|
|
// FOR DEV
|
|
// botName = "M_AND_M_Test_bot"
|
|
// token = "6273217607:AAGvMIU6bpc7xFWKDhd4llPH6f9zhgUzirs"
|
|
// alertChatIdsArrStr = []string{"648385994", "-894368809", "-893895409"}
|
|
// whitelistChatIdsArrStr = []string{"648385994", "-894368809", "-893895409"}
|
|
|
|
if len(alertChatIdsArrStr) == 0 {
|
|
return nil, fmt.Errorf("missing TELEGRAM_ALERT_CHAT_IDS environment variable")
|
|
}
|
|
|
|
if len(whitelistChatIdsArrStr) == 0 {
|
|
return nil, fmt.Errorf("missing TELEGRAM_WHITELIST_CHAT_IDS environment variable")
|
|
}
|
|
|
|
if botName == "" {
|
|
return nil, fmt.Errorf("missing TELEGRAM_BOT_NAME environment variable")
|
|
}
|
|
|
|
if token == "" {
|
|
return nil, fmt.Errorf("missing TELEGRAM_TOKEN environment variable")
|
|
}
|
|
|
|
config := botConfig{
|
|
AlertChatIds: []int64{},
|
|
BotName: botName,
|
|
Token: token,
|
|
WhiteListChatIds: []int64{},
|
|
}
|
|
|
|
for _, chatIdStr := range alertChatIdsArrStr {
|
|
chatId, err := strconv.ParseInt(chatIdStr, 10, 64)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("impossible to use TELEGRAM_ALERT_CHAT_IDS. format expected: int64,int64")
|
|
}
|
|
|
|
config.AlertChatIds = append(config.AlertChatIds, chatId)
|
|
}
|
|
|
|
for _, chatIdStr := range whitelistChatIdsArrStr {
|
|
chatId, err := strconv.ParseInt(chatIdStr, 10, 64)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("impossible to use TELEGRAM_WHITELIST_CHAT_IDS. format expected: int64,int64")
|
|
}
|
|
|
|
config.WhiteListChatIds = append(config.WhiteListChatIds, chatId)
|
|
}
|
|
|
|
return &config, nil
|
|
}
|