setupcomfyuimodels 1.0.3 → 1.1.0

package/dist/index.js

@@ -12998,21 +12998,18 @@ var chalkStderr = createChalk({ level: stderrColor ? stderrColor.level : 0 });
 var source_default = chalk;
 
 // src/downloadAssets.ts
-var { WORKSPACE, CIVITAI_TOKEN, HF_TOKEN } = process.env;
-if (!WORKSPACE) {
-  throw new Error("WORKSPACE environment variable is not set.");
-}
+var { CIVITAI_TOKEN, HF_TOKEN } = process.env;
 var MAX_CONCURRENT_DOWNLOADS = 8;
 var concurrentDownloads = 0;
-var downloadAssets = async (configLocation) => {
+var downloadAssets = async (configLocation, targetFolder) => {
   const config = await getConfig(configLocation);
   const filePromises = [
     ...Object.entries(config.files).map(([targetLocation, url]) => {
       const fileName = import_path.default.basename(targetLocation), dir = import_path.default.dirname(targetLocation);
-      return scheduleDownload(dir, fileName, url);
+      return scheduleDownload(targetFolder, dir, fileName, url);
     }),
     ...Object.entries(config.folders).flatMap(
-      ([folder, urls]) => urls.map((url) => scheduleDownload(folder, null, url))
+      ([folder, urls]) => urls.map((url) => scheduleDownload(targetFolder, folder, null, url))
     )
   ];
   setTotalProgress(0, filePromises.length);
@@ -13035,7 +13032,7 @@ var downloadAssets = async (configLocation) => {
 };
 var scheduledDownloads = [];
 var results = [];
-var scheduleDownload = async (targetLocation, targetFileName, url) => {
+var scheduleDownload = async (globalTargetLocation, targetLocation, targetFileName, url) => {
   scheduledDownloads.push({ targetLocation, url, targetFileName });
   if (concurrentDownloads < MAX_CONCURRENT_DOWNLOADS) {
     while (scheduledDownloads.length > 0) {
@@ -13044,6 +13041,7 @@ var scheduleDownload = async (targetLocation, targetFileName, url) => {
         concurrentDownloads++;
         try {
           const skipped = await downloadFile(
+            globalTargetLocation,
             nextDownload.targetLocation,
             nextDownload.targetFileName,
             nextDownload.url
@@ -13065,10 +13063,12 @@ var scheduleDownload = async (targetLocation, targetFileName, url) => {
     }
   }
 };
-var downloadFile = async (targetLocation, targetFileName, url) => {
+var downloadFile = async (globalTargetLocation, targetLocation, targetFileName, url) => {
   const getTargetId = () => import_path.default.join(targetLocation, targetFileName || "");
   try {
-    const targetDir = import_path.default.normalize(`${WORKSPACE}/models/${targetLocation}`);
+    const targetDir = import_path.default.normalize(
+      `${globalTargetLocation}/${targetLocation}`
+    );
     fs.mkdirSync(targetDir, { recursive: true });
     if (targetFileName && fs.existsSync(import_path.default.join(targetDir, targetFileName))) {
       log2(`File already exists, skipping: ${getTargetId()}`);
@@ -13136,8 +13136,8 @@ var downloadFile = async (targetLocation, targetFileName, url) => {
 // src/index.ts
 var program2 = new Command();
 program2.name("Setup ComfyUi Models");
-program2.command("download").argument("<config>", "JSON config file location").action(async (configLocation) => {
-  await downloadAssets(configLocation);
+program2.command("download").argument("<config>", "JSON config file location").option("-l, --location <path>", "Target download location").action(async (configLocation, options) => {
+  await downloadAssets(configLocation, options.location || "./");
 });
 program2.parse();
 //# sourceMappingURL=index.js.map