zaobot/BotService.cs

using System;
using System.Collections.Generic;
using System.Net.Http;
using System.Text;
using System.Text.Json;
using System.Threading;
using System.Threading.Tasks;
using Telegram.Bot;
using Telegram.Bot.Args;
using Telegram.Bot.Types.ReplyMarkups;

namespace ZaoBot
{
	class BotService
	{
		private JsonDocument config;
		static ITelegramBotClient botClient;
		HttpClient http = new HttpClient();

		const string CMD_WEATHER = "Погода";
		const string CMD_LEFT_GATE = "Открой левые ворота";
		const string CMD_RIGHT_GATE = "Открой правые ворота";

		bool shouldRestart = false;

		public BotService(JsonDocument config)
		{
			this.config = config;
		}

		public void Run()
		{
			var token = config.RootElement.GetProperty("token").GetString();
			shouldRestart = false;
			botClient = new TelegramBotClient(token);
			var me = botClient.GetMeAsync().Result;
			Console.WriteLine($"Hello, World! I am user {me.Id} and my name is {me.FirstName}.");
			botClient.OnMessage += OnBotMessage;
			botClient.StartReceiving();
			while (!shouldRestart)
			{
				Thread.Sleep(1000);
			}
		}

		private async void OnBotMessage(object sender, MessageEventArgs e)
		{
			try
			{
				if (e.Message.Text != null)
				{
					if (!VerifyUser(e.Message.Chat.Id, e.Message.Chat.Username))
					{
						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 + " сообщите его администратору"
						);
						return;
					}
					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
					);
				}
			}
			catch (Exception ex)
			{
				Console.WriteLine("Exception: " + ex.Message);
				Console.WriteLine(ex.StackTrace);
				Console.WriteLine("Restarting bot");
				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;
		}
	}
}