From 19d9c8b53068c8bc5da9878c38ddc7b1353b86cd Mon Sep 17 00:00:00 2001
From: orbit <orbit@rulingcom.com>
Date: Wed, 27 Mar 2019 18:29:11 +0800
Subject: [PATCH] added widget for site count

---
 app/controllers/registered_sites_controller.rb | 13 +++++++++++++
 lib/registered_sites/engine.rb                 |  2 ++
 2 files changed, 15 insertions(+)
 create mode 100644 app/controllers/registered_sites_controller.rb

diff --git a/app/controllers/registered_sites_controller.rb b/app/controllers/registered_sites_controller.rb
new file mode 100644
index 0000000..58698f4
--- /dev/null
+++ b/app/controllers/registered_sites_controller.rb
@@ -0,0 +1,13 @@
+class RegisteredSitesController < ApplicationController
+
+  def site_count_widget
+    count = RegisteredSite.count
+    {
+      {
+        "extras" => {
+          "total_sites" => count
+        }
+      }
+    }
+  end
+end
\ No newline at end of file
diff --git a/lib/registered_sites/engine.rb b/lib/registered_sites/engine.rb
index 11138fe..13cbb06 100644
--- a/lib/registered_sites/engine.rb
+++ b/lib/registered_sites/engine.rb
@@ -4,6 +4,8 @@ module RegisteredSites
       OrbitApp.registration "RegisteredSites", :type => "ModuleApp" do
         module_label "registered_sites.registered_sites"
         base_url File.expand_path File.dirname(__FILE__)
+        widget_methods ["site_count_widget"]
+        widget_settings [{"data_count"=>30}]
         authorizable
         taggable "RegisteredSite"
         categorizable