From ff65a4b93592d7da25a2693042656c89461e6d54 Mon Sep 17 00:00:00 2001
From: Eliot Berriot
Date: Fri, 18 May 2018 18:47:35 +0200
Subject: [PATCH 1/6] See #152: added permission fields on user model and
corresponding API permission
---
.../migrations/0006_auto_20180517_2324.py | 28 ++++++++++
api/funkwhale_api/users/models.py | 39 +++++++------
api/funkwhale_api/users/permissions.py | 19 +++++++
api/tests/users/test_models.py | 34 +++++++++++
api/tests/users/test_permissions.py | 56 +++++++++++++++++++
5 files changed, 159 insertions(+), 17 deletions(-)
create mode 100644 api/funkwhale_api/users/migrations/0006_auto_20180517_2324.py
create mode 100644 api/funkwhale_api/users/permissions.py
create mode 100644 api/tests/users/test_permissions.py
diff --git a/api/funkwhale_api/users/migrations/0006_auto_20180517_2324.py b/api/funkwhale_api/users/migrations/0006_auto_20180517_2324.py
new file mode 100644
index 000000000..7c9ab0fad
--- /dev/null
+++ b/api/funkwhale_api/users/migrations/0006_auto_20180517_2324.py
@@ -0,0 +1,28 @@
+# Generated by Django 2.0.4 on 2018-05-17 23:24
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+ dependencies = [
+ ('users', '0005_user_subsonic_api_token'),
+ ]
+
+ operations = [
+ migrations.AddField(
+ model_name='user',
+ name='permission_federation',
+ field=models.BooleanField(default=False),
+ ),
+ migrations.AddField(
+ model_name='user',
+ name='permission_library',
+ field=models.BooleanField(default=False),
+ ),
+ migrations.AddField(
+ model_name='user',
+ name='permission_settings',
+ field=models.BooleanField(default=False),
+ ),
+ ]
diff --git a/api/funkwhale_api/users/models.py b/api/funkwhale_api/users/models.py
index 8273507c4..1de23092a 100644
--- a/api/funkwhale_api/users/models.py
+++ b/api/funkwhale_api/users/models.py
@@ -19,6 +19,13 @@ def get_token():
return binascii.b2a_hex(os.urandom(15)).decode('utf-8')
+PERMISSIONS = [
+ 'federation',
+ 'library',
+ 'settings',
+]
+
+
@python_2_unicode_compatible
class User(AbstractUser):
@@ -28,20 +35,6 @@ class User(AbstractUser):
# updated on logout or password change, to invalidate JWT
secret_key = models.UUIDField(default=uuid.uuid4, null=True)
- # permissions that are used for API access and that worth serializing
- relevant_permissions = {
- # internal_codename : {external_codename}
- 'music.add_importbatch': {
- 'external_codename': 'import.launch',
- },
- 'dynamic_preferences.change_globalpreferencemodel': {
- 'external_codename': 'settings.change',
- },
- 'federation.change_library': {
- 'external_codename': 'federation.manage',
- },
- }
-
privacy_level = fields.get_privacy_field()
# Unfortunately, Subsonic API assumes a MD5/password authentication
@@ -52,12 +45,24 @@ class User(AbstractUser):
subsonic_api_token = models.CharField(
blank=True, null=True, max_length=255)
+ # permissions
+ permission_federation = models.BooleanField(default=False)
+ permission_library = models.BooleanField(default=False)
+ permission_settings = models.BooleanField(default=False)
+
def __str__(self):
return self.username
- def add_permission(self, codename):
- p = Permission.objects.get(codename=codename)
- self.user_permissions.add(p)
+ def get_permissions(self):
+ perms = {}
+ for p in PERMISSIONS:
+ v = self.is_superuser or getattr(self, 'permission_{}'.format(p))
+ perms[p] = v
+ return perms
+
+ def has_permissions(self, *perms):
+ permissions = self.get_permissions()
+ return all([permissions[p] for p in perms])
def get_absolute_url(self):
return reverse('users:detail', kwargs={'username': self.username})
diff --git a/api/funkwhale_api/users/permissions.py b/api/funkwhale_api/users/permissions.py
new file mode 100644
index 000000000..2ff49ff3f
--- /dev/null
+++ b/api/funkwhale_api/users/permissions.py
@@ -0,0 +1,19 @@
+from rest_framework.permissions import BasePermission
+
+
+class HasUserPermission(BasePermission):
+ """
+ Ensure the request user has the proper permissions.
+
+ Usage:
+
+ class MyView(APIView):
+ permission_classes = [HasUserPermission]
+ required_permissions = ['federation']
+ """
+ def has_permission(self, request, view):
+ if not hasattr(request, 'user') or not request.user:
+ return False
+ if request.user.is_anonymous:
+ return False
+ return request.user.has_permissions(*view.required_permissions)
diff --git a/api/tests/users/test_models.py b/api/tests/users/test_models.py
index c7cd12e9e..49199e0a7 100644
--- a/api/tests/users/test_models.py
+++ b/api/tests/users/test_models.py
@@ -1,3 +1,7 @@
+import pytest
+
+from funkwhale_api.users import models
+
def test__str__(factories):
user = factories['users.User'](username='hello')
@@ -16,3 +20,33 @@ def test_changing_password_updates_subsonic_api_token(factories):
assert user.subsonic_api_token is not None
assert user.subsonic_api_token != 'test'
+
+
+def test_get_permissions_superuser(factories):
+ user = factories['users.User'](is_superuser=True)
+
+ perms = user.get_permissions()
+ for p in models.PERMISSIONS:
+ assert perms[p] is True
+
+
+def test_get_permissions_regular(factories):
+ user = factories['users.User'](permission_library=True)
+
+ perms = user.get_permissions()
+ for p in models.PERMISSIONS:
+ if p == 'library':
+ assert perms[p] is True
+ else:
+ assert perms[p] is False
+
+
+@pytest.mark.parametrize('args,perms,expected', [
+ ({'is_superuser': True}, ['federation', 'library'], True),
+ ({'is_superuser': False}, ['federation'], False),
+ ({'permission_library': True}, ['library'], True),
+ ({'permission_library': True}, ['library', 'federation'], False),
+])
+def test_has_permissions(args, perms, expected, factories):
+ user = factories['users.User'](**args)
+ assert user.has_permissions(*perms) is expected
diff --git a/api/tests/users/test_permissions.py b/api/tests/users/test_permissions.py
new file mode 100644
index 000000000..1564c761d
--- /dev/null
+++ b/api/tests/users/test_permissions.py
@@ -0,0 +1,56 @@
+import pytest
+from rest_framework.views import APIView
+
+from funkwhale_api.users import permissions
+
+
+def test_has_user_permission_no_user(api_request):
+ view = APIView.as_view()
+ permission = permissions.HasUserPermission()
+ request = api_request.get('/')
+ assert permission.has_permission(request, view) is False
+
+
+def test_has_user_permission_anonymous(anonymous_user, api_request):
+ view = APIView.as_view()
+ permission = permissions.HasUserPermission()
+ request = api_request.get('/')
+ setattr(request, 'user', anonymous_user)
+ assert permission.has_permission(request, view) is False
+
+
+@pytest.mark.parametrize('value', [True, False])
+def test_has_user_permission_logged_in_single(value, factories, api_request):
+ user = factories['users.User'](permission_federation=value)
+
+ class View(APIView):
+ required_permissions = ['federation']
+ view = View()
+ permission = permissions.HasUserPermission()
+ request = api_request.get('/')
+ setattr(request, 'user', user)
+ result = permission.has_permission(request, view)
+ assert result == user.has_permissions('federation') == value
+
+
+@pytest.mark.parametrize('federation,library,expected', [
+ (True, False, False),
+ (False, True, False),
+ (False, False, False),
+ (True, True, True),
+])
+def test_has_user_permission_logged_in_single(
+ federation, library, expected, factories, api_request):
+ user = factories['users.User'](
+ permission_federation=federation,
+ permission_library=library,
+ )
+
+ class View(APIView):
+ required_permissions = ['federation', 'library']
+ view = View()
+ permission = permissions.HasUserPermission()
+ request = api_request.get('/')
+ setattr(request, 'user', user)
+ result = permission.has_permission(request, view)
+ assert result == user.has_permissions('federation', 'library') == expected
From 6fc4275b681ac65e1ebd289b55e99856408af12f Mon Sep 17 00:00:00 2001
From: Eliot Berriot
Date: Fri, 18 May 2018 18:48:46 +0200
Subject: [PATCH 2/6] See #152: use new user permissions on relevant viewsets
---
api/funkwhale_api/common/permissions.py | 13 +---------
api/funkwhale_api/federation/views.py | 12 ++++-----
api/funkwhale_api/instance/views.py | 4 ++-
api/funkwhale_api/music/views.py | 18 ++++++--------
api/funkwhale_api/users/serializers.py | 9 ++-----
api/tests/conftest.py | 9 +++++++
api/tests/federation/test_views.py | 9 +++++++
api/tests/instance/test_views.py | 14 ++++++++++-
api/tests/music/test_views.py | 8 ++++++
api/tests/users/test_views.py | 33 ++++++++++---------------
10 files changed, 71 insertions(+), 58 deletions(-)
diff --git a/api/funkwhale_api/common/permissions.py b/api/funkwhale_api/common/permissions.py
index cab4b699d..e9e8b8819 100644
--- a/api/funkwhale_api/common/permissions.py
+++ b/api/funkwhale_api/common/permissions.py
@@ -3,7 +3,7 @@ import operator
from django.conf import settings
from django.http import Http404
-from rest_framework.permissions import BasePermission, DjangoModelPermissions
+from rest_framework.permissions import BasePermission
from funkwhale_api.common import preferences
@@ -16,17 +16,6 @@ class ConditionalAuthentication(BasePermission):
return True
-class HasModelPermission(DjangoModelPermissions):
- """
- Same as DjangoModelPermissions, but we pin the model:
-
- class MyModelPermission(HasModelPermission):
- model = User
- """
- def get_required_permissions(self, method, model_cls):
- return super().get_required_permissions(method, self.model)
-
-
class OwnerPermission(BasePermission):
"""
Ensure the request user is the owner of the object.
diff --git a/api/funkwhale_api/federation/views.py b/api/funkwhale_api/federation/views.py
index 37ad9ebfd..06a2cd040 100644
--- a/api/funkwhale_api/federation/views.py
+++ b/api/funkwhale_api/federation/views.py
@@ -15,8 +15,8 @@ from rest_framework.serializers import ValidationError
from funkwhale_api.common import preferences
from funkwhale_api.common import utils as funkwhale_utils
-from funkwhale_api.common.permissions import HasModelPermission
from funkwhale_api.music.models import TrackFile
+from funkwhale_api.users.permissions import HasUserPermission
from . import activity
from . import actors
@@ -187,16 +187,13 @@ class MusicFilesViewSet(FederationMixin, viewsets.GenericViewSet):
return response.Response(data)
-class LibraryPermission(HasModelPermission):
- model = models.Library
-
-
class LibraryViewSet(
mixins.RetrieveModelMixin,
mixins.UpdateModelMixin,
mixins.ListModelMixin,
viewsets.GenericViewSet):
- permission_classes = [LibraryPermission]
+ permission_classes = (HasUserPermission,)
+ required_permissions = ['federation']
queryset = models.Library.objects.all().select_related(
'actor',
'follow',
@@ -291,7 +288,8 @@ class LibraryViewSet(
class LibraryTrackViewSet(
mixins.ListModelMixin,
viewsets.GenericViewSet):
- permission_classes = [LibraryPermission]
+ permission_classes = (HasUserPermission,)
+ required_permissions = ['federation']
queryset = models.LibraryTrack.objects.all().select_related(
'library__actor',
'library__follow',
diff --git a/api/funkwhale_api/instance/views.py b/api/funkwhale_api/instance/views.py
index e6725e248..b905acd3e 100644
--- a/api/funkwhale_api/instance/views.py
+++ b/api/funkwhale_api/instance/views.py
@@ -6,6 +6,7 @@ from dynamic_preferences.api import viewsets as preferences_viewsets
from dynamic_preferences.registries import global_preferences_registry
from funkwhale_api.common import preferences
+from funkwhale_api.users.permissions import HasUserPermission
from . import nodeinfo
from . import stats
@@ -18,7 +19,8 @@ NODEINFO_2_CONTENT_TYPE = (
class AdminSettings(preferences_viewsets.GlobalPreferencesViewSet):
pagination_class = None
-
+ permission_classes = (HasUserPermission,)
+ required_permissions = ['settings']
class InstanceSettings(views.APIView):
permission_classes = []
diff --git a/api/funkwhale_api/music/views.py b/api/funkwhale_api/music/views.py
index f2ab72c5a..e71d3555e 100644
--- a/api/funkwhale_api/music/views.py
+++ b/api/funkwhale_api/music/views.py
@@ -25,8 +25,8 @@ from rest_framework import permissions
from musicbrainzngs import ResponseError
from funkwhale_api.common import utils as funkwhale_utils
-from funkwhale_api.common.permissions import (
- ConditionalAuthentication, HasModelPermission)
+from funkwhale_api.common.permissions import ConditionalAuthentication
+from funkwhale_api.users.permissions import HasUserPermission
from taggit.models import Tag
from funkwhale_api.federation import actors
from funkwhale_api.federation.authentication import SignatureAuthentication
@@ -107,25 +107,22 @@ class ImportBatchViewSet(
.annotate(job_count=Count('jobs'))
)
serializer_class = serializers.ImportBatchSerializer
- permission_classes = (permissions.DjangoModelPermissions, )
+ permission_classes = (HasUserPermission,)
+ required_permissions = ['library']
filter_class = filters.ImportBatchFilter
def perform_create(self, serializer):
serializer.save(submitted_by=self.request.user)
-class ImportJobPermission(HasModelPermission):
- # not a typo, perms on import job is proxied to import batch
- model = models.ImportBatch
-
-
class ImportJobViewSet(
mixins.CreateModelMixin,
mixins.ListModelMixin,
viewsets.GenericViewSet):
queryset = (models.ImportJob.objects.all().select_related())
serializer_class = serializers.ImportJobSerializer
- permission_classes = (ImportJobPermission, )
+ permission_classes = (HasUserPermission,)
+ required_permissions = ['library']
filter_class = filters.ImportJobFilter
@list_route(methods=['get'])
@@ -442,7 +439,8 @@ class Search(views.APIView):
class SubmitViewSet(viewsets.ViewSet):
queryset = models.ImportBatch.objects.none()
- permission_classes = (permissions.DjangoModelPermissions, )
+ permission_classes = (HasUserPermission,)
+ required_permissions = ['library']
@list_route(methods=['post'])
@transaction.non_atomic_requests
diff --git a/api/funkwhale_api/users/serializers.py b/api/funkwhale_api/users/serializers.py
index eadce6154..3a095e78a 100644
--- a/api/funkwhale_api/users/serializers.py
+++ b/api/funkwhale_api/users/serializers.py
@@ -55,16 +55,11 @@ class UserReadSerializer(serializers.ModelSerializer):
'is_superuser',
'permissions',
'date_joined',
- 'privacy_level'
+ 'privacy_level',
]
def get_permissions(self, o):
- perms = {}
- for internal_codename, conf in o.relevant_permissions.items():
- perms[conf['external_codename']] = {
- 'status': o.has_perm(internal_codename)
- }
- return perms
+ return o.get_permissions()
class PasswordResetSerializer(PRS):
diff --git a/api/tests/conftest.py b/api/tests/conftest.py
index dda537801..b7a7d071a 100644
--- a/api/tests/conftest.py
+++ b/api/tests/conftest.py
@@ -14,6 +14,7 @@ from rest_framework.test import APIClient
from rest_framework.test import APIRequestFactory
from funkwhale_api.activity import record
+from funkwhale_api.users.permissions import HasUserPermission
from funkwhale_api.taskapp import celery
@@ -224,3 +225,11 @@ def authenticated_actor(factories, mocker):
'funkwhale_api.federation.authentication.SignatureAuthentication.authenticate_actor',
return_value=actor)
yield actor
+
+
+@pytest.fixture
+def assert_user_permission():
+ def inner(view, permissions):
+ assert HasUserPermission in view.permission_classes
+ assert set(view.required_permissions) == set(permissions)
+ return inner
diff --git a/api/tests/federation/test_views.py b/api/tests/federation/test_views.py
index fd6ac2eb2..10237ed9f 100644
--- a/api/tests/federation/test_views.py
+++ b/api/tests/federation/test_views.py
@@ -9,9 +9,18 @@ from funkwhale_api.federation import activity
from funkwhale_api.federation import models
from funkwhale_api.federation import serializers
from funkwhale_api.federation import utils
+from funkwhale_api.federation import views
from funkwhale_api.federation import webfinger
+@pytest.mark.parametrize('view,permissions', [
+ (views.LibraryViewSet, ['federation']),
+ (views.LibraryTrackViewSet, ['federation']),
+])
+def test_permissions(assert_user_permission, view, permissions):
+ assert_user_permission(view, permissions)
+
+
@pytest.mark.parametrize('system_actor', actors.SYSTEM_ACTORS.keys())
def test_instance_actors(system_actor, db, api_client):
actor = actors.SYSTEM_ACTORS[system_actor].get_actor_instance()
diff --git a/api/tests/instance/test_views.py b/api/tests/instance/test_views.py
index 6d8dcac3e..daf54db51 100644
--- a/api/tests/instance/test_views.py
+++ b/api/tests/instance/test_views.py
@@ -1,5 +1,16 @@
+import pytest
+
from django.urls import reverse
+from funkwhale_api.instance import views
+
+
+@pytest.mark.parametrize('view,permissions', [
+ (views.AdminSettings, ['settings']),
+])
+def test_permissions(assert_user_permission, view, permissions):
+ assert_user_permission(view, permissions)
+
def test_nodeinfo_endpoint(db, api_client, mocker):
payload = {
@@ -43,7 +54,8 @@ def test_admin_settings_restrict_access(db, logged_in_api_client, preferences):
def test_admin_settings_correct_permission(
db, logged_in_api_client, preferences):
user = logged_in_api_client.user
- user.add_permission('change_globalpreferencemodel')
+ user.permission_settings = True
+ user.save()
url = reverse('api:v1:instance:admin-settings-list')
response = logged_in_api_client.get(url)
diff --git a/api/tests/music/test_views.py b/api/tests/music/test_views.py
index e641b45d5..030fc3a73 100644
--- a/api/tests/music/test_views.py
+++ b/api/tests/music/test_views.py
@@ -8,6 +8,14 @@ from funkwhale_api.music import views
from funkwhale_api.federation import actors
+@pytest.mark.parametrize('view,permissions', [
+ (views.ImportBatchViewSet, ['library']),
+ (views.ImportJobViewSet, ['library']),
+])
+def test_permissions(assert_user_permission, view, permissions):
+ assert_user_permission(view, permissions)
+
+
@pytest.mark.parametrize('param,expected', [
('true', 'full'),
('false', 'empty'),
diff --git a/api/tests/users/test_views.py b/api/tests/users/test_views.py
index fffc762fd..1bbf8b9a2 100644
--- a/api/tests/users/test_views.py
+++ b/api/tests/users/test_views.py
@@ -53,33 +53,24 @@ def test_can_disable_registration_view(preferences, client, db):
assert response.status_code == 403
-def test_can_fetch_data_from_api(client, factories):
+def test_can_fetch_data_from_api(api_client, factories):
url = reverse('api:v1:users:users-me')
- response = client.get(url)
+ response = api_client.get(url)
# login required
assert response.status_code == 401
user = factories['users.User'](
- is_staff=True,
- perms=[
- 'music.add_importbatch',
- 'dynamic_preferences.change_globalpreferencemodel',
- ]
+ permission_library=True
)
- assert user.has_perm('music.add_importbatch')
- client.login(username=user.username, password='test')
- response = client.get(url)
+ api_client.login(username=user.username, password='test')
+ response = api_client.get(url)
assert response.status_code == 200
-
- payload = json.loads(response.content.decode('utf-8'))
-
- assert payload['username'] == user.username
- assert payload['is_staff'] == user.is_staff
- assert payload['is_superuser'] == user.is_superuser
- assert payload['email'] == user.email
- assert payload['name'] == user.name
- assert payload['permissions']['import.launch']['status']
- assert payload['permissions']['settings.change']['status']
+ assert response.data['username'] == user.username
+ assert response.data['is_staff'] == user.is_staff
+ assert response.data['is_superuser'] == user.is_superuser
+ assert response.data['email'] == user.email
+ assert response.data['name'] == user.name
+ assert response.data['permissions'] == user.get_permissions()
def test_can_get_token_via_api(client, factories):
@@ -202,6 +193,8 @@ def test_user_can_get_new_subsonic_token(logged_in_api_client):
assert response.data == {
'subsonic_api_token': 'test'
}
+
+
def test_user_can_request_new_subsonic_token(logged_in_api_client):
user = logged_in_api_client.user
user.subsonic_api_token = 'test'
From 4ce6715dc7a3cf6946dc38b12e02790a9424d2fa Mon Sep 17 00:00:00 2001
From: Eliot Berriot
Date: Fri, 18 May 2018 19:09:47 +0200
Subject: [PATCH 3/6] See #152: updated front-end to use new permissions
---
front/src/components/About.vue | 2 +-
front/src/components/Sidebar.vue | 14 +++++++-------
front/src/components/library/Library.vue | 4 ++--
front/src/components/requests/Card.vue | 2 +-
front/src/store/auth.js | 2 +-
front/test/unit/specs/store/auth.spec.js | 4 +---
6 files changed, 13 insertions(+), 15 deletions(-)
diff --git a/front/src/components/About.vue b/front/src/components/About.vue
index b0ae67ef7..59c8411ac 100644
--- a/front/src/components/About.vue
+++ b/front/src/components/About.vue
@@ -15,7 +15,7 @@
{{ $t('Edit instance info') }}
diff --git a/front/src/components/Sidebar.vue b/front/src/components/Sidebar.vue
index 9fbc5605c..9f3134c2a 100644
--- a/front/src/components/Sidebar.vue
+++ b/front/src/components/Sidebar.vue
@@ -60,7 +60,7 @@