|
|
|
using System;
|
|
|
|
using System.Net;
|
|
|
|
using System.Web;
|
|
|
|
using FluentValidation.Results;
|
|
|
|
using NLog;
|
|
|
|
using NzbDrone.Common.Extensions;
|
|
|
|
using NzbDrone.Common.Serializer;
|
|
|
|
using RestSharp;
|
|
|
|
using NzbDrone.Core.Rest;
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Notifications.Telegram
|
|
|
|
{
|
|
|
|
public interface ITelegramProxy
|
|
|
|
{
|
|
|
|
void SendNotification(string title, string message, TelegramSettings settings);
|
|
|
|
ValidationFailure Test(TelegramSettings settings);
|
|
|
|
}
|
|
|
|
|
|
|
|
public class TelegramProxy : ITelegramProxy
|
|
|
|
{
|
|
|
|
private readonly Logger _logger;
|
|
|
|
private const string URL = "https://api.telegram.org";
|
|
|
|
|
|
|
|
public TelegramProxy(Logger logger)
|
|
|
|
{
|
|
|
|
_logger = logger;
|
|
|
|
}
|
|
|
|
|
|
|
|
public void SendNotification(string title, string message, TelegramSettings settings)
|
|
|
|
{
|
|
|
|
//Format text to add the title before and bold using markdown
|
|
|
|
var text = $"<b>{HttpUtility.HtmlEncode(title)}</b>\n{HttpUtility.HtmlEncode(message)}";
|
|
|
|
var client = RestClientFactory.BuildClient(URL);
|
|
|
|
var request = new RestRequest("bot{token}/sendmessage", Method.POST);
|
|
|
|
|
|
|
|
request.AddUrlSegment("token", settings.BotToken);
|
|
|
|
request.AddParameter("chat_id", settings.ChatId);
|
|
|
|
request.AddParameter("parse_mode", "HTML");
|
|
|
|
request.AddParameter("text", text);
|
|
|
|
|
|
|
|
client.ExecuteAndValidate(request);
|
|
|
|
}
|
|
|
|
|
|
|
|
public ValidationFailure Test(TelegramSettings settings)
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
const string title = "Test Notification";
|
|
|
|
const string body = "This is a test message from Lidarr";
|
|
|
|
|
|
|
|
SendNotification(title, body, settings);
|
|
|
|
}
|
|
|
|
catch (Exception ex)
|
|
|
|
{
|
|
|
|
_logger.Error(ex, "Unable to send test message");
|
|
|
|
|
|
|
|
var restException = ex as RestException;
|
|
|
|
|
|
|
|
if (restException != null && restException.Response.StatusCode == HttpStatusCode.BadRequest)
|
|
|
|
{
|
|
|
|
var error = Json.Deserialize<TelegramError>(restException.Response.Content);
|
|
|
|
var property = error.Description.ContainsIgnoreCase("chat not found") ? "ChatId" : "BotToken";
|
|
|
|
|
|
|
|
return new ValidationFailure(property, error.Description);
|
|
|
|
}
|
|
|
|
|
|
|
|
return new ValidationFailure("BotToken", "Unable to send test message");
|
|
|
|
}
|
|
|
|
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|