[WIP] Convert image links to shortened URLs

app/controllers/users_controller.rb

@@ -131,5 +131,4 @@ def allow_iframe_request
     # cannot-display-my-rails-4-app-in-iframe-even-if-x-frame-options-is-allowall
   end
 
-
 end

app/models/company.rb

@@ -8,6 +8,8 @@ class Company < ApplicationRecord
 
   include Dinkurs::Errors
 
+  include Rails.application.routes.url_helpers
+
   before_destroy :destroy_checks
 
   validates_presence_of :company_number
@@ -177,6 +179,19 @@ def se_mailing_csv_str
       AddressExporter.se_mailing_csv_str( main_address )
   end
 
+  def get_short_h_brand_url
+    found = self.short_h_brand_url
+    return found if found
+    url = company_h_brand_url(0, company_id: self.id)
+    short_url = ShortenUrl.short(url)
+    if short_url
+      self.update_attribute(:short_h_brand_url, short_url)
+      short_url
+    else
+      url
+    end
+  end
+
 
   private
 

app/models/user.rb

@@ -1,6 +1,8 @@
 class User < ApplicationRecord
   include PaymentUtility
 
+  include Rails.application.routes.url_helpers
+
   # Include default devise modules. Others available are:
   # :confirmable, :lockable, :timeoutable and :omniauthable
   devise :database_authenticatable, :registerable,
@@ -117,6 +119,19 @@ def grant_membership(send_email: true)
     Arel.sql("lpad(membership_number, 20, '0')")
   end
 
+  def get_short_proof_of_membership_url
+    found = self.short_proof_of_membership_url
+    return found if found
+    url = proof_of_membership_url(self.id)
+    short_url = ShortenUrl.short(url)
+    if short_url
+      self.update_attribute(:short_proof_of_membership_url, short_url)
+      short_url
+    else
+      url
+    end
+  end
+
   private
 
   def issue_membership_number

app/services/shorten_url.rb

@@ -0,0 +1,20 @@
+class ShortenUrl
+
+  SHORTEN_URL_LOG = 'log/tinyurl.log'
+  SUCCESS_CODES = [200, 201, 202].freeze
+
+  def self.short(url)
+    response = HTTParty.get("http://tinyurl.com/api-create.php?url=#{url}")
+    raise HTTParty::Error if response.match?(/error/i) || !SUCCESS_CODES.include?(response.code)
+    response
+  rescue HTTParty::Error => error
+    ActivityLogger.open(SHORTEN_URL_LOG, 'TINYURL_API', 'shortening url', false) do |log|
+      log.record('error', "Exception: #{error.message}")
+      log.record('error', "Attempted URL: #{url}")
+      log.record('error', "Response body: #{response.body}")
+      log.record('error', "HTTP code: #{response.code}")
+    end
+    nil
+  end
+end
+

app/views/users/_company_h_brand.html.haml

@@ -35,7 +35,7 @@
         %br
         %br
         = t('users.show.use_this_image_link_html')
-        = company_h_brand_url(user, company_id: company.id)
+        = company.get_short_h_brand_url
 
       :javascript
         window.onload = function(){

app/views/users/_proof_of_membership.html.haml

@@ -57,7 +57,7 @@
         %br
         %br
         = t('users.show.use_this_image_link_html')
-        = proof_of_membership_url(user)
+        = user.get_short_proof_of_membership_url
 
       :javascript
         window.onload = function(){

config/application.rb

@@ -61,6 +61,7 @@ class Application < Rails::Application
 
     # Default for development and testing.  production.rb overrides this
     config.action_mailer.default_url_options = { host: 'localhost', port: '3000' }
+    routes.default_url_options = { host: 'localhost', port: '3000' }
 
     # Ensure this is false by default (to be secure).
     # Only change in development or test environments where really needed

config/environments/production.rb

@@ -81,6 +81,16 @@
   #
   ###
 
+  ###
+  #
+  # Routes
+  #
+
+  # Needed to create shortened urls in the ShortLinksContainer model.
+  routes.default_url_options = { host: ENV['DEFAULT_HOST'] }
+
+  #
+  ###
 
   # Enable locale fallbacks for I18n (makes lookups for any locale fall back to
   # the I18n.default_locale when a translation cannot be found).

db/migrate/20180717043851_add_short_h_brand_url_to_companies.rb

@@ -0,0 +1,5 @@
+class AddShortHBrandUrlToCompanies < ActiveRecord::Migration[5.1]
+  def change
+    add_column :companies, :short_h_brand_url, :string
+  end
+end

db/migrate/20180719021503_add_short_proof_of_membership_url_to_users.rb

@@ -0,0 +1,5 @@
+class AddShortProofOfMembershipUrlToUsers < ActiveRecord::Migration[5.1]
+  def change
+    add_column :users, :short_proof_of_membership_url, :string
+  end
+end

db/schema.rb

@@ -10,7 +10,7 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema.define(version: 20180428103625) do
+ActiveRecord::Schema.define(version: 20180719021503) do
 
   # These are extensions that must be enabled in order to support this database
   enable_extension "plpgsql"
@@ -94,6 +94,8 @@
     t.datetime "updated_at", null: false
     t.text "description"
     t.string "dinkurs_company_id"
+    t.boolean "show_dinkurs_events"
+    t.string "short_h_brand_url"
     t.index ["company_number"], name: "index_companies_on_company_number", unique: true
   end
 
@@ -229,6 +231,7 @@
     t.string "member_photo_content_type"
     t.integer "member_photo_file_size"
     t.datetime "member_photo_updated_at"
+    t.string "short_proof_of_membership_url"
     t.index ["email"], name: "index_users_on_email", unique: true
     t.index ["membership_number"], name: "index_users_on_membership_number", unique: true
     t.index ["reset_password_token"], name: "index_users_on_reset_password_token", unique: true
@@ -240,9 +243,6 @@
   add_foreign_key "company_applications", "companies"
   add_foreign_key "company_applications", "shf_applications"
   add_foreign_key "events", "companies"
-  add_foreign_key "company_applications", "companies"
-  add_foreign_key "company_applications", "shf_applications"
-  add_foreign_key "events", "companies"
   add_foreign_key "payments", "companies"
   add_foreign_key "payments", "users"
   add_foreign_key "shf_applications", "member_app_waiting_reasons", column: "member_app_waiting_reasons_id"

db/structure.sql

@@ -239,7 +239,8 @@ CREATE TABLE public.companies (
     updated_at timestamp without time zone NOT NULL,
     description text,
     dinkurs_company_id character varying,
-    show_dinkurs_events boolean
+    show_dinkurs_events boolean,
+    short_h_brand_url character varying
 );
 
 
@@ -700,7 +701,8 @@ CREATE TABLE public.users (
     member_photo_file_name character varying,
     member_photo_content_type character varying,
     member_photo_file_size integer,
-    member_photo_updated_at timestamp without time zone
+    member_photo_updated_at timestamp without time zone,
+    short_proof_of_membership_url character varying
 );
 
 
@@ -1318,6 +1320,8 @@ INSERT INTO "schema_migrations" (version) VALUES
 ('20180326103433'),
 ('20180328105100'),
 ('20180428103625'),
-('20180624155644');
+('20180624155644'),
+('20180717043851'),
+('20180719021503');
 
 

spec/factories/companies.rb

@@ -16,6 +16,7 @@
     phone_number '123123123'
     email 'thiscompany@example.com'
     website 'http://www.example.com'
+    short_h_brand_url nil
 
     transient do
       num_addresses 1

spec/factories/users.rb

@@ -8,6 +8,7 @@
     password 'my_password'
     admin false
     member false
+    short_proof_of_membership_url nil
     member_photo do
       File.new("#{Rails.root}/spec/fixtures/member_photos/photo_unavailable.png")
     end
@@ -44,4 +45,5 @@
     end
   end
 
+
 end

spec/models/company_spec.rb

@@ -5,6 +5,12 @@
 
 RSpec.describe Company, type: :model do
 
+  include Rails.application.routes.url_helpers
+
+  let(:with_short_h_brand_url) do
+    create(:company, short_h_brand_url: 'http://www.tinyurl.com/hbrand')
+  end
+
   let(:no_name) do
     create(:company, name: '', company_number: '2120000142')
   end
@@ -94,6 +100,7 @@
     it { is_expected.to have_db_column :description }
     it { is_expected.to have_db_column :dinkurs_company_id }
     it { is_expected.to have_db_column :show_dinkurs_events }
+    it { is_expected.to have_db_column :short_h_brand_url }
   end
 
   describe 'Validations' do
@@ -697,4 +704,25 @@
     end
   end
 
+  describe '#get_short_h_brand_url' do
+    context 'there is already a shortened url in the table' do
+      it 'returns shortened url' do
+        expect(with_short_h_brand_url.get_short_h_brand_url).to eq('http://www.tinyurl.com/hbrand')
+      end
+    end
+    context 'there is no shortened url in the table and ShortenUrl.short is called' do
+      it 'saves the result if the result is not nil and returns shortened url' do
+        url = company_h_brand_url(0, company_id: complete_co.id)
+        allow(ShortenUrl).to receive(:short).with(url).and_return('http://tinyurl.com/hbrand2')
+        expect(complete_co.get_short_h_brand_url).to eq(ShortenUrl.short(url))
+        expect(complete_co.short_h_brand_url).to eq(ShortenUrl.short(url))
+      end
+      it 'does not save anything if the result is nil and returns unshortened url' do
+        url = company_h_brand_url(0, company_id: complete_co.id)
+        allow(ShortenUrl).to receive(:short).with(url).and_return(nil)
+        expect(complete_co.get_short_h_brand_url).to eq(url)
+        expect(complete_co.short_h_brand_url).to eq(nil)
+      end
+    end
+  end
 end

spec/models/user_spec.rb

@@ -2,10 +2,17 @@
 require 'email_spec/rspec'
 
 RSpec.describe User, type: :model do
+
+  include Rails.application.routes.url_helpers
+
   let(:user) { create(:user) }
   let(:user_with_app) { create(:user_with_membership_app) }
   let(:member) { create(:member_with_membership_app) }
 
+  let(:with_short_proof_of_membership_url) do
+    create(:user, short_proof_of_membership_url: 'http://www.tinyurl.com/proofofmembership')
+  end
+
   let(:application) do
     create(:shf_application, user: user, state: :accepted)
   end
@@ -535,4 +542,26 @@
       end
     end
   end
+
+  describe '#get_short_proof_of_membership_url' do
+    context 'there is already a shortened url in the table' do
+      it 'returns shortened url' do
+        expect(with_short_proof_of_membership_url.get_short_proof_of_membership_url).to eq('http://www.tinyurl.com/proofofmembership')
+      end
+    end
+    context 'there is no shortened url in the table and ShortenUrl.short is called' do
+      it 'saves the result if the result is not nil and returns shortened url' do
+        url = proof_of_membership_url(user.id)
+        allow(ShortenUrl).to receive(:short).with(url).and_return('http://tinyurl.com/proofofmembership2')
+        expect(user.get_short_proof_of_membership_url).to eq(ShortenUrl.short(url))
+        expect(user.short_proof_of_membership_url).to eq(ShortenUrl.short(url))
+      end
+      it 'does not save anything if the result is nil and returns unshortened url' do
+        url = proof_of_membership_url(user.id)
+        allow(ShortenUrl).to receive(:short).with(url).and_return(nil)
+        expect(user.get_short_proof_of_membership_url).to eq(url)
+        expect(user.short_proof_of_membership_url).to eq(nil)
+      end
+    end
+  end
 end

spec/services/shorten_url_spec.rb

@@ -0,0 +1,19 @@
+require 'rails_helper'
+
+describe ShortenUrl do
+  describe '.short' do
+    it 'creates shortened link' do
+      VCR.use_cassette('shorten_url/short') do
+        shortened_url = ShortenUrl.short 'http://sverigeshundforetagare.se/anvandare/2/proof_of_membership'
+        expect(shortened_url).to match 'tinyurl.com'
+      end
+    end
+    it 'if the service raises an error, returns nil and writes to the log' do
+      VCR.use_cassette('shorten_url/short') do
+        expect(ActivityLogger).to receive(:open)
+        shortened_url = ShortenUrl.short '/'
+        expect(shortened_url).to eq nil
+      end
+    end
+  end
+end