From 1e90dc2022c5a1261f9e4b47c4f3828ed90b72a8 Mon Sep 17 00:00:00 2001
From: Hayden <64056131+hay-kot@users.noreply.github.com>
Date: Sun, 27 Mar 2022 15:12:18 -0800
Subject: [PATCH] feat: add group statistics on profile page

* resolve file not found error and add constants

* add group stats and storage functionality

* generate new types

* add statistics and storage cap graphs

* fix: add loadFood query param #1103

* refactor to flex view
---
 frontend/api/class-interfaces/groups.ts       |  11 ++
 frontend/components/global/StatsCards.vue     |  53 ++++++++
 frontend/composables/recipes/use-recipes.ts   |   2 +-
 frontend/pages/user/profile/index.vue         | 124 +++++++++++++++++-
 frontend/types/api-types/group.ts             |  13 ++
 frontend/types/components.d.ts                |   2 +
 mealie/pkgs/stats/fs_stats.py                 |   9 +-
 mealie/repos/repository_group.py              |  33 +++--
 mealie/repos/repository_recipes.py            |   6 +-
 .../groups/controller_group_self_service.py   |  38 ++++--
 mealie/schema/group/__init__.py               |   1 +
 mealie/schema/group/group_statistics.py       |  26 ++++
 .../services/group_services/group_service.py  |  40 ++++++
 13 files changed, 326 insertions(+), 32 deletions(-)
 create mode 100644 frontend/components/global/StatsCards.vue
 create mode 100644 mealie/schema/group/group_statistics.py
 create mode 100644 mealie/services/group_services/group_service.py

diff --git a/frontend/api/class-interfaces/groups.ts b/frontend/api/class-interfaces/groups.ts
index 26446bb9636e..1bd4d872f184 100644
--- a/frontend/api/class-interfaces/groups.ts
+++ b/frontend/api/class-interfaces/groups.ts
@@ -1,5 +1,6 @@
 import { BaseCRUDAPI } from "../_base";
 import { GroupInDB, UserOut } from "~/types/api-types/user";
+import { GroupStatistics, GroupStorage } from "~/types/api-types/group";
 
 const prefix = "/api";
 
@@ -11,6 +12,8 @@ const routes = {
   permissions: `${prefix}/groups/permissions`,
 
   preferences: `${prefix}/groups/preferences`,
+  statistics: `${prefix}/groups/statistics`,
+  storage: `${prefix}/groups/storage`,
 
   invitation: `${prefix}/groups/invitations`,
 
@@ -103,4 +106,12 @@ export class GroupAPI extends BaseCRUDAPI<GroupInDB, CreateGroup> {
     // TODO: This should probably be a patch request, which isn't offered by the API currently
     return await this.requests.put<Permissions, SetPermissions>(routes.permissions, payload);
   }
+
+  async statistics() {
+    return await this.requests.get<GroupStatistics>(routes.statistics);
+  }
+
+  async storage() {
+    return await this.requests.get<GroupStorage>(routes.storage);
+  }
 }
diff --git a/frontend/components/global/StatsCards.vue b/frontend/components/global/StatsCards.vue
new file mode 100644
index 000000000000..ca3d8589cc69
--- /dev/null
+++ b/frontend/components/global/StatsCards.vue
@@ -0,0 +1,53 @@
+<template>
+  <v-card :min-width="minWidth" :to="to" :hover="to ? true : false">
+    <div class="d-flex flex-no-wrap">
+      <v-avatar class="ml-3 mr-0 mt-3" color="primary" size="36">
+        <v-icon color="white" class="pa-1">
+          {{ activeIcon }}
+        </v-icon>
+      </v-avatar>
+      <div>
+        <v-card-title class="text-subtitle-1 pt-2 pb-2">
+          <slot name="title"></slot>
+        </v-card-title>
+        <v-card-subtitle class="pb-2">
+          <slot name="value"></slot>
+        </v-card-subtitle>
+      </div>
+    </div>
+  </v-card>
+</template>
+
+<script lang="ts">
+import { computed, defineComponent, useContext } from "@nuxtjs/composition-api";
+
+export default defineComponent({
+  props: {
+    icon: {
+      type: String,
+      default: null,
+    },
+    minWidth: {
+      type: String,
+      default: "",
+    },
+    to: {
+      type: String,
+      default: null,
+    },
+  },
+  setup(props) {
+    const { $globals } = useContext();
+
+    const activeIcon = computed(() => {
+      return props.icon ?? $globals.icons.primary;
+    });
+
+    return {
+      activeIcon,
+    };
+  },
+});
+</script>
+
+<style scoped></style>
diff --git a/frontend/composables/recipes/use-recipes.ts b/frontend/composables/recipes/use-recipes.ts
index 17931d4a28cc..a9c15d52712e 100644
--- a/frontend/composables/recipes/use-recipes.ts
+++ b/frontend/composables/recipes/use-recipes.ts
@@ -97,7 +97,7 @@ export const useRecipes = (all = false, fetchRecipes = true) => {
   })();
 
   async function refreshRecipes() {
-    const { data } = await api.recipes.getAll(start, end);
+    const { data } = await api.recipes.getAll(start, end, { loadFood: true });
     if (data) {
       recipes.value = data;
     }
diff --git a/frontend/pages/user/profile/index.vue b/frontend/pages/user/profile/index.vue
index ab48e9f4e3b5..e37c72ab014f 100644
--- a/frontend/pages/user/profile/index.vue
+++ b/frontend/pages/user/profile/index.vue
@@ -38,6 +38,51 @@
         </div>
       </v-card>
     </section>
+    <section class="my-3">
+      <div>
+        <h3 class="headline">Account Summary</h3>
+        <p>Here's a summary of your group's information</p>
+      </div>
+      <v-row tag="section">
+        <v-col cols="12" sm="12" md="6">
+          <v-card outlined>
+            <v-card-title class="headline pb-0"> Group Statistics </v-card-title>
+            <v-card-text class="py-0">
+              Your Group Statistics provide some insight how you're using Mealie.
+            </v-card-text>
+            <v-card-text class="d-flex flex-wrap justify-center align-center" style="gap: 0.8rem">
+              <StatsCards
+                v-for="(value, key) in stats"
+                :key="`${key}-${value}`"
+                :min-width="$vuetify.breakpoint.xs ? '100%' : '158'"
+                :icon="getStatsIcon(key)"
+                :to="getStatsTo(key)"
+              >
+                <template #title> {{ getStatsTitle(key) }}</template>
+                <template #value> {{ value }}</template>
+              </StatsCards>
+            </v-card-text>
+          </v-card>
+        </v-col>
+        <v-col cols="12" sm="12" md="6" class="d-flex align-strart">
+          <v-card outlined>
+            <v-card-title class="headline pb-0"> Storage Capacity </v-card-title>
+            <v-card-text class="py-0">
+              Your storage capacity is a calculation of the images and assets you have uploaded.
+              <strong> This feature is currently inactive</strong>
+            </v-card-text>
+            <v-card-text>
+              <v-progress-linear :value="storageUsedPercentage" color="accent" class="rounded" height="30">
+                <template #default>
+                  <strong> {{ storageText }} </strong>
+                </template>
+              </v-progress-linear>
+            </v-card-text>
+          </v-card>
+        </v-col>
+      </v-row>
+    </section>
+    <v-divider class="my-7"></v-divider>
     <section>
       <div>
         <h3 class="headline">Personal</h3>
@@ -149,18 +194,21 @@
 </template>
 
 <script lang="ts">
-import { computed, defineComponent, useContext, ref, toRefs, reactive } from "@nuxtjs/composition-api";
+import { computed, defineComponent, useContext, ref, toRefs, reactive, useAsync } from "@nuxtjs/composition-api";
 import UserProfileLinkCard from "@/components/Domain/User/UserProfileLinkCard.vue";
 import { useUserApi } from "~/composables/api";
 import { validators } from "~/composables/use-validators";
 import { alert } from "~/composables/use-toast";
 import UserAvatar from "@/components/Domain/User/UserAvatar.vue";
+import { useAsyncKey } from "~/composables/use-utils";
+import StatsCards from "~/components/global/StatsCards.vue";
 
 export default defineComponent({
   name: "UserProfile",
   components: {
     UserProfileLinkCard,
     UserAvatar,
+    StatsCards,
   },
   scrollToTop: true,
   setup() {
@@ -218,7 +266,81 @@ export default defineComponent({
       return false;
     });
 
+    const stats = useAsync(async () => {
+      const { data } = await api.groups.statistics();
+
+      if (data) {
+        return data;
+      }
+    }, useAsyncKey());
+
+    const statsText: { [key: string]: string } = {
+      totalRecipes: "Recipes",
+      totalUsers: "Users",
+      totalCategories: "Categories",
+      totalTags: "Tags",
+      totalTools: "Tools",
+    };
+
+    function getStatsTitle(key: string) {
+      return statsText[key] ?? "unknown";
+    }
+
+    const { $globals } = useContext();
+
+    const iconText: { [key: string]: string } = {
+      totalUsers: $globals.icons.user,
+      totalCategories: $globals.icons.tags,
+      totalTags: $globals.icons.tags,
+      totalTools: $globals.icons.potSteam,
+    };
+
+    function getStatsIcon(key: string) {
+      return iconText[key] ?? $globals.icons.primary;
+    }
+
+    const statsTo: { [key: string]: string } = {
+      totalRecipes: "/recipes/all",
+      totalUsers: "/group/members",
+      totalCategories: "/recipes/categories",
+      totalTags: "/recipes/tags",
+      totalTools: "/recipes/tools",
+    };
+
+    function getStatsTo(key: string) {
+      return statsTo[key] ?? "unknown";
+    }
+
+    const storage = useAsync(async () => {
+      const { data } = await api.groups.storage();
+
+      if (data) {
+        return data;
+      }
+    }, useAsyncKey());
+
+    const storageUsedPercentage = computed(() => {
+      if (!storage.value) {
+        return 0;
+      }
+
+      return (storage.value?.usedStorageBytes / storage.value?.totalStorageBytes) * 100 ?? 0;
+    });
+
+    const storageText = computed(() => {
+      if (!storage.value) {
+        return "Loading...";
+      }
+      return `${storage.value.usedStorageStr} / ${storage.value.totalStorageStr}`;
+    });
+
     return {
+      storageText,
+      storageUsedPercentage,
+      getStatsTitle,
+      getStatsIcon,
+      getStatsTo,
+      stats,
       user,
       constructLink,
       generatedLink,
diff --git a/frontend/types/api-types/group.ts b/frontend/types/api-types/group.ts
index 695f0edcf716..f42d2b13dc1e 100644
--- a/frontend/types/api-types/group.ts
+++ b/frontend/types/api-types/group.ts
@@ -176,6 +176,19 @@ export interface GroupEventNotifierUpdate {
   options?: GroupEventNotifierOptions;
   id: string;
 }
+export interface GroupStatistics {
+  totalRecipes: number;
+  totalUsers: number;
+  totalCategories: number;
+  totalTags: number;
+  totalTools: number;
+}
+export interface GroupStorage {
+  usedStorageBytes: number;
+  usedStorageStr: string;
+  totalStorageBytes: number;
+  totalStorageStr: string;
+}
 export interface IngredientFood {
   name: string;
   description?: string;
diff --git a/frontend/types/components.d.ts b/frontend/types/components.d.ts
index afb5aa3f6f3d..146de564aa12 100644
--- a/frontend/types/components.d.ts
+++ b/frontend/types/components.d.ts
@@ -10,6 +10,7 @@
  import BannerExperimental from "@/components/global/BannerExperimental.vue";
  import BaseDialog from "@/components/global/BaseDialog.vue";
  import RecipeJsonEditor from "@/components/global/RecipeJsonEditor.vue";
+ import StatsCards from "@/components/global/StatsCards.vue";
  import InputLabelType from "@/components/global/InputLabelType.vue";
  import BaseStatCard from "@/components/global/BaseStatCard.vue";
  import DevDumpJson from "@/components/global/DevDumpJson.vue";
@@ -45,6 +46,7 @@ declare module "vue" {
      BannerExperimental: typeof BannerExperimental;
      BaseDialog: typeof BaseDialog;
      RecipeJsonEditor: typeof RecipeJsonEditor;
+     StatsCards: typeof StatsCards;
      InputLabelType: typeof InputLabelType;
      BaseStatCard: typeof BaseStatCard;
      DevDumpJson: typeof DevDumpJson;
diff --git a/mealie/pkgs/stats/fs_stats.py b/mealie/pkgs/stats/fs_stats.py
index 77ef2c0d0102..1594db0d9536 100644
--- a/mealie/pkgs/stats/fs_stats.py
+++ b/mealie/pkgs/stats/fs_stats.py
@@ -1,6 +1,9 @@
 import os
 from pathlib import Path
 
+megabyte = 1_048_576
+gigabyte = 1_073_741_824
+
 
 def pretty_size(size: int) -> str:
     """
@@ -23,7 +26,11 @@ def get_dir_size(path: Path | str) -> int:
     """
     Get the size of a directory
     """
-    total_size = os.path.getsize(path)
+    try:
+        total_size = os.path.getsize(path)
+    except FileNotFoundError:
+        return 0
+
     for item in os.listdir(path):
         itempath = os.path.join(path, item)
         if os.path.isfile(itempath):
diff --git a/mealie/repos/repository_group.py b/mealie/repos/repository_group.py
index a6a24eb2c23d..19cbbaae3a47 100644
--- a/mealie/repos/repository_group.py
+++ b/mealie/repos/repository_group.py
@@ -1,32 +1,31 @@
 from typing import Union
 
-from sqlalchemy.orm.session import Session
+from pydantic import UUID4
 
 from mealie.db.models.group import Group
-from mealie.schema.meal_plan.meal import MealPlanOut
+from mealie.db.models.recipe.category import Category
+from mealie.db.models.recipe.recipe import RecipeModel
+from mealie.db.models.recipe.tag import Tag
+from mealie.db.models.recipe.tool import Tool
+from mealie.db.models.users.users import User
+from mealie.schema.group.group_statistics import GroupStatistics
 from mealie.schema.user.user import GroupInDB
 
 from .repository_generic import RepositoryGeneric
 
 
 class RepositoryGroup(RepositoryGeneric[GroupInDB, Group]):
-    def get_meals(self, session: Session, match_value: str, match_key: str = "name") -> list[MealPlanOut]:
-        """A Helper function to get the group from the database and return a sorted list of
-
-        Args:
-            session (Session): SqlAlchemy Session
-            match_value (str): Match Value
-            match_key (str, optional): Match Key. Defaults to "name".
-
-        Returns:
-            list[MealPlanOut]: [description]
-        """
-        group: GroupInDB = session.query(self.sql_model).filter_by(**{match_key: match_value}).one_or_none()
-
-        return group.mealplans
-
     def get_by_name(self, name: str, limit=1) -> Union[GroupInDB, Group, None]:
         dbgroup = self.session.query(self.sql_model).filter_by(**{"name": name}).one_or_none()
         if dbgroup is None:
             return None
         return self.schema.from_orm(dbgroup)
+
+    def statistics(self, group_id: UUID4) -> GroupStatistics:
+        return GroupStatistics(
+            total_recipes=self.session.query(RecipeModel).filter_by(group_id=group_id).count(),
+            total_users=self.session.query(User).filter_by(group_id=group_id).count(),
+            total_categories=self.session.query(Category).filter_by(group_id=group_id).count(),
+            total_tags=self.session.query(Tag).filter_by(group_id=group_id).count(),
+            total_tools=self.session.query(Tool).filter_by(group_id=group_id).count(),
+        )
diff --git a/mealie/repos/repository_recipes.py b/mealie/repos/repository_recipes.py
index c229c786f550..a14c345d9733 100644
--- a/mealie/repos/repository_recipes.py
+++ b/mealie/repos/repository_recipes.py
@@ -2,6 +2,7 @@ from random import randint
 from typing import Any, Optional
 from uuid import UUID
 
+from pydantic import UUID4
 from slugify import slugify
 from sqlalchemy import and_, func
 from sqlalchemy.exc import IntegrityError
@@ -163,7 +164,7 @@ class RepositoryRecipes(RepositoryGeneric[Recipe, RecipeModel]):
             .limit(limit)
         ]
 
-    def get_by_slug(self, group_id: UUID, slug: str, limit=1) -> Optional[Recipe]:
+    def get_by_slug(self, group_id: UUID4, slug: str, limit=1) -> Optional[Recipe]:
         dbrecipe = (
             self.session.query(RecipeModel)
             .filter(RecipeModel.group_id == group_id, RecipeModel.slug == slug)
@@ -172,3 +173,6 @@ class RepositoryRecipes(RepositoryGeneric[Recipe, RecipeModel]):
         if dbrecipe is None:
             return None
         return self.schema.from_orm(dbrecipe)
+
+    def all_ids(self, group_id: UUID4) -> list[UUID4]:
+        return [tpl[0] for tpl in self.session.query(RecipeModel.id).filter(RecipeModel.group_id == group_id).all()]
diff --git a/mealie/routes/groups/controller_group_self_service.py b/mealie/routes/groups/controller_group_self_service.py
index 3add0e8f9678..816f8a74e149 100644
--- a/mealie/routes/groups/controller_group_self_service.py
+++ b/mealie/routes/groups/controller_group_self_service.py
@@ -1,3 +1,5 @@
+from functools import cached_property
+
 from fastapi import HTTPException, status
 
 from mealie.routes._base.abc_controller import BaseUserController
@@ -5,13 +7,29 @@ from mealie.routes._base.controller import controller
 from mealie.routes._base.routers import UserAPIRouter
 from mealie.schema.group.group_permissions import SetPermissions
 from mealie.schema.group.group_preferences import ReadGroupPreferences, UpdateGroupPreferences
+from mealie.schema.group.group_statistics import GroupStatistics, GroupStorage
 from mealie.schema.user.user import GroupInDB, UserOut
+from mealie.services.group_services.group_service import GroupService
 
 router = UserAPIRouter(prefix="/groups", tags=["Groups: Self Service"])
 
 
 @controller(router)
 class GroupSelfServiceController(BaseUserController):
+    @cached_property
+    def service(self) -> GroupService:
+        return GroupService(self.group_id, self.repos)
+
+    @router.get("/self", response_model=GroupInDB)
+    def get_logged_in_user_group(self):
+        """Returns the Group Data for the Current User"""
+        return self.group
+
+    @router.get("/members", response_model=list[UserOut])
+    def get_group_members(self):
+        """Returns the Group of user lists"""
+        return self.repos.users.multi_query(query_by={"group_id": self.group.id}, override_schema=UserOut)
+
     @router.get("/preferences", response_model=ReadGroupPreferences)
     def get_group_preferences(self):
         return self.group.preferences
@@ -20,18 +38,8 @@ class GroupSelfServiceController(BaseUserController):
     def update_group_preferences(self, new_pref: UpdateGroupPreferences):
         return self.repos.group_preferences.update(self.group_id, new_pref)
 
-    @router.get("/self", response_model=GroupInDB)
-    async def get_logged_in_user_group(self):
-        """Returns the Group Data for the Current User"""
-        return self.group
-
-    @router.get("/members", response_model=list[UserOut])
-    async def get_group_members(self):
-        """Returns the Group of user lists"""
-        return self.repos.users.multi_query(query_by={"group_id": self.group.id}, override_schema=UserOut)
-
     @router.put("/permissions", response_model=UserOut)
-    async def set_member_permissions(self, permissions: SetPermissions):
+    def set_member_permissions(self, permissions: SetPermissions):
         self.checks.can_manage()
 
         target_user = self.repos.users.get(permissions.user_id)
@@ -47,3 +55,11 @@ class GroupSelfServiceController(BaseUserController):
         target_user.can_organize = permissions.can_organize
 
         return self.repos.users.update(permissions.user_id, target_user)
+
+    @router.get("/statistics", response_model=GroupStatistics)
+    def get_statistics(self):
+        return self.service.calculate_statistics()
+
+    @router.get("/storage", response_model=GroupStorage)
+    def get_storage(self):
+        return self.service.calculate_group_storage()
diff --git a/mealie/schema/group/__init__.py b/mealie/schema/group/__init__.py
index 508495e2f0e1..224c1d967e7a 100644
--- a/mealie/schema/group/__init__.py
+++ b/mealie/schema/group/__init__.py
@@ -6,5 +6,6 @@ from .group_migration import *
 from .group_permissions import *
 from .group_preferences import *
 from .group_shopping_list import *
+from .group_statistics import *
 from .invite_token import *
 from .webhook import *
diff --git a/mealie/schema/group/group_statistics.py b/mealie/schema/group/group_statistics.py
new file mode 100644
index 000000000000..ba2fefbe9332
--- /dev/null
+++ b/mealie/schema/group/group_statistics.py
@@ -0,0 +1,26 @@
+from mealie.pkgs.stats import fs_stats
+from mealie.schema._mealie import MealieModel
+
+
+class GroupStatistics(MealieModel):
+    total_recipes: int
+    total_users: int
+    total_categories: int
+    total_tags: int
+    total_tools: int
+
+
+class GroupStorage(MealieModel):
+    used_storage_bytes: int
+    used_storage_str: str
+    total_storage_bytes: int
+    total_storage_str: str
+
+    @classmethod
+    def bytes(cls, used_storage_bytes: int, total_storage_bytes: int) -> "GroupStorage":
+        return cls(
+            used_storage_bytes=used_storage_bytes,
+            used_storage_str=fs_stats.pretty_size(used_storage_bytes),
+            total_storage_bytes=total_storage_bytes,
+            total_storage_str=fs_stats.pretty_size(total_storage_bytes),
+        )
diff --git a/mealie/services/group_services/group_service.py b/mealie/services/group_services/group_service.py
new file mode 100644
index 000000000000..48a299c95480
--- /dev/null
+++ b/mealie/services/group_services/group_service.py
@@ -0,0 +1,40 @@
+from pydantic import UUID4
+
+from mealie.pkgs.stats import fs_stats
+from mealie.repos.repository_factory import AllRepositories
+from mealie.schema.group.group_statistics import GroupStatistics, GroupStorage
+from mealie.services._base_service import BaseService
+
+ALLOWED_SIZE = 500 * fs_stats.megabyte
+
+
+class GroupService(BaseService):
+    def __init__(self, group_id: UUID4, repos: AllRepositories):
+        self.group_id = group_id
+        self.repos = repos
+        super().__init__()
+
+    def calculate_statistics(self, group_id: None | UUID4 = None) -> GroupStatistics:
+        """
+        calculate_statistics calculates the statistics for the group and returns
+        a GroupStatistics object.
+        """
+        target_id = group_id or self.group_id
+
+        return self.repos.groups.statistics(target_id)
+
+    def calculate_group_storage(self, group_id: None | UUID4 = None) -> GroupStorage:
+        """
+        calculate_group_storage calculates the storage used by the group and returns
+        a GroupStorage object.
+        """
+
+        target_id = group_id or self.group_id
+
+        all_ids = self.repos.recipes.all_ids(target_id)
+
+        used_size = sum(
+            fs_stats.get_dir_size(f"{self.directories.RECIPE_DATA_DIR}/{str(recipe_id)}") for recipe_id in all_ids
+        )
+
+        return GroupStorage.bytes(used_size, ALLOWED_SIZE)