react-native-stallion 2.1.0-alpha.2 → 2.1.0-alpha.3

package/android/src/main/java/com/stallion/events/StallionEventConstants.java

@@ -3,6 +3,7 @@ package com.stallion.events;
 public class StallionEventConstants {
   public enum NativeProdEventTypes {
     DOWNLOAD_STARTED_PROD,
+    DOWNLOAD_RESUME_PROD,
     DOWNLOAD_ERROR_PROD,
     DOWNLOAD_COMPLETE_PROD,
     SYNC_ERROR_PROD,
@@ -20,6 +21,7 @@ public class StallionEventConstants {
     DOWNLOAD_COMPLETE_STAGE,
     EXCEPTION_STAGE,
     DOWNLOAD_STARTED_STAGE,
+    DOWNLOAD_RESUME_STAGE,
     DOWNLOAD_ERROR_STAGE,
     INSTALLED_STAGE,
   }

package/android/src/main/java/com/stallion/networkmanager/StallionDownloadCacheManager.java

@@ -0,0 +1,48 @@
+package com.stallion.networkmanager;
+
+import com.stallion.storage.StallionConfig;
+import com.stallion.utils.StallionFileManager;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.util.Objects;
+
+public class StallionDownloadCacheManager {
+  private static final String metaFilePath = "/download-cache.meta";
+
+  public static long getDownloadCache(StallionConfig config, String downloadUrl, String downloadPath) {
+    String lastDownloadingUrl = config.getLastDownloadingUrl();
+    long alreadyDownloaded = readMetaFile(downloadPath);
+    if(!Objects.equals(lastDownloadingUrl, downloadUrl) || alreadyDownloaded <= 0) {
+      config.setLastDownloadingUrl(downloadUrl);
+      StallionFileManager.deleteFileOrFolderSilently(new File(downloadPath));
+      return 0;
+    } else {
+      return alreadyDownloaded;
+    }
+  }
+
+  public static void saveDownloadCache(String path, long bytes) {
+    try (FileOutputStream fos = new FileOutputStream(path + metaFilePath)) {
+      fos.write(Long.toString(bytes).getBytes());
+    } catch (Exception ignored) {}
+  }
+
+  private static long readMetaFile(String path) {
+    File meta = new File(path + metaFilePath);
+    if (!meta.exists()) return 0;
+    try (FileInputStream fis = new FileInputStream(meta)) {
+      byte[] data = new byte[(int) meta.length()];
+      fis.read(data);
+      return Long.parseLong(new String(data));
+    } catch (Exception e) {
+      return 0;
+    }
+  }
+
+  public static void deleteDownloadCache(String path) {
+    File meta = new File(path + metaFilePath);
+    if (meta.exists()) meta.delete();
+  }
+}

package/android/src/main/java/com/stallion/networkmanager/StallionFileDownloader.java

@@ -8,6 +8,7 @@ import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileOutputStream;
 import java.io.IOException;
+import java.io.RandomAccessFile;
 import java.net.HttpURLConnection;
 import java.net.URL;
 import java.nio.ByteBuffer;
@@ -28,6 +29,7 @@ public class StallionFileDownloader {
   public static void downloadBundle(
     String downloadUrl,
     String downloadDirectory,
+    long alreadyDownloaded,
     StallionDownloadCallback stallionDownloadCallback
   ) {
     executor.execute(() -> {
@@ -61,7 +63,7 @@ public class StallionFileDownloader {
         }
 
         // Download file
-        downloadFile(downloadUrl, downloadedZip, appToken, sdkToken, stallionDownloadCallback);
+        downloadFile(downloadUrl, downloadedZip, appToken, sdkToken, stallionDownloadCallback, alreadyDownloaded, downloadDirectory);
 
         // Validate and unzip the downloaded file
         validateAndUnzip(downloadedZip, downloadDirectory, stallionDownloadCallback);
@@ -75,7 +77,7 @@ public class StallionFileDownloader {
   private static long getFileSize(String downloadUrl, String appToken, String apiKey) throws IOException {
     HttpURLConnection connection = null;
     try {
-      connection = setupConnection(downloadUrl, appToken, apiKey);
+      connection = setupConnection(downloadUrl, appToken, apiKey, 0);
       return connection.getContentLength();
     } finally {
       if (connection != null) {
@@ -96,11 +98,10 @@ public class StallionFileDownloader {
 
   private static File prepareForDownload(String downloadDirectory) throws IOException {
     File downloadFolder = new File(downloadDirectory);
-    if (downloadFolder.exists()) {
-      StallionFileManager.deleteFileOrFolderSilently(downloadFolder);
-    }
-    if (!downloadFolder.mkdirs()) {
-      throw new IOException("Failed to create download directory: " + downloadDirectory);
+    if (!downloadFolder.exists()) {
+      if (!downloadFolder.mkdirs()) {
+        throw new IOException("Failed to create download directory: " + downloadDirectory);
+      }
     }
     return new File(downloadFolder, StallionApiConstants.ZIP_FILE_NAME);
   }
@@ -110,21 +111,21 @@ public class StallionFileDownloader {
     File destinationFile,
     String appToken,
     String sdkToken,
-    StallionDownloadCallback callback
+    StallionDownloadCallback callback,
+    long alreadyDownloaded,
+    String downloadDirectory
   ) throws IOException {
-    HttpURLConnection connection = null;
+    HttpURLConnection connection = setupConnection(downloadUrl, appToken, sdkToken, alreadyDownloaded);
     try (
-      BufferedInputStream inputStream = new BufferedInputStream(setupConnection(downloadUrl, appToken, sdkToken).getInputStream());
-      FileOutputStream fout = new FileOutputStream(destinationFile);
-      BufferedOutputStream bout = new BufferedOutputStream(fout, StallionApiConstants.DOWNLOAD_BUFFER_SIZE)
+      BufferedInputStream inputStream = new BufferedInputStream(connection.getInputStream());
+      RandomAccessFile raf = new RandomAccessFile(destinationFile, "rw")
     ) {
-      connection = setupConnection(downloadUrl, appToken, sdkToken);
-
+      raf.seek(alreadyDownloaded);
       byte[] buffer = new byte[StallionApiConstants.DOWNLOAD_BUFFER_SIZE];
-      long totalBytes = connection.getContentLength();
-      long receivedBytes = 0;
+      long totalBytes = connection.getContentLength() + alreadyDownloaded;
+      long receivedBytes = alreadyDownloaded;
       int bytesRead;
-      double lastProgress = 0;
+      double lastProgress = (double) receivedBytes / totalBytes;
 
       // Ensure totalBytes is valid
       if (totalBytes <= 0) {
@@ -133,9 +134,11 @@ public class StallionFileDownloader {
       }
 
       while ((bytesRead = inputStream.read(buffer)) != -1) {
-        bout.write(buffer, 0, bytesRead);
+        raf.write(buffer, 0, bytesRead);
         receivedBytes += bytesRead;
 
+        StallionDownloadCacheManager.saveDownloadCache(downloadDirectory, receivedBytes);
+
         double progress = (double) receivedBytes / totalBytes;
         if (Double.isNaN(progress) || Double.isInfinite(progress)) {
           callback.onReject(StallionApiConstants.DOWNLOAD_ERROR_PREFIX, "Invalid progress calculation");
@@ -148,8 +151,7 @@ public class StallionFileDownloader {
         }
       }
 
-      bout.close();
-      fout.close();
+      raf.close();
       inputStream.close();
 
       // Check for incomplete download
@@ -161,18 +163,25 @@ public class StallionFileDownloader {
       callback.onReject(StallionApiConstants.DOWNLOAD_ERROR_PREFIX, "IOException occurred: ");
       throw e;
     } finally {
-      if (connection != null) {
-        connection.disconnect();
-      }
+      connection.disconnect();
     }
   }
 
 
-  private static HttpURLConnection setupConnection(String downloadUrl, String appToken, String sdkToken) throws IOException {
+  private static HttpURLConnection setupConnection(
+    String downloadUrl,
+    String appToken,
+    String sdkToken,
+    long offset
+  ) throws IOException {
     URL url = new URL(downloadUrl);
     HttpURLConnection connection = (HttpURLConnection) url.openConnection();
     connection.setRequestMethod("GET");
 
+    if (offset > 0) {
+      connection.setRequestProperty("Range", "bytes=" + offset + "-");
+    }
+
     if(!appToken.isEmpty()) {
       connection.setRequestProperty(StallionApiConstants.STALLION_APP_TOKEN_KEY, appToken);
     }

package/android/src/main/java/com/stallion/networkmanager/StallionStageManager.java

@@ -4,6 +4,7 @@ import com.facebook.react.bridge.Promise;
 import com.facebook.react.bridge.ReadableMap;
 import com.stallion.events.StallionEventConstants;
 import com.stallion.events.StallionEventManager;
+import com.stallion.storage.StallionConfig;
 import com.stallion.storage.StallionConfigConstants;
 import com.stallion.storage.StallionMetaConstants;
 import com.stallion.storage.StallionStateManager;
@@ -21,14 +22,18 @@ public class StallionStageManager {
         && receivedHash != null
         && !receivedHash.isEmpty()
     ) {
-      String downloadPath = stallionStateManager.getStallionConfig().getFilesDirectory()
+      StallionConfig config = stallionStateManager.getStallionConfig();
+      String downloadPath = config.getFilesDirectory()
         + StallionConfigConstants.STAGE_DIRECTORY
         + StallionConfigConstants.TEMP_FOLDER_SLOT;
+      long alreadyDownloaded = StallionDownloadCacheManager.getDownloadCache(config, receivedDownloadUrl, downloadPath);
+
+      emitDownloadStartedStage(receivedHash, alreadyDownloaded > 0);
 
-      emitDownloadStartedStage(receivedHash);
       StallionFileDownloader.downloadBundle(
         receivedDownloadUrl,
         downloadPath,
+        alreadyDownloaded,
         new StallionDownloadCallback() {
           @Override
           public void onReject(String prefix, String error) {
@@ -42,6 +47,7 @@ public class StallionStageManager {
             stallionStateManager.stallionMeta.setStageTempHash(receivedHash);
             stallionStateManager.syncStallionMeta();
             emitDownloadSuccessStage(receivedHash);
+            StallionDownloadCacheManager.deleteDownloadCache(downloadPath);
             promise.resolve(successPayload);
           }
 
@@ -77,13 +83,15 @@ public class StallionStageManager {
     );
   }
 
-  private static void emitDownloadStartedStage(String releaseHash) {
+  private static void emitDownloadStartedStage(String releaseHash, Boolean isResume) {
     JSONObject startedPayload = new JSONObject();
     try {
       startedPayload.put("releaseHash", releaseHash);
     } catch (Exception ignored) { }
     StallionEventManager.getInstance().sendEvent(
-      StallionEventConstants.NativeStageEventTypes.DOWNLOAD_STARTED_STAGE.toString(),
+      isResume ?
+        StallionEventConstants.NativeStageEventTypes.DOWNLOAD_RESUME_STAGE.toString()
+        : StallionEventConstants.NativeStageEventTypes.DOWNLOAD_STARTED_STAGE.toString(),
       startedPayload
     );
   }

package/android/src/main/java/com/stallion/networkmanager/StallionSyncHandler.java

@@ -106,16 +106,21 @@ public class StallionSyncHandler {
     }
     try {
       StallionStateManager stateManager = StallionStateManager.getInstance();
-      String downloadPath = stateManager.getStallionConfig().getFilesDirectory()
+      StallionConfig config = stateManager.getStallionConfig();
+      String downloadPath = config.getFilesDirectory()
         + StallionConfigConstants.PROD_DIRECTORY
         + StallionConfigConstants.TEMP_FOLDER_SLOT;
-      String projectId = stateManager.getStallionConfig().getProjectId();
+      String projectId = config.getProjectId();
+      String downloadUrl = newReleaseUrl + "?projectId=" + projectId;
 
-      emitDownloadStarted(newReleaseHash);
+      long alreadyDownloaded = StallionDownloadCacheManager.getDownloadCache(config, downloadUrl, downloadPath);
+
+      emitDownloadStarted(newReleaseHash, alreadyDownloaded > 0);
 
       StallionFileDownloader.downloadBundle(
-        newReleaseUrl + "?projectId=" + projectId,
+        downloadUrl,
         downloadPath,
+        alreadyDownloaded,
         new StallionDownloadCallback() {
           @Override
           public void onReject(String prefix, String error) {
@@ -133,6 +138,7 @@ public class StallionSyncHandler {
               StallionSlotManager.stabilizeProd();
             }
             stateManager.syncStallionMeta();
+            StallionDownloadCacheManager.deleteDownloadCache(downloadPath);
             emitDownloadSuccess(newReleaseHash);
           }
 
@@ -182,13 +188,13 @@ public class StallionSyncHandler {
     );
   }
 
-  private static void emitDownloadStarted(String releaseHash) {
+  private static void emitDownloadStarted(String releaseHash, Boolean isResume) {
     JSONObject successPayload = new JSONObject();
     try {
       successPayload.put("releaseHash", releaseHash);
     } catch (Exception ignored) { }
     StallionEventManager.getInstance().sendEvent(
-      NativeProdEventTypes.DOWNLOAD_STARTED_PROD.toString(),
+      isResume ? NativeProdEventTypes.DOWNLOAD_RESUME_PROD.toString(): NativeProdEventTypes.DOWNLOAD_STARTED_PROD.toString(),
       successPayload
     );
   }

package/android/src/main/java/com/stallion/storage/StallionConfig.java

@@ -19,6 +19,7 @@ public class StallionConfig {
   private final String appVersion;
   private final SharedPreferences sharedPreferences;
   private final String filesDirectory;
+  private String lastDownloadingUrl;
 
   public StallionConfig(Context context, SharedPreferences sharedPreferences) {
     this.sharedPreferences = sharedPreferences;
@@ -63,6 +64,18 @@ public class StallionConfig {
 
     this.appVersion = fetchAppVersion(context);
     this.filesDirectory = context.getFilesDir().getAbsolutePath();
+    this.lastDownloadingUrl = sharedPreferences.getString(StallionConfigConstants.LAST_DOWNLOADING_URL_IDENTIFIER, "");
+  }
+
+  public String getLastDownloadingUrl() {
+    return this.lastDownloadingUrl;
+  }
+
+  public void setLastDownloadingUrl(String newUrl) {
+    this.lastDownloadingUrl = newUrl;
+    SharedPreferences.Editor editor = sharedPreferences.edit();
+    editor.putString(StallionConfigConstants.LAST_DOWNLOADING_URL_IDENTIFIER, newUrl);
+    editor.apply();
   }
 
   private String fetchAppVersion(Context context) {

package/android/src/main/java/com/stallion/storage/StallionConfigConstants.java

@@ -9,6 +9,7 @@ public class StallionConfigConstants {
   public static final String STALLION_APP_TOKEN_IDENTIFIER = "StallionAppToken";
   public static final String UNIQUE_ID_IDENTIFIER = "stallionDeviceId";
   public static final String API_KEY_IDENTIFIER = "x-sdk-access-token";
+  public static final String LAST_DOWNLOADING_URL_IDENTIFIER = "StallionLastDownloadingUrl";
 
   public static final String PROD_DIRECTORY = "/StallionProd";
   public static final String STAGE_DIRECTORY = "/StallionStage";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "react-native-stallion",
-  "version": "2.1.0-alpha.2",
+  "version": "2.1.0-alpha.3",
   "description": "Offical React Native SDK for Stallion",
   "main": "index",
   "types": "types/index.d.ts",