chore: endpoint to create sources

chatwoot · sojan-official · Jul 21, 2023 · May 22, 2023 · May 29, 2023 · Mar 20, 2023
commit dc01127857d7fc7348b9e8a80bfc31ae16c1f570
diff --git a/app/controllers/api/v1/accounts/response_sources_controller.rb b/app/controllers/api/v1/accounts/response_sources_controller.rb
@@ -0,0 +1,17 @@
+class Api::V1::Accounts::ResponseSourcesController < Api::V1::Accounts::BaseController
+  before_action :current_account
+  before_action :check_authorization
+
+  def create
+    @response_source = Current.account.response_sources.new(response_source_params)
+    @response_source.save!
+    @response_source
+  end
+
+  private
+
+  def response_source_params
+    params.require(:response_source).permit(:name, :source_link, :inbox_id,
+                                            response_documents_attributes: [:document_link])
+  end
+end
diff --git a/app/jobs/response_source_job.rb b/app/jobs/response_source_job.rb
diff --git a/app/models/account.rb b/app/models/account.rb
@@ -65,6 +65,9 @@ class Account < ApplicationRecord
   has_many :notification_settings, dependent: :destroy_async
   has_many :notifications, dependent: :destroy_async
   has_many :portals, dependent: :destroy_async, class_name: '::Portal'
+  has_many :response_sources, dependent: :destroy_async
+  has_many :response_documents, dependent: :destroy_async
+  has_many :responses, dependent: :destroy_async
   has_many :sms_channels, dependent: :destroy_async, class_name: '::Channel::Sms'
   has_many :teams, dependent: :destroy_async
   has_many :telegram_bots, dependent: :destroy_async

diff --git a/app/models/inbox.rb b/app/models/inbox.rb
@@ -70,6 +70,7 @@ class Inbox < ApplicationRecord
   has_one :agent_bot, through: :agent_bot_inbox
   has_many :webhooks, dependent: :destroy_async
   has_many :hooks, dependent: :destroy_async, class_name: 'Integrations::Hook'
+  has_many :response_sources, dependent: :destroy_async
 
   enum sender_name_type: { friendly: 0, professional: 1 }
 

diff --git a/app/models/response_document.rb b/app/models/response_document.rb
@@ -14,19 +14,23 @@
 #
 # Indexes
 #
-#  index_response_documents_on_account_id_and_document_link  (account_id,document_link) UNIQUE
-#  index_response_documents_on_document                      (document_type,document_id)
-#  index_response_documents_on_response_source_id            (response_source_id)
+#  index_response_documents_on_document            (document_type,document_id)
+#  index_response_documents_on_response_source_id  (response_source_id)
 #
 class ResponseDocument < ApplicationRecord
   has_many :responses, dependent: :destroy
   belongs_to :account
   belongs_to :response_source
 
+  before_validation :set_account
   after_update :handle_content_change
 
   private
 
+  def set_account
+    self.account = response_source.account
+  end
+
   def handle_content_change
     return unless saved_change_to_content? && content_was.nil? && !content.nil?
 

diff --git a/app/models/response_source.rb b/app/models/response_source.rb
@@ -10,24 +10,18 @@
 #  created_at        :datetime         not null
 #  updated_at        :datetime         not null
 #  account_id        :bigint           not null
+#  inbox_id          :bigint           not null
 #  source_model_id   :bigint
 #
 # Indexes
 #
-#  index_response_sources_on_account_id_and_source_link  (account_id,source_link) UNIQUE
-#  index_response_sources_on_source_model                (source_model_type,source_model_id)
+#  index_response_sources_on_source_model  (source_model_type,source_model_id)
 #
 class ResponseSource < ApplicationRecord
   enum source_type: { external: 0, kbase: 1, inbox: 2 }
   belongs_to :account
   has_many :response_documents, dependent: :destroy
   has_many :responses, through: :response_documents
 
-  after_save :process_response_source
-
-  private
-
-  def process_response_source
-    ResponseSourceJob.perform_later(id)
-  end
+  accepts_nested_attributes_for :response_documents
 end
diff --git a/app/policies/response_source_policy.rb b/app/policies/response_source_policy.rb
@@ -0,0 +1,5 @@
+class ResponseSourcePolicy < ApplicationPolicy
+  def create?
+    @account_user.administrator?
+  end
+end
diff --git a/config/routes.rb b/config/routes.rb
@@ -151,6 +151,7 @@
             end
           end
           resources :labels, only: [:index, :show, :create, :update, :destroy]
+          resources :response_sources, only: [:create]
 
           resources :notifications, only: [:index, :update] do
             collection do

diff --git a/db/migrate/20230709113830_create_responses.rb b/db/migrate/20230709113830_create_responses.rb
@@ -14,10 +14,9 @@ def create_response_sources
       t.string :source_link
       t.references :source_model, polymorphic: true
       t.bigint :account_id, null: false
+      t.bigint :inbox_id, null: false
       t.timestamps
     end
-
-    add_index :response_sources, [:account_id, :source_link], unique: true
   end
 
   def create_response_documents
@@ -30,7 +29,6 @@ def create_response_documents
       t.timestamps
     end
 
-    add_index :response_documents, [:account_id, :document_link], unique: true
     add_index :response_documents, :response_source_id
   end
 
@@ -40,9 +38,11 @@ def create_responses
       t.string :question, null: false
       t.text :answer, null: false
       t.bigint :account_id, null: false
+      t.vector :embedding, limit: 1536
       t.timestamps
     end
 
     add_index :responses, :response_document_id
+    add_index :responses, :embedding, using: :ivfflat, opclass: :vector_l2_ops
   end
 end
diff --git a/db/migrate/20230710155231_add_embeddding_to_responses.rb b/db/migrate/20230710155231_add_embeddding_to_responses.rb
diff --git a/db/schema.rb b/db/schema.rb
@@ -10,7 +10,7 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema[7.0].define(version: 2023_07_10_155231) do
+ActiveRecord::Schema[7.0].define(version: 2023_07_09_113830) do
   # These are extensions that must be enabled in order to support this database
   enable_extension "pg_stat_statements"
   enable_extension "pg_trgm"
@@ -823,7 +823,6 @@
     t.bigint "account_id", null: false
     t.datetime "created_at", null: false
     t.datetime "updated_at", null: false
-    t.index ["account_id", "document_link"], name: "index_response_documents_on_account_id_and_document_link", unique: true
     t.index ["document_type", "document_id"], name: "index_response_documents_on_document"
     t.index ["response_source_id"], name: "index_response_documents_on_response_source_id"
   end
@@ -835,9 +834,9 @@
     t.string "source_model_type"
     t.bigint "source_model_id"
     t.bigint "account_id", null: false
+    t.bigint "inbox_id", null: false
     t.datetime "created_at", null: false
     t.datetime "updated_at", null: false
-    t.index ["account_id", "source_link"], name: "index_response_sources_on_account_id_and_source_link", unique: true
     t.index ["source_model_type", "source_model_id"], name: "index_response_sources_on_source_model"
   end
 
@@ -846,9 +845,9 @@
     t.string "question", null: false
     t.text "answer", null: false
     t.bigint "account_id", null: false
+    t.vector "embedding", limit: 1536
     t.datetime "created_at", null: false
     t.datetime "updated_at", null: false
-    t.vector "embedding", limit: 1536
     t.index ["embedding"], name: "index_responses_on_embedding", using: :ivfflat
     t.index ["response_document_id"], name: "index_responses_on_response_document_id"
   end

diff --git a/spec/controllers/api/v1/accounts/response_sources_controller_spec.rb b/spec/controllers/api/v1/accounts/response_sources_controller_spec.rb
@@ -0,0 +1,44 @@
+require 'rails_helper'
+
+RSpec.describe 'Response Sources API', type: :request do
+  let!(:account) { create(:account) }
+  let!(:inbox) { create(:inbox, account: account) }
+
+  describe 'POST /api/v1/accounts/{account.id}/response_sources' do
+    let(:valid_params) do
+      {
+        response_source: {
+          name: 'Test',
+          source_link: 'http://test.test',
+          inbox_id: inbox.id,
+          response_documents_attributes: [
+            { document_link: 'http://test1.test' },
+            { document_link: 'http://test2.test' }
+          ]
+        }
+      }
+    end
+
+    context 'when it is an unauthenticated user' do
+      it 'returns unauthorized' do
+        expect { post "/api/v1/accounts/#{account.id}/response_sources", params: valid_params }.not_to change(Label, :count)
+
+        expect(response).to have_http_status(:unauthorized)
+      end
+    end
+
+    context 'when it is an authenticated user' do
+      let(:admin) { create(:user, account: account, role: :administrator) }
+
+      it 'creates the contact' do
+        expect do
+          post "/api/v1/accounts/#{account.id}/response_sources", headers: admin.create_new_auth_token,
+                                                                  params: valid_params
+        end.to change(ResponseSource, :count).by(1)
+
+        expect(ResponseDocument.count).to eq(2)
+        expect(response).to have_http_status(:success)
+      end
+    end
+  end
+end