android/src/com/mapswithme/maps/downloader/MapDownloadManager.java - master - githubmirror/omim

android/src/com/mapswithme/maps/downloader/MapDownloadManager.java
167 строк4 KB
Arsentiy Milchakov
[android][downloader] review fixes
30 мар 2021, 13:59
30 мар 2021, 13:5929882b4
package com.mapswithme.maps.downloader;

import android.app.DownloadManager;
import android.content.Context;
import android.database.Cursor;
import android.net.Uri;
import android.text.TextUtils;

import androidx.annotation.MainThread;
import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import com.mapswithme.maps.MwmApplication;
import com.mapswithme.util.Utils;
import com.mapswithme.util.log.Logger;
import com.mapswithme.util.log.LoggerFactory;

import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.HashMap;
import java.util.Map;

public class MapDownloadManager
{
  private static final Logger LOGGER = LoggerFactory.INSTANCE.getLogger(LoggerFactory.Type.DOWNLOADER);
  private static final String TAG = MapDownloadManager.class.getSimpleName();

  @NonNull
  private DownloadManager mDownloadManager;
  @Nullable
  private Map<String, Long> mRestoredRequests;
  @NonNull
  private MapDownloadProgressTracker mProgressTracker;

  public MapDownloadManager(@NonNull Context context)
  {
    DownloadManager downloadManager =
        (DownloadManager) context.getSystemService(Context.DOWNLOAD_SERVICE);

    if (downloadManager == null)
      throw new NullPointerException("Download manager is null, failed to create MapDownloadManager");

    mDownloadManager = downloadManager;
    mProgressTracker = new MapDownloadProgressTracker(context);
  }

  @NonNull
  private Map<String, Long> loadEnqueued()
  {
    Map<String, Long> result = new HashMap<>();
    DownloadManager.Query query = new DownloadManager.Query();
    query.setFilterByStatus(DownloadManager.STATUS_PENDING | DownloadManager.STATUS_RUNNING |
                            DownloadManager.STATUS_PAUSED | DownloadManager.STATUS_SUCCESSFUL);
    Cursor cursor = null;
    try
    {
      cursor = mDownloadManager.query(query);

      cursor.moveToFirst();
      int count = cursor.getCount();

      for (int i = 0; i < count; ++i)
      {
        long id = cursor.getLong(cursor.getColumnIndex(DownloadManager.COLUMN_ID));
        String url = cursor.getString(cursor.getColumnIndex(DownloadManager.COLUMN_URI));
        String urlPath = getUrlPath(url);

        if (!TextUtils.isEmpty(urlPath))
          result.put(urlPath, id);

        cursor.moveToNext();
      }
    }
    catch (Exception e)
    {
      LOGGER.e(TAG, "Failed to load enqueued requests. Exception ", e);
    }
    finally
    {
      Utils.closeSafely(cursor);
    }

    return result;
  }

  @Nullable
  private String getUrlPath(@Nullable String url)
  {
    if (TextUtils.isEmpty(url))
      return null;

    String path = Uri.parse(url).getPath();
    if (TextUtils.isEmpty(path)|| !MapManager.nativeIsUrlSupported(path))
      return null;

    try
    {
      return URLDecoder.decode(path, "UTF-8");
    }
    catch (UnsupportedEncodingException ignored)
    {
      return null;
    }
  }

  @MainThread
  public long enqueue(@NonNull String url)
  {
    Uri uri = Uri.parse(url);
    String uriPath = uri.getPath();
    if (uriPath == null)
      throw new AssertionError("The path must be not null");

    if (mRestoredRequests == null)
      mRestoredRequests = loadEnqueued();

    Long id = mRestoredRequests.get(uriPath);
    long requestId = 0;

    if (id == null)
    {
      DownloadManager.Request request = new DownloadManager
          .Request(uri)
          .setNotificationVisibility(DownloadManager.Request.VISIBILITY_HIDDEN);

      requestId = mDownloadManager.enqueue(request);
    }
    else
    {
      mRestoredRequests.remove(uriPath);
      requestId = id;
    }

    mProgressTracker.add(requestId);

    return requestId;
  }

  @MainThread
  public void remove(long requestId)
  {
    mProgressTracker.remove(requestId);
    mDownloadManager.remove(requestId);
  }

  public void onDownloadFinished(boolean status, long id)
  {
    mProgressTracker.remove(id);
    MapManager.nativeOnDownloadFinished(status, id);
  }

  public void startProgressTracking()
  {
    mProgressTracker.start();
  }

  public void stopProgressTracking()
  {
    mProgressTracker.stop();
  }

  @NonNull
  public static MapDownloadManager from(@NonNull Context context)
  {
    MwmApplication app = (MwmApplication) context.getApplicationContext();
    return app.getMapDownloadManager();
  }
}