# frozen_string_literal: true require 'rails_helper' describe Plugin::Instance do after do DiscoursePluginRegistry.reset! end context "find_all" do it "can find plugins correctly" do plugins = Plugin::Instance.find_all("#{Rails.root}/spec/fixtures/plugins") expect(plugins.count).to eq(5) plugin = plugins[3] expect(plugin.name).to eq("plugin-name") expect(plugin.path).to eq("#{Rails.root}/spec/fixtures/plugins/my_plugin/plugin.rb") end it "does not blow up on missing directory" do plugins = Plugin::Instance.find_all("#{Rails.root}/frank_zappa") expect(plugins.count).to eq(0) end end context "enabling/disabling" do it "is enabled by default" do expect(Plugin::Instance.new.enabled?).to eq(true) end context "with a plugin that extends things" do class Trout attr_accessor :data end class TroutSerializer < ApplicationSerializer attribute :name def name "a trout" end end class TroutJuniorSerializer < TroutSerializer attribute :i_am_child def name "a trout jr" end def i_am_child true end end class TroutPlugin < Plugin::Instance attr_accessor :enabled def enabled? @enabled end end before do @plugin = TroutPlugin.new @trout = Trout.new poison = TroutSerializer.new(@trout) poison.attributes poison = TroutJuniorSerializer.new(@trout) poison.attributes # New method @plugin.add_to_class(:trout, :status?) { "evil" } # DiscourseEvent @hello_count = 0 @increase_count = -> { @hello_count += 1 } @set = @plugin.on(:hello, &@increase_count) # Serializer @plugin.add_to_serializer(:trout, :scales) { 1024 } @serializer = TroutSerializer.new(@trout) @child_serializer = TroutJuniorSerializer.new(@trout) end after do DiscourseEvent.off(:hello, &@set.first) end it "checks enabled/disabled functionality for extensions" do # with an enabled plugin @plugin.enabled = true expect(@trout.status?).to eq("evil") DiscourseEvent.trigger(:hello) expect(@hello_count).to eq(1) expect(@serializer.scales).to eq(1024) expect(@serializer.include_scales?).to eq(true) expect(@child_serializer.attributes[:scales]).to eq(1024) # When a plugin is disabled @plugin.enabled = false expect(@trout.status?).to eq(nil) DiscourseEvent.trigger(:hello) expect(@hello_count).to eq(1) expect(@serializer.scales).to eq(1024) expect(@serializer.include_scales?).to eq(false) expect(@serializer.name).to eq("a trout") expect(@child_serializer.scales).to eq(1024) expect(@child_serializer.include_scales?).to eq(false) expect(@child_serializer.name).to eq("a trout jr") end it "only returns HTML if enabled" do ctx = Trout.new ctx.data = "hello" @plugin.register_html_builder('test:html') { |c| "<div>#{c.data}</div>" } @plugin.enabled = false expect(DiscoursePluginRegistry.build_html('test:html', ctx)).to eq("") @plugin.enabled = true expect(DiscoursePluginRegistry.build_html('test:html', ctx)).to eq("<div>hello</div>") end end end context "register asset" do it "populates the DiscoursePluginRegistry" do plugin = Plugin::Instance.new nil, "/tmp/test.rb" plugin.register_asset("test.css") plugin.register_asset("test2.css") plugin.send :register_assets! expect(DiscoursePluginRegistry.mobile_stylesheets[plugin.directory_name]).to be_nil expect(DiscoursePluginRegistry.stylesheets[plugin.directory_name].count).to eq(2) end it "remaps vendored_core_pretty_text asset" do plugin = Plugin::Instance.new nil, "/tmp/test.rb" plugin.register_asset("moment.js", :vendored_core_pretty_text) plugin.send :register_assets! expect(DiscoursePluginRegistry.vendored_core_pretty_text.first).to eq("vendor/assets/javascripts/moment.js") end end context "register service worker" do it "populates the DiscoursePluginRegistry" do plugin = Plugin::Instance.new nil, "/tmp/test.rb" plugin.register_service_worker("test.js") plugin.register_service_worker("test2.js") plugin.send :register_service_workers! expect(DiscoursePluginRegistry.service_workers.count).to eq(2) end end context "#add_report" do it "adds a report" do plugin = Plugin::Instance.new nil, "/tmp/test.rb" plugin.add_report("readers") {} expect(Report.respond_to?(:report_readers)).to eq(true) end end it 'patches the enabled? function for auth_providers if not defined' do SimpleAuthenticator = Class.new(Auth::Authenticator) do def name "my_authenticator" end end plugin = Plugin::Instance.new # lets piggy back on another boolean setting, so we don't dirty our SiteSetting object SiteSetting.enable_badges = false # No enabled_site_setting authenticator = SimpleAuthenticator.new plugin.auth_provider(authenticator: authenticator) plugin.notify_before_auth expect(authenticator.enabled?).to eq(true) # With enabled site setting plugin = Plugin::Instance.new authenticator = SimpleAuthenticator.new plugin.auth_provider(enabled_setting: 'enable_badges', authenticator: authenticator) plugin.notify_before_auth expect(authenticator.enabled?).to eq(false) # Defines own method plugin = Plugin::Instance.new SiteSetting.enable_badges = true authenticator = Class.new(SimpleAuthenticator) do def enabled? false end end.new plugin.auth_provider(enabled_setting: 'enable_badges', authenticator: authenticator) plugin.notify_before_auth expect(authenticator.enabled?).to eq(false) end context "activate!" do before do # lets piggy back on another boolean setting, so we don't dirty our SiteSetting object SiteSetting.enable_badges = false end it "can activate plugins correctly" do plugin = Plugin::Instance.new plugin.path = "#{Rails.root}/spec/fixtures/plugins/my_plugin/plugin.rb" junk_file = "#{plugin.auto_generated_path}/junk" plugin.ensure_directory(junk_file) File.open("#{plugin.auto_generated_path}/junk", "w") { |f| f.write("junk") } plugin.activate! # calls ensure_assets! make sure they are there expect(plugin.assets.count).to eq(1) plugin.assets.each do |a, opts| expect(File.exist?(a)).to eq(true) end # ensure it cleans up all crap in autogenerated directory expect(File.exist?(junk_file)).to eq(false) end it "registers auth providers correctly" do plugin = Plugin::Instance.new plugin.path = "#{Rails.root}/spec/fixtures/plugins/my_plugin/plugin.rb" plugin.activate! expect(DiscoursePluginRegistry.auth_providers.count).to eq(0) plugin.notify_before_auth expect(DiscoursePluginRegistry.auth_providers.count).to eq(1) auth_provider = DiscoursePluginRegistry.auth_providers.to_a[0] expect(auth_provider.authenticator.name).to eq('facebook') end it "finds all the custom assets" do plugin = Plugin::Instance.new plugin.path = "#{Rails.root}/spec/fixtures/plugins/my_plugin/plugin.rb" plugin.register_asset("test.css") plugin.register_asset("test2.scss") plugin.register_asset("mobile.css", :mobile) plugin.register_asset("desktop.css", :desktop) plugin.register_asset("desktop2.css", :desktop) plugin.register_asset("code.js") plugin.register_asset("my_admin.js", :admin) plugin.register_asset("my_admin2.js", :admin) plugin.activate! expect(DiscoursePluginRegistry.javascripts.count).to eq(2) expect(DiscoursePluginRegistry.admin_javascripts.count).to eq(2) expect(DiscoursePluginRegistry.desktop_stylesheets[plugin.directory_name].count).to eq(2) expect(DiscoursePluginRegistry.stylesheets[plugin.directory_name].count).to eq(2) expect(DiscoursePluginRegistry.mobile_stylesheets[plugin.directory_name].count).to eq(1) end end context "serialized_current_user_fields" do before do DiscoursePluginRegistry.serialized_current_user_fields << "has_car" end after do DiscoursePluginRegistry.serialized_current_user_fields.delete "has_car" end it "correctly serializes custom user fields" do DiscoursePluginRegistry.serialized_current_user_fields << "has_car" user = Fabricate(:user) user.custom_fields["has_car"] = "true" user.save! payload = JSON.parse(CurrentUserSerializer.new(user, scope: Guardian.new(user)).to_json) expect(payload["current_user"]["custom_fields"]["has_car"]).to eq("true") payload = JSON.parse(UserSerializer.new(user, scope: Guardian.new(user)).to_json) expect(payload["user"]["custom_fields"]["has_car"]).to eq("true") UserCustomField.destroy_all user.reload payload = JSON.parse(CurrentUserSerializer.new(user, scope: Guardian.new(user)).to_json) expect(payload["current_user"]["custom_fields"]).to eq({}) payload = JSON.parse(UserSerializer.new(user, scope: Guardian.new(user)).to_json) expect(payload["user"]["custom_fields"]).to eq({}) end end context "register_color_scheme" do it "can add a color scheme for the first time" do plugin = Plugin::Instance.new nil, "/tmp/test.rb" expect { plugin.register_color_scheme("Purple", primary: 'EEE0E5') plugin.notify_after_initialize }.to change { ColorScheme.count }.by(1) expect(ColorScheme.where(name: "Purple")).to be_present end it "doesn't add the same color scheme twice" do Fabricate(:color_scheme, name: "Halloween") plugin = Plugin::Instance.new nil, "/tmp/test.rb" expect { plugin.register_color_scheme("Halloween", primary: 'EEE0E5') plugin.notify_after_initialize }.to_not change { ColorScheme.count } end end describe '.register_seedfu_fixtures' do it "should add the new path to SeedFu's fixtures path" do plugin = Plugin::Instance.new nil, "/tmp/test.rb" plugin.register_seedfu_fixtures(['some_path']) plugin.register_seedfu_fixtures('some_path2') expect(SeedFu.fixture_paths).to include('some_path') expect(SeedFu.fixture_paths).to include('some_path2') end end describe '#add_model_callback' do let(:metadata) do metadata = Plugin::Metadata.new metadata.name = 'test' metadata end let(:plugin_instance) do plugin = Plugin::Instance.new(nil, "/tmp/test.rb") plugin.metadata = metadata plugin end it 'should add the right callback' do called = 0 plugin_instance.add_model_callback(User, :after_create) do called += 1 end user = Fabricate(:user) expect(called).to eq(1) user.update!(username: 'some_username') expect(called).to eq(1) end it 'should add the right callback with options' do called = 0 plugin_instance.add_model_callback(User, :after_commit, on: :create) do called += 1 end user = Fabricate(:user) expect(called).to eq(1) user.update!(username: 'some_username') expect(called).to eq(1) end end context "locales" do let(:plugin_path) { "#{Rails.root}/spec/fixtures/plugins/custom_locales" } let!(:plugin) { Plugin::Instance.new(nil, "#{plugin_path}/plugin.rb") } let(:plural) do { keys: [:one, :few, :other], rule: lambda do |n| return :one if n == 1 return :few if n < 10 :other end } end def register_locale(locale, opts) plugin.register_locale(locale, opts) plugin.activate! DiscoursePluginRegistry.locales[locale] end it "enables the registered locales only on activate" do plugin.register_locale("foo_BAR", name: "Foo", nativeName: "Foo Bar", plural: plural) plugin.register_locale("tup", name: "Tupi", nativeName: "Tupi", fallbackLocale: "pt_BR") expect(DiscoursePluginRegistry.locales.count).to eq(0) plugin.activate! expect(DiscoursePluginRegistry.locales.count).to eq(2) end it "allows finding the locale by string and symbol" do register_locale("foo_BAR", name: "Foo", nativeName: "Foo Bar", plural: plural) expect(DiscoursePluginRegistry.locales).to have_key(:foo_BAR) expect(DiscoursePluginRegistry.locales).to have_key('foo_BAR') end it "correctly registers a new locale" do locale = register_locale("foo_BAR", name: "Foo", nativeName: "Foo Bar", plural: plural) expect(DiscoursePluginRegistry.locales.count).to eq(1) expect(DiscoursePluginRegistry.locales).to have_key(:foo_BAR) expect(locale[:fallbackLocale]).to be_nil expect(locale[:message_format]).to eq(["foo_BAR", "#{plugin_path}/lib/javascripts/locale/message_format/foo_BAR.js"]) expect(locale[:moment_js]).to eq(["foo_BAR", "#{plugin_path}/lib/javascripts/locale/moment_js/foo_BAR.js"]) expect(locale[:moment_js_timezones]).to eq(["foo_BAR", "#{plugin_path}/lib/javascripts/locale/moment_js_timezones/foo_BAR.js"]) expect(locale[:plural]).to eq(plural.with_indifferent_access) expect(Rails.configuration.assets.precompile).to include("locales/foo_BAR.js") expect(JsLocaleHelper.find_message_format_locale(["foo_BAR"], fallback_to_english: true)).to eq(locale[:message_format]) expect(JsLocaleHelper.find_moment_locale(["foo_BAR"])).to eq (locale[:moment_js]) expect(JsLocaleHelper.find_moment_locale(["foo_BAR"], timezone_names: true)).to eq (locale[:moment_js_timezones]) end it "correctly registers a new locale using a fallback locale" do locale = register_locale("tup", name: "Tupi", nativeName: "Tupi", fallbackLocale: "pt_BR") expect(DiscoursePluginRegistry.locales.count).to eq(1) expect(DiscoursePluginRegistry.locales).to have_key(:tup) expect(locale[:fallbackLocale]).to eq("pt_BR") expect(locale[:message_format]).to eq(["pt_BR", "#{Rails.root}/lib/javascripts/locale/pt_BR.js"]) expect(locale[:moment_js]).to eq(["pt-br", "#{Rails.root}/vendor/assets/javascripts/moment-locale/pt-br.js"]) expect(locale[:moment_js_timezones]).to eq(["pt", "#{Rails.root}/vendor/assets/javascripts/moment-timezone-names-locale/pt.js"]) expect(locale[:plural]).to be_nil expect(Rails.configuration.assets.precompile).to include("locales/tup.js") expect(JsLocaleHelper.find_message_format_locale(["tup"], fallback_to_english: true)).to eq(locale[:message_format]) expect(JsLocaleHelper.find_moment_locale(["tup"])).to eq (locale[:moment_js]) end it "correctly registers a new locale when some files exist in core" do locale = register_locale("tlh", name: "Klingon", nativeName: "tlhIngan Hol", plural: plural) expect(DiscoursePluginRegistry.locales.count).to eq(1) expect(DiscoursePluginRegistry.locales).to have_key(:tlh) expect(locale[:fallbackLocale]).to be_nil expect(locale[:message_format]).to eq(["tlh", "#{plugin_path}/lib/javascripts/locale/message_format/tlh.js"]) expect(locale[:moment_js]).to eq(["tlh", "#{Rails.root}/vendor/assets/javascripts/moment-locale/tlh.js"]) expect(locale[:plural]).to eq(plural.with_indifferent_access) expect(Rails.configuration.assets.precompile).to include("locales/tlh.js") expect(JsLocaleHelper.find_message_format_locale(["tlh"], fallback_to_english: true)).to eq(locale[:message_format]) expect(JsLocaleHelper.find_moment_locale(["tlh"])).to eq (locale[:moment_js]) end it "does not register a new locale when the fallback locale does not exist" do register_locale("bar", name: "Bar", nativeName: "Bar", fallbackLocale: "foo") expect(DiscoursePluginRegistry.locales.count).to eq(0) end [ "config/locales/client.foo_BAR.yml", "config/locales/server.foo_BAR.yml", "lib/javascripts/locale/message_format/foo_BAR.js", "lib/javascripts/locale/moment_js/foo_BAR.js", "assets/locales/foo_BAR.js.erb" ].each do |path| it "does not register a new locale when #{path} is missing" do path = "#{plugin_path}/#{path}" File.stubs('exist?').returns(false) File.stubs('exist?').with(regexp_matches(/#{Regexp.quote(plugin_path)}.*/)).returns(true) File.stubs('exist?').with(path).returns(false) register_locale("foo_BAR", name: "Foo", nativeName: "Foo Bar", plural: plural) expect(DiscoursePluginRegistry.locales.count).to eq(0) end end end describe '#register_reviewable_types' do it 'Overrides the existing Reviewable types adding new ones' do current_types = Reviewable.types new_type_class = Class Plugin::Instance.new.register_reviewable_type new_type_class expect(Reviewable.types).to match_array(current_types << new_type_class.name) end end describe '#extend_list_method' do it 'Overrides the existing list appending new elements' do current_list = Reviewable.types new_element = Class.name Plugin::Instance.new.extend_list_method Reviewable, :types, [new_element] expect(Reviewable.types).to match_array(current_list << new_element) end end describe '#register_emoji' do before do Plugin::CustomEmoji.clear_cache end after do Plugin::CustomEmoji.clear_cache end it 'allows to register an emoji' do Plugin::Instance.new.register_emoji("foo", "/foo/bar.png") custom_emoji = Emoji.custom.first expect(custom_emoji.name).to eq("foo") expect(custom_emoji.url).to eq("/foo/bar.png") expect(custom_emoji.group).to eq(Emoji::DEFAULT_GROUP) end it 'allows to register an emoji with a group' do Plugin::Instance.new.register_emoji("bar", "/baz/bar.png", "baz") custom_emoji = Emoji.custom.first expect(custom_emoji.name).to eq("bar") expect(custom_emoji.url).to eq("/baz/bar.png") expect(custom_emoji.group).to eq("baz") end end describe '#replace_flags' do after do PostActionType.replace_flag_settings(nil) ReviewableScore.reload_types end let(:original_flags) { PostActionType.flag_settings } it 'adds a new flag' do highest_flag_id = ReviewableScore.types.values.max flag_name = :new_flag subject.replace_flags(settings: original_flags) do |settings, next_flag_id| settings.add( next_flag_id, flag_name ) end expect(PostActionType.flag_settings.flag_types.keys).to include(flag_name) expect(PostActionType.flag_settings.flag_types.values.max).to eq(highest_flag_id + 1) end it 'adds a new score type after adding a new flag' do highest_flag_id = ReviewableScore.types.values.max new_score_type = :new_score_type subject.replace_flags(settings: original_flags, score_type_names: [new_score_type]) do |settings, next_flag_id| settings.add( next_flag_id, :new_flag ) end expect(PostActionType.flag_settings.flag_types.values.max).to eq(highest_flag_id + 1) expect(ReviewableScore.types.keys).to include(new_score_type) expect(ReviewableScore.types.values.max).to eq(highest_flag_id + 2) end end describe '#add_api_key_scope' do after { DiscoursePluginRegistry.reset! } it 'adds a custom api key scope' do actions = %w[admin/groups#create] subject.add_api_key_scope(:groups, create: { actions: actions }) expect(ApiKeyScope.scope_mappings.dig(:groups, :create, :actions)).to contain_exactly(*actions) end end describe '#add_directory_column' do let!(:plugin) { Plugin::Instance.new } before do DirectoryItem.clear_plugin_queries end after do DirectoryColumn.clear_plugin_directory_columns end describe "with valid column name" do let(:column_name) { "random_c" } before do DB.exec("ALTER TABLE directory_items ADD COLUMN IF NOT EXISTS #{column_name} integer") end after do DB.exec("ALTER TABLE directory_items DROP COLUMN IF EXISTS #{column_name}") DiscourseEvent.all_off("before_directory_refresh") end it 'creates a directory column record when directory items are refreshed' do plugin.add_directory_column(column_name, query: "SELECT COUNT(*) FROM users", icon: 'recycle') expect(DirectoryColumn.find_by(name: column_name, icon: 'recycle', enabled: false)).not_to be_present DirectoryItem.refresh! expect(DirectoryColumn.find_by(name: column_name, icon: 'recycle', enabled: false)).to be_present end end it 'errors when the column_name contains invalid characters' do expect { plugin.add_directory_column('Capital', query: "SELECT COUNT(*) FROM users", icon: 'recycle') }.to raise_error(RuntimeError) expect { plugin.add_directory_column('has space', query: "SELECT COUNT(*) FROM users", icon: 'recycle') }.to raise_error(RuntimeError) expect { plugin.add_directory_column('has_number_1', query: "SELECT COUNT(*) FROM users", icon: 'recycle') }.to raise_error(RuntimeError) end end describe '#register_site_categories_callback' do fab!(:category) { Fabricate(:category) } it 'adds a callback to the Site#categories' do instance = Plugin::Instance.new instance.register_site_categories_callback do |categories| categories.each do |category| category[:test_field] = "test" end end site = Site.new(Guardian.new) expect(site.categories.first[:test_field]).to eq("test") ensure Site.clear_cache Site.categories_callbacks.clear end end describe '#register_notification_consolidation_plan' do let(:plugin) { Plugin::Instance.new } fab!(:topic) { Fabricate(:topic) } after do DiscoursePluginRegistry.reset_register!(:notification_consolidation_plans) end it 'fails when the received object is not a consolidation plan' do expect { plugin.register_notification_consolidation_plan(Object.new) }.to raise_error(ArgumentError) end it 'registers a consolidation plan and uses it' do plan = Notifications::ConsolidateNotifications.new( from: Notification.types[:code_review_commit_approved], to: Notification.types[:code_review_commit_approved], threshold: 1, consolidation_window: 1.minute, unconsolidated_query_blk: ->(notifications, _data) do notifications.where("(data::json ->> 'consolidated') IS NULL") end, consolidated_query_blk: ->(notifications, _data) do notifications.where("(data::json ->> 'consolidated') IS NOT NULL") end ).set_mutations( set_data_blk: ->(notification) do notification.data_hash.merge(consolidated: true) end ) plugin.register_notification_consolidation_plan(plan) create_notification! create_notification! expect(commit_approved_notifications.count).to eq(1) consolidated_notification = commit_approved_notifications.last expect(consolidated_notification.data_hash[:consolidated]).to eq(true) end def commit_approved_notifications Notification.where(user: topic.user, notification_type: Notification.types[:code_review_commit_approved]) end def create_notification! Notification.consolidate_or_create!( notification_type: Notification.types[:code_review_commit_approved], topic_id: topic.id, user: topic.user, data: {} ) end end end