Check free space on disk before download the file

Author: danielesergio

ddiclient-api/build.gradle

@@ -9,7 +9,7 @@
  */
 
 group 'com.kynetics.updatefactory'
-version '0.4.6'
+version '0.4.7'
 
 apply plugin: 'java'
 apply plugin: 'maven'

ddiclient-core/build.gradle

@@ -9,7 +9,7 @@
  */
 
 group 'com.kynetics.updatefactory'
-version '0.4.6'
+version '0.4.7'
 
 apply plugin: 'java'
 apply plugin: 'maven'

ddiclient-core/src/main/java/com/kynetics/updatefactory/ddiclient/core/UFService.java

@@ -229,12 +229,22 @@ private void handlerState(AbstractState currentState, long forceDelay){
                 break;
             case UPDATE_DOWNLOAD:
                 final UpdateDownloadState updateDownloadState = (UpdateDownloadState) currentState;
-                callToAbort = client.downloadArtifact(
-                        tenant,
-                        controllerId,
-                        updateDownloadState.getFileInfo().getLinkInfo().getSoftwareModules(),
-                        updateDownloadState.getFileInfo().getLinkInfo().getFileName());
-                execute(callToAbort, new DownloadArtifact(updateDownloadState), forceDelay);
+                long spaceNeeded = 0;
+                for(int i = updateDownloadState.getNextFileToDownload(); i<updateDownloadState.getFileInfoList().size(); i++){
+                    spaceNeeded += updateDownloadState.getFileInfoList().get(i).getSize();
+                }
+
+                if(systemOperation.checkSpace(spaceNeeded)){
+                    callToAbort = client.downloadArtifact(
+                            tenant,
+                            controllerId,
+                            updateDownloadState.getFileInfo().getLinkInfo().getSoftwareModules(),
+                            updateDownloadState.getFileInfo().getLinkInfo().getFileName());
+                    execute(callToAbort, new DownloadArtifact(updateDownloadState), forceDelay);
+                } else {
+                    onEvent(new InsufficientSpaceEvent());
+                }
+
                 break;
             case UPDATE_READY:
                 final Call checkCancellation = client.getControllerBase(tenant, controllerId);

ddiclient-core/src/main/java/com/kynetics/updatefactory/ddiclient/core/model/event/AbstractEvent.java

@@ -22,7 +22,7 @@ public abstract class AbstractEvent implements Serializable{
     public enum EventName{
         SLEEP_REQUEST,  UPDATE_CONFIG_REQUEST, SUCCESS, FAILURE, ERROR, UPDATE_FOUND, DOWNLOAD_REQUEST, DOWNLOAD_STARTED,
         FILE_CORRUPTED, CANCEL, UPDATE_ERROR, AUTHORIZATION_GRANTED, AUTHORIZATION_DENIED, RESUME, DOWNLOAD_PENDING,
-        FORCE_CANCEL, AUTHORIZATION_PENDING
+        FORCE_CANCEL, AUTHORIZATION_PENDING, INSUFFICIENT_SPACE
     }
 
     private final EventName eventName;

ddiclient-core/src/main/java/com/kynetics/updatefactory/ddiclient/core/model/event/InsufficientSpaceEvent.java

@@ -0,0 +1,22 @@
+/*
+ * Copyright © 2017-2019 Kynetics LLC
+ *
+ *  All rights reserved. This program and the accompanying materials
+ *  are made available under the terms of the Eclipse Public License v1.0
+ *  which accompanies this distribution, and is available at
+ *  http://www.eclipse.org/legal/epl-v10.html
+ *
+ */
+
+package com.kynetics.updatefactory.ddiclient.core.model.event;
+
+import static com.kynetics.updatefactory.ddiclient.core.model.event.AbstractEvent.EventName.INSUFFICIENT_SPACE;
+
+/**
+ * @author Daniele Sergio
+ */
+public class InsufficientSpaceEvent extends AbstractEvent {
+    public InsufficientSpaceEvent() {
+        super(INSUFFICIENT_SPACE);
+    }
+}

ddiclient-core/src/main/java/com/kynetics/updatefactory/ddiclient/core/model/state/UpdateDownloadState.java

@@ -43,7 +43,9 @@ public AbstractState onEvent(AbstractEvent event) {
         switch (event.getEventName()) {
             case DOWNLOAD_STARTED:
                 return new SavingFileState(getActionId(), isForced(), getFileInfoList(), getNextFileToDownload(), getLastHash(), ((DownloadStartedEvent) event).getInputStream());
-            default:
+            case INSUFFICIENT_SPACE:
+                return new UpdateEndedState(getActionId(), false, new String[]{"Not enough space"});
+                default:
                 return super.onEvent(event);
         }
     }

ddiclient-core/src/main/java/com/kynetics/updatefactory/ddiclient/core/servicecallback/SystemOperation.java

@@ -24,6 +24,8 @@ public interface SystemOperation {
 
     UpdateStatus updateStatus();
 
+    boolean checkSpace(long spaceNeeded);
+
     class UpdateStatus{
         private final StatusName statusName;
         private final String[] messages;

ddiclient-example/build.gradle

@@ -9,7 +9,7 @@
  */
 
 group 'com.kynetics.updatefactory'
-version '0.4.6'
+version '0.4.7'
 apply plugin: 'java'
 
 sourceCompatibility = 1.8

ddiclient-example/src/main/java/com/kynetics/updatefactory/ddiclient/example/callback/SystemOperationMock.java

@@ -49,4 +49,9 @@ public void executeUpdate(long actionId) {
     public UpdateStatus updateStatus() {
         return status;
     }
+
+    @Override
+    public boolean checkSpace(long spaceNeeded) {
+        return true;
+    }
 }