import 'dart:async';
import 'dart:io';

import 'package:dio/dio.dart';
import 'package:flutter/foundation.dart';
import 'package:spotube/models/logger.dart';

final logger = getLogger("ChunkedDownload");

/// Downloading by spiting as file in chunks
extension ChunkDownload on Dio {
  Future<Response> chunkedDownload(
    url, {
    Map<String, dynamic>? queryParameters,
    required String savePath,
    ProgressCallback? onReceiveProgress,
    CancelToken? cancelToken,
    bool deleteOnError = true,
    int chunkSize = 102400, // 100KB
    int maxConcurrentChunk = 3,
    String tempExtension = ".temp",
  }) async {
    int total = 0;
    var progress = <int>[];

    ProgressCallback createCallback(int chunkIndex) {
      return (int received, _) {
        progress[chunkIndex] = received;
        if (onReceiveProgress != null && total != 0) {
          onReceiveProgress(progress.reduce((a, b) => a + b), total);
        }
      };
    }

    // this is the last response
    // status & headers will the last chunk's status & headers
    final completer = Completer<Response>();

    Future<Response> downloadChunk(
      String url, {
      required int start,
      required int end,
      required int chunkIndex,
    }) async {
      progress.add(0);
      --end;
      final res = await download(
        url,
        savePath + tempExtension + chunkIndex.toString(),
        onReceiveProgress: createCallback(chunkIndex),
        cancelToken: cancelToken,
        queryParameters: queryParameters,
        deleteOnError: deleteOnError,
        options: Options(
          responseType: ResponseType.bytes,
          headers: {"range": "bytes=$start-$end"},
        ),
      );

      return res;
    }

    Future<void> mergeTempFiles(int chunk) async {
      File headFile = File("$savePath${tempExtension}0");
      var raf = await headFile.open(mode: FileMode.writeOnlyAppend);
      for (int i = 1; i < chunk; ++i) {
        File chunkFile = File(savePath + tempExtension + i.toString());
        raf = await raf.writeFrom(await chunkFile.readAsBytes());
        await chunkFile.delete();
      }
      await raf.close();

      logger.d("Downloaded file path: ${headFile.path}");

      headFile = await headFile.rename(savePath);

      logger.d("Renamed file path: ${headFile.path}");
    }

    final firstResponse = await downloadChunk(
      url,
      start: 0,
      end: chunkSize,
      chunkIndex: 0,
    );

    final responses = <Response>[firstResponse];

    if (firstResponse.statusCode == HttpStatus.partialContent) {
      total = int.parse(
        firstResponse.headers
                .value(HttpHeaders.contentRangeHeader)
                ?.split("/")
                .lastOrNull ??
            '0',
      );

      final reserved = total -
          int.parse(
            firstResponse.headers.value(HttpHeaders.contentLengthHeader) ??
                // since its a partial content, the content length will be the chunk size
                chunkSize.toString(),
          );

      int chunk = (reserved / chunkSize).ceil() + 1;

      if (chunk > 1) {
        int currentChunkSize = chunkSize;
        if (chunk > maxConcurrentChunk + 1) {
          chunk = maxConcurrentChunk + 1;
          currentChunkSize = (reserved / maxConcurrentChunk).ceil();
        }

        responses.addAll(
          await Future.wait(
            List.generate(maxConcurrentChunk, (i) {
              int start = chunkSize + i * currentChunkSize;
              return downloadChunk(
                url,
                start: start,
                end: start + currentChunkSize,
                chunkIndex: i + 1,
              );
            }),
          ),
        );
      }

      await mergeTempFiles(chunk).then((_) {
        final response = responses.last;
        final isPartialStatus =
            response.statusCode == HttpStatus.partialContent;

        completer.complete(
          Response(
            data: response.data,
            headers: response.headers,
            requestOptions: response.requestOptions,
            statusCode: isPartialStatus ? HttpStatus.ok : response.statusCode,
            statusMessage: isPartialStatus ? 'Ok' : response.statusMessage,
            extra: response.extra,
            isRedirect: response.isRedirect,
            redirects: response.redirects,
          ),
        );
      }).catchError((e) {
        completer.completeError(e);
      });
    }

    return completer.future;
  }
}