From 86c423b07dec419f308d9c8fcfe05d45793bf8cc Mon Sep 17 00:00:00 2001
From: grimhilt <grimhilt@users.noreply.github.com>
Date: Mon, 7 Aug 2023 16:52:11 +0200
Subject: [PATCH] imporve grant access

---
 src/pages/playlist/content.jsx |  3 +++
 src/pages/playlist/index.jsx   | 39 +++++++++++++++++++++++-----------
 src/tools/grant-access.jsx     | 23 +++++++++++++++-----
 3 files changed, 48 insertions(+), 17 deletions(-)

diff --git a/src/pages/playlist/content.jsx b/src/pages/playlist/content.jsx
index 172649f..412a1c6 100644
--- a/src/pages/playlist/content.jsx
+++ b/src/pages/playlist/content.jsx
@@ -25,6 +25,7 @@ const Content = ({ form, playlistId }) => {
 
             file.position = max_position;
             file.seconds = 10;
+            const index = form.values.files.length;
             form.insertListItem('files', file);
             API.addFileToPlaylist(playlistId, { position: file.position, file_id: file.id, seconds: file.seconds })
                 .then((res) => {
@@ -33,6 +34,8 @@ const Content = ({ form, playlistId }) => {
                     }
                 })
                 .catch((err) => {
+                    console.log("here")
+                    form.removeListItem('files', index)
                     setNotification(true, err);
                 });
         });
diff --git a/src/pages/playlist/index.jsx b/src/pages/playlist/index.jsx
index b5ab43a..cff0f55 100644
--- a/src/pages/playlist/index.jsx
+++ b/src/pages/playlist/index.jsx
@@ -7,6 +7,7 @@ import ModalUpdate from '../playlists/update';
 import { useForm } from '@mantine/form';
 import Content from './content';
 import { useNavigate } from 'react-router-dom';
+import GrantAccess, { Perm } from '../../tools/grant-access';
 
 const Playlist = (item) => {
     const id = window.location.href.split('/').slice(-1)[0];
@@ -68,6 +69,8 @@ const Playlist = (item) => {
                     setNotification(true, err);
                     if (err.response.status === 404) {
                         navigate('/playlists');
+                    } else if (err.response.status === 401) {
+                        navigate('/login');
                     }
                 });
         }
@@ -87,18 +90,30 @@ const Playlist = (item) => {
                     </Text>
                 </Text>
                 <Group>
-                    <Button
-                        variant="light"
-                        mt="sm"
-                        color={isActive ? 'red' : 'green'}
-                        onClick={toggleActivate}
-                        loading={isLoading}
-                    >
-                        {isActive ? 'Stop' : 'Activate'}
-                    </Button>
-                    <Button variant="light" mt="sm" onClick={toggleUpdate}>
-                        Edit
-                    </Button>
+                    <GrantAccess
+                        roles={[Perm.OWN_PLAYLIST, Perm.ACTIVATE_PLAYLIST]}
+                        item={playlist}
+                        children={
+                            <Button
+                                variant="light"
+                                mt="sm"
+                                color={isActive ? 'red' : 'green'}
+                                onClick={toggleActivate}
+                                loading={isLoading}
+                            >
+                                {isActive ? 'Stop' : 'Activate'}
+                            </Button>
+                        }
+                    />
+                    <GrantAccess
+                        role={Perm.OWN_PLAYLIST}
+                        item={playlist}
+                        children={
+                            <Button variant="light" mt="sm" onClick={toggleUpdate}>
+                                Edit
+                            </Button>
+                        }
+                    />
                 </Group>
             </Group>
             <Paper p="xs" radius="sm" shadow="sm" withBorder my="md">
diff --git a/src/tools/grant-access.jsx b/src/tools/grant-access.jsx
index baa7197..2037266 100644
--- a/src/tools/grant-access.jsx
+++ b/src/tools/grant-access.jsx
@@ -1,5 +1,6 @@
 import { useAuth } from './auth-provider';
 import Authentication from '../pages/auth';
+import { useEffect } from 'react';
 
 export const Perm = {
     CREATE_ROLE: 0,
@@ -7,26 +8,38 @@ export const Perm = {
     VIEW_PLAYLIST: 2,
     OWN_PLAYLIST: 3,
     EDIT_PLAYLIST: 4,
+    ACTIVATE_PLAYLIST: 5,
 };
 
-const checkPerm = (perm, user) => {
+const checkPerm = (perm, user, item = {}) => {
     console.log(user);
+    console.log(item);
     switch (perm) {
         case Perm.CREATE_ROLE:
             return false;
         case Perm.CREATE_PLAYLIST:
             return user.roles.findIndex((role) => role.can_create_playlist) !== -1;
+        case Perm.OWN_PLAYLIST:
+            return item?.owner_id === user.id;
         default:
             return false;
     }
 };
 
-const GrantAccess = ({ role, roles, children }) => {
+const GrantAccess = ({ role, roles, children, item }) => {
     const { user } = useAuth();
-    if (role && checkPerm(role, user)) {
-        return children;
-    } else if (roles && roles.includes(user)) {
+    if (role && checkPerm(role, user, item)) {
         return children;
+    } else if (roles) {
+        let flag = false;
+        let i = 0;
+        while (!flag && i < roles.length) {
+            if (checkPerm(roles[i], user, item)) {
+                flag = true;
+            }
+            i++;
+        }
+        if (flag) return children;
     }
     return null;
 };