From e94f767e93acb1c430ae9e98b67f91c7b701dc17 Mon Sep 17 00:00:00 2001
From: SebClem <seb6596@gmail.com>
Date: Mon, 22 Jul 2024 11:20:43 +0200
Subject: [PATCH] Add ha to cloud manual upload

---
 .../backend/src/routes/homeAssistant.ts       | 13 ++++
 .../backend/src/services/orchestrator.ts      | 59 ++++++++++++++++++-
 .../src/components/homeAssistant/HaList.vue   | 19 +++++-
 .../components/homeAssistant/HaListItem.vue   |  8 ++-
 .../src/services/homeAssistantService.ts      |  4 ++
 5 files changed, 99 insertions(+), 4 deletions(-)

diff --git a/nextcloud_backup/backend/src/routes/homeAssistant.ts b/nextcloud_backup/backend/src/routes/homeAssistant.ts
index e50cf43..62e8b0b 100644
--- a/nextcloud_backup/backend/src/routes/homeAssistant.ts
+++ b/nextcloud_backup/backend/src/routes/homeAssistant.ts
@@ -1,5 +1,7 @@
 import express from "express";
 import * as haOsService from "../services/homeAssistantService.js";
+import { uploadToCloud } from "../services/orchestrator.js";
+import logger from "../config/winston.js";
 
 const homeAssistantRouter = express.Router();
 
@@ -29,6 +31,17 @@ homeAssistantRouter.get("/backup/:slug", (req, res) => {
     });
 });
 
+homeAssistantRouter.post("/backup/:slug/upload", (req, res) => {
+  uploadToCloud(req.params.slug)
+    .then(() => {
+      logger.info("All good !");
+    })
+    .catch(() => {
+      logger.error("Something wrong !");
+    });
+  res.sendStatus(202);
+});
+
 homeAssistantRouter.get("/addons", (req, res) => {
   haOsService
     .getAddonList()
diff --git a/nextcloud_backup/backend/src/services/orchestrator.ts b/nextcloud_backup/backend/src/services/orchestrator.ts
index 9a3e2bd..0eb7065 100644
--- a/nextcloud_backup/backend/src/services/orchestrator.ts
+++ b/nextcloud_backup/backend/src/services/orchestrator.ts
@@ -4,7 +4,10 @@ import logger from "../config/winston.js";
 import messageManager from "../tools/messageManager.js";
 import * as statusTools from "../tools/status.js";
 import { BackupType } from "../types/services/backupConfig.js";
-import type { AddonModel } from "../types/services/ha_os_response.js";
+import type {
+  AddonModel,
+  BackupDetailModel,
+} from "../types/services/ha_os_response.js";
 import { WorkflowType } from "../types/services/orchecstrator.js";
 import * as backupConfigService from "./backupConfigService.js";
 import * as homeAssistantService from "./homeAssistantService.js";
@@ -89,7 +92,10 @@ export function doBackupWorkflow(type: WorkflowType) {
     })
     .then(() => {
       logger.info("Backup workflow finished successfully !");
-      messageManager.info("Backup workflow finished successfully !");
+      messageManager.info(
+        "Backup workflow finished successfully !",
+        `name: ${name}`
+      );
       const status = statusTools.getStatus();
       status.last_backup.success = true;
       status.last_backup.last_try = DateTime.now();
@@ -105,6 +111,55 @@ export function doBackupWorkflow(type: WorkflowType) {
     });
 }
 
+export function uploadToCloud(slug: string) {
+  const webdavConfig = getWebdavConfig();
+  let tmpBackupFile = "";
+  let backupInfo = {} as BackupDetailModel;
+
+  return webDavService
+    .checkWebdavLogin(webdavConfig)
+    .then(() => {
+      return homeAssistantService.getBackupInfo(slug);
+    })
+    .then((response) => {
+      backupInfo = response.body.data;
+      return homeAssistantService.downloadSnapshot(slug);
+    })
+    .then((tmpFile) => {
+      tmpBackupFile = tmpFile;
+      if (webdavConfig.chunckedUpload) {
+        return webDavService.chunkedUpload(
+          tmpFile,
+          getBackupFolder(WorkflowType.MANUAL, webdavConfig) +
+            backupInfo.name +
+            ".tar",
+          webdavConfig
+        );
+      } else {
+        return webDavService.webdavUploadFile(
+          tmpFile,
+          getBackupFolder(WorkflowType.MANUAL, webdavConfig) +
+            backupInfo.name +
+            ".tar",
+          webdavConfig
+        );
+      }
+    })
+    .then(() => {
+      logger.info(`Successfully uploaded ${backupInfo.name} to cloud.`);
+      messageManager.info(
+        "Successfully uploaded backup to cloud.",
+        `Name: ${backupInfo.name}`
+      );
+    })
+    .catch(() => {
+      if (tmpBackupFile != "") {
+        unlinkSync(tmpBackupFile);
+      }
+      return Promise.reject(new Error());
+    });
+}
+
 // This methods remove addon that are no installed in HA from the conf array
 function sanitizeAddonList(addonInConf: string[], addonInHA: AddonModel[]) {
   return addonInConf.filter((value) => addonInHA.some((v) => v.slug == value));
diff --git a/nextcloud_backup/frontend/src/components/homeAssistant/HaList.vue b/nextcloud_backup/frontend/src/components/homeAssistant/HaList.vue
index f9e4c6a..00e8952 100644
--- a/nextcloud_backup/frontend/src/components/homeAssistant/HaList.vue
+++ b/nextcloud_backup/frontend/src/components/homeAssistant/HaList.vue
@@ -37,6 +37,7 @@
                     :key="item.slug"
                     :item="item"
                     :index="index"
+                    @upload="upload"
                   >
                   </ha-list-item>
                 </v-list>
@@ -52,12 +53,18 @@
 <script lang="ts" setup>
 import type { BackupModel } from "@/types/homeAssistant";
 import { ref, onBeforeUnmount } from "vue";
-import { getBackups } from "@/services/homeAssistantService";
+import {
+  getBackups,
+  uploadHomeAssistantBackup,
+} from "@/services/homeAssistantService";
 import HaListItem from "./HaListItem.vue";
+import { useAlertStore } from "@/store/alert";
 
 const backups = ref<BackupModel[]>([]);
 const loading = ref<boolean>(true);
 
+const alertStore = useAlertStore();
+
 function refreshBackup() {
   loading.value = true;
   getBackups()
@@ -70,6 +77,16 @@ function refreshBackup() {
     });
 }
 
+function upload(item: BackupModel) {
+  uploadHomeAssistantBackup(item.slug)
+    .then(() => {
+      alertStore.add("success", "Backup upload as started.");
+    })
+    .catch(() => {
+      alertStore.add("error", "Fail to start backup upload !");
+    });
+}
+
 refreshBackup();
 
 defineExpose({ refreshBackup });
diff --git a/nextcloud_backup/frontend/src/components/homeAssistant/HaListItem.vue b/nextcloud_backup/frontend/src/components/homeAssistant/HaListItem.vue
index 6761d40..5dd9b81 100644
--- a/nextcloud_backup/frontend/src/components/homeAssistant/HaListItem.vue
+++ b/nextcloud_backup/frontend/src/components/homeAssistant/HaListItem.vue
@@ -156,7 +156,12 @@
       <v-card-actions class="justify-center">
         <v-tooltip text="Upload to Cloud" location="bottom">
           <template v-slot:activator="{ props }">
-            <v-btn variant="outlined" color="success" v-bind="props">
+            <v-btn
+              variant="outlined"
+              color="success"
+              v-bind="props"
+              @click="emits('upload', item)"
+            >
               <v-icon>mdi-cloud-upload</v-icon>
             </v-btn>
           </template>
@@ -193,5 +198,6 @@ watch(detail, (value) => {
 
 const emits = defineEmits<{
   (e: "delete", item: BackupModel): void;
+  (e: "upload", item: BackupModel): void;
 }>();
 </script>
diff --git a/nextcloud_backup/frontend/src/services/homeAssistantService.ts b/nextcloud_backup/frontend/src/services/homeAssistantService.ts
index f78ce91..c99dcba 100644
--- a/nextcloud_backup/frontend/src/services/homeAssistantService.ts
+++ b/nextcloud_backup/frontend/src/services/homeAssistantService.ts
@@ -21,3 +21,7 @@ export function getBackups() {
 export function getBackupDetail(slug: string) {
   return kyClient.get(`homeAssistant/backup/${slug}`).json<BackupDetailModel>();
 }
+
+export function uploadHomeAssistantBackup(slug: string) {
+  return kyClient.post(`homeAssistant/backup/${slug}/upload`);
+}