diff --git a/app/controllers/chat_controller.rb b/app/controllers/chat_controller.rb index 63e9b9e..42879f8 100644 --- a/app/controllers/chat_controller.rb +++ b/app/controllers/chat_controller.rb @@ -36,6 +36,7 @@ class DiscourseChat::ChatController < ApplicationController rescue Discourse::InvalidParameters, ActiveRecord::RecordNotFound => e render json: {errors: [e.message]}, status: 422 rescue DiscourseChat::ProviderError => e + Rails.logger.error("Test provider failed #{e.info}") if e.info.key?(:error_key) and !e.info[:error_key].nil? render json: {error_key: e.info[:error_key]}, status: 422 else diff --git a/app/models/channel.rb b/app/models/channel.rb index 2337ade..ca97194 100644 --- a/app/models/channel.rb +++ b/app/models/channel.rb @@ -50,6 +50,6 @@ class DiscourseChat::Channel < DiscourseChat::PluginModel scope :with_provider, ->(provider) { where("value::json->>'provider'=?", provider)} - scope :with_data_value, ->(key, value) { where("(value::json->>'data')::json->>?=?", key, value)} + scope :with_data_value, ->(key, value) { where("(value::json->>'data')::json->>?=?", key.to_s, value.to_s)} end \ No newline at end of file diff --git a/config/locales/client.en.yml b/config/locales/client.en.yml index 63ac6ce..c6158ab 100644 --- a/config/locales/client.en.yml +++ b/config/locales/client.en.yml @@ -63,4 +63,13 @@ en: ####################################### telegram: title: "Telegram" - channel_instructions: "Enter a chat_id. You will need to add the bot to the chat first." \ No newline at end of file + param: + name: + title: "Name" + help: "A name to describe the channel. It is not used for the connection for telegram." + chat_id: + title: Chat ID + help: A number given to you by the bot, or a broadcast channel identifier in the form @channelname + errors: + channel_not_found: "The specified channel does not exist on Telegram" + forbidden: "The bot does not have permission to post to this channel" \ No newline at end of file diff --git a/config/locales/server.en.yml b/config/locales/server.en.yml index 26b32d0..e2c496a 100644 --- a/config/locales/server.en.yml +++ b/config/locales/server.en.yml @@ -60,4 +60,12 @@ en: *List rules:* `/discourse status` - *Help:* `/discourse help` \ No newline at end of file + *Help:* `/discourse help` + + telegram: + unknown_chat: "This chat isn't setup on %{site_title}. Ask an administrator to add a channel with 'Chat ID' %{chat_id}." + known_chat: "This chat is setup on %{site_title}. Configure it in the admin panel. (Chat ID: %{chat_id})" + message: |- + %{user} posted in %{title} + +
%{post_excerpt}
\ No newline at end of file diff --git a/config/settings.yml b/config/settings.yml index 61caed7..07b2cbd 100644 --- a/config/settings.yml +++ b/config/settings.yml @@ -30,3 +30,11 @@ plugins: ####################################### chat_integration_telegram_enabled: default: false + chat_integration_telegram_access_token: + default: '' + chat_integration_telegram_excerpt_length: + default: 400 + chat_integration_telegram_enable_slash_commands: + default: true + chat_integration_telegram_secret: + hidden: true \ No newline at end of file diff --git a/lib/discourse_chat/provider/telegram/telegram_command_controller.rb b/lib/discourse_chat/provider/telegram/telegram_command_controller.rb index c63c22e..2df2195 100644 --- a/lib/discourse_chat/provider/telegram/telegram_command_controller.rb +++ b/lib/discourse_chat/provider/telegram/telegram_command_controller.rb @@ -2,9 +2,80 @@ module DiscourseChat::Provider::TelegramProvider class TelegramCommandController < DiscourseChat::Provider::HookController requires_provider ::DiscourseChat::Provider::TelegramProvider::PROVIDER_NAME - def say_hello + before_filter :telegram_token_valid?, only: :command - render json: {hello: "from telegram"} + skip_before_filter :check_xhr, + :preload_json, + :verify_authenticity_token, + :redirect_to_login_if_required, + only: :command + + def command + + # If it's a new message (telegram also sends hooks for other reasons that we don't care about) + if params.key?('message') + chat_id = params['message']['chat']['id'] + + message_text = process_command(params['message']) + + message = { + chat_id: chat_id, + text: message_text, + parse_mode: "html", + disable_web_page_preview: true, + } + + DiscourseChat::Provider::TelegramProvider.sendMessage(message) + + end + + # Always give telegram a success message, otherwise we'll stop receiving webhooks + data = { + success: true + } + render json: data + end + + def process_command(message) + chat_id = params['message']['chat']['id'] + + provider = DiscourseChat::Provider::TelegramProvider::PROVIDER_NAME + + channel = DiscourseChat::Channel.with_provider(provider).with_data_value('chat_id',chat_id).first + + if channel.nil? + return I18n.t( + "chat_integration.provider.telegram.unknown_chat", + site_title: CGI::escapeHTML(SiteSetting.title), + chat_id: chat_id, + ) + end + + # If slash commands disabled, send a generic message + if !SiteSetting.chat_integration_telegram_enable_slash_commands + return I18n.t( + "chat_integration.provider.telegram.known_chat", + site_title: CGI::escapeHTML(SiteSetting.title), + chat_id: chat_id, + ) + end + + tokens = message['text'].split(" ") + + return "I don't understand how to do that yet" + + + + end + + def telegram_token_valid? + params.require(:token) + + if SiteSetting.chat_integration_telegram_secret.blank? || + SiteSetting.chat_integration_telegram_secret != params[:token] + + raise Discourse::InvalidAccess.new + end end end @@ -14,6 +85,6 @@ module DiscourseChat::Provider::TelegramProvider end TelegramEngine.routes.draw do - get "command" => "telegram_command#say_hello" + post "command/:token" => "telegram_command#command" end end \ No newline at end of file diff --git a/lib/discourse_chat/provider/telegram/telegram_initializer.rb b/lib/discourse_chat/provider/telegram/telegram_initializer.rb new file mode 100644 index 0000000..1922814 --- /dev/null +++ b/lib/discourse_chat/provider/telegram/telegram_initializer.rb @@ -0,0 +1,15 @@ +Rails.logger.error("LOADED") +DiscourseEvent.on(:site_setting_saved) do |sitesetting| + isEnabledSetting = sitesetting.name == 'chat_integration_telegram_enabled' + isAccessToken = sitesetting.name == 'chat_integration_telegram_access_token' + + if (isEnabledSetting or isAccessToken) + enabled = isEnabledSetting ? sitesetting.value == 't' : SiteSetting.chat_integration_telegram_enabled + # Rails.logger.error("JOB ENQUEUED"+sitesetting.value+SiteSetting.chat_integration_telegram_enabled.to_s) + if enabled + Scheduler::Defer.later("Setup Telegram Webhook") do + DiscourseChat::Provider::TelegramProvider.setup_webhook() + end + end + end +end \ No newline at end of file diff --git a/lib/discourse_chat/provider/telegram/telegram_provider.rb b/lib/discourse_chat/provider/telegram/telegram_provider.rb index a681603..2bf505a 100644 --- a/lib/discourse_chat/provider/telegram/telegram_provider.rb +++ b/lib/discourse_chat/provider/telegram/telegram_provider.rb @@ -3,10 +3,107 @@ module DiscourseChat module TelegramProvider PROVIDER_NAME = "telegram".freeze PROVIDER_ENABLED_SETTING = :chat_integration_telegram_enabled - CHANNEL_PARAMETERS = [] + CHANNEL_PARAMETERS = [ + {key: "name", regex: '^\S+'}, + {key: "chat_id", regex: '^(-?[0-9]+|@\S+)$'} + ] + + def self.setup_webhook + newSecret = SecureRandom.hex + SiteSetting.chat_integration_telegram_secret = newSecret + + message = { + url: Discourse.base_url+'/chat-integration/telegram/command/'+newSecret, + } + + response = self.do_api_request('setWebhook', message) + + if not response['ok'] == true + # If setting up webhook failed, disable provider + SiteSetting.chat_integration_telegram_enabled = false + Rails.logger.error("Failed to setup telegram webhook. Message data= "+message.to_json+ " response="+response.to_json) + end + + end + + def self.sendMessage(message) + return self.do_api_request('sendMessage', message) + end + + def self.do_api_request(methodName, message) + http = Net::HTTP.new("api.telegram.org", 443) + http.use_ssl = true + + access_token = SiteSetting.chat_integration_telegram_access_token + + uri = URI("https://api.telegram.org/bot#{access_token}/#{methodName}") + + req = Net::HTTP::Post.new(uri, 'Content-Type' =>'application/json') + req.body = message.to_json + response = http.request(req) + + responseData = JSON.parse(response.body) + + return responseData + end + + def self.message_text(post) + display_name = "@#{post.user.username}" + full_name = post.user.name || "" + + if !(full_name.strip.empty?) && (full_name.strip.gsub(' ', '_').casecmp(post.user.username) != 0) && (full_name.strip.gsub(' ', '').casecmp(post.user.username) != 0) + display_name = "#{full_name} @#{post.user.username}" + end + + topic = post.topic + + category = '' + if topic.category + category = (topic.category.parent_category) ? "[#{topic.category.parent_category.name}/#{topic.category.name}]": "[#{topic.category.name}]" + end + + tags = '' + if topic.tags.present? + tags = topic.tags.map(&:name).join(', ') + end + + return I18n.t( + "chat_integration.provider.telegram.message", + user: display_name, + post_url: "https://meta.discourse.org", + title: CGI::escapeHTML(topic.title), + post_excerpt: post.excerpt(SiteSetting.chat_integration_telegram_excerpt_length, text_entities: true, strip_links: true, remap_emoji: true), + ) + + end + + def self.trigger_notification(post, channel) + chat_id = channel.data['chat_id'] + + message = { + chat_id: chat_id, + text: message_text(post), + parse_mode: "html", + disable_web_page_preview: true, + } + + response = sendMessage(message) + + if not response['ok'] == true + error_key = nil + if response['description'].include? 'chat not found' + error_key = 'chat_integration.provider.telegram.channel_not_found' + elsif response['description'].include? 'Forbidden' + error_key = 'chat_integration.provider.telegram.forbidden' + end + raise ::DiscourseChat::ProviderError.new info: {error_key: error_key, message: message, response_body:response} + end + + end end end end -require_relative "telegram_command_controller.rb" \ No newline at end of file +require_relative "telegram_command_controller.rb" +require_relative "telegram_initializer.rb" \ No newline at end of file