auth-wip

resolvers/admin.py

@@ -1,11 +1,12 @@
 from math import ceil
-from sqlalchemy import or_
+from sqlalchemy import or_, cast, String
 from graphql.error import GraphQLError
 
 from auth.decorators import admin_auth_required
 from services.db import local_session
-from services.schema import query
-from auth.orm import Author, Role
+from services.schema import query, mutation
+from auth.orm import Author, Role, AuthorRole
+from services.env import EnvManager, EnvVariable
 from utils.logger import root_logger as logger
 
 
@@ -40,7 +41,7 @@ async def admin_get_users(_, info, limit=10, offset=0, search=None):
                     or_(
                         Author.email.ilike(search_term),
                         Author.name.ilike(search_term),
-                        Author.id.cast(str).ilike(search_term),
+                        cast(Author.id, String).ilike(search_term),
                     )
                 )
 
@@ -67,9 +68,7 @@ async def admin_get_users(_, info, limit=10, offset=0, search=None):
                         if hasattr(user, "roles") and user.roles
                         else [],
                         "created_at": user.created_at,
-                        "last_seen": user.last_seen,
-                        "muted": user.muted or False,
-                        "is_active": not user.blocked if hasattr(user, "blocked") else True,
+                        "last_seen": user.last_seen
                     }
                     for user in users
                 ],
@@ -120,3 +119,179 @@ async def admin_get_roles(_, info):
     except Exception as e:
         logger.error(f"Ошибка при получении списка ролей: {str(e)}")
         raise GraphQLError(f"Не удалось получить список ролей: {str(e)}")
+
+
+@query.field("getEnvVariables")
+@admin_auth_required
+async def get_env_variables(_, info):
+    """
+    Получает список переменных окружения, сгруппированных по секциям
+    
+    Args:
+        info: Контекст GraphQL запроса
+        
+    Returns:
+        Список секций с переменными окружения
+    """
+    try:
+        # Создаем экземпляр менеджера переменных окружения
+        env_manager = EnvManager()
+        
+        # Получаем все переменные
+        sections = env_manager.get_all_variables()
+        
+        # Преобразуем к формату GraphQL API
+        result = [
+            {
+                "name": section.name,
+                "description": section.description,
+                "variables": [
+                    {
+                        "key": var.key,
+                        "value": var.value,
+                        "description": var.description,
+                        "type": var.type,
+                        "isSecret": var.is_secret,
+                    }
+                    for var in section.variables
+                ]
+            }
+            for section in sections
+        ]
+        
+        return result
+    except Exception as e:
+        logger.error(f"Ошибка при получении переменных окружения: {str(e)}")
+        raise GraphQLError(f"Не удалось получить переменные окружения: {str(e)}")
+
+
+@mutation.field("updateEnvVariable")
+@admin_auth_required
+async def update_env_variable(_, info, key, value):
+    """
+    Обновляет значение переменной окружения
+    
+    Args:
+        info: Контекст GraphQL запроса
+        key: Ключ переменной
+        value: Новое значение
+        
+    Returns:
+        Boolean: результат операции
+    """
+    try:
+        # Создаем экземпляр менеджера переменных окружения
+        env_manager = EnvManager()
+        
+        # Обновляем переменную
+        result = env_manager.update_variable(key, value)
+        
+        if result:
+            logger.info(f"Переменная окружения '{key}' успешно обновлена")
+        else:
+            logger.error(f"Не удалось обновить переменную окружения '{key}'")
+            
+        return result
+    except Exception as e:
+        logger.error(f"Ошибка при обновлении переменной окружения: {str(e)}")
+        return False
+
+
+@mutation.field("updateEnvVariables")
+@admin_auth_required
+async def update_env_variables(_, info, variables):
+    """
+    Массовое обновление переменных окружения
+    
+    Args:
+        info: Контекст GraphQL запроса
+        variables: Список переменных для обновления
+        
+    Returns:
+        Boolean: результат операции
+    """
+    try:
+        # Создаем экземпляр менеджера переменных окружения
+        env_manager = EnvManager()
+        
+        # Преобразуем входные данные в формат для менеджера
+        env_variables = [
+            EnvVariable(
+                key=var.get("key", ""),
+                value=var.get("value", ""),
+                type=var.get("type", "string")
+            )
+            for var in variables
+        ]
+        
+        # Обновляем переменные
+        result = env_manager.update_variables(env_variables)
+        
+        if result:
+            logger.info(f"Переменные окружения успешно обновлены ({len(variables)} шт.)")
+        else:
+            logger.error(f"Не удалось обновить переменные окружения")
+            
+        return result
+    except Exception as e:
+        logger.error(f"Ошибка при массовом обновлении переменных окружения: {str(e)}")
+        return False
+
+
+@mutation.field("adminUpdateUser")
+@admin_auth_required
+async def admin_update_user(_, info, user):
+    """
+    Обновляет роли пользователя
+    
+    Args:
+        info: Контекст GraphQL запроса
+        user: Данные для обновления пользователя (содержит id и roles)
+        
+    Returns:
+        Boolean: результат операции
+    """
+    try:
+        user_id = user.get("id")
+        roles = user.get("roles", [])
+        
+        if not roles:
+            logger.warning(f"Пользователю {user_id} не назначено ни одной роли. Доступ в систему будет заблокирован.")
+        
+        with local_session() as session:
+            # Получаем пользователя из базы данных
+            author = session.query(Author).filter(Author.id == user_id).first()
+            
+            if not author:
+                logger.error(f"Пользователь с ID {user_id} не найден")
+                return False
+            
+            # Получаем текущие роли пользователя
+            current_roles = {role.id for role in author.roles} if author.roles else set()
+            
+            # Обновляем роли только если они изменились
+            if set(roles) != current_roles:
+                # Получаем все существующие роли, которые указаны для обновления
+                role_objects = session.query(Role).filter(Role.id.in_(roles)).all()
+                
+                # Очищаем текущие роли и добавляем новые
+                author.roles = role_objects
+                
+                # Сохраняем изменения в базе данных
+                session.commit()
+                
+                # Проверяем, добавлена ли пользователю роль reader
+                has_reader = 'reader' in roles
+                if not has_reader:
+                    logger.warning(f"Пользователю {author.email or author.id} не назначена роль 'reader'. Доступ в систему будет ограничен.")
+                
+                logger.info(f"Роли пользователя {author.email or author.id} обновлены: {', '.join(roles)}")
+            else:
+                logger.info(f"Роли пользователя {author.email or author.id} не изменились")
+            
+            return True
+    except Exception as e:
+        import traceback
+        logger.error(f"Ошибка при обновлении ролей пользователя: {str(e)}")
+        logger.error(traceback.format_exc())
+        return False