From 8eae8e624f417ced25294d9cd6ebf7b528de45bf Mon Sep 17 00:00:00 2001
From: jorgepastorr <jorge@pangea.org>
Date: Wed, 3 Jan 2024 18:53:46 +0100
Subject: [PATCH] quota nextcloud

---
 orchestra/contrib/saas/backends/nextcloud.py | 40 +++++++++++++++++---
 1 file changed, 34 insertions(+), 6 deletions(-)

diff --git a/orchestra/contrib/saas/backends/nextcloud.py b/orchestra/contrib/saas/backends/nextcloud.py
index 3c3ef0a1..b11650a7 100644
--- a/orchestra/contrib/saas/backends/nextcloud.py
+++ b/orchestra/contrib/saas/backends/nextcloud.py
@@ -10,6 +10,8 @@ from django.utils.translation import gettext_lazy as _
 
 from orchestra.contrib.orchestration import ServiceController
 from orchestra.contrib.resources import ServiceMonitor
+from orchestra.contrib.resources.models import ResourceData
+from orchestra.contrib.saas.models import SaaS
 
 from . import ApacheTrafficByName
 from .. import settings
@@ -52,11 +54,31 @@ class NextCloudAPIMixin(object):
     def create(self, saas):
         data = {
             'userid': saas.name,
-            'password': saas.password
+            'password': saas.password,
         }
         self.api_post('users', data)
+
+    def update_group(self, saas):
+        data = {
+            'groupid': saas.account.username
+        }
+        try:
+            self.api_get('groups/%s' % saas.account.username)
+        except RuntimeError:
+            self.api_post('groups', data)
+        self.api_post(f'users/{saas.name}/groups', data)
     
-    def update(self, saas):
+    def update_quota(self, saas):
+        if hasattr(saas, 'resources') and hasattr(saas.resources, 'nextcloud-disk'):
+            resource = getattr(saas.resources, 'nextcloud-disk')
+            quotaValue = f"{resource.allocated}G" if resource.allocated > 0 else "default"
+            data = {
+                'key': "quota",
+                'value': quotaValue
+            }
+            self.api_put(f'users/{saas.name}', data)
+
+    def update_password(self, saas):
         """
         key: email|quota|display|password
         value: el valor a modificar.
@@ -70,6 +92,7 @@ class NextCloudAPIMixin(object):
         }
         self.api_put('users/%s' % saas.name, data)
     
+    
     def get_user(self, saas):
         """
         {
@@ -112,13 +135,18 @@ class NextCloudController(NextCloudAPIMixin, ServiceController):
         try:
             self.api_get('users/%s' % saas.name)
         except RuntimeError:
-            if getattr(saas, 'password'):
+            if getattr(saas, 'password', None):
                 self.create(saas)
+                self.update_group(saas)
+                self.update_quota(saas)
             else:
                 raise
         else:
-            if getattr(saas, 'password'):
-                self.update(saas)
+            if getattr(saas, 'password', None):
+                self.update_password(saas)
+            else:
+                self.update_group(saas)
+                self.update_quota(saas)
     
     def remove(self, saas, server):
         self.api_delete('users/%s' % saas.name)
@@ -126,7 +154,7 @@ class NextCloudController(NextCloudAPIMixin, ServiceController):
     def save(self, saas):
         # TODO disable user https://github.com/owncloud/core/issues/12601
         self.append(self.update_or_create, saas)
-    
+          
     def delete(self, saas):
         self.append(self.remove, saas)