superport/lib/data/datasources/remote/warehouse_remote_datasource.dart

import 'package:injectable/injectable.dart';
import 'package:superport/core/constants/api_endpoints.dart';
import 'package:superport/core/errors/exceptions.dart';
import 'package:superport/data/datasources/remote/api_client.dart';
import 'package:superport/data/models/warehouse/warehouse_dto.dart';

abstract class WarehouseRemoteDataSource {
  Future<WarehouseLocationListDto> getWarehouseLocations({
    int page = 1,
    int perPage = 10,
    bool? isActive,
    String? search,
    bool includeInactive = false,
  });

  Future<WarehouseDto> getWarehouseLocationById(int id);
  Future<WarehouseDto> createWarehouseLocation(WarehouseRequestDto request);
  Future<WarehouseDto> updateWarehouseLocation(int id, WarehouseUpdateRequestDto request);
  Future<void> deleteWarehouseLocation(int id);
  Future<WarehouseEquipmentListDto> getWarehouseEquipment(
    int warehouseId, {
    int page = 1,
    int perPage = 10,
  });
  Future<WarehouseCapacityInfo> getWarehouseCapacity(int id);
  Future<List<WarehouseDto>> getInUseWarehouseLocations();
}

@LazySingleton(as: WarehouseRemoteDataSource)
class WarehouseRemoteDataSourceImpl implements WarehouseRemoteDataSource {
  final ApiClient _apiClient;

  WarehouseRemoteDataSourceImpl({
    required ApiClient apiClient,
  }) : _apiClient = apiClient;

  @override
  Future<WarehouseLocationListDto> getWarehouseLocations({
    int page = 1,
    int perPage = 10,
    bool? isActive,
    String? search,
    bool includeInactive = false,
  }) async {
    try {
      final queryParams = <String, dynamic>{
        'page': page,
        'per_page': perPage,
      };

      if (isActive != null) queryParams['is_active'] = isActive;
      if (search != null && search.isNotEmpty) queryParams['search'] = search;
      queryParams['include_inactive'] = includeInactive;

      final response = await _apiClient.get(
        ApiEndpoints.warehouses,
        queryParameters: queryParams,
      );

      // 백엔드 응답을 직접 처리 (success 필드 없음)
      if (response.data != null && response.data['data'] != null) {
        final List<dynamic> dataList = response.data['data'];
        
        final listData = {
          'items': dataList,
          'total': response.data['total'] ?? 0,
          'page': response.data['page'] ?? 1,
          'per_page': response.data['page_size'] ?? 10,
          'total_pages': response.data['total_pages'] ?? 1,
        };
        
        return WarehouseLocationListDto.fromJson(listData);
      } else {
        throw ApiException(
          message: 'Failed to fetch warehouse locations',
        );
      }
    } catch (e) {
      throw _handleError(e);
    }
  }

  @override
  Future<WarehouseDto> getWarehouseLocationById(int id) async {
    try {
      final response = await _apiClient.get(
        '${ApiEndpoints.warehouses}/$id',
      );

      // 백엔드가 직접 데이터를 반환하는 경우 처리
      if (response.data != null) {
        // success 필드가 없으면 직접 데이터로 간주
        if (response.data is Map && !response.data.containsKey('success')) {
          return WarehouseDto.fromJson(response.data);
        }
        // success 필드가 있는 경우 기존 방식 처리
        else if (response.data['success'] == true && response.data['data'] != null) {
          return WarehouseDto.fromJson(response.data['data']);
        }
      }
      
      throw ApiException(
        message: response.data?['error']?['message'] ?? 'Failed to fetch warehouse location',
      );
    } catch (e) {
      throw _handleError(e);
    }
  }

  @override
  Future<WarehouseDto> createWarehouseLocation(WarehouseRequestDto request) async {
    try {
      final response = await _apiClient.post(
        ApiEndpoints.warehouses,
        data: request.toJson(),
      );

      // 백엔드가 직접 데이터를 반환하는 경우 처리
      if (response.data != null) {
        // success 필드가 없으면 직접 데이터로 간주
        if (response.data is Map && !response.data.containsKey('success')) {
          return WarehouseDto.fromJson(response.data);
        }
        // success 필드가 있는 경우 기존 방식 처리
        else if (response.data['success'] == true && response.data['data'] != null) {
          return WarehouseDto.fromJson(response.data['data']);
        }
      }
      
      throw ApiException(
        message: response.data?['error']?['message'] ?? 'Failed to create warehouse location',
      );
    } catch (e) {
      throw _handleError(e);
    }
  }

  @override
  Future<WarehouseDto> updateWarehouseLocation(int id, WarehouseUpdateRequestDto request) async {
    try {
      final response = await _apiClient.put(
        '${ApiEndpoints.warehouses}/$id',
        data: request.toJson(),
      );

      // 백엔드가 직접 데이터를 반환하는 경우 처리
      if (response.data != null) {
        // success 필드가 없으면 직접 데이터로 간주
        if (response.data is Map && !response.data.containsKey('success')) {
          return WarehouseDto.fromJson(response.data);
        }
        // success 필드가 있는 경우 기존 방식 처리
        else if (response.data['success'] == true && response.data['data'] != null) {
          return WarehouseDto.fromJson(response.data['data']);
        }
      }
      
      throw ApiException(
        message: response.data?['error']?['message'] ?? 'Failed to update warehouse location',
      );
    } catch (e) {
      throw _handleError(e);
    }
  }

  @override
  Future<void> deleteWarehouseLocation(int id) async {
    try {
      await _apiClient.delete(
        '${ApiEndpoints.warehouses}/$id',
      );
    } catch (e) {
      throw _handleError(e);
    }
  }

  @override
  Future<WarehouseEquipmentListDto> getWarehouseEquipment(
    int warehouseId, {
    int page = 1,
    int perPage = 10,
  }) async {
    try {
      final queryParams = <String, dynamic>{
        'page': page,
        'per_page': perPage,
      };

      final response = await _apiClient.get(
        '${ApiEndpoints.warehouseLocations}/$warehouseId/equipment',
        queryParameters: queryParams,
      );

      if (response.data != null && response.data['success'] == true && response.data['data'] != null) {
        return WarehouseEquipmentListDto.fromJson(response.data['data']);
      } else {
        throw ApiException(
          message: response.data?['error']?['message'] ?? 'Failed to fetch warehouse equipment',
        );
      }
    } catch (e) {
      throw _handleError(e);
    }
  }

  @override
  Future<WarehouseCapacityInfo> getWarehouseCapacity(int id) async {
    try {
      final response = await _apiClient.get(
        '${ApiEndpoints.warehouseLocations}/$id/capacity',
      );

      if (response.data != null && response.data['success'] == true && response.data['data'] != null) {
        return WarehouseCapacityInfo.fromJson(response.data['data']);
      } else {
        throw ApiException(
          message: response.data?['error']?['message'] ?? 'Failed to fetch warehouse capacity',
        );
      }
    } catch (e) {
      throw _handleError(e);
    }
  }

  @override
  Future<List<WarehouseDto>> getInUseWarehouseLocations() async {
    try {
      final response = await _apiClient.get(
        '${ApiEndpoints.warehouses}/in-use',
      );

      if (response.data != null && response.data['success'] == true && response.data['data'] is List) {
        return (response.data['data'] as List)
            .map((item) => WarehouseDto.fromJson(item))
            .toList();
      } else {
        throw ApiException(message: 'Invalid response format');
      }
    } catch (e) {
      throw _handleError(e);
    }
  }

  Exception _handleError(dynamic error) {
    if (error is ApiException) {
      return error;
    }
    return ApiException(
      message: error.toString(),
    );
  }
}