chatgpt_assistant 0.1.2 → 0.1.4

data/lib/bots/helpers/messenger_helper.rb

@@ -0,0 +1,134 @@
+# frozen_string_literal: true
+
+module ChatgptAssistant
+  # Helper for messenger
+  module MessengerHelper
+    def chat_success(chat_id)
+      Message.create(chat_id: chat_id, content: msg.text, role: "user")
+      text = chatter.chat(msg.text, chat_id)
+      mess = parse_message(text, 4096)
+      mess.each { |m| send_message m, msg.chat.id }
+    end
+
+    def respond_with_success
+      user.update(current_chat_id: chat.id)
+      send_message success_messages[:chat_created]
+    end
+
+    def parse_message(message, max_length)
+      if message.length > max_length
+        array = message.scan(/.{1,#{max_length}}/) if max_length.positive?
+        array = ["Something went wrong! Try again later"] if max_length <= 0
+      else
+        array = [message]
+      end
+      array
+    end
+
+    def send_message(text, chat_id = nil)
+      @send_message = bot.respond_to?(:api) ? telegram_send_message(text, chat_id) : discord_send_message(text)
+    end
+
+    def telegram_send_message(text, chat_id)
+      messages = parse_message(text, 4096)
+      messages.each { |m| bot.api.send_message(chat_id: chat_id, text: m) }
+    end
+
+    def discord_send_message(text)
+      messages = parse_message(text, 2000)
+      messages.each { |m| evnt.respond m }
+    end
+
+    def discord_help_message
+      help_messages.join("\n").gsub(" /", discord_prefix)
+                   .gsub("register/", "#{discord_prefix}register ")
+                   .gsub("login/", "#{discord_prefix}login ")
+                   .gsub("new_chat/", "#{discord_prefix}new_chat/")
+                   .gsub("sl_chat/", "#{discord_prefix}sl_chat/")
+    end
+
+    def user_logged_message
+      register_user_action("login", user.id)
+      user.update(telegram_id: msg.chat.id)
+      bot.api&.send_message(chat_id: msg.chat.id, text: success_messages[:user_logged_in])
+      evnt&.respond success_messages[:user_logged_in] if evnt.present?
+    end
+
+    def invalid_command_error_message
+      register_user_action("invalid_command", user.id) if user
+      register_visitor_action("invalid_command", visitor.id) if visitor&.tel_user.nil?
+      bot.api&.send_message(chat_id: msg.chat.id, text: error_messages[:invalid_command])
+      evnt&.respond error_messages[:invalid_command]
+    end
+
+    def user_not_found_error_message
+      register_visitor_action("error: user not found", visitor.id)
+      bot.api&.send_message(chat_id: msg.chat.id, text: error_messages[:user_not_found]) if bot.api.present?
+      evnt&.respond error_messages[:user_not_found] if evnt.present?
+    end
+
+    def user_created_message
+      register_visitor_action("user_created", visitor.id)
+      bot.api&.send_message(chat_id: msg.chat.id, text: success_messages[:user_created]) if bot.api.present?
+      evnt&.respond success_messages[:user_created] if evnt.present?
+    end
+
+    def user_creation_error_message
+      register_visitor_action("error: user creation", visitor.id)
+      bot.api&.send_message(chat_id: msg.chat.id, text: error_messages[:user_creation]) if bot.api.present?
+      evnt&.respond error_messages[:user_creation] if evnt.present?
+    end
+
+    def chat_created_message(chat)
+      register_user_action("chat_created", user.id) if user
+      user&.update(current_chat_id: chat.id)
+      bot.api&.send_message(chat_id: msg.chat.id, text: success_messages[:chat_created]) if bot.api.present?
+    end
+
+    def not_logged_in_message
+      register_visitor_action("error: not logged in", visitor.id) if visitor_user?
+      bot.api&.send_message(chat_id: msg.chat.id, text: error_messages[:user_not_logged_in])
+      evnt&.respond(error_messages[:user_not_logged_in])
+    end
+
+    def wrong_password_error_message
+      register_visitor_action("error: wrong password", visitor.id) if visitor_user?
+      bot.api&.send_message(chat_id: msg.chat.id, text: error_messages[:password])
+      evnt&.respond(error_messages[:password])
+    end
+
+    def chat_not_found_message
+      register_visitor_action("error: visitors cant select chat", visitor.id) if visitor_user?
+      register_user_action("error: chat not found", user.id) if user
+      bot.api&.send_message(chat_id: msg.chat.id, text: error_messages[:chat_not_found])
+      evnt&.respond(error_messages[:chat_not_found])
+    end
+
+    def no_chat_selected_message
+      register_visitor_action("error: visitors cant select chat", visitor.id) if visitor_user?
+      register_user_action("error: no chat selected", user.id) if user
+      bot&.api&.send_message(chat_id: msg.chat.id, text: error_messages[:no_chat_selected])
+      evnt&.respond(error_messages[:no_chat_selected])
+    end
+
+    def no_messages_founded_message
+      register_visitor_action("error: visitors cant view the chat history", visitor.id) if visitor_user?
+      register_user_action("error: no messages founded", user.id) if user
+      bot&.api&.send_message(chat_id: msg.chat.id, text: error_messages[:no_messages_founded])
+      evnt&.respond(error_messages[:no_messages_founded])
+    end
+
+    def chat_creation_failed_message
+      register_visitor_action("error: chat creation failed", visitor.id) if visitor_user?
+      register_user_action("error: chat creation failed", user.id) if user
+      bot&.api&.send_message(chat_id: msg.chat.id, text: error_messages[:chat_creation_failed])
+      evnt&.respond(error_messages[:chat_creation_failed])
+    end
+
+    def user_logged_in_message
+      register_visitor_action("user_logged_in", visitor.id) if visitor
+      bot.api&.send_message(chat_id: msg.chat.id, text: success_messages[:user_logged_in])
+      evnt&.respond(success_messages[:user_logged_in])
+    end
+  end
+end

data/lib/bots/helpers/search_helper.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module ChatgptAssistant
+  # Helper for search
+  module SearchHelper
+    def find_visitor(telegram_id: nil, discord_id: nil)
+      if telegram_id
+        Visitor.find_by(telegram_id: telegram_id)
+      elsif discord_id
+        Visitor.find_by(discord_id: discord_id)
+      end
+    end
+
+    def find_user(telegram_id: nil, discord_id: nil, email: nil)
+      if telegram_id
+        User.find_by(telegram_id: telegram_id)
+      elsif discord_id
+        User.find_by(discord_id: discord_id)
+      elsif email
+        User.find_by(email: email)
+      end
+    end
+
+    def where_user(telegram_id: nil, discord_id: nil, email: nil)
+      if telegram_id
+        User.where(telegram_id: telegram_id)
+      elsif discord_id
+        User.where(discord_id: discord_id)
+      elsif email
+        User.where(email: email)
+      end
+    end
+  end
+end

data/lib/bots/helpers/telegram_helper.rb

@@ -0,0 +1,89 @@
+# frozen_string_literal: true
+
+module ChatgptAssistant
+  # Helper for telegram
+  module TelegramHelper
+    def user
+      @user = find_user(telegram_id: msg.chat.id)
+    end
+
+    def bot
+      @bot ||= Telegram::Bot::Client.new(telegram_token)
+    end
+
+    def telegram_chat_event
+      user ? chat_if_exists : not_logged_in_message
+    end
+
+    def telegram_audio_info
+      bot.api.get_file(file_id: telegram_audio.file_id)
+    end
+
+    def telegram_audio_url
+      "https://api.telegram.org/file/bot#{telegram_token}/#{telegram_audio_info["result"]["file_path"]}"
+    end
+
+    def telegram_audio
+      msg.audio || msg.voice
+    end
+
+    def telegram_process_ai_voice(user_audio)
+      ai_text = chatter.chat(user_audio["text"], user.current_chat_id)
+      ai_voice = synthesis.synthesize_text(ai_text)
+      {
+        voice: ai_voice,
+        text: ai_text
+      }
+    end
+
+    def telegram_send_voice_message(voice: nil, text: nil)
+      messages = parse_message text, 4096
+      bot.api.send_voice(chat_id: msg.chat.id, voice: Faraday::UploadIO.new(voice, "audio/mp3"))
+      messages.each { |message| send_message message, msg.chat.id }
+    end
+
+    def telegram_user_create(visitor_id, email, password)
+      visitor = Visitor.find_by(id: visitor_id)
+      return false unless visitor
+
+      visitor.tel_user.update(telegram_id: nil) if visitor.tel_user.present?
+      user = User.new(telegram_id: visitor.telegram_id, email: email, password: password)
+      user.save ? user.email : user.errors.full_messages
+    end
+
+    def telegram_send_start_message
+      send_message commom_messages[:start], msg.chat.id
+      help_message = help_messages.join("\n").to_s
+      send_message help_message, msg.chat.id
+      send_message commom_messages[:start_helper], msg.chat.id
+      send_message commom_messages[:start_sec_helper], msg.chat.id
+    end
+
+    def telegram_create_chat
+      text = msg.text
+      title = text.split("/").last
+      chat = Chat.new(user_id: user.id, status: 0, title: title)
+      chat.save ? chat_created_message(chat) : chat_creation_failed_message
+    end
+
+    def telegram_user_history
+      user.current_chat.messages.last(10).map { |m| "#{m.role}: #{m.content}\nat: #{m.created_at}" }
+    end
+
+    def telegram_text_or_audio?
+      msg.respond_to?(:text) || msg.respond_to?(:audio) || msg.respond_to?(:voice)
+    end
+
+    def telegram_message_has_text?
+      msg.text.present?
+    end
+
+    def telegram_message_has_audio?
+      msg.audio.present? || msg.voice.present?
+    end
+
+    def telegram_actions?
+      msg.text.include?("new_chat/") || msg.text.include?("sl_chat/") || msg.text.include?("login/") || msg.text.include?("register/")
+    end
+  end
+end

data/lib/bots/helpers/validation_helper.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module ChatgptAssistant
+  # This module is responsible for the validation of the bot actions
+  module ValidationHelper
+    def valid_for_list_action?
+      not_logged_in_message if user.nil?
+      chat_not_found_message if user.chats.count.zero?
+      !user.nil? && user.chats.count.positive?
+    end
+
+    def chat_if_exists
+      chat = Chat.find_by(user_id: user.id, id: user.current_chat_id)
+      chat ? chat_success(chat.id) : no_chat_selected_message
+    end
+
+    def visitor_user?
+      visitor&.tel_user.nil? && visitor&.dis_user.nil?
+    end
+
+    def discord_voice_bot_disconnected?
+      user && evnt.user.voice_channel && !evnt.voice && !chat.nil?
+    end
+
+    def discord_voice_bot_connected?
+      user && evnt.user.voice_channel && evnt.voice && !chat.nil?
+    end
+  end
+end

data/lib/bots/helpers/visit_helper.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module ChatgptAssistant
+  # Helper for visit
+  module VisitHelper
+    def telegram_visited?(chat_id)
+      visitor = Visitor.find_by(telegram_id: chat_id, name: msg.from.first_name)
+      if visitor.nil?
+        Visitor.create(telegram_id: chat_id, name: msg.from.first_name)
+      else
+        visitor
+      end
+    end
+
+    def discord_visited?(user_id)
+      visitor = Visitor.find_by(discord_id: user_id, name: evnt.user.name)
+      if visitor.nil?
+        Visitor.create(discord_id: user_id, name: evnt.user.name)
+      else
+        visitor
+      end
+    end
+  end
+end