Make it play nice with CE

Author: zoldar

lib/plausible_web/controllers/admin_controller.ex

@@ -1,10 +1,9 @@
 defmodule PlausibleWeb.AdminController do
   use PlausibleWeb, :controller
+  use Plausible
 
   alias Plausible.Teams
 
-  alias PlausibleWeb.Router.Helpers, as: Routes
-
   def usage(conn, params) do
     user_id = String.to_integer(params["user_id"])
 
@@ -73,35 +72,14 @@ defmodule PlausibleWeb.AdminController do
   end
 
   defp usage_and_limits_html(team, usage, limits, embed?) do
-    sites_row =
-      if team do
-        sites_count =
-          team
-          |> Ecto.assoc(:sites)
-          |> Plausible.Repo.aggregate(:count)
-
-        sites_link =
-          Routes.kaffy_resource_url(PlausibleWeb.Endpoint, :index, :sites, :site,
-            custom_search: team.owner.email
-          )
-
-        """
-        <li>Owner of <a href="#{sites_link}">#{sites_count} site#{if sites_count != 1, do: "s", else: ""}</a></li>
-        """
-      else
-        """
-        <li>Owner of 0 sites</li>
-        """
-      end
-
     content = """
       <ul>
         <li>Team: <b>#{team && team.name}</b></li>
         <li>Sites: <b>#{usage.sites}</b> / #{limits.sites}</li>
         <li>Team members: <b>#{usage.team_members}</b> / #{limits.team_members}</li>
         <li>Features: #{features_usage(usage.features)}</li>
         <li>Monthly pageviews: #{monthly_pageviews_usage(usage.monthly_pageviews, limits.monthly_pageviews)}</li>
-        #{sites_row}
+        #{sites_count_row(team)}
       </ul>
     """
 
@@ -131,6 +109,32 @@ defmodule PlausibleWeb.AdminController do
     end
   end
 
+  on_ee do
+    alias PlausibleWeb.Router.Helpers, as: Routes
+
+    defp sites_count_row(%Plausible.Teams.Team{} = team) do
+      sites_count =
+        team
+        |> Ecto.assoc(:sites)
+        |> Plausible.Repo.aggregate(:count)
+
+      sites_link =
+        Routes.kaffy_resource_url(PlausibleWeb.Endpoint, :index, :sites, :site,
+          custom_search: team.owner.email
+        )
+
+      """
+      <li>Owner of <a href="#{sites_link}">#{sites_count} site#{if sites_count != 1, do: "s", else: ""}</a></li>
+      """
+    end
+  end
+
+  defp sites_count_row(_) do
+    """
+    <li>Owner of 0 sites</li>
+    """
+  end
+
   defp features_usage(features_module_list) do
     list_items =
       features_module_list