import type { AddonModel } 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"; import { getBackupFolder, getWebdavConfig } from "./webdavConfigService.js"; import * as webDavService from "./webdavService.js"; import * as statusTools from "../tools/status.js"; import { stat, unlinkSync } from "fs"; import logger from "../config/winston.js"; import { BackupType } from "../types/services/backupConfig.js"; import { DateTime } from "luxon"; import messageManager from "../tools/messageManager.js"; export function doBackupWorkflow(type: WorkflowType) { let name = ""; let addonsToStartStop = [] as string[]; let addonInHa = [] as AddonModel[]; let tmpBackupFile = ""; const backupConfig = backupConfigService.getBackupConfig(); const webdavConfig = getWebdavConfig(); return homeAssistantService .getVersion() .then((value) => { const version = value.body.data.version; name = backupConfigService.getFormatedName(type, version); return homeAssistantService.getAddonList(); }) .then((response) => { addonInHa = response.body.data.addons; addonsToStartStop = sanitizeAddonList( backupConfig.autoStopAddon, response.body.data.addons ); return webDavService.checkWebdavLogin(webdavConfig); }) .then(() => { return homeAssistantService.stopAddons(addonsToStartStop); }) .then((response) => { if (backupConfig.backupType == BackupType.FULL) { return homeAssistantService.createNewBackup( name, backupConfig.backupType, backupConfig.password.enabled, backupConfig.password.value ); } else { const addons = getAddonToBackup( backupConfig.exclude?.addon as string[], addonInHa ); const folders = getFolderToBackup( backupConfig.exclude?.folder as string[], homeAssistantService.getFolderList() ); return homeAssistantService.createNewBackup( name, backupConfig.backupType, backupConfig.password.enabled, backupConfig.password.value, addons, folders ); } }) .then((response) => { response.body.data.slug; return homeAssistantService.downloadSnapshot(response.body.data.slug); }) .then((tmpFile) => { tmpBackupFile = tmpFile; return webDavService.chunkedUpload( tmpFile, getBackupFolder(type, webdavConfig) + name, webdavConfig ); // return webDavService.webdavUploadFile( // tmpFile, // getBackupFolder(type, webdavConfig) + name, // webdavConfig // ); }) .then(() => { logger.info("Backup workflow finished successfully !"); messageManager.info("Backup workflow finished successfully !"); const status = statusTools.getStatus(); status.last_backup.success = true; status.last_backup.last_try = DateTime.now(); status.last_backup.last_success = DateTime.now(); statusTools.setStatus(status); }) .catch(() => { backupFail(); if (tmpBackupFile != "") { unlinkSync(tmpBackupFile); } return Promise.reject(); }); } // This methods remove addon that are no installed in HA from the conf array function sanitizeAddonList(addonInConf: string[], addonInHA: AddonModel[]) { addonInConf.filter((value) => addonInHA.some((v) => v.slug == value)); return addonInConf; } function getAddonToBackup(excludedAddon: string[], addonInHA: AddonModel[]) { const slugs: string[] = []; for (const addon of addonInHA) { if (!excludedAddon.includes(addon.slug)) slugs.push(addon.slug); } logger.debug("Addon to backup:"); logger.debug(slugs); return slugs; } function getFolderToBackup( excludedFolder: string[], folderInHA: { name: string; slug: string }[] ) { const slugs = []; for (const folder of folderInHA) { if (!excludedFolder.includes(folder.slug)) slugs.push(folder.slug); } logger.debug("Folders to backup:"); logger.debug(slugs); return slugs; } function backupFail() { const status = statusTools.getStatus(); status.last_backup.success = false; status.last_backup.last_try = DateTime.now(); statusTools.setStatus(status); messageManager.error("Last backup as failed !"); }