tweb-i2p/src/lib/serviceWorker/stream.ts

/*
 * https://github.com/morethanwords/tweb
 * Copyright (C) 2019-2021 Eduard Kuzmenko
 * https://github.com/morethanwords/tweb/blob/master/LICENSE
 */

import { deferredPromise } from "../../helpers/cancellablePromise";
import { notifySomeone } from "../../helpers/context";
import { isSafari } from "../../helpers/userAgent";
import { UploadFile } from "../../layer";
import { DownloadOptions } from "../mtproto/apiFileManager";
import { RequestFilePartTask, deferredPromises, incrementTaskId } from "./index.service";
import timeout from "./timeout";

export default function onStreamFetch(event: FetchEvent, params: string) {
  const range = parseRange(event.request.headers.get('Range'));
  let [offset, end] = range;

  const info: DownloadOptions = JSON.parse(decodeURIComponent(params));
  //const fileName = getFileNameByLocation(info.location);

  // ! если грузить очень большое видео чанками по 512Кб в мобильном Safari, то стрим не запустится
  const limitPart = info.size > (75 * 1024 * 1024) ? STREAM_CHUNK_UPPER_LIMIT : STREAM_CHUNK_MIDDLE_LIMIT;

  /* if(info.size > limitPart && isSafari && offset === limitPart) {
    //end = info.size - 1;
    //offset = info.size - 1 - limitPart;
    offset = info.size - (info.size % limitPart);
  } */

  //log.debug('[stream]', url, offset, end);

  event.respondWith(Promise.race([
    timeout(45 * 1000),

    new Promise<Response>((resolve, reject) => {
      // safari workaround
      const possibleResponse = responseForSafariFirstRange(range, info.mimeType, info.size);
      if(possibleResponse) {
        return resolve(possibleResponse);
      }

      const limit = end && end < limitPart ? alignLimit(end - offset + 1) : limitPart;
      const alignedOffset = alignOffset(offset, limit);

      //log.debug('[stream] requestFilePart:', /* info.dcId, info.location, */ alignedOffset, limit);

      const task: RequestFilePartTask = {
        type: 'requestFilePart',
        id: incrementTaskId(),
        payload: [info.dcId, info.location, alignedOffset, limit]
      };

      
      const deferred = deferredPromises[task.id] = deferredPromise<UploadFile.uploadFile>();
      deferred.then(result => {
        let ab = result.bytes as Uint8Array;
        
        //log.debug('[stream] requestFilePart result:', result);

        const headers: Record<string, string> = {
          'Accept-Ranges': 'bytes',
          'Content-Range': `bytes ${alignedOffset}-${alignedOffset + ab.byteLength - 1}/${info.size || '*'}`,
          'Content-Length': `${ab.byteLength}`,
        };

        if(info.mimeType) headers['Content-Type'] = info.mimeType;

        if(isSafari) {
          ab = ab.slice(offset - alignedOffset, end - alignedOffset + 1);
          headers['Content-Range'] = `bytes ${offset}-${offset + ab.byteLength - 1}/${info.size || '*'}`;
          headers['Content-Length'] = `${ab.byteLength}`;
        }

        // simulate slow connection
        //setTimeout(() => {
          resolve(new Response(ab, {
            status: 206,
            statusText: 'Partial Content',
            headers,
          }));
        //}, 2.5e3);
      }).catch(err => {});

      notifySomeone(task);
    })
  ]));
}

function responseForSafariFirstRange(range: [number, number], mimeType: string, size: number): Response {
  if(range[0] === 0 && range[1] === 1) {
    return new Response(new Uint8Array(2).buffer, {
      status: 206,
      statusText: 'Partial Content',
      headers: {
        'Accept-Ranges': 'bytes',
        'Content-Range': `bytes 0-1/${size || '*'}`,
        'Content-Length': '2',
        'Content-Type': mimeType || 'video/mp4',
      },
    });
  }

  return null;
}

/* const STREAM_CHUNK_UPPER_LIMIT = 256 * 1024;
const SMALLEST_CHUNK_LIMIT = 256 * 4; */
/* const STREAM_CHUNK_UPPER_LIMIT = 1024 * 1024;
const SMALLEST_CHUNK_LIMIT = 1024 * 4; */
const STREAM_CHUNK_MIDDLE_LIMIT = 512 * 1024;
const STREAM_CHUNK_UPPER_LIMIT = 1024 * 1024;
const SMALLEST_CHUNK_LIMIT = 512 * 4;

function parseRange(header: string): [number, number] {
  if(!header) return [0, 0];
  const [, chunks] = header.split('=');
  const ranges = chunks.split(', ');
  const [offset, end] = ranges[0].split('-');

  return [+offset, +end || 0];
}

function alignOffset(offset: number, base = SMALLEST_CHUNK_LIMIT) {
  return offset - (offset % base);
}

function alignLimit(limit: number) {
  return 2 ** Math.ceil(Math.log(limit) / Math.log(2));
}