Added user state (nested menus) and sub-menu to control different potential problems and system status

This commit is contained in:
ftc
2022-04-28 09:10:41 +03:00
parent 23b6ae390e
commit 2e03e8b747
7 changed files with 375 additions and 132 deletions
+19 -132
View File
@@ -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;
}
}
}