feat: multi tenancy support

app/apps/transactions/views/entities.py

@@ -8,6 +8,8 @@ from django.views.decorators.http import require_http_methods
 from apps.common.decorators.htmx import only_htmx
 from apps.transactions.forms import TransactionEntityForm
 from apps.transactions.models import TransactionEntity
+from apps.common.models import SharedObject
+from apps.common.forms import SharedObjectForm
 
 
 @login_required
@@ -85,6 +87,16 @@ def entity_add(request, **kwargs):
 def entity_edit(request, entity_id):
     entity = get_object_or_404(TransactionEntity, id=entity_id)
 
+    if entity.owner and entity.owner != request.user:
+        messages.error(request, _("Only the owner can edit this"))
+
+        return HttpResponse(
+            status=204,
+            headers={
+                "HX-Trigger": "updated, hide_offcanvas",
+            },
+        )
+
     if request.method == "POST":
         form = TransactionEntityForm(request.POST, instance=entity)
         if form.is_valid():
@@ -113,9 +125,12 @@ def entity_edit(request, entity_id):
 def entity_delete(request, entity_id):
     entity = get_object_or_404(TransactionEntity, id=entity_id)
 
-    entity.delete()
-
-    messages.success(request, _("Entity deleted successfully"))
+    if entity.owner != request.user and request.user in entity.shared_with.all():
+        entity.shared_with.remove(request.user)
+        messages.success(request, _("Item no longer shared with you"))
+    else:
+        entity.delete()
+        messages.success(request, _("Entity deleted successfully"))
 
     return HttpResponse(
         status=204,
@@ -123,3 +138,62 @@ def entity_delete(request, entity_id):
             "HX-Trigger": "updated, hide_offcanvas",
         },
     )
+
+
+@only_htmx
+@login_required
+@require_http_methods(["GET"])
+def entity_take_ownership(request, entity_id):
+    entity = get_object_or_404(TransactionEntity, id=entity_id)
+
+    if not entity.owner:
+        entity.owner = request.user
+        entity.visibility = SharedObject.Visibility.private
+        entity.save()
+
+        messages.success(request, _("Ownership taken successfully"))
+
+    return HttpResponse(
+        status=204,
+        headers={
+            "HX-Trigger": "updated, hide_offcanvas",
+        },
+    )
+
+
+@only_htmx
+@login_required
+@require_http_methods(["GET", "POST"])
+def entity_share(request, pk):
+    obj = get_object_or_404(TransactionEntity, id=pk)
+
+    if obj.owner and obj.owner != request.user:
+        messages.error(request, _("Only the owner can edit this"))
+
+        return HttpResponse(
+            status=204,
+            headers={
+                "HX-Trigger": "updated, hide_offcanvas",
+            },
+        )
+
+    if request.method == "POST":
+        form = SharedObjectForm(request.POST, instance=obj, user=request.user)
+        if form.is_valid():
+            form.save()
+            messages.success(request, _("Configuration saved successfully"))
+
+            return HttpResponse(
+                status=204,
+                headers={
+                    "HX-Trigger": "updated, hide_offcanvas",
+                },
+            )
+    else:
+        form = SharedObjectForm(instance=obj, user=request.user)
+
+    return render(
+        request,
+        "entities/fragments/share.html",
+        {"form": form, "object": obj},
+    )