Integração com o banco de dados

src/services/imageGeneration.ts

@@ -16,6 +16,35 @@ export interface ImageGenerationResponse {
   message: string; // Descrição original
 }
 
+/**
+ * Interface para uma imagem armazenada no banco de dados
+ */
+export interface ImageRecord {
+  id: string;
+  user_email: string;
+  estabelecimento_id: number;
+  description: string;
+  model: string;
+  image_url: string;
+  size: ImageSize;
+  cost_usd: string;
+  total_tokens: number;
+  input_tokens: number;
+  output_tokens: number;
+  created_at: string;
+}
+
+/**
+ * Interface para resposta paginada de imagens
+ */
+export interface GetImagesResponse {
+  images: ImageRecord[];
+  total: number;
+  page: number;
+  per_page: number;
+  total_pages: number;
+}
+
 /**
  * Interface para os dados necessários para geração de imagem
  */
@@ -55,6 +84,8 @@ export const IMAGE_SIZE_OPTIONS = {
  */
 class ImageGenerationService {
   private readonly IMAGE_GENERATION_ENDPOINT = '/webhook/codex/image_generator';
+  private readonly GET_IMAGES_ENDPOINT = '/webhook/fd073ec0-81ab-4681-8acd-b0da71f6ad30/codex/get_images';
+  private readonly DELETE_IMAGE_ENDPOINT = '/webhook/fd073ec0-81ab-4681-8acd-b0da71f6ad30/codex/delete_images';
 
   /**
    * Gera uma imagem a partir de uma descrição em texto
@@ -213,6 +244,150 @@ class ImageGenerationService {
       throw new Error('Não foi possível baixar a imagem');
     }
   }
+
+  /**
+   * Lista imagens do banco de dados com paginação
+   *
+   * @param userEmail - Email do usuário
+   * @param page - Número da página (padrão: 1)
+   * @param perPage - Quantidade de itens por página (padrão: 10)
+   * @returns Promise com a resposta paginada
+   */
+  async getImages(
+    userEmail?: string,
+    page: number = 1,
+    perPage: number = 10
+  ): Promise<ImageRecord[]> {
+    // Usa valores do GlobalFunctions se não forem fornecidos
+    const email = userEmail || GlobalFunctions.getTransferProperty(TransferAreaProperties.UsuarioEmail);
+
+    if (!email) {
+      throw {
+        success: false,
+        message: 'Email do usuário não fornecido',
+      };
+    }
+
+    console.log('Buscando imagens:', {
+      userEmail: email,
+      page,
+      perPage,
+    });
+
+    try {
+      // Faz a requisição GET com parâmetros na URL e query
+      const response = await apiService.get<ImageRecord[]>(
+        `${this.GET_IMAGES_ENDPOINT}/${email}`,
+        {
+          params: {
+            page: page.toString(),
+            per_page: perPage.toString(),
+          },
+        }
+      );
+
+      console.log('Resposta completa da API:', response);
+      console.log('response.data:', response.data);
+      console.log('É array?:', Array.isArray(response.data));
+
+      // A API retorna diretamente o array de imagens
+      // Garante que sempre retorna um array
+      if (Array.isArray(response.data)) {
+        return response.data;
+      } else if (response.data && typeof response.data === 'object') {
+        // Se a resposta for um objeto com uma propriedade 'images' ou similar
+        console.warn('API retornou objeto em vez de array:', response.data);
+
+        // Tenta encontrar o array dentro do objeto
+        if (Array.isArray((response.data as any).images)) {
+          return (response.data as any).images;
+        } else if (Array.isArray((response.data as any).data)) {
+          return (response.data as any).data;
+        }
+      }
+
+      // Se não conseguir extrair array, retorna vazio
+      console.warn('Não foi possível extrair array de imagens da resposta');
+      return [];
+    } catch (error: any) {
+      console.error('Erro ao buscar imagens:', error);
+
+      throw {
+        success: false,
+        message: error.message || 'Erro ao buscar imagens',
+        status: error.status,
+      };
+    }
+  }
+
+  /**
+   * Deleta uma imagem do banco de dados
+   *
+   * @param imageId - ID da imagem a ser deletada
+   * @param userEmail - Email do usuário (opcional)
+   * @returns Promise com sucesso ou erro
+   */
+  async deleteImage(imageId: string, userEmail?: string): Promise<{ success: boolean; message?: string }> {
+    // Usa valores do GlobalFunctions se não forem fornecidos
+    const email = userEmail || GlobalFunctions.getTransferProperty(TransferAreaProperties.UsuarioEmail);
+
+    if (!email) {
+      throw {
+        success: false,
+        message: 'Email do usuário não fornecido',
+      };
+    }
+
+    if (!imageId) {
+      throw {
+        success: false,
+        message: 'ID da imagem não fornecido',
+      };
+    }
+
+    console.log('Deletando imagem:', {
+      imageId,
+      userEmail: email,
+    });
+
+    try {
+      // Faz a requisição DELETE com parâmetros na URL
+      const response = await apiService.delete<{ success: boolean; message?: string } | Array<{ success: boolean; message?: string }>>(
+        `${this.DELETE_IMAGE_ENDPOINT}/${email}/${imageId}`
+      );
+
+      console.log('Resposta completa do DELETE:', response);
+      console.log('response.data:', response.data);
+
+      // A API pode retornar um array com um objeto: [{"success":true}]
+      // ou diretamente um objeto: {"success":true}
+      let result: { success: boolean; message?: string };
+
+      if (Array.isArray(response.data)) {
+        // Se for array, pega o primeiro elemento
+        result = response.data[0];
+        console.log('API retornou array, usando primeiro elemento:', result);
+      } else {
+        // Se for objeto direto
+        result = response.data;
+        console.log('API retornou objeto direto:', result);
+      }
+
+      // Garante que tem a estrutura mínima
+      return {
+        success: result.success ?? true,
+        message: result.message || 'Imagem deletada com sucesso',
+      };
+    } catch (error: any) {
+      console.error('Erro ao deletar imagem:', error);
+
+      throw {
+        success: false,
+        message: error.message || 'Erro ao deletar imagem',
+        status: error.status,
+      };
+    }
+  }
 }
 
 // Exporta instância única (Singleton)