From 3f075c7794ca7cc6e3c566591e9929859c6c325c Mon Sep 17 00:00:00 2001 From: Eugen Rochko Date: Sun, 29 Jan 2017 01:30:32 +0100 Subject: [PATCH] API for apps to register for push notifications --- Gemfile | 1 + Gemfile.lock | 7 +++++ app/controllers/api/v1/devices_controller.rb | 18 ++++++++++++ app/models/device.rb | 7 +++++ app/services/notify_service.rb | 5 ++++ .../send_push_notification_service.rb | 28 +++++++++++++++++++ app/workers/push_notification_worker.rb | 11 ++++++++ config/routes.rb | 3 ++ db/migrate/20170129000348_create_devices.rb | 13 +++++++++ db/schema.rb | 11 +++++++- spec/fabricators/device_fabricator.rb | 3 ++ spec/fabricators/domain_block_fabricator.rb | 2 +- spec/models/device_spec.rb | 5 ++++ 13 files changed, 112 insertions(+), 2 deletions(-) create mode 100644 app/controllers/api/v1/devices_controller.rb create mode 100644 app/models/device.rb create mode 100644 app/services/send_push_notification_service.rb create mode 100644 app/workers/push_notification_worker.rb create mode 100644 db/migrate/20170129000348_create_devices.rb create mode 100644 spec/fabricators/device_fabricator.rb create mode 100644 spec/models/device_spec.rb diff --git a/Gemfile b/Gemfile index fb9a11ba1..4472f3306 100644 --- a/Gemfile +++ b/Gemfile @@ -50,6 +50,7 @@ gem 'pg_search' gem 'simple-navigation' gem 'statsd-instrument' gem 'ruby-oembed', require: 'oembed' +gem 'fcm' gem 'react-rails' gem 'browserify-rails' diff --git a/Gemfile.lock b/Gemfile.lock index 5361b2a05..bfed29072 100644 --- a/Gemfile.lock +++ b/Gemfile.lock @@ -127,6 +127,9 @@ GEM execjs (2.7.0) fabrication (2.15.2) fast_blank (1.0.0) + fcm (0.0.2) + httparty + json font-awesome-rails (4.6.3.1) railties (>= 3.2, < 5.1) fuubar (2.1.1) @@ -160,6 +163,8 @@ GEM domain_name (~> 0.5) http-form_data (1.0.1) http_parser.rb (0.6.0) + httparty (0.14.0) + multi_xml (>= 0.5.2) httplog (0.3.2) colorize i18n (0.7.0) @@ -207,6 +212,7 @@ GEM mini_portile2 (2.1.0) minitest (5.10.1) multi_json (1.12.1) + multi_xml (0.6.0) nio4r (1.2.1) nokogiri (1.7.0.1) mini_portile2 (~> 2.1.0) @@ -434,6 +440,7 @@ DEPENDENCIES dotenv-rails fabrication fast_blank + fcm font-awesome-rails fuubar goldfinger diff --git a/app/controllers/api/v1/devices_controller.rb b/app/controllers/api/v1/devices_controller.rb new file mode 100644 index 000000000..c565e972b --- /dev/null +++ b/app/controllers/api/v1/devices_controller.rb @@ -0,0 +1,18 @@ +# frozen_string_literal: true + +class Api::V1::DevicesController < ApiController + before_action -> { doorkeeper_authorize! :read } + before_action :require_user! + + respond_to :json + + def register + Device.where(account: current_account, registration_id: params[:registration_id]).first_or_create!(account: current_account, registration_id: params[:registration_id]) + render_empty + end + + def unregister + Device.where(account: current_account, registration_id: params[:registration_id]).delete_all + render_empty + end +end diff --git a/app/models/device.rb b/app/models/device.rb new file mode 100644 index 000000000..2782a7f38 --- /dev/null +++ b/app/models/device.rb @@ -0,0 +1,7 @@ +# frozen_string_literal: true + +class Device < ApplicationRecord + belongs_to :account + + validates :account, :registration_id, presence: true +end diff --git a/app/services/notify_service.rb b/app/services/notify_service.rb index 1ec36637c..d1504cadf 100644 --- a/app/services/notify_service.rb +++ b/app/services/notify_service.rb @@ -10,6 +10,7 @@ class NotifyService < BaseService create_notification send_email if email_enabled? + send_push_notification rescue ActiveRecord::RecordInvalid return end @@ -57,6 +58,10 @@ class NotifyService < BaseService NotificationMailer.send(@notification.type, @recipient, @notification).deliver_later end + def send_push_notification + PushNotificationWorker.perform_async(@notification.id) + end + def email_enabled? @recipient.user.settings.notification_emails[@notification.type] end diff --git a/app/services/send_push_notification_service.rb b/app/services/send_push_notification_service.rb new file mode 100644 index 000000000..802614fce --- /dev/null +++ b/app/services/send_push_notification_service.rb @@ -0,0 +1,28 @@ +# frozen_string_literal: true + +class SendPushNotificationService < BaseService + def call(notification) + return if ENV['FCM_API_KEY'].blank? + + devices = Device.where(account: notification.account).pluck(:registration_id) + fcm = FCM.new(ENV['FCM_API_KEY']) + + response = fcm.send(devices, data: { notification_id: notification.id }, collapse_key: :notifications, priority: :high) + handle_response(response) + end + + private + + def handle_response(response) + update_canonical_ids(response[:canonical_ids]) if response[:canonical_ids] + remove_bad_ids(response[:not_registered_ids]) if response[:not_registered_ids] + end + + def update_canonical_ids(ids) + ids.each { |pair| Device.find_by(registration_id: pair[:old]).update(registration_id: pair[:new]) } + end + + def remove_bad_ids(bad_ids) + Device.where(registration_id: bad_ids).delete_all + end +end diff --git a/app/workers/push_notification_worker.rb b/app/workers/push_notification_worker.rb new file mode 100644 index 000000000..a61d0e349 --- /dev/null +++ b/app/workers/push_notification_worker.rb @@ -0,0 +1,11 @@ +# frozen_string_literal: true + +class PushNotificationWorker + include Sidekiq::Worker + + def perform(notification_id) + SendPushNotificationService.new.call(Notification.find(notification_id)) + rescue ActiveRecord::RecordNotFound + true + end +end diff --git a/config/routes.rb b/config/routes.rb index 87f35770a..bce345f2e 100644 --- a/config/routes.rb +++ b/config/routes.rb @@ -118,6 +118,9 @@ Rails.application.routes.draw do resources :blocks, only: [:index] resources :favourites, only: [:index] + post '/devices/register', to: 'devices#register', as: :register_device + post '/devices/unregister', to: 'devices#unregister', as: :unregister_device + resources :follow_requests, only: [:index] do member do post :authorize diff --git a/db/migrate/20170129000348_create_devices.rb b/db/migrate/20170129000348_create_devices.rb new file mode 100644 index 000000000..bf8f5fc6e --- /dev/null +++ b/db/migrate/20170129000348_create_devices.rb @@ -0,0 +1,13 @@ +class CreateDevices < ActiveRecord::Migration[5.0] + def change + create_table :devices do |t| + t.integer :account_id, null: false + t.string :registration_id, null: false, default: '' + + t.timestamps + end + + add_index :devices, :registration_id + add_index :devices, :account_id + end +end diff --git a/db/schema.rb b/db/schema.rb index 7a7fea86b..448a7b861 100644 --- 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.define(version: 20170127165745) do +ActiveRecord::Schema.define(version: 20170129000348) do # These are extensions that must be enabled in order to support this database enable_extension "plpgsql" @@ -54,6 +54,15 @@ ActiveRecord::Schema.define(version: 20170127165745) do t.index ["account_id", "target_account_id"], name: "index_blocks_on_account_id_and_target_account_id", unique: true, using: :btree end + create_table "devices", force: :cascade do |t| + t.integer "account_id", null: false + t.string "registration_id", default: "", null: false + t.datetime "created_at", null: false + t.datetime "updated_at", null: false + t.index ["account_id"], name: "index_devices_on_account_id", using: :btree + t.index ["registration_id"], name: "index_devices_on_registration_id", using: :btree + end + create_table "domain_blocks", force: :cascade do |t| t.string "domain", default: "", null: false t.datetime "created_at", null: false diff --git a/spec/fabricators/device_fabricator.rb b/spec/fabricators/device_fabricator.rb new file mode 100644 index 000000000..02b24e8b3 --- /dev/null +++ b/spec/fabricators/device_fabricator.rb @@ -0,0 +1,3 @@ +Fabricator(:device) do + registration_id "12345678" +end diff --git a/spec/fabricators/domain_block_fabricator.rb b/spec/fabricators/domain_block_fabricator.rb index 540ddcacd..563a0f65b 100644 --- a/spec/fabricators/domain_block_fabricator.rb +++ b/spec/fabricators/domain_block_fabricator.rb @@ -1,3 +1,3 @@ Fabricator(:domain_block) do - domain "MyString" + domain "example.com" end diff --git a/spec/models/device_spec.rb b/spec/models/device_spec.rb new file mode 100644 index 000000000..f56fbf978 --- /dev/null +++ b/spec/models/device_spec.rb @@ -0,0 +1,5 @@ +require 'rails_helper' + +RSpec.describe Device, type: :model do + +end