Added backup functions, device status (recovery,device,fastboot), fol…

src/adb.js

@@ -32,6 +32,7 @@ const DEFAULT_EXEC = (args, callback) => {
     callback
   );
 };
+
 const DEFAULT_LOG = console.log;
 const DEFAULT_PORT = 5037;
 
@@ -487,6 +488,220 @@ class Adb {
         );
     });
   }
+
+  // AW : return the file size of a complete folder
+  getFileSize(file) {
+    var _this = this;
+    return new Promise(function(resolve, reject) {
+      _this.getAdbState().then(stdout => {
+        if (stdout == 1) {
+          // Recovery
+          _this
+            .shell("du -shk " + file)
+            .then(stdout => {
+              stdout = parseFloat(stdout);
+              _this.log("FileSize is " + stdout + " Ko");
+              resolve(stdout);
+            })
+            .catch(reject);
+        } else if (stdout == 2) {
+          // device (type : used, size)
+          _this
+            .shell("df -hBK " + file + " --output=used|tail -n1")
+            .then(stdout => {
+              stdout = parseFloat(stdout);
+              _this.log("FileSize is " + stdout + " Ko");
+              resolve(stdout);
+            })
+            .catch(reject);
+        }
+      });
+    }).catch(e => {
+      _this.log(e);
+      reject;
+    });
+  }
+
+  // AW : Return the status of the device (0 : bootloader, 1 : recovery, 2 : device ...)
+  getAdbState() {
+    var _this = this;
+    return new Promise(function(resolve, reject) {
+      _this
+        .execCommand(["get-state"])
+        .then(stdout => {
+          _this.log("Device state is " + stdout);
+          switch (stdout) {
+            case "device":
+              resolve(2);
+            case "bootloader":
+              resolve(0);
+            case "recovery":
+              resolve(1);
+            default:
+              resolve(3); // Unknown
+          }
+        })
+        .catch(e => {
+          // Try with recovery command
+          _this.log("Unknown error :" + e);
+          reject;
+        });
+    });
+  }
+
+  // AW : Backup file "srcfile" from the phone to "destfile" localy
+  // TO-DO : This function should be moved to devices.js
+  backupremote(srcfile, destfile, adbpath, progress) {
+    var _this = this;
+    _this.log(
+      "Backup Function Entry, will backup " +
+        srcfile +
+        " to " +
+        destfile +
+        " adbpath:" +
+        adbpath
+    );
+    return new Promise(function(resolve, reject) {
+      _this.fileSize = 9999999999999; // Init to file size to a very high value while waiting for the real size.
+      // Get file size
+      _this
+        .getFileSize(srcfile)
+        .then(stdout => {
+          _this.log("Returned FileSize is " + stdout + " Ko");
+          _this.fileSize = stdout;
+          // Creating pipe
+          _this
+            .shell("mkfifo /backup.pipe")
+            .then(stdout => {
+              _this.log("Pipe created !");
+              var lastSize = 0;
+              var progressInterval = setInterval(() => {
+                const stats = fs.statSync(destfile);
+                const fileSizeInBytes = stats.size;
+                progress((lastSize / _this.fileSize) * 100);
+                lastSize = fileSizeInBytes / 1024;
+              }, 1000);
+
+              // Start the backup
+              _this.log("Starting Backup...");
+              _this
+                .execCommand([
+                  "exec-out 'tar -cvp ",
+                  srcfile,
+                  " 2>/backup.pipe' | dd of=" + destfile,
+                  " & ",
+                  adbpath + "/adb -P",
+                  _this.port,
+                  " shell cat /backup.pipe",
+                  process.platform == "win32"
+                    ? ' | findstr /v "%]"'
+                    : ' | grep -v "%]"'
+                ])
+                .then((error, stdout, stderr) => {
+                  _this.log("Backup Ended");
+                  clearInterval(progressInterval);
+                  _this
+                    .shell("rm /backup.pipe")
+                    .then(() => {
+                      _this.log("Pipe released.");
+                      resolve(); // Everything's Fine !
+                    })
+                    .catch(e => {
+                      utils.log.warn;
+                      reject(e + " Unable to delete the pipe " + e.length);
+                    }); // Pipe Deletion
+                })
+                .catch(e => {
+                  utils.log.warn;
+                  reject(e + " Unable to backuping the device " + e.length);
+                }); // Backup start
+            })
+            .catch(e => {
+              utils.log.warn;
+              reject(e + " Pipe creation failed " + e.length);
+            }); // Pipe Creation
+        })
+        .catch(e => {
+          utils.log.warn;
+          reject(e + " Unable to get the partition's filesize " + e.length);
+        }); // Get file size
+    });
+  }
+
+  // AW : Restore file "srcfile" from the computer to "destfile" on the phone
+  // TO-DO : This function should be moved to devices.js
+  restoreremote(srcfile, destfile, adbpath, progress) {
+    var _this = this;
+    _this.log(
+      "Restore Function Entry, will restore " +
+        srcfile +
+        " on " +
+        destfile +
+        " adbpath:" +
+        adbpath
+    );
+    return new Promise(function(resolve, reject) {
+      _this.fileSize = 9999999999999; // Init to file size to a very high value while waiting for the real size.
+      // Get file size
+      _this
+        .getFileSize(srcfile)
+        .then(stdout => {
+          _this.log("Returned FileSize is " + stdout + " Ko");
+          _this.fileSize = stdout;
+          // Creating pipe
+          _this
+            .shell("mkfifo /restore.pipe")
+            .then(stdout => {
+              _this.log("Pipe created !");
+              var lastSize = 0;
+              var progressInterval = setInterval(() => {
+                const stats = fs.statSync(destfile);
+                const fileSizeInBytes = stats.size;
+                progress((lastSize / _this.fileSize) * 100);
+                lastSize = fileSizeInBytes / 1024;
+              }, 1000);
+
+              // Start the restoration
+              _this.log("Starting Restore..."); //adb push user.tar /resto.pipe & adb shell 'cd /; cat /resto.pipe | tar -xv'
+              _this
+                .execCommand([
+                  "push",
+                  srcfile,
+                  "/resto.pipe &",
+                  adbpath + "/adb -P",
+                  _this.port,
+                  " shell 'cd /; cat /resto.pipe | tar -xv'"
+                ])
+                .then((error, stdout, stderr) => {
+                  _this.log("Restore Ended");
+                  clearInterval(progressInterval);
+                  _this
+                    .shell("rm /restore.pipe")
+                    .then(() => {
+                      _this.log("Pipe released.");
+                      resolve(); // Everything's Fine !
+                    })
+                    .catch(e => {
+                      utils.log.warn;
+                      reject("Unable to delete the pipe " + e);
+                    }); // Pipe Deletion
+                })
+                .catch(e => {
+                  utils.log.warn;
+                  reject("Unable to restore the backup " + e);
+                }); // Backup start
+            })
+            .catch(e => {
+              utils.log.warn;
+              reject("Pipe creation failed " + e);
+            }); // Pipe Creation
+        })
+        .catch(e => {
+          utils.log.warn;
+          reject("Unable to get the partition's filesize " + e);
+        }); // Get file size
+    });
+  }
 }
 
 module.exports = Adb;