From 297c11dba2864b20992cd3f98282f5ce35d5d144 Mon Sep 17 00:00:00 2001
From: Matt Jankowski <mjankowski@thoughtbot.com>
Date: Tue, 18 Apr 2017 16:20:12 -0400
Subject: [PATCH] Language detection refactor (#2099)

* Extract detect_language to separate class

* Use default locale, not just en

* Add spec to confirm that whatlanguage cant identify empty string

* Allow account locale to override default in language detector

* PostStatusService supplies an account to detect language
---
 app/lib/language_detector.rb              | 20 +++++++
 app/services/post_status_service.rb       |  6 +-
 spec/lib/language_detector_spec.rb        | 71 +++++++++++++++++++++++
 spec/services/post_status_service_spec.rb | 12 ++++
 4 files changed, 106 insertions(+), 3 deletions(-)
 create mode 100644 app/lib/language_detector.rb
 create mode 100644 spec/lib/language_detector_spec.rb

diff --git a/app/lib/language_detector.rb b/app/lib/language_detector.rb
new file mode 100644
index 0000000000..b6f81923b5
--- /dev/null
+++ b/app/lib/language_detector.rb
@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+class LanguageDetector
+  attr_reader :text, :account
+
+  def initialize(text, account = nil)
+    @text = text
+    @account = account
+  end
+
+  def to_iso_s
+    WhatLanguage.new(:all).language_iso(text) || default_locale.to_sym
+  end
+
+  private
+
+  def default_locale
+    account&.user&.locale || I18n.default_locale
+  end
+end
diff --git a/app/services/post_status_service.rb b/app/services/post_status_service.rb
index 00af28edde..6ce434a13b 100644
--- a/app/services/post_status_service.rb
+++ b/app/services/post_status_service.rb
@@ -19,7 +19,7 @@ class PostStatusService < BaseService
                                       sensitive: options[:sensitive],
                                       spoiler_text: options[:spoiler_text] || '',
                                       visibility: options[:visibility],
-                                      language: detect_language(text),
+                                      language: detect_language_for(text, account),
                                       application: options[:application])
 
     attach_media(status, media)
@@ -52,8 +52,8 @@ class PostStatusService < BaseService
     media.update(status_id: status.id)
   end
 
-  def detect_language(text)
-    WhatLanguage.new(:all).language_iso(text) || 'en'
+  def detect_language_for(text, account)
+    LanguageDetector.new(text, account).to_iso_s
   end
 
   def process_mentions_service
diff --git a/spec/lib/language_detector_spec.rb b/spec/lib/language_detector_spec.rb
new file mode 100644
index 0000000000..74b8b6c48d
--- /dev/null
+++ b/spec/lib/language_detector_spec.rb
@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+require 'rails_helper'
+
+describe LanguageDetector do
+  describe 'to_iso_s' do
+    it 'detects english language' do
+      string = 'Hello and welcome to mastadon'
+      result = described_class.new(string).to_iso_s
+
+      expect(result).to eq :en
+    end
+
+    it 'detects spanish language' do
+      string = 'Obtener un Hola y bienvenidos a Mastadon'
+      result = described_class.new(string).to_iso_s
+
+      expect(result).to eq :es
+    end
+
+    describe 'when language cant be detected' do
+      it 'confirm language engine cant detect' do
+        result = WhatLanguage.new(:all).language_iso('')
+        expect(result).to be_nil
+      end
+
+      describe 'with an account' do
+        it 'uses the account locale when present' do
+          user = double(:user, locale: 'fr')
+          account = double(:account, user: user)
+          result = described_class.new('', account).to_iso_s
+
+          expect(result).to eq :fr
+        end
+
+        it 'uses default locale when account is present but has no locale' do
+          user = double(:user, locale: nil)
+          account = double(:accunt, user: user)
+          result = described_class.new('', account).to_iso_s
+
+          expect(result).to eq :en
+        end
+      end
+
+      describe 'with an `en` default locale' do
+        it 'uses the default locale' do
+          string = ''
+          result = described_class.new(string).to_iso_s
+
+          expect(result).to eq :en
+        end
+      end
+
+      describe 'with a non-`en` default locale' do
+        around(:each) do |example|
+          before = I18n.default_locale
+          I18n.default_locale = :ja
+          example.run
+          I18n.default_locale = before
+        end
+
+        it 'uses the default locale' do
+          string = ''
+          result = described_class.new(string).to_iso_s
+
+          expect(result).to eq :ja
+        end
+      end
+    end
+  end
+end
diff --git a/spec/services/post_status_service_spec.rb b/spec/services/post_status_service_spec.rb
index 0e39cd969a..c9d80257ff 100644
--- a/spec/services/post_status_service_spec.rb
+++ b/spec/services/post_status_service_spec.rb
@@ -64,6 +64,18 @@ RSpec.describe PostStatusService do
     expect(status.application).to eq application
   end
 
+  it 'creates a status with a language set' do
+    detector = double(to_iso_s: :en)
+    allow(LanguageDetector).to receive(:new).and_return(detector)
+
+    account = Fabricate(:account)
+    text = 'test status text'
+
+    subject.call(account, text)
+
+    expect(LanguageDetector).to have_received(:new).with(text, account)
+  end
+
   it 'processes mentions' do
     mention_service = double(:process_mentions_service)
     allow(mention_service).to receive(:call)