chore: APIs for add and remove documents

chatwoot · sojan-official · Jul 21, 2023 · May 22, 2023 · May 29, 2023 · Mar 20, 2023
commit b7c7b0978b35fb3c78675cb6d88421e6aa213385
diff --git a/app/controllers/api/v1/accounts/response_documents_controller.rb b/app/controllers/api/v1/accounts/response_documents_controller.rb
@@ -0,0 +1,22 @@
+class Api::V1::Accounts::ResponseSourcesController < Api::V1::Accounts::BaseController
+  before_action :current_account
+  before_action :check_authorization
+
+  def parse
+    links = PageCrawlerService.new(params[:link]).get_links
+    render json: { links: links }
+  end
+
+  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/controllers/api/v1/accounts/response_sources_controller.rb b/app/controllers/api/v1/accounts/response_sources_controller.rb
@@ -1,6 +1,7 @@
 class Api::V1::Accounts::ResponseSourcesController < Api::V1::Accounts::BaseController
   before_action :current_account
   before_action :check_authorization
+  before_action :find_response_source, only: [:add_document, :remove_document]
 
   def parse
     links = PageCrawlerService.new(params[:link]).get_links
@@ -13,8 +14,22 @@ def create
     @response_source
   end
 
+  def add_document
+    @response_source.response_documents.create!(document_link: params[:document_link])
+    @response_source
+  end
+
+  def remove_document
+    @response_source.response_documents.find(params[:document_id]).destroy!
+    @response_source
+  end
+
   private
 
+  def find_response_source
+    @response_source = Current.account.response_sources.find(params[:id])
+  end
+
   def response_source_params
     params.require(:response_source).permit(:name, :source_link, :inbox_id,
                                             response_documents_attributes: [:document_link])

diff --git a/app/models/response_source.rb b/app/models/response_source.rb
@@ -20,6 +20,7 @@
 class ResponseSource < ApplicationRecord
   enum source_type: { external: 0, kbase: 1, inbox: 2 }
   belongs_to :account
+  belongs_to :inbox
   has_many :response_documents, dependent: :destroy
   has_many :responses, through: :response_documents
 

diff --git a/app/policies/response_source_policy.rb b/app/policies/response_source_policy.rb
@@ -6,4 +6,12 @@ def parse?
   def create?
     @account_user.administrator?
   end
+
+  def add_document?
+    @account_user.administrator?
+  end
+
+  def remove_document?
+    @account_user.administrator?
+  end
 end
diff --git a/app/views/api/v1/accounts/response_sources/add_document.json.jbuilder b/app/views/api/v1/accounts/response_sources/add_document.json.jbuilder
@@ -0,0 +1,5 @@
+json.id @label.id
+json.title @label.title
+json.description @label.description
+json.color @label.color
+json.show_on_sidebar @label.show_on_sidebar
diff --git a/app/views/api/v1/accounts/response_sources/create.json.jbuilder b/app/views/api/v1/accounts/response_sources/create.json.jbuilder
@@ -0,0 +1,5 @@
+json.id @label.id
+json.title @label.title
+json.description @label.description
+json.color @label.color
+json.show_on_sidebar @label.show_on_sidebar
diff --git a/app/views/api/v1/accounts/response_sources/remove_document.json.jbuilder b/app/views/api/v1/accounts/response_sources/remove_document.json.jbuilder
@@ -0,0 +1,5 @@
+json.id @label.id
+json.title @label.title
+json.description @label.description
+json.color @label.color
+json.show_on_sidebar @label.show_on_sidebar
diff --git a/config/routes.rb b/config/routes.rb
@@ -155,6 +155,10 @@
             collection do
               post :parse
             end
+            member do
+              post :add_document
+              post :remove_document
+            end
           end
 
           resources :notifications, only: [:index, :update] do

diff --git a/spec/controllers/api/v1/accounts/response_sources_controller_spec.rb b/spec/controllers/api/v1/accounts/response_sources_controller_spec.rb
@@ -2,6 +2,7 @@
 
 RSpec.describe 'Response Sources API', type: :request do
   let!(:account) { create(:account) }
+  let!(:admin) { create(:user, account: account, role: :administrator) }
   let!(:inbox) { create(:inbox, account: account) }
 
   describe 'POST /api/v1/accounts/{account.id}/response_sources/parse' do
@@ -13,15 +14,13 @@
 
     context 'when it is an unauthenticated user' do
       it 'returns unauthorized' do
-        expect { post "/api/v1/accounts/#{account.id}/response_sources/parse", params: valid_params }.not_to change(Label, :count)
+        post "/api/v1/accounts/#{account.id}/response_sources/parse", params: valid_params
 
         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 'returns links in the webpage' do
         crawler = double
         allow(PageCrawlerService).to receive(:new).and_return(crawler)
@@ -52,15 +51,13 @@
 
     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 { post "/api/v1/accounts/#{account.id}/response_sources", params: valid_params }.not_to change(ResponseSource, :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 response sources and documents' do
         expect do
           post "/api/v1/accounts/#{account.id}/response_sources", headers: admin.create_new_auth_token,
@@ -72,4 +69,63 @@
       end
     end
   end
+
+  describe 'POST /api/v1/accounts/{account.id}/response_sources/{response_source.id}/add_document' do
+    let!(:response_source) { create(:response_source, account: account, inbox: inbox) }
+    let(:valid_params) do
+      { document_link: 'http://test.test' }
+    end
+
+    context 'when it is an unauthenticated user' do
+      it 'returns unauthorized' do
+        expect do
+          post "/api/v1/accounts/#{account.id}/response_sources/#{response_source.id}/add_document",
+               params: valid_params
+        end.not_to change(ResponseDocument, :count)
+
+        expect(response).to have_http_status(:unauthorized)
+      end
+    end
+
+    context 'when it is an authenticated user' do
+      it 'creates the response sources and documents' do
+        expect do
+          post "/api/v1/accounts/#{account.id}/response_sources/#{response_source.id}/add_document", headers: admin.create_new_auth_token,
+                                                                                                     params: valid_params
+        end.to change(ResponseDocument, :count).by(1)
+        expect(response).to have_http_status(:success)
+      end
+    end
+  end
+
+  describe 'POST /api/v1/accounts/{account.id}/response_sources/{response_source.id}/remove_document' do
+    let!(:response_source) { create(:response_source, account: account, inbox: inbox) }
+    let!(:response_document) { response_source.response_documents.create!(document_link: 'http://test.test') }
+    let(:valid_params) do
+      { document_id: response_document.id }
+    end
+
+    context 'when it is an unauthenticated user' do
+      it 'returns unauthorized' do
+        expect do
+          post "/api/v1/accounts/#{account.id}/response_sources/#{response_source.id}/remove_document",
+               params: valid_params
+        end.not_to change(ResponseDocument, :count)
+
+        expect(response).to have_http_status(:unauthorized)
+      end
+    end
+
+    context 'when it is an authenticated user' do
+      it 'creates the response sources and documents' do
+        expect do
+          post "/api/v1/accounts/#{account.id}/response_sources/#{response_source.id}/remove_document", headers: admin.create_new_auth_token,
+                                                                                                        params: valid_params
+        end.to change(ResponseDocument, :count).by(-1)
+        expect(response).to have_http_status(:success)
+
+        expect { response_document.reload }.to raise_error(ActiveRecord::RecordNotFound)
+      end
+    end
+  end
 end
diff --git a/spec/factories/response_source.rb b/spec/factories/response_source.rb
@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :response_source do
+    name { Faker::Name.name }
+    source_link { Faker::Internet.url }
+    account
+  end
+end