From c6ff7f819ce5fd3b6dc35a50eb31b3d9585ded40 Mon Sep 17 00:00:00 2001
From: SebClem <seb6596@gmail.com>
Date: Sat, 1 Oct 2022 17:18:30 +0200
Subject: [PATCH] :hammer: Add file upload for webdav

---
 .../backend/src/services/webdavService.ts     | 108 +++++++++++++-----
 1 file changed, 82 insertions(+), 26 deletions(-)

diff --git a/nextcloud_backup/backend/src/services/webdavService.ts b/nextcloud_backup/backend/src/services/webdavService.ts
index 6564be3..6a0b8ab 100644
--- a/nextcloud_backup/backend/src/services/webdavService.ts
+++ b/nextcloud_backup/backend/src/services/webdavService.ts
@@ -1,15 +1,17 @@
+import { XMLParser } from "fast-xml-parser";
+import { createReadStream, statSync, unlinkSync } from "fs";
 import got, { HTTPError, Method } from "got";
+import { DateTime } from "luxon";
 import logger from "../config/winston.js";
 import messageManager from "../tools/messageManager.js";
+import * as pathTools from "../tools/pathTools.js";
+import * as statusTools from "../tools/status.js";
+import { WebdavBackup } from "../types/services/webdav.js";
 import { WebdavConfig } from "../types/services/webdavConfig.js";
 import { getEndpoint } from "./webdavConfigService.js";
-import * as pathTools from "../tools/pathTools.js";
-import { XMLParser } from "fast-xml-parser";
-import { WebdavBackup } from "../types/services/webdav.js";
-import { DateTime } from "luxon";
 
 const PROPFIND_BODY =
-'<?xml version="1.0" encoding="utf-8" ?>\
+  '<?xml version="1.0" encoding="utf-8" ?>\
 <d:propfind xmlns:d="DAV:">\
   <d:prop>\
         <d:getlastmodified />\
@@ -118,33 +120,99 @@ export function getBackups(folder: string, config: WebdavConfig) {
   );
 }
 
-
-function parseXmlBackupData(body: string){
+function parseXmlBackupData(body: string) {
   const parser = new XMLParser();
   const data = parser.parse(body);
   const multistatus = data["d:multistatus"];
   const backups: WebdavBackup[] = [];
-  if(Array.isArray(multistatus["d:response"])){
-    for(const elem of multistatus["d:response"]){
+  if (Array.isArray(multistatus["d:response"])) {
+    for (const elem of multistatus["d:response"]) {
       // If array -> base folder, ignoring it
-      if(!Array.isArray(elem["d:propstat"])){
+      if (!Array.isArray(elem["d:propstat"])) {
         const propstat = elem["d:propstat"];
-        const id = propstat["d:prop"]["d:getetag"].replaceAll("\"", "");
+        const id = propstat["d:prop"]["d:getetag"].replaceAll('"', "");
         const href = decodeURI(elem["d:href"]);
         const name = href.split("/").slice(-1)[0];
-        const lastEdit = DateTime.fromHTTP(propstat["d:prop"]["d:getlastmodified"]);
+        const lastEdit = DateTime.fromHTTP(
+          propstat["d:prop"]["d:getlastmodified"]
+        );
         backups.push({
           id: id,
           lastEdit: lastEdit,
           size: propstat["d:prop"]["d:getcontentlenght"],
-          name: name
-        })
+          name: name,
+        });
       }
     }
   }
   return backups;
 }
 
+export function webdabUploadFile(
+  localPath: string,
+  webdavPath: string,
+  config: WebdavConfig
+) {
+  return new Promise((resolve, reject) => {
+    logger.info(`Uploading ${localPath} to webdav...`);
+    const stats = statSync(localPath);
+    const stream = createReadStream(localPath);
+    const options = {
+      body: stream,
+      headers: {
+        authorization:
+          "Basic " +
+          Buffer.from(config.username + ":" + config.password).toString(
+            "base64"
+          ),
+        "content-length": String(stats.size),
+      },
+      https: { rejectUnauthorized: !config.allowSelfSignedCerts },
+    };
+    const url =
+      config.url + getEndpoint(config) + config.backupDir + webdavPath;
+
+    logger.debug(`...URI: ${encodeURI(url)}`);
+    logger.debug(`...rejectUnauthorized: ${options.https?.rejectUnauthorized}`);
+    const status = statusTools.getStatus();
+    got.stream
+      .put(encodeURI(url), options)
+      .on("uploadProgress", (e) => {
+        const percent = e.percent;
+        if (status.progress !== percent) {
+          status.progress = percent;
+          statusTools.setStatus(status);
+        }
+        if (percent >= 1) {
+          logger.info("Upload done...");
+        }
+      })
+      .on("response", (res) => {
+        if (res.statusCode != 201 && res.statusCode != 204) {
+          messageManager.error(
+            "Fail to upload file to webdav.",
+            `Status code: ${res.statusCode}`
+          );
+          logger.error(
+            `Fail to upload file to webdav, Status code: ${res.statusCode}`
+          );
+          logger.error(status.message);
+          unlinkSync(localPath);
+          reject(res);
+        } else {
+          logger.info(`...Upload finish ! (status: ${res.statusCode})`);
+          unlinkSync(localPath);
+          resolve(undefined);
+        }
+      })
+      .on("error", (err) => {
+        logger.error(status.message);
+        logger.error(err.stack);
+        reject(status.message);
+      });
+  });
+}
+
 // import fs from "fs";
 // import got from "got";
 // import https from "https";
@@ -610,17 +678,5 @@ function parseXmlBackupData(body: string){
 //   }
 // }
 
-// function cleanTempFolder() {
-//   fs.readdir("./temp/", (err, files) => {
-//     if (err) throw err;
-
-//     for (const file of files) {
-//       fs.unlink(path.join("./temp/", file), (err) => {
-//         if (err) throw err;
-//       });
-//     }
-//   });
-// }
-
 // const INSTANCE = new WebdavTools();
 // export default INSTANCE;