Added user state (nested menus) and sub-menu to control different potential problems and system status
This commit is contained in:
+19
-132
@@ -15,11 +15,8 @@ namespace ZaoBot
|
||||
{
|
||||
private JsonDocument config;
|
||||
static ITelegramBotClient botClient;
|
||||
HttpClient http = new HttpClient();
|
||||
|
||||
const string CMD_WEATHER = "Погода";
|
||||
const string CMD_LEFT_GATE = "Открой левые ворота";
|
||||
const string CMD_RIGHT_GATE = "Открой правые ворота";
|
||||
Dictionary<long, IUserState> userStates = new Dictionary<long, IUserState>();
|
||||
private OpenHab.OpenHabClient openhab;
|
||||
|
||||
bool shouldRestart = false;
|
||||
|
||||
@@ -33,6 +30,7 @@ namespace ZaoBot
|
||||
var token = config.RootElement.GetProperty("token").GetString();
|
||||
shouldRestart = false;
|
||||
botClient = new TelegramBotClient(token);
|
||||
openhab = new OpenHab.OpenHabClient(config.RootElement.GetProperty("openhab_url").GetString());
|
||||
var me = botClient.GetMeAsync().Result;
|
||||
Console.WriteLine($"Hello, World! I am user {me.Id} and my name is {me.FirstName}.");
|
||||
botClient.OnMessage += OnBotMessage;
|
||||
@@ -49,53 +47,27 @@ namespace ZaoBot
|
||||
{
|
||||
if (e.Message.Text != null)
|
||||
{
|
||||
if (!VerifyUser(e.Message.Chat.Id, e.Message.Chat.Username))
|
||||
var id = e.Message.Chat.Id;
|
||||
IUserState state = null;
|
||||
if (userStates.ContainsKey(id))
|
||||
{
|
||||
state = userStates[id];
|
||||
}
|
||||
if (state == null)
|
||||
{
|
||||
state = new MainMenuUserState(botClient, config, openhab);
|
||||
}
|
||||
state = await state.ProcessMessage(state, e);
|
||||
if (state == null)
|
||||
{
|
||||
Console.WriteLine($"Received message {e.Message.Text} from invalid user {e.Message.Chat.Username}");
|
||||
await botClient.SendTextMessageAsync(
|
||||
chatId: e.Message.Chat,
|
||||
text: "Ваш ID " + e.Message.Chat.Id + " сообщите его администратору"
|
||||
text: "Я запутался, давайте сначала"
|
||||
);
|
||||
return;
|
||||
state = new MainMenuUserState(botClient, config, openhab);
|
||||
state = await state.ProcessMessage(state, e);
|
||||
}
|
||||
Console.WriteLine($"Received a text message in chat {e.Message.Chat.Id}.");
|
||||
|
||||
var response = "";
|
||||
if (e.Message.Text == CMD_LEFT_GATE)
|
||||
{
|
||||
response = await OpenLeftGate();
|
||||
}
|
||||
else if (e.Message.Text == CMD_RIGHT_GATE)
|
||||
{
|
||||
response = await OpenRightGate();
|
||||
}
|
||||
else if (e.Message.Text == CMD_WEATHER)
|
||||
{
|
||||
response = await ShowWeather();
|
||||
}
|
||||
else if (e.Message.Text == "")
|
||||
{
|
||||
response = "Превед!";
|
||||
}
|
||||
else response = "Я не понял :-(";
|
||||
|
||||
if (e.Message.Text != "") response += "\nЧто дальше?";
|
||||
else response = "\nЧего изволите?";
|
||||
|
||||
var replyKeyboardMarkup = new ReplyKeyboardMarkup(
|
||||
new KeyboardButton[][]
|
||||
{
|
||||
new KeyboardButton[] { CMD_WEATHER },
|
||||
new KeyboardButton[] { CMD_LEFT_GATE, CMD_RIGHT_GATE },
|
||||
},
|
||||
resizeKeyboard: true
|
||||
);
|
||||
|
||||
await botClient.SendTextMessageAsync(
|
||||
chatId: e.Message.Chat,
|
||||
text: response,
|
||||
replyMarkup: replyKeyboardMarkup
|
||||
);
|
||||
userStates[id] = state;
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
@@ -106,90 +78,5 @@ namespace ZaoBot
|
||||
shouldRestart = true;
|
||||
}
|
||||
}
|
||||
|
||||
private async Task<string> ShowWeather()
|
||||
{
|
||||
try
|
||||
{
|
||||
string response = "";
|
||||
foreach (var element in config.RootElement.GetProperty("items_weather").EnumerateObject())
|
||||
{
|
||||
var url = config.RootElement.GetProperty("openhab_url").GetString();
|
||||
url += element.Name;
|
||||
var result = await http.GetAsync(url);
|
||||
var content = await result.Content.ReadAsStringAsync();
|
||||
response += element.Value + ": " + ParseContent(content) + "\n";
|
||||
}
|
||||
return response;
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
return "Fail: " + e.Message;
|
||||
}
|
||||
}
|
||||
|
||||
private string ParseContent(string content)
|
||||
{
|
||||
var doc = JsonDocument.Parse(content);
|
||||
float value;
|
||||
if (!float.TryParse(doc.RootElement.GetProperty("state").GetString(), out value))
|
||||
{
|
||||
return "null";
|
||||
}
|
||||
try
|
||||
{
|
||||
var stateDescription = doc.RootElement.GetProperty("stateDescription");
|
||||
var pattern = stateDescription.GetProperty("pattern");
|
||||
return printf.Printf.sprintf(pattern.GetString(), value);
|
||||
}
|
||||
catch (KeyNotFoundException)
|
||||
{
|
||||
return value.ToString();
|
||||
}
|
||||
}
|
||||
|
||||
private async Task<string> OpenRightGate()
|
||||
{
|
||||
var url = config.RootElement.GetProperty("openhab_url").GetString();
|
||||
url += config.RootElement.GetProperty("item_right_gate").GetString();
|
||||
return await Post(url, "ON");
|
||||
}
|
||||
|
||||
private async Task<string> Post(string url, string v)
|
||||
{
|
||||
var content = new ByteArrayContent(Encoding.UTF8.GetBytes(v));
|
||||
content.Headers.Add("Content-Type", "text/plain");
|
||||
var result = await http.PostAsync(url, content);
|
||||
if (result.IsSuccessStatusCode) return "OK";
|
||||
else return "Fail. " + result.ReasonPhrase;
|
||||
|
||||
}
|
||||
|
||||
private async Task<string> OpenLeftGate()
|
||||
{
|
||||
var url = config.RootElement.GetProperty("openhab_url").GetString();
|
||||
url += config.RootElement.GetProperty("item_left_gate").GetString();
|
||||
return await Post(url, "ON");
|
||||
}
|
||||
|
||||
private bool VerifyUser(long id, string username)
|
||||
{
|
||||
Console.WriteLine("Verify user " + username + " id " + id);
|
||||
foreach (var element in config.RootElement.GetProperty("valid_users").EnumerateArray())
|
||||
{
|
||||
if (username == element.GetString())
|
||||
{
|
||||
return true;
|
||||
}
|
||||
}
|
||||
foreach (var element in config.RootElement.GetProperty("valid_user_ids").EnumerateArray())
|
||||
{
|
||||
if (id.ToString() == element.GetString())
|
||||
{
|
||||
return true;
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user